arcfour.c 2.87 KB
Newer Older
Niels Möller's avatar
Niels Möller committed
1
/* arcfour.c
Niels Möller's avatar
Niels Möller committed
2
 *
Niels Möller's avatar
Niels Möller committed
3
4
5
6
7
 * This implements the Arcfour stream cipher with 128 bit keys. 
 *
 * The Arcfour cipher is believed to be compatible with the RC4 cipher. 
 * RC4 is a registered trademark of RSA Data Security Inc.
 *   
Niels Möller's avatar
Niels Möller committed
8
9
 */

Niels Möller's avatar
Niels Möller committed
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
/* lsh, an implementation of the ssh protocol
 *
 * Copyright (C) 1998 Niels Mller
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

Niels Möller's avatar
src/    
Niels Möller committed
29
#include "arcfour.h"
Niels Möller's avatar
Niels Möller committed
30

31
32
#include <assert.h>

Niels Möller's avatar
Niels Möller committed
33
34
35
36
37
38
#ifdef RCSID
RCSID("$Id$");
#endif

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

39
40
41
42
43
44
45
46
47
48
void arcfour_init(struct arcfour_ctx *ctx)
{
  unsigned i;

  /* Initialize context */

  for (i = 0; i<256; i++)
    ctx->S[i] = i;
}

49
50
/* This mode of operation is non-standard and possibly insecure. */
#if 0
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
void arcfour_update_key(struct arcfour_ctx *ctx,
			UINT32 length, const UINT8 *key)
{
  register UINT8 i = ctx->i;
  register UINT8 j = ctx->j;

  unsigned k;

  for (k = 0; k<length; k++)
    {
      i++; i &= 0xff;
      j += ctx->S[i] + key[k]; j &= 0xff;
      SWAP(ctx->S[i], ctx->S[j]);
    }
  ctx->i = i; ctx->j = j;
}
67
#endif
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87

void arcfour_stream(struct arcfour_ctx *ctx,
		    UINT32 length, UINT8 *dest)
{
  register UINT8 i = ctx->i;
  register UINT8 j = ctx->j;
  unsigned k;

  for (k = 0; k<length; k++)
    {
      i++; i &= 0xff;
      j += ctx->S[i]; j &= 0xff;
      SWAP(ctx->S[i], ctx->S[j]);
      dest[k] = ctx->S[ (ctx->S[i] + ctx->S[j]) & 0xff ];
    }
  
  ctx->i = i; ctx->j = j;
}

void arcfour_set_key(struct arcfour_ctx *ctx, UINT32 length, const UINT8 *key)
Niels Möller's avatar
Niels Möller committed
88
{
89
  register UINT8 j; /* Depends on the eight-bitness of these variables. */
Niels Möller's avatar
Niels Möller committed
90
  unsigned i;
91
  UINT32 k;
Niels Möller's avatar
Niels Möller committed
92
93

  /* Initialize context */
94
  arcfour_init(ctx);
Niels Möller's avatar
Niels Möller committed
95

96
97
  assert(length);
  
Niels Möller's avatar
Niels Möller committed
98
99
  /* Expand key */
  i = j = k = 0;
100
101
102
103
104
105
  for ( ; i<256; i++)
    {
      j += ctx->S[i] + key[k]; j &= 0xff;
      SWAP(ctx->S[i], ctx->S[j]);
      k = (k+1) % length; /* Repeat key if needed */
    } 
Niels Möller's avatar
Niels Möller committed
106
107
108
109
  
  ctx->i = ctx->j = 0;
}

Niels Möller's avatar
Niels Möller committed
110
void arcfour_crypt(struct arcfour_ctx *ctx, UINT8 *dest,
111
		   UINT32 length, const UINT8 *src)
Niels Möller's avatar
Niels Möller committed
112
{
113
  register UINT8 i, j;
Niels Möller's avatar
Niels Möller committed
114
115

  i = ctx->i; j = ctx->j;
116
  while(length--)
Niels Möller's avatar
Niels Möller committed
117
    {
118
119
      i++; i &= 0xff;
      j += ctx->S[i]; j &= 0xff;
Niels Möller's avatar
Niels Möller committed
120
121
122
123
124
      SWAP(ctx->S[i], ctx->S[j]);
      *dest++ = *src++ ^ ctx->S[ (ctx->S[i] + ctx->S[j]) & 0xff ];
    }
  ctx->i = i; ctx->j = j;
}