18d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt/* 28d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * AES (Rijndael) cipher 39bce59c7fef20e34a05f04d1e33a4076083dca0cDmitry Shmidt * Copyright (c) 2003-2012, Jouni Malinen <j@w1.fi> 48d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt * 5c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt * This software may be distributed under the terms of the BSD license. 6c5ec7f57ead87efa365800228aa0b09a12d9e6c4Dmitry Shmidt * See README for more details. 78d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt */ 88d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 98d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#ifndef AES_I_H 108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define AES_I_H 118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#include "aes.h" 138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt/* #define FULL_UNROLL */ 158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define AES_SMALL_TABLES 168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtextern const u32 Te0[256]; 188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtextern const u32 Te1[256]; 198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtextern const u32 Te2[256]; 208d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtextern const u32 Te3[256]; 218d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtextern const u32 Te4[256]; 228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtextern const u32 Td0[256]; 238d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtextern const u32 Td1[256]; 248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtextern const u32 Td2[256]; 258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtextern const u32 Td3[256]; 268d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtextern const u32 Td4[256]; 278d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtextern const u32 rcon[10]; 288d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtextern const u8 Td4s[256]; 298d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtextern const u8 rcons[10]; 308d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 318d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#ifndef AES_SMALL_TABLES 328d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 338d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define RCON(i) rcon[(i)] 348d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 358d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE0(i) Te0[((i) >> 24) & 0xff] 368d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE1(i) Te1[((i) >> 16) & 0xff] 378d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE2(i) Te2[((i) >> 8) & 0xff] 388d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE3(i) Te3[(i) & 0xff] 398d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE41(i) (Te4[((i) >> 24) & 0xff] & 0xff000000) 408d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE42(i) (Te4[((i) >> 16) & 0xff] & 0x00ff0000) 418d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE43(i) (Te4[((i) >> 8) & 0xff] & 0x0000ff00) 428d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE44(i) (Te4[(i) & 0xff] & 0x000000ff) 438d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE421(i) (Te4[((i) >> 16) & 0xff] & 0xff000000) 448d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE432(i) (Te4[((i) >> 8) & 0xff] & 0x00ff0000) 458d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE443(i) (Te4[(i) & 0xff] & 0x0000ff00) 468d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE414(i) (Te4[((i) >> 24) & 0xff] & 0x000000ff) 479bce59c7fef20e34a05f04d1e33a4076083dca0cDmitry Shmidt#define TE411(i) (Te4[((i) >> 24) & 0xff] & 0xff000000) 489bce59c7fef20e34a05f04d1e33a4076083dca0cDmitry Shmidt#define TE422(i) (Te4[((i) >> 16) & 0xff] & 0x00ff0000) 499bce59c7fef20e34a05f04d1e33a4076083dca0cDmitry Shmidt#define TE433(i) (Te4[((i) >> 8) & 0xff] & 0x0000ff00) 509bce59c7fef20e34a05f04d1e33a4076083dca0cDmitry Shmidt#define TE444(i) (Te4[(i) & 0xff] & 0x000000ff) 518d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE4(i) (Te4[(i)] & 0x000000ff) 528d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 538d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD0(i) Td0[((i) >> 24) & 0xff] 548d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD1(i) Td1[((i) >> 16) & 0xff] 558d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD2(i) Td2[((i) >> 8) & 0xff] 568d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD3(i) Td3[(i) & 0xff] 578d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD41(i) (Td4[((i) >> 24) & 0xff] & 0xff000000) 588d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD42(i) (Td4[((i) >> 16) & 0xff] & 0x00ff0000) 598d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD43(i) (Td4[((i) >> 8) & 0xff] & 0x0000ff00) 608d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD44(i) (Td4[(i) & 0xff] & 0x000000ff) 618d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD0_(i) Td0[(i) & 0xff] 628d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD1_(i) Td1[(i) & 0xff] 638d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD2_(i) Td2[(i) & 0xff] 648d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD3_(i) Td3[(i) & 0xff] 658d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 668d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#else /* AES_SMALL_TABLES */ 678d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 688d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define RCON(i) (rcons[(i)] << 24) 698d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 708d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidtstatic inline u32 rotr(u32 val, int bits) 718d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt{ 728d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt return (val >> bits) | (val << (32 - bits)); 738d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt} 748d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 758d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE0(i) Te0[((i) >> 24) & 0xff] 768d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE1(i) rotr(Te0[((i) >> 16) & 0xff], 8) 778d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE2(i) rotr(Te0[((i) >> 8) & 0xff], 16) 788d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE3(i) rotr(Te0[(i) & 0xff], 24) 798d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE41(i) ((Te0[((i) >> 24) & 0xff] << 8) & 0xff000000) 808d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE42(i) (Te0[((i) >> 16) & 0xff] & 0x00ff0000) 818d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE43(i) (Te0[((i) >> 8) & 0xff] & 0x0000ff00) 828d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE44(i) ((Te0[(i) & 0xff] >> 8) & 0x000000ff) 838d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE421(i) ((Te0[((i) >> 16) & 0xff] << 8) & 0xff000000) 848d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE432(i) (Te0[((i) >> 8) & 0xff] & 0x00ff0000) 858d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE443(i) (Te0[(i) & 0xff] & 0x0000ff00) 868d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE414(i) ((Te0[((i) >> 24) & 0xff] >> 8) & 0x000000ff) 879bce59c7fef20e34a05f04d1e33a4076083dca0cDmitry Shmidt#define TE411(i) ((Te0[((i) >> 24) & 0xff] << 8) & 0xff000000) 889bce59c7fef20e34a05f04d1e33a4076083dca0cDmitry Shmidt#define TE422(i) (Te0[((i) >> 16) & 0xff] & 0x00ff0000) 899bce59c7fef20e34a05f04d1e33a4076083dca0cDmitry Shmidt#define TE433(i) (Te0[((i) >> 8) & 0xff] & 0x0000ff00) 909bce59c7fef20e34a05f04d1e33a4076083dca0cDmitry Shmidt#define TE444(i) ((Te0[(i) & 0xff] >> 8) & 0x000000ff) 918d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TE4(i) ((Te0[(i)] >> 8) & 0x000000ff) 928d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 938d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD0(i) Td0[((i) >> 24) & 0xff] 948d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD1(i) rotr(Td0[((i) >> 16) & 0xff], 8) 958d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD2(i) rotr(Td0[((i) >> 8) & 0xff], 16) 968d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD3(i) rotr(Td0[(i) & 0xff], 24) 978d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD41(i) (Td4s[((i) >> 24) & 0xff] << 24) 988d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD42(i) (Td4s[((i) >> 16) & 0xff] << 16) 998d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD43(i) (Td4s[((i) >> 8) & 0xff] << 8) 1008d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD44(i) (Td4s[(i) & 0xff]) 1018d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD0_(i) Td0[(i) & 0xff] 1028d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD1_(i) rotr(Td0[(i) & 0xff], 8) 1038d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD2_(i) rotr(Td0[(i) & 0xff], 16) 1048d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define TD3_(i) rotr(Td0[(i) & 0xff], 24) 1058d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 1068d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#endif /* AES_SMALL_TABLES */ 1078d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 1088d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#ifdef _MSC_VER 1098d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define SWAP(x) (_lrotl(x, 8) & 0x00ff00ff | _lrotr(x, 8) & 0xff00ff00) 1108d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define GETU32(p) SWAP(*((u32 *)(p))) 1118d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define PUTU32(ct, st) { *((u32 *)(ct)) = SWAP((st)); } 1128d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#else 1138d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define GETU32(pt) (((u32)(pt)[0] << 24) ^ ((u32)(pt)[1] << 16) ^ \ 1148d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt((u32)(pt)[2] << 8) ^ ((u32)(pt)[3])) 1158d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#define PUTU32(ct, st) { \ 1168d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt(ct)[0] = (u8)((st) >> 24); (ct)[1] = (u8)((st) >> 16); \ 1178d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt(ct)[2] = (u8)((st) >> 8); (ct)[3] = (u8)(st); } 1188d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#endif 1198d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 1209bce59c7fef20e34a05f04d1e33a4076083dca0cDmitry Shmidt#define AES_PRIV_SIZE (4 * 4 * 15 + 4) 1219bce59c7fef20e34a05f04d1e33a4076083dca0cDmitry Shmidt#define AES_PRIV_NR_POS (4 * 15) 1228d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 1239bce59c7fef20e34a05f04d1e33a4076083dca0cDmitry Shmidtint rijndaelKeySetupEnc(u32 rk[], const u8 cipherKey[], int keyBits); 1248d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt 1258d520ff1dc2da35cdca849e982051b86468016d8Dmitry Shmidt#endif /* AES_I_H */ 126