#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#ifdef HAVE_SYS_TYPES_H
# include <sys/types.h>
#endif
#include <string.h>
#include "dpxcrypt.h"
static void _gcry_burn_stack (int bytes)
{
char buf[64];
memset(buf, 0, sizeof buf);
bytes -= sizeof buf;
if (bytes > 0) _gcry_burn_stack(bytes);
}
#define rol(x,n) ( ((x) << (n)) | ((x) >> (32-(n))) )
void texpdf_MD5_init (MD5_CONTEXT *ctx)
{
ctx->A = 0x67452301;
ctx->B = 0xefcdab89;
ctx->C = 0x98badcfe;
ctx->D = 0x10325476;
ctx->nblocks = 0;
ctx->count = 0;
}
#define FF(b, c, d) (d ^ (b & (c ^ d)))
#define FG(b, c, d) FF(d, b, c)
#define FH(b, c, d) (b ^ c ^ d)
#define FI(b, c, d) (c ^ (b | ~d))
static void transform (MD5_CONTEXT *ctx, const unsigned char *data)
{
uint32_t correct_words[16];
register uint32_t A = ctx->A;
register uint32_t B = ctx->B;
register uint32_t C = ctx->C;
register uint32_t D = ctx->D;
uint32_t *cwp = correct_words;
#ifdef WORDS_BIGENDIAN
{ int i; const unsigned char *p1; unsigned char *p2;
for (i = 0, p1 = data, p2 = (unsigned char *)correct_words; i < 16; i++, p2 += 4 ) {
p2[3] = *p1++; p2[2] = *p1++; p2[1] = *p1++; p2[0] = *p1++;
}
}
#else
memcpy(correct_words, data, sizeof(uint32_t) * 16);
#endif
#define OP(a, b, c, d, s, T) \
do { a += FF(b, c, d) + (*cwp++) + T; a = rol(a, s); a += b; } while (0)
OP(A, B, C, D, 7, 0xd76aa478);
OP(D, A, B, C, 12, 0xe8c7b756);
OP(C, D, A, B, 17, 0x242070db);
OP(B, C, D, A, 22, 0xc1bdceee);
OP(A, B, C, D, 7, 0xf57c0faf);
OP(D, A, B, C, 12, 0x4787c62a);
OP(C, D, A, B, 17, 0xa8304613);
OP(B, C, D, A, 22, 0xfd469501);
OP(A, B, C, D, 7, 0x698098d8);
OP(D, A, B, C, 12, 0x8b44f7af);
OP(C, D, A, B, 17, 0xffff5bb1);
OP(B, C, D, A, 22, 0x895cd7be);
OP(A, B, C, D, 7, 0x6b901122);
OP(D, A, B, C, 12, 0xfd987193);
OP(C, D, A, B, 17, 0xa679438e);
OP(B, C, D, A, 22, 0x49b40821);
#undef OP
#define OP(f, a, b, c, d, k, s, T) \
do { a += f(b, c, d) + correct_words[k] + T; a = rol(a, s); a += b; } while (0)
OP(FG, A, B, C, D, 1, 5, 0xf61e2562);
OP(FG, D, A, B, C, 6, 9, 0xc040b340);
OP(FG, C, D, A, B, 11, 14, 0x265e5a51);
OP(FG, B, C, D, A, 0, 20, 0xe9b6c7aa);
OP(FG, A, B, C, D, 5, 5, 0xd62f105d);
OP(FG, D, A, B, C, 10, 9, 0x02441453);
OP(FG, C, D, A, B, 15, 14, 0xd8a1e681);
OP(FG, B, C, D, A, 4, 20, 0xe7d3fbc8);
OP(FG, A, B, C, D, 9, 5, 0x21e1cde6);
OP(FG, D, A, B, C, 14, 9, 0xc33707d6);
OP(FG, C, D, A, B, 3, 14, 0xf4d50d87);
OP(FG, B, C, D, A, 8, 20, 0x455a14ed);
OP(FG, A, B, C, D, 13, 5, 0xa9e3e905);
OP(FG, D, A, B, C, 2, 9, 0xfcefa3f8);
OP(FG, C, D, A, B, 7, 14, 0x676f02d9);
OP(FG, B, C, D, A, 12, 20, 0x8d2a4c8a);
OP(FH, A, B, C, D, 5, 4, 0xfffa3942);
OP(FH, D, A, B, C, 8, 11, 0x8771f681);
OP(FH, C, D, A, B, 11, 16, 0x6d9d6122);
OP(FH, B, C, D, A, 14, 23, 0xfde5380c);
OP(FH, A, B, C, D, 1, 4, 0xa4beea44);
OP(FH, D, A, B, C, 4, 11, 0x4bdecfa9);
OP(FH, C, D, A, B, 7, 16, 0xf6bb4b60);
OP(FH, B, C, D, A, 10, 23, 0xbebfbc70);
OP(FH, A, B, C, D, 13, 4, 0x289b7ec6);
OP(FH, D, A, B, C, 0, 11, 0xeaa127fa);
OP(FH, C, D, A, B, 3, 16, 0xd4ef3085);
OP(FH, B, C, D, A, 6, 23, 0x04881d05);
OP(FH, A, B, C, D, 9, 4, 0xd9d4d039);
OP(FH, D, A, B, C, 12, 11, 0xe6db99e5);
OP(FH, C, D, A, B, 15, 16, 0x1fa27cf8);
OP(FH, B, C, D, A, 2, 23, 0xc4ac5665);
OP(FI, A, B, C, D, 0, 6, 0xf4292244);
OP(FI, D, A, B, C, 7, 10, 0x432aff97);
OP(FI, C, D, A, B, 14, 15, 0xab9423a7);
OP(FI, B, C, D, A, 5, 21, 0xfc93a039);
OP(FI, A, B, C, D, 12, 6, 0x655b59c3);
OP(FI, D, A, B, C, 3, 10, 0x8f0ccc92);
OP(FI, C, D, A, B, 10, 15, 0xffeff47d);
OP(FI, B, C, D, A, 1, 21, 0x85845dd1);
OP(FI, A, B, C, D, 8, 6, 0x6fa87e4f);
OP(FI, D, A, B, C, 15, 10, 0xfe2ce6e0);
OP(FI, C, D, A, B, 6, 15, 0xa3014314);
OP(FI, B, C, D, A, 13, 21, 0x4e0811a1);
OP(FI, A, B, C, D, 4, 6, 0xf7537e82);
OP(FI, D, A, B, C, 11, 10, 0xbd3af235);
OP(FI, C, D, A, B, 2, 15, 0x2ad7d2bb);
OP(FI, B, C, D, A, 9, 21, 0xeb86d391);
ctx->A += A;
ctx->B += B;
ctx->C += C;
ctx->D += D;
}
void texpdf_MD5_write (MD5_CONTEXT *hd, const unsigned char *inbuf, unsigned long inlen)
{
if (hd->count == 64) {
transform(hd, hd->buf);
_gcry_burn_stack(80+6*sizeof(void*));
hd->count = 0;
hd->nblocks++;
}
if (!inbuf) return;
if (hd->count) {
for (; inlen && hd->count < 64; inlen--)
hd->buf[hd->count++] = *inbuf++;
texpdf_MD5_write(hd, NULL, 0);
if (!inlen) return;
}
_gcry_burn_stack(80+6*sizeof(void*));
while (inlen >= 64) {
transform(hd, inbuf);
hd->count = 0;
hd->nblocks++;
inlen -= 64;
inbuf += 64;
}
for (; inlen && hd->count < 64; inlen--)
hd->buf[hd->count++] = *inbuf++;
}
void texpdf_MD5_final (unsigned char *outbuf, MD5_CONTEXT *hd)
{
unsigned long t, msb, lsb;
unsigned char *p;
texpdf_MD5_write(hd, NULL, 0);
t = hd->nblocks;
lsb = t << 6;
msb = t >> 26;
t = lsb;
if ((lsb += hd->count) < t) msb++;
t = lsb;
lsb <<= 3;
msb <<= 3;
msb |= t >> 29;
if (hd->count < 56) {
hd->buf[hd->count++] = 0x80;
while (hd->count < 56) hd->buf[hd->count++] = 0;
} else {
hd->buf[hd->count++] = 0x80;
while (hd->count < 64) hd->buf[hd->count++] = 0;
texpdf_MD5_write(hd, NULL, 0);
memset(hd->buf, 0, 56);
}
hd->buf[56] = lsb & 0xff;
hd->buf[57] = (lsb >> 8) & 0xff;
hd->buf[58] = (lsb >> 16) & 0xff;
hd->buf[59] = (lsb >> 24) & 0xff;
hd->buf[60] = msb & 0xff;
hd->buf[61] = (msb >> 8) & 0xff;
hd->buf[62] = (msb >> 16) & 0xff;
hd->buf[63] = (msb >> 24) & 0xff;
transform(hd, hd->buf);
_gcry_burn_stack(80+6*sizeof(void*));
p = outbuf;
#ifdef WORDS_BIGENDIAN
#define X(a) do { *p++ = hd->a; *p++ = hd->a >> 8; \
*p++ = hd->a >> 16; *p++ = hd->a >> 24; } while (0)
#else
#define X(a) do { *(uint32_t *)p = (*hd).a ; p += sizeof(uint32_t); } while (0)
#endif
X(A);
X(B);
X(C);
X(D);
#undef X
}
static void do_encrypt_stream (ARC4_KEY *ctx, unsigned char *outbuf, const unsigned char *inbuf, unsigned long len)
{
register int i = ctx->idx_i;
register int j = ctx->idx_j;
register unsigned char *sbox = ctx->sbox;
register int t;
while (len--) {
i++;
i = i & 255;
j += sbox[i];
j &= 255;
t = sbox[i]; sbox[i] = sbox[j]; sbox[j] = t;
*outbuf++ = *inbuf++ ^ sbox[(sbox[i] + sbox[j]) & 255];
}
ctx->idx_i = i;
ctx->idx_j = j;
}
void ARC4 (ARC4_KEY *ctx, unsigned long len, const unsigned char *inbuf, unsigned char *outbuf)
{
do_encrypt_stream(ctx, outbuf, inbuf, len);
_gcry_burn_stack(64);
}
static void do_arcfour_setkey (ARC4_KEY *ctx, const unsigned char *key, unsigned int keylen)
{
int i, j;
unsigned char karr[256];
ctx->idx_i = ctx->idx_j = 0;
for (i = 0; i < 256; i++) ctx->sbox[i] = i;
for (i = 0; i < 256; i++) karr[i] = key[i%keylen];
for (i = j = 0; i < 256; i++) {
int t;
j = (j + ctx->sbox[i] + karr[i]) % 256;
t = ctx->sbox[i];
ctx->sbox[i] = ctx->sbox[j];
ctx->sbox[j] = t;
}
memset(karr, 0, 256);
}
void ARC4_set_key (ARC4_KEY *ctx, unsigned int keylen, const unsigned char *key)
{
do_arcfour_setkey(ctx, key, keylen);
_gcry_burn_stack(300);
}