Files
2020-05-24 10:16:09 +08:00

62 lines
1.3 KiB
C++

#include <Windows.h>
#include "md5.h"
typedef struct _RC4_CONTEXT
{
unsigned char state[256];
unsigned char x, y;
} RC4_CONTEXT;
void rc4_init(RC4_CONTEXT *a4i, const unsigned char *key, unsigned int keyLen);
void rc4_crypt(RC4_CONTEXT *a4i, unsigned char *inoutString, unsigned int length);
void rc4_init(RC4_CONTEXT *a4i, const unsigned char *key, unsigned int keyLen)
{
unsigned int keyIndex = 0, stateIndex = 0;
unsigned int i, a;
a4i->x = a4i->y = 0;
for (i = 0; i<256; i++)
a4i->state[i] = i;
for (i = 0; i<256; i++)
{
a = a4i->state[i];
stateIndex += key[keyIndex] + a;
stateIndex &= 0xff;
a4i->state[i] = a4i->state[stateIndex];
a4i->state[stateIndex] = a;
if (++keyIndex >= keyLen)
keyIndex = 0;
}
}
void rc4_crypt(RC4_CONTEXT *a4i, unsigned char *inoutString, unsigned int length)
{
unsigned char *const s = a4i->state;
unsigned int x = a4i->x;
unsigned int y = a4i->y;
unsigned int a, b;
while (length--)
{
x = (x + 1) & 0xff;
a = s[x];
y = (y + a) & 0xff;
b = s[y];
s[x] = b;
s[y] = a;
*inoutString++ ^= s[(a + b) & 0xff];
}
a4i->x = x;
a4i->y = y;
}
void rc4Decrypt(CRYPTO_BUFFER *data, CRYPTO_BUFFER *key)
{
RC4_CONTEXT a4i;
rc4_init(&a4i, key->Buffer, key->Length);
rc4_crypt(&a4i, data->Buffer, data->Length);
}