#include "stdafx.h"#include "AES.h"void AES_ecb_encrypt(const unsigned char *in, unsigned char *out,					 const AES_KEY *key, const int enc) {						 assert(in && out && key);						 assert((AES_ENCRYPT == enc)||(AES_DECRYPT == enc));						 if (AES_ENCRYPT == enc)							 AES_encrypt(in, out, key);						 else							 AES_decrypt(in, out, key);}int AES_set_encrypt_key(const unsigned char *userKey, const int bits,						AES_KEY *key) {							u32 *rk;							int i = 0;							u32 temp;							if (!userKey || !key)								return -1;							if (bits != 128 && bits != 192 && bits != 256)								return -2;							rk = key->rd_key;							if (bits==128)								key->rounds = 10;							else if (bits==192)								key->rounds = 12;							else								key->rounds = 14;							rk[0] = GETU32(userKey     );							rk[1] = GETU32(userKey +  4);							rk[2] = GETU32(userKey +  8);							rk[3] = GETU32(userKey + 12);							if (bits == 128) {								for (;;) {									temp  = rk[3];									rk[4] = rk[0] ^										(Te4[(temp >> 16) & 0xff] & 0xff000000) ^										(Te4[(temp >>  8) & 0xff] & 0x00ff0000) ^										(Te4[(temp      ) & 0xff] & 0x0000ff00) ^										(Te4[(temp >> 24)       ] & 0x000000ff) ^										rcon[i];									rk[5] = rk[1] ^ rk[4];									rk[6] = rk[2] ^ rk[5];									rk[7] = rk[3] ^ rk[6];									if (++i == 10) {										return 0;									}									rk += 4;								}							}							rk[4] = GETU32(userKey + 16);							rk[5] = GETU32(userKey + 20);							if (bits == 192) {								for (;;) {									temp = rk[ 5];									rk[ 6] = rk[ 0] ^										(Te4[(temp >> 16) & 0xff] & 0xff000000) ^										(Te4[(temp >>  8) & 0xff] & 0x00ff0000) ^										(Te4[(temp      ) & 0xff] & 0x0000ff00) ^										(Te4[(temp >> 24)       ] & 0x000000ff) ^										rcon[i];									rk[ 7] = rk[ 1] ^ rk[ 6];									rk[ 8] = rk[ 2] ^ rk[ 7];									rk[ 9] = rk[ 3] ^ rk[ 8];									if (++i == 8) {										return 0;									}									rk[10] = rk[ 4] ^ rk[ 9];									rk[11] = rk[ 5] ^ rk[10];									rk += 6;								}							}							rk[6] = GETU32(userKey + 24);							rk[7] = GETU32(userKey + 28);							if (bits == 256) {								for (;;) {									temp = rk[ 7];									rk[ 8] = rk[ 0] ^										(Te4[(temp >> 16) & 0xff] & 0xff000000) ^										(Te4[(temp >>  8) & 0xff] & 0x00ff0000) ^										(Te4[(temp      ) & 0xff] & 0x0000ff00) ^										(Te4[(temp >> 24)       ] & 0x000000ff) ^										rcon[i];									rk[ 9] = rk[ 1] ^ rk[ 8];									rk[10] = rk[ 2] ^ rk[ 9];									rk[11] = rk[ 3] ^ rk[10];									if (++i == 7) {										return 0;									}									temp = rk[11];									rk[12] = rk[ 4] ^										(Te4[(temp >> 24)       ] & 0xff000000) ^										(Te4[(temp >> 16) & 0xff] & 0x00ff0000) ^										(Te4[(temp >>  8) & 0xff] & 0x0000ff00) ^										(Te4[(temp      ) & 0xff] & 0x000000ff);									rk[13] = rk[ 5] ^ rk[12];									rk[14] = rk[ 6] ^ rk[13];									rk[15] = rk[ 7] ^ rk[14];									rk += 8;								}							}							return 0;}int AES_set_decrypt_key(const unsigned char *userKey, const int bits,						AES_KEY *key) {							u32 *rk;							int i, j, status;							u32 temp;							/* first, start with an encryption schedule */							status = AES_set_encrypt_key(userKey, bits, key);							if (status < 0)								return status;							rk = key->rd_key;							/* invert the order of the round keys: */							for (i = 0, j = 4*(key->rounds); i < j; i += 4, j -= 4) {								temp = rk[i    ]; rk[i    ] = rk[j    ]; rk[j    ] = temp;								temp = rk[i + 1]; rk[i + 1] = rk[j + 1]; rk[j + 1] = temp;								temp = rk[i + 2]; rk[i + 2] = rk[j + 2]; rk[j + 2] = temp;								temp = rk[i + 3]; rk[i + 3] = rk[j + 3]; rk[j + 3] = temp;							}							/* apply the inverse MixColumn transform to all round keys but the first and the last: */							for (i = 1; i < (key->rounds); i++) {								rk += 4;								rk[0] =									Td0[Te4[(rk[0] >> 24)       ] & 0xff] ^									Td1[Te4[(rk[0] >> 16) & 0xff] & 0xff] ^									Td2[Te4[(rk[0] >>  8) & 0xff] & 0xff] ^									Td3[Te4[(rk[0]      ) & 0xff] & 0xff];								rk[1] =									Td0[Te4[(rk[1] >> 24)       ] & 0xff] ^									Td1[Te4[(rk[1] >> 16) & 0xff] & 0xff] ^									Td2[Te4[(rk[1] >>  8) & 0xff] & 0xff] ^									Td3[Te4[(rk[1]      ) & 0xff] & 0xff];								rk[2] =									Td0[Te4[(rk[2] >> 24)       ] & 0xff] ^									Td1[Te4[(rk[2] >> 16) & 0xff] & 0xff] ^									Td2[Te4[(rk[2] >>  8) & 0xff] & 0xff] ^									Td3[Te4[(rk[2]      ) & 0xff] & 0xff];								rk[3] =									Td0[Te4[(rk[3] >> 24)       ] & 0xff] ^									Td1[Te4[(rk[3] >> 16) & 0xff] & 0xff] ^									Td2[Te4[(rk[3] >>  8) & 0xff] & 0xff] ^									Td3[Te4[(rk[3]      ) & 0xff] & 0xff];							}							return 0;}void AES_encrypt(const unsigned char *in, unsigned char *out,				 const AES_KEY *key) {					 const u32 *rk;					 u32 s0, s1, s2, s3, t0, t1, t2, t3;#ifndef FULL_UNROLL					 int r;#endif /* ?FULL_UNROLL */					 assert(in && out && key);					 rk = key->rd_key;					 /*					 * map byte array block to cipher state					 * and add initial round key:					 */					 s0 = GETU32(in     ) ^ rk[0];					 s1 = GETU32(in +  4) ^ rk[1];					 s2 = GETU32(in +  8) ^ rk[2];					 s3 = GETU32(in + 12) ^ rk[3];#ifdef FULL_UNROLL					 /* round 1: */					 t0 = Te0[s0 >> 24] ^ Te1[(s1 >> 16) & 0xff] ^ Te2[(s2 >>  8) & 0xff] ^ Te3[s3 & 0xff] ^ rk[ 4];					 t1 = Te0[s1 >> 24] ^ Te1[(s2 >> 16) & 0xff] ^ Te2[(s3 >>  8) & 0xff] ^ Te3[s0 & 0xff] ^ rk[ 5];					 t2 = Te0[s2 >> 24] ^ Te1[(s3 >> 16) & 0xff] ^ Te2[(s0 >>  8) & 0xff] ^ Te3[s1 & 0xff] ^ rk[ 6];					 t3 = Te0[s3 >> 24] ^ Te1[(s0 >> 16) & 0xff] ^ Te2[(s1 >>  8) & 0xff] ^ Te3[s2 & 0xff] ^ rk[ 7];					 /* round 2: */					 s0 = Te0[t0 >> 24] ^ Te1[(t1 >> 16) & 0xff] ^ Te2[(t2 >>  8) & 0xff] ^ Te3[t3 & 0xff] ^ rk[ 8];					 s1 = Te0[t1 >> 24] ^ Te1[(t2 >> 16) & 0xff] ^ Te2[(t3 >>  8) & 0xff] ^ Te3[t0 & 0xff] ^ rk[ 9];					 s2 = Te0[t2 >> 24] ^ Te1[(t3 >> 16) & 0xff] ^ Te2[(t0 >>  8) & 0xff] ^ Te3[t1 & 0xff] ^ rk[10];					 s3 = Te0[t3 >> 24] ^ Te1[(t0 >> 16) & 0xff] ^ Te2[(t1 >>  8) & 0xff] ^ Te3[t2 & 0xff] ^ rk[11];					 /* round 3: */					 t0 = Te0[s0 >> 24] ^ Te1[(s1 >> 16) & 0xff] ^ Te2[(s2 >>  8) & 0xff] ^ Te3[s3 & 0xff] ^ rk[12];					 t1 = Te0[s1 >> 24] ^ Te1[(s2 >> 16) & 0xff] ^ Te2[(s3 >>  8) & 0xff] ^ Te3[s0 & 0xff] ^ rk[13];					 t2 = Te0[s2 >> 24] ^ Te1[(s3 >> 16) & 0xff] ^ Te2[(s0 >>  8) & 0xff] ^ Te3[s1 & 0xff] ^ rk[14];					 t3 = Te0[s3 >> 24] ^ Te1[(s0 >> 16) & 0xff] ^ Te2[(s1 >>  8) & 0xff] ^ Te3[s2 & 0xff] ^ rk[15];					 /* round 4: */					 s0 = Te0[t0 >> 24] ^ Te1[(t1 >> 16) & 0xff] ^ Te2[(t2 >>  8) & 0xff] ^ Te3[t3 & 0xff] ^ rk[16];					 s1 = Te0[t1 >> 24] ^ Te1[(t2 >> 16) & 0xff] ^ Te2[(t3 >>  8) & 0xff] ^ Te3[t0 & 0xff] ^ rk[17];					 s2 = Te0[t2 >> 24] ^ Te1[(t3 >> 16) & 0xff] ^ Te2[(t0 >>  8) & 0xff] ^ Te3[t1 & 0xff] ^ rk[18];					 s3 = Te0[t3 >> 24] ^ Te1[(t0 >> 16) & 0xff] ^ Te2[(t1 >>  8) & 0xff] ^ Te3[t2 & 0xff] ^ rk[19];					 /* round 5: */					 t0 = Te0[s0 >> 24] ^ Te1[(s1 >> 16) & 0xff] ^ Te2[(s2 >>  8) & 0xff] ^ Te3[s3 & 0xff] ^ rk[20];					 t1 = Te0[s1 >> 24] ^ Te1[(s2 >> 16) & 0xff] ^ Te2[(s3 >>  8) & 0xff] ^ Te3[s0 & 0xff] ^ rk[21];					 t2 = Te0[s2 >> 24] ^ Te1[(s3 >> 16) & 0xff] ^ Te2[(s0 >>  8) & 0xff] ^ Te3[s1 & 0xff] ^ rk[22];					 t3 = Te0[s3 >> 24] ^ Te1[(s0 >> 16) & 0xff] ^ Te2[(s1 >>  8) & 0xff] ^ Te3[s2 & 0xff] ^ rk[23];					 /* round 6: */					 s0 = Te0[t0 >> 24] ^ Te1[(t1 >> 16) & 0xff] ^ Te2[(t2 >>  8) & 0xff] ^ Te3[t3 & 0xff] ^ rk[24];					 s1 = Te0[t1 >> 24] ^ Te1[(t2 >> 16) & 0xff] ^ Te2[(t3 >>  8) & 0xff] ^ Te3[t0 & 0xff] ^ rk[25];					 s2 = Te0[t2 >> 24] ^ Te1[(t3 >> 16) & 0xff] ^ Te2[(t0 >>  8) & 0xff] ^ Te3[t1 & 0xff] ^ rk[26];					 s3 = Te0[t3 >> 24] ^ Te1[(t0 >> 16) & 0xff] ^ Te2[(t1 >>  8) & 0xff] ^ Te3[t2 & 0xff] ^ rk[27];					 /* round 7: */					 t0 = Te0[s0 >> 24] ^ Te1[(s1 >> 16) & 0xff] ^ Te2[(s2 >>  8) & 0xff] ^ Te3[s3 & 0xff] ^ rk[28];					 t1 = Te0[s1 >> 24] ^ Te1[(s2 >> 16) & 0xff] ^ Te2[(s3 >>  8) & 0xff] ^ Te3[s0 & 0xff] ^ rk[29];					 t2 = Te0[s2 >> 24] ^ Te1[(s3 >> 16) & 0xff] ^ Te2[(s0 >>  8) & 0xff] ^ Te3[s1 & 0xff] ^ rk[30];					 t3 = Te0[s3 >> 24] ^ Te1[(s0 >> 16) & 0xff] ^ Te2[(s1 >>  8) & 0xff] ^ Te3[s2 & 0xff] ^ rk[31];					 /* round 8: */					 s0 = Te0[t0 >> 24] ^ Te1[(t1 >> 16) & 0xff] ^ Te2[(t2 >>  8) & 0xff] ^ Te3[t3 & 0xff] ^ rk[32];					 s1 = Te0[t1 >> 24] ^ Te1[(t2 >> 16) & 0xff] ^ Te2[(t3 >>  8) & 0xff] ^ Te3[t0 & 0xff] ^ rk[33];					 s2 = Te0[t2 >> 24] ^ Te1[(t3 >> 16) & 0xff] ^ Te2[(t0 >>  8) & 0xff] ^ Te3[t1 & 0xff] ^ rk[34];					 s3 = Te0[t3 >> 24] ^ Te1[(t0 >> 16) & 0xff] ^ Te2[(t1 >>  8) & 0xff] ^ Te3[t2 & 0xff] ^ rk[35];					 /* round 9: */					 t0 = Te0[s0 >> 24] ^ Te1[(s1 >> 16) & 0xff] ^ Te2[(s2 >>  8) & 0xff] ^ Te3[s3 & 0xff] ^ rk[36];					 t1 = Te0[s1 >> 24] ^ Te1[(s2 >> 16) & 0xff] ^ Te2[(s3 >>  8) & 0xff] ^ Te3[s0 & 0xff] ^ rk[37];					 t2 = Te0[s2 >> 24] ^ Te1[(s3 >> 16) & 0xff] ^ Te2[(s0 >>  8) & 0xff] ^ Te3[s1 & 0xff] ^ rk[38];					 t3 = Te0[s3 >> 24] ^ Te1[(s0 >> 16) & 0xff] ^ Te2[(s1 >>  8) & 0xff] ^ Te3[s2 & 0xff] ^ rk[39];					 if (key->rounds > 10) {						 /* round 10: */						 s0 = Te0[t0 >> 24] ^ Te1[(t1 >> 16) & 0xff] ^ Te2[(t2 >>  8) & 0xff] ^ Te3[t3 & 0xff] ^ rk[40];						 s1 = Te0[t1 >> 24] ^ Te1[(t2 >> 16) & 0xff] ^ Te2[(t3 >>  8) & 0xff] ^ Te3[t0 & 0xff] ^ rk[41];						 s2 = Te0[t2 >> 24] ^ Te1[(t3 >> 16) & 0xff] ^ Te2[(t0 >>  8) & 0xff] ^ Te3[t1 & 0xff] ^ rk[42];						 s3 = Te0[t3 >> 24] ^ Te1[(t0 >> 16) & 0xff] ^ Te2[(t1 >>  8) & 0xff] ^ Te3[t2 & 0xff] ^ rk[43];						 /* round 11: */						 t0 = Te0[s0 >> 24] ^ Te1[(s1 >> 16) & 0xff] ^ Te2[(s2 >>  8) & 0xff] ^ Te3[s3 & 0xff] ^ rk[44];						 t1 = Te0[s1 >> 24] ^ Te1[(s2 >> 16) & 0xff] ^ Te2[(s3 >>  8) & 0xff] ^ Te3[s0 & 0xff] ^ rk[45];						 t2 = Te0[s2 >> 24] ^ Te1[(s3 >> 16) & 0xff] ^ Te2[(s0 >>  8) & 0xff] ^ Te3[s1 & 0xff] ^ rk[46];						 t3 = Te0[s3 >> 24] ^ Te1[(s0 >> 16) & 0xff] ^ Te2[(s1 >>  8) & 0xff] ^ Te3[s2 & 0xff] ^ rk[47];						 if (key->rounds > 12) {							 /* round 12: */							 s0 = Te0[t0 >> 24] ^ Te1[(t1 >> 16) & 0xff] ^ Te2[(t2 >>  8) & 0xff] ^ Te3[t3 & 0xff] ^ rk[48];							 s1 = Te0[t1 >> 24] ^ Te1[(t2 >> 16) & 0xff] ^ Te2[(t3 >>  8) & 0xff] ^ Te3[t0 & 0xff] ^ rk[49];							 s2 = Te0[t2 >> 24] ^ Te1[(t3 >> 16) & 0xff] ^ Te2[(t0 >>  8) & 0xff] ^ Te3[t1 & 0xff] ^ rk[50];							 s3 = Te0[t3 >> 24] ^ Te1[(t0 >> 16) & 0xff] ^ Te2[(t1 >>  8) & 0xff] ^ Te3[t2 & 0xff] ^ rk[51];							 /* round 13: */							 t0 = Te0[s0 >> 24] ^ Te1[(s1 >> 16) & 0xff] ^ Te2[(s2 >>  8) & 0xff] ^ Te3[s3 & 0xff] ^ rk[52];							 t1 = Te0[s1 >> 24] ^ Te1[(s2 >> 16) & 0xff] ^ Te2[(s3 >>  8) & 0xff] ^ Te3[s0 & 0xff] ^ rk[53];							 t2 = Te0[s2 >> 24] ^ Te1[(s3 >> 16) & 0xff] ^ Te2[(s0 >>  8) & 0xff] ^ Te3[s1 & 0xff] ^ rk[54];							 t3 = Te0[s3 >> 24] ^ Te1[(s0 >> 16) & 0xff] ^ Te2[(s1 >>  8) & 0xff] ^ Te3[s2 & 0xff] ^ rk[55];						 }					 }					 rk += key->rounds << 2;#else  /* !FULL_UNROLL */					 /*					 * Nr - 1 full rounds:					 */					 r = key->rounds >> 1;					 for (;;) {						 t0 =							 Te0[(s0 >> 24)       ] ^							 Te1[(s1 >> 16) & 0xff] ^							 Te2[(s2 >>  8) & 0xff] ^							 Te3[(s3      ) & 0xff] ^							 rk[4];						 t1 =							 Te0[(s1 >> 24)       ] ^							 Te1[(s2 >> 16) & 0xff] ^