aes_i.h revision 252726
1214501Srpaulo/* 2214501Srpaulo * AES (Rijndael) cipher 3252726Srpaulo * Copyright (c) 2003-2012, Jouni Malinen <j@w1.fi> 4214501Srpaulo * 5252726Srpaulo * This software may be distributed under the terms of the BSD license. 6252726Srpaulo * See README for more details. 7214501Srpaulo */ 8214501Srpaulo 9214501Srpaulo#ifndef AES_I_H 10214501Srpaulo#define AES_I_H 11214501Srpaulo 12214501Srpaulo#include "aes.h" 13214501Srpaulo 14214501Srpaulo/* #define FULL_UNROLL */ 15214501Srpaulo#define AES_SMALL_TABLES 16214501Srpaulo 17214501Srpauloextern const u32 Te0[256]; 18214501Srpauloextern const u32 Te1[256]; 19214501Srpauloextern const u32 Te2[256]; 20214501Srpauloextern const u32 Te3[256]; 21214501Srpauloextern const u32 Te4[256]; 22214501Srpauloextern const u32 Td0[256]; 23214501Srpauloextern const u32 Td1[256]; 24214501Srpauloextern const u32 Td2[256]; 25214501Srpauloextern const u32 Td3[256]; 26214501Srpauloextern const u32 Td4[256]; 27214501Srpauloextern const u32 rcon[10]; 28214501Srpauloextern const u8 Td4s[256]; 29214501Srpauloextern const u8 rcons[10]; 30214501Srpaulo 31214501Srpaulo#ifndef AES_SMALL_TABLES 32214501Srpaulo 33214501Srpaulo#define RCON(i) rcon[(i)] 34214501Srpaulo 35214501Srpaulo#define TE0(i) Te0[((i) >> 24) & 0xff] 36214501Srpaulo#define TE1(i) Te1[((i) >> 16) & 0xff] 37214501Srpaulo#define TE2(i) Te2[((i) >> 8) & 0xff] 38214501Srpaulo#define TE3(i) Te3[(i) & 0xff] 39214501Srpaulo#define TE41(i) (Te4[((i) >> 24) & 0xff] & 0xff000000) 40214501Srpaulo#define TE42(i) (Te4[((i) >> 16) & 0xff] & 0x00ff0000) 41214501Srpaulo#define TE43(i) (Te4[((i) >> 8) & 0xff] & 0x0000ff00) 42214501Srpaulo#define TE44(i) (Te4[(i) & 0xff] & 0x000000ff) 43214501Srpaulo#define TE421(i) (Te4[((i) >> 16) & 0xff] & 0xff000000) 44214501Srpaulo#define TE432(i) (Te4[((i) >> 8) & 0xff] & 0x00ff0000) 45214501Srpaulo#define TE443(i) (Te4[(i) & 0xff] & 0x0000ff00) 46214501Srpaulo#define TE414(i) (Te4[((i) >> 24) & 0xff] & 0x000000ff) 47252726Srpaulo#define TE411(i) (Te4[((i) >> 24) & 0xff] & 0xff000000) 48252726Srpaulo#define TE422(i) (Te4[((i) >> 16) & 0xff] & 0x00ff0000) 49252726Srpaulo#define TE433(i) (Te4[((i) >> 8) & 0xff] & 0x0000ff00) 50252726Srpaulo#define TE444(i) (Te4[(i) & 0xff] & 0x000000ff) 51214501Srpaulo#define TE4(i) (Te4[(i)] & 0x000000ff) 52214501Srpaulo 53214501Srpaulo#define TD0(i) Td0[((i) >> 24) & 0xff] 54214501Srpaulo#define TD1(i) Td1[((i) >> 16) & 0xff] 55214501Srpaulo#define TD2(i) Td2[((i) >> 8) & 0xff] 56214501Srpaulo#define TD3(i) Td3[(i) & 0xff] 57214501Srpaulo#define TD41(i) (Td4[((i) >> 24) & 0xff] & 0xff000000) 58214501Srpaulo#define TD42(i) (Td4[((i) >> 16) & 0xff] & 0x00ff0000) 59214501Srpaulo#define TD43(i) (Td4[((i) >> 8) & 0xff] & 0x0000ff00) 60214501Srpaulo#define TD44(i) (Td4[(i) & 0xff] & 0x000000ff) 61214501Srpaulo#define TD0_(i) Td0[(i) & 0xff] 62214501Srpaulo#define TD1_(i) Td1[(i) & 0xff] 63214501Srpaulo#define TD2_(i) Td2[(i) & 0xff] 64214501Srpaulo#define TD3_(i) Td3[(i) & 0xff] 65214501Srpaulo 66214501Srpaulo#else /* AES_SMALL_TABLES */ 67214501Srpaulo 68214501Srpaulo#define RCON(i) (rcons[(i)] << 24) 69214501Srpaulo 70214501Srpaulostatic inline u32 rotr(u32 val, int bits) 71214501Srpaulo{ 72214501Srpaulo return (val >> bits) | (val << (32 - bits)); 73214501Srpaulo} 74214501Srpaulo 75214501Srpaulo#define TE0(i) Te0[((i) >> 24) & 0xff] 76214501Srpaulo#define TE1(i) rotr(Te0[((i) >> 16) & 0xff], 8) 77214501Srpaulo#define TE2(i) rotr(Te0[((i) >> 8) & 0xff], 16) 78214501Srpaulo#define TE3(i) rotr(Te0[(i) & 0xff], 24) 79214501Srpaulo#define TE41(i) ((Te0[((i) >> 24) & 0xff] << 8) & 0xff000000) 80214501Srpaulo#define TE42(i) (Te0[((i) >> 16) & 0xff] & 0x00ff0000) 81214501Srpaulo#define TE43(i) (Te0[((i) >> 8) & 0xff] & 0x0000ff00) 82214501Srpaulo#define TE44(i) ((Te0[(i) & 0xff] >> 8) & 0x000000ff) 83214501Srpaulo#define TE421(i) ((Te0[((i) >> 16) & 0xff] << 8) & 0xff000000) 84214501Srpaulo#define TE432(i) (Te0[((i) >> 8) & 0xff] & 0x00ff0000) 85214501Srpaulo#define TE443(i) (Te0[(i) & 0xff] & 0x0000ff00) 86214501Srpaulo#define TE414(i) ((Te0[((i) >> 24) & 0xff] >> 8) & 0x000000ff) 87252726Srpaulo#define TE411(i) ((Te0[((i) >> 24) & 0xff] << 8) & 0xff000000) 88252726Srpaulo#define TE422(i) (Te0[((i) >> 16) & 0xff] & 0x00ff0000) 89252726Srpaulo#define TE433(i) (Te0[((i) >> 8) & 0xff] & 0x0000ff00) 90252726Srpaulo#define TE444(i) ((Te0[(i) & 0xff] >> 8) & 0x000000ff) 91214501Srpaulo#define TE4(i) ((Te0[(i)] >> 8) & 0x000000ff) 92214501Srpaulo 93214501Srpaulo#define TD0(i) Td0[((i) >> 24) & 0xff] 94214501Srpaulo#define TD1(i) rotr(Td0[((i) >> 16) & 0xff], 8) 95214501Srpaulo#define TD2(i) rotr(Td0[((i) >> 8) & 0xff], 16) 96214501Srpaulo#define TD3(i) rotr(Td0[(i) & 0xff], 24) 97214501Srpaulo#define TD41(i) (Td4s[((i) >> 24) & 0xff] << 24) 98214501Srpaulo#define TD42(i) (Td4s[((i) >> 16) & 0xff] << 16) 99214501Srpaulo#define TD43(i) (Td4s[((i) >> 8) & 0xff] << 8) 100214501Srpaulo#define TD44(i) (Td4s[(i) & 0xff]) 101214501Srpaulo#define TD0_(i) Td0[(i) & 0xff] 102214501Srpaulo#define TD1_(i) rotr(Td0[(i) & 0xff], 8) 103214501Srpaulo#define TD2_(i) rotr(Td0[(i) & 0xff], 16) 104214501Srpaulo#define TD3_(i) rotr(Td0[(i) & 0xff], 24) 105214501Srpaulo 106214501Srpaulo#endif /* AES_SMALL_TABLES */ 107214501Srpaulo 108214501Srpaulo#ifdef _MSC_VER 109214501Srpaulo#define SWAP(x) (_lrotl(x, 8) & 0x00ff00ff | _lrotr(x, 8) & 0xff00ff00) 110214501Srpaulo#define GETU32(p) SWAP(*((u32 *)(p))) 111214501Srpaulo#define PUTU32(ct, st) { *((u32 *)(ct)) = SWAP((st)); } 112214501Srpaulo#else 113214501Srpaulo#define GETU32(pt) (((u32)(pt)[0] << 24) ^ ((u32)(pt)[1] << 16) ^ \ 114214501Srpaulo((u32)(pt)[2] << 8) ^ ((u32)(pt)[3])) 115214501Srpaulo#define PUTU32(ct, st) { \ 116214501Srpaulo(ct)[0] = (u8)((st) >> 24); (ct)[1] = (u8)((st) >> 16); \ 117214501Srpaulo(ct)[2] = (u8)((st) >> 8); (ct)[3] = (u8)(st); } 118214501Srpaulo#endif 119214501Srpaulo 120252726Srpaulo#define AES_PRIV_SIZE (4 * 4 * 15 + 4) 121252726Srpaulo#define AES_PRIV_NR_POS (4 * 15) 122214501Srpaulo 123252726Srpauloint rijndaelKeySetupEnc(u32 rk[], const u8 cipherKey[], int keyBits); 124214501Srpaulo 125214501Srpaulo#endif /* AES_I_H */ 126