git.sur5r.net Git - freertos/blob - FreeRTOS-Plus/Source/CyaSSL/ctaocrypt/src/arc4.c

   1 /* arc4.c
   2  *
   3  * Copyright (C) 2006-2012 Sawtooth Consulting Ltd.
   4  *
   5  * This file is part of CyaSSL.
   6  *
   7  * CyaSSL is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * CyaSSL is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; if not, write to the Free Software
  19  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
  20  */
  21
  22 #ifdef HAVE_CONFIG_H
  23     #include <config.h>
  24 #endif
  25
  26 #include <cyassl/ctaocrypt/arc4.h>
  27
  28
  29 void Arc4SetKey(Arc4* arc4, const byte* key, word32 length)
  30 {
  31     word32 i;
  32     word32 keyIndex = 0, stateIndex = 0;
  33
  34     arc4->x = 1;
  35     arc4->y = 0;
  36
  37     for (i = 0; i < ARC4_STATE_SIZE; i++)
  38         arc4->state[i] = (byte)i;
  39
  40     for (i = 0; i < ARC4_STATE_SIZE; i++) {
  41         word32 a = arc4->state[i];
  42         stateIndex += key[keyIndex] + a;
  43         stateIndex &= 0xFF;
  44         arc4->state[i] = arc4->state[stateIndex];
  45         arc4->state[stateIndex] = (byte)a;
  46
  47         if (++keyIndex >= length)
  48             keyIndex = 0;
  49     }
  50 }
  51
  52
  53 static INLINE byte MakeByte(word32* x, word32* y, byte* s)
  54 {
  55     word32 a = s[*x], b;
  56     *y = (*y+a) & 0xff;
  57
  58     b = s[*y];
  59     s[*x] = (byte)b;
  60     s[*y] = (byte)a;
  61     *x = (*x+1) & 0xff;
  62
  63     return s[(a+b) & 0xff];
  64 }
  65
  66
  67 void Arc4Process(Arc4* arc4, byte* out, const byte* in, word32 length)
  68 {
  69     word32 x = arc4->x;
  70     word32 y = arc4->y;
  71
  72     while(length--)
  73         *out++ = *in++ ^ MakeByte(&x, &y, arc4->state);
  74
  75     arc4->x = (byte)x;
  76     arc4->y = (byte)y;
  77 }
  78