aes_i.h revision 214501
1216115Slstewart/* 2216115Slstewart * AES (Rijndael) cipher 3216115Slstewart * Copyright (c) 2003-2005, Jouni Malinen <j@w1.fi> 4216115Slstewart * 5216115Slstewart * This program is free software; you can redistribute it and/or modify 6216115Slstewart * it under the terms of the GNU General Public License version 2 as 7216115Slstewart * published by the Free Software Foundation. 8216115Slstewart * 9220560Slstewart * Alternatively, this software may be distributed under the terms of BSD 10220560Slstewart * license. 11220560Slstewart * 12216115Slstewart * See README and COPYING for more details. 13216115Slstewart */ 14216115Slstewart 15216115Slstewart#ifndef AES_I_H 16216115Slstewart#define AES_I_H 17216115Slstewart 18216115Slstewart#include "aes.h" 19216115Slstewart 20216115Slstewart/* #define FULL_UNROLL */ 21216115Slstewart#define AES_SMALL_TABLES 22216115Slstewart 23216115Slstewartextern const u32 Te0[256]; 24216115Slstewartextern const u32 Te1[256]; 25216115Slstewartextern const u32 Te2[256]; 26216115Slstewartextern const u32 Te3[256]; 27216115Slstewartextern const u32 Te4[256]; 28216115Slstewartextern const u32 Td0[256]; 29216115Slstewartextern const u32 Td1[256]; 30216115Slstewartextern const u32 Td2[256]; 31216115Slstewartextern const u32 Td3[256]; 32216115Slstewartextern const u32 Td4[256]; 33216115Slstewartextern const u32 rcon[10]; 34216115Slstewartextern const u8 Td4s[256]; 35216115Slstewartextern const u8 rcons[10]; 36216115Slstewart 37216115Slstewart#ifndef AES_SMALL_TABLES 38216115Slstewart 39216115Slstewart#define RCON(i) rcon[(i)] 40216115Slstewart 41216115Slstewart#define TE0(i) Te0[((i) >> 24) & 0xff] 42216115Slstewart#define TE1(i) Te1[((i) >> 16) & 0xff] 43220560Slstewart#define TE2(i) Te2[((i) >> 8) & 0xff] 44220560Slstewart#define TE3(i) Te3[(i) & 0xff] 45220560Slstewart#define TE41(i) (Te4[((i) >> 24) & 0xff] & 0xff000000) 46220560Slstewart#define TE42(i) (Te4[((i) >> 16) & 0xff] & 0x00ff0000) 47216115Slstewart#define TE43(i) (Te4[((i) >> 8) & 0xff] & 0x0000ff00) 48216115Slstewart#define TE44(i) (Te4[(i) & 0xff] & 0x000000ff) 49216115Slstewart#define TE421(i) (Te4[((i) >> 16) & 0xff] & 0xff000000) 50216115Slstewart#define TE432(i) (Te4[((i) >> 8) & 0xff] & 0x00ff0000) 51216115Slstewart#define TE443(i) (Te4[(i) & 0xff] & 0x0000ff00) 52216115Slstewart#define TE414(i) (Te4[((i) >> 24) & 0xff] & 0x000000ff) 53216115Slstewart#define TE4(i) (Te4[(i)] & 0x000000ff) 54216115Slstewart 55216115Slstewart#define TD0(i) Td0[((i) >> 24) & 0xff] 56216115Slstewart#define TD1(i) Td1[((i) >> 16) & 0xff] 57216115Slstewart#define TD2(i) Td2[((i) >> 8) & 0xff] 58216115Slstewart#define TD3(i) Td3[(i) & 0xff] 59216115Slstewart#define TD41(i) (Td4[((i) >> 24) & 0xff] & 0xff000000) 60216115Slstewart#define TD42(i) (Td4[((i) >> 16) & 0xff] & 0x00ff0000) 61216115Slstewart#define TD43(i) (Td4[((i) >> 8) & 0xff] & 0x0000ff00) 62216115Slstewart#define TD44(i) (Td4[(i) & 0xff] & 0x000000ff) 63216115Slstewart#define TD0_(i) Td0[(i) & 0xff] 64216115Slstewart#define TD1_(i) Td1[(i) & 0xff] 65216115Slstewart#define TD2_(i) Td2[(i) & 0xff] 66216115Slstewart#define TD3_(i) Td3[(i) & 0xff] 67216115Slstewart 68216115Slstewart#else /* AES_SMALL_TABLES */ 69216115Slstewart 70216115Slstewart#define RCON(i) (rcons[(i)] << 24) 71216115Slstewart 72216115Slstewartstatic inline u32 rotr(u32 val, int bits) 73216115Slstewart{ 74216115Slstewart return (val >> bits) | (val << (32 - bits)); 75216115Slstewart} 76216115Slstewart 77216115Slstewart#define TE0(i) Te0[((i) >> 24) & 0xff] 78216115Slstewart#define TE1(i) rotr(Te0[((i) >> 16) & 0xff], 8) 79216115Slstewart#define TE2(i) rotr(Te0[((i) >> 8) & 0xff], 16) 80216115Slstewart#define TE3(i) rotr(Te0[(i) & 0xff], 24) 81216115Slstewart#define TE41(i) ((Te0[((i) >> 24) & 0xff] << 8) & 0xff000000) 82216115Slstewart#define TE42(i) (Te0[((i) >> 16) & 0xff] & 0x00ff0000) 83216115Slstewart#define TE43(i) (Te0[((i) >> 8) & 0xff] & 0x0000ff00) 84216115Slstewart#define TE44(i) ((Te0[(i) & 0xff] >> 8) & 0x000000ff) 85216115Slstewart#define TE421(i) ((Te0[((i) >> 16) & 0xff] << 8) & 0xff000000) 86216115Slstewart#define TE432(i) (Te0[((i) >> 8) & 0xff] & 0x00ff0000) 87216115Slstewart#define TE443(i) (Te0[(i) & 0xff] & 0x0000ff00) 88216115Slstewart#define TE414(i) ((Te0[((i) >> 24) & 0xff] >> 8) & 0x000000ff) 89216115Slstewart#define TE4(i) ((Te0[(i)] >> 8) & 0x000000ff) 90216115Slstewart 91216115Slstewart#define TD0(i) Td0[((i) >> 24) & 0xff] 92216115Slstewart#define TD1(i) rotr(Td0[((i) >> 16) & 0xff], 8) 93216115Slstewart#define TD2(i) rotr(Td0[((i) >> 8) & 0xff], 16) 94216115Slstewart#define TD3(i) rotr(Td0[(i) & 0xff], 24) 95216115Slstewart#define TD41(i) (Td4s[((i) >> 24) & 0xff] << 24) 96216115Slstewart#define TD42(i) (Td4s[((i) >> 16) & 0xff] << 16) 97216115Slstewart#define TD43(i) (Td4s[((i) >> 8) & 0xff] << 8) 98216115Slstewart#define TD44(i) (Td4s[(i) & 0xff]) 99216115Slstewart#define TD0_(i) Td0[(i) & 0xff] 100216115Slstewart#define TD1_(i) rotr(Td0[(i) & 0xff], 8) 101216115Slstewart#define TD2_(i) rotr(Td0[(i) & 0xff], 16) 102216115Slstewart#define TD3_(i) rotr(Td0[(i) & 0xff], 24) 103216115Slstewart 104216115Slstewart#endif /* AES_SMALL_TABLES */ 105216115Slstewart 106216115Slstewart#ifdef _MSC_VER 107216115Slstewart#define SWAP(x) (_lrotl(x, 8) & 0x00ff00ff | _lrotr(x, 8) & 0xff00ff00) 108216115Slstewart#define GETU32(p) SWAP(*((u32 *)(p))) 109216115Slstewart#define PUTU32(ct, st) { *((u32 *)(ct)) = SWAP((st)); } 110216115Slstewart#else 111216115Slstewart#define GETU32(pt) (((u32)(pt)[0] << 24) ^ ((u32)(pt)[1] << 16) ^ \ 112216115Slstewart((u32)(pt)[2] << 8) ^ ((u32)(pt)[3])) 113216115Slstewart#define PUTU32(ct, st) { \ 114216115Slstewart(ct)[0] = (u8)((st) >> 24); (ct)[1] = (u8)((st) >> 16); \ 115216115Slstewart(ct)[2] = (u8)((st) >> 8); (ct)[3] = (u8)(st); } 116216115Slstewart#endif 117216115Slstewart 118216115Slstewart#define AES_PRIV_SIZE (4 * 44) 119216115Slstewart 120216115Slstewartvoid rijndaelKeySetupEnc(u32 rk[/*44*/], const u8 cipherKey[]); 121216115Slstewart 122216115Slstewart#endif /* AES_I_H */ 123216115Slstewart