arcfour/arcfour.c

   1 /* arcfour.c */
   2 /*
   3  This file is part of the AVR-Crypto-Lib.
   4  Copyright (C) 2006-2015 Daniel Otte (bg@nerilex.org)
   5
   6  This program is free software: you can redistribute it and/or modify
   7  it under the terms of the GNU General Public License as published by
   8  the Free Software Foundation, either version 3 of the License, or
   9  (at your option) any later version.
  10
  11  This program is distributed in the hope that it will be useful,
  12  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  GNU General Public License for more details.
  15
  16  You should have received a copy of the GNU General Public License
  17  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18  */
  19 /*
  20  * File:        arcfour.c
  21  * Author:      Daniel Otte
  22  * email:       bg@nerilex.org
  23  * Date:        2006-06-07
  24  * License:     GPLv3 or later
  25  * Description: Implementation of the ARCFOUR (RC4 compatible) stream cipher algorithm.
  26  *
  27  */
  28
  29 #include <stdint.h>
  30 #include "arcfour.h"
  31
  32 /*
  33  * length is length of key in bytes!
  34  */
  35
  36 void arcfour_init(const void *key, uint16_t length_b, arcfour_ctx_t *ctx)
  37 {
  38     uint8_t t;
  39     const uint8_t length_B = length_b / 8;
  40     uint8_t nidx = length_B;
  41     uint8_t x = 0, y = 0;
  42     const uint8_t *kptr = (const uint8_t*) key;
  43     do {
  44         ctx->s[x] = x;
  45     } while ((uint8_t) ++x);
  46
  47     do {
  48         y += ctx->s[x] + *kptr++;
  49         if (!--nidx) {
  50             kptr = (const uint8_t*) key;
  51             nidx = length_B;
  52         }
  53         y &= 0xff;
  54         /* ctx->s[y] <--> ctx->s[x] */
  55         t = ctx->s[y];
  56         ctx->s[y] = ctx->s[x];
  57         ctx->s[x] = t;
  58     } while ((uint8_t) ++x);
  59
  60     ctx->i = ctx->j = 0;
  61 }
  62
  63 uint8_t arcfour_gen(arcfour_ctx_t *ctx)
  64 {
  65     uint8_t t;
  66     ctx->i++;
  67     ctx->j += ctx->s[ctx->i];
  68     /* ctx->s[i] <--> ctx->s[j] */
  69     t = ctx->s[ctx->j];
  70     ctx->s[ctx->j] = ctx->s[ctx->i];
  71     ctx->s[ctx->i] = t;
  72     return ctx->s[(ctx->s[ctx->j] + ctx->s[ctx->i]) & 0xff];
  73 }
  74