arcfour.c 981 Bytes
Newer Older
Niels Möller's avatar
Niels Möller committed
1
/* arcfour.c
Niels Möller's avatar
Niels Möller committed
2
3
4
5
 *
 */

#include "crypto_types.h"
Niels Möller's avatar
Niels Möller committed
6
#include <arcfour.h>
Niels Möller's avatar
Niels Möller committed
7
8
9
10
11
12
13

#ifdef RCSID
RCSID("$Id$");
#endif

#define SWAP(a,b) do { int _t = a; a = b; b = _t; } while(0)

Niels Möller's avatar
Niels Möller committed
14
void arcfour_set_key(struct arcfour_ctx *ctx, const UINT8 *key, UINT32 len)
Niels Möller's avatar
Niels Möller committed
15
{
16
  register UINT8 j; /* Depends on the eight-bitness of these variables. */
Niels Möller's avatar
Niels Möller committed
17
  unsigned i;
18
  UINT32 k;
Niels Möller's avatar
Niels Möller committed
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

  /* Initialize context */
  i = 0;
  do ctx->S[i] = i; while (++i < 256);

  /* Expand key */
  i = j = k = 0;
  do {
    j += ctx->S[i] + key[k];
    SWAP(ctx->S[i], ctx->S[j]);
    k = (k+1) % len; /* Repeat key if needed */
  } while(++i < 256);
  
  ctx->i = ctx->j = 0;
}

Niels Möller's avatar
Niels Möller committed
35
36
void arcfour_crypt(struct arcfour_ctx *ctx, UINT8 *dest,
		   const UINT8 *src, UINT32 len)
Niels Möller's avatar
Niels Möller committed
37
{
38
  register UINT8 i, j;
Niels Möller's avatar
Niels Möller committed
39
40
41
42

  i = ctx->i; j = ctx->j;
  while(len--)
    {
43
44
      i++; i &= 0xff;
      j += ctx->S[i]; j &= 0xff;
Niels Möller's avatar
Niels Möller committed
45
46
47
48
49
      SWAP(ctx->S[i], ctx->S[j]);
      *dest++ = *src++ ^ ctx->S[ (ctx->S[i] + ctx->S[j]) & 0xff ];
    }
  ctx->i = i; ctx->j = j;
}