aes_i.h revision c5ec7f57ead87efa365800228aa0b09a12d9e6c4
1/* 2 * AES (Rijndael) cipher 3 * Copyright (c) 2003-2005, Jouni Malinen <j@w1.fi> 4 * 5 * This software may be distributed under the terms of the BSD license. 6 * See README for more details. 7 */ 8 9#ifndef AES_I_H 10#define AES_I_H 11 12#include "aes.h" 13 14/* #define FULL_UNROLL */ 15#define AES_SMALL_TABLES 16 17extern const u32 Te0[256]; 18extern const u32 Te1[256]; 19extern const u32 Te2[256]; 20extern const u32 Te3[256]; 21extern const u32 Te4[256]; 22extern const u32 Td0[256]; 23extern const u32 Td1[256]; 24extern const u32 Td2[256]; 25extern const u32 Td3[256]; 26extern const u32 Td4[256]; 27extern const u32 rcon[10]; 28extern const u8 Td4s[256]; 29extern const u8 rcons[10]; 30 31#ifndef AES_SMALL_TABLES 32 33#define RCON(i) rcon[(i)] 34 35#define TE0(i) Te0[((i) >> 24) & 0xff] 36#define TE1(i) Te1[((i) >> 16) & 0xff] 37#define TE2(i) Te2[((i) >> 8) & 0xff] 38#define TE3(i) Te3[(i) & 0xff] 39#define TE41(i) (Te4[((i) >> 24) & 0xff] & 0xff000000) 40#define TE42(i) (Te4[((i) >> 16) & 0xff] & 0x00ff0000) 41#define TE43(i) (Te4[((i) >> 8) & 0xff] & 0x0000ff00) 42#define TE44(i) (Te4[(i) & 0xff] & 0x000000ff) 43#define TE421(i) (Te4[((i) >> 16) & 0xff] & 0xff000000) 44#define TE432(i) (Te4[((i) >> 8) & 0xff] & 0x00ff0000) 45#define TE443(i) (Te4[(i) & 0xff] & 0x0000ff00) 46#define TE414(i) (Te4[((i) >> 24) & 0xff] & 0x000000ff) 47#define TE4(i) (Te4[(i)] & 0x000000ff) 48 49#define TD0(i) Td0[((i) >> 24) & 0xff] 50#define TD1(i) Td1[((i) >> 16) & 0xff] 51#define TD2(i) Td2[((i) >> 8) & 0xff] 52#define TD3(i) Td3[(i) & 0xff] 53#define TD41(i) (Td4[((i) >> 24) & 0xff] & 0xff000000) 54#define TD42(i) (Td4[((i) >> 16) & 0xff] & 0x00ff0000) 55#define TD43(i) (Td4[((i) >> 8) & 0xff] & 0x0000ff00) 56#define TD44(i) (Td4[(i) & 0xff] & 0x000000ff) 57#define TD0_(i) Td0[(i) & 0xff] 58#define TD1_(i) Td1[(i) & 0xff] 59#define TD2_(i) Td2[(i) & 0xff] 60#define TD3_(i) Td3[(i) & 0xff] 61 62#else /* AES_SMALL_TABLES */ 63 64#define RCON(i) (rcons[(i)] << 24) 65 66static inline u32 rotr(u32 val, int bits) 67{ 68 return (val >> bits) | (val << (32 - bits)); 69} 70 71#define TE0(i) Te0[((i) >> 24) & 0xff] 72#define TE1(i) rotr(Te0[((i) >> 16) & 0xff], 8) 73#define TE2(i) rotr(Te0[((i) >> 8) & 0xff], 16) 74#define TE3(i) rotr(Te0[(i) & 0xff], 24) 75#define TE41(i) ((Te0[((i) >> 24) & 0xff] << 8) & 0xff000000) 76#define TE42(i) (Te0[((i) >> 16) & 0xff] & 0x00ff0000) 77#define TE43(i) (Te0[((i) >> 8) & 0xff] & 0x0000ff00) 78#define TE44(i) ((Te0[(i) & 0xff] >> 8) & 0x000000ff) 79#define TE421(i) ((Te0[((i) >> 16) & 0xff] << 8) & 0xff000000) 80#define TE432(i) (Te0[((i) >> 8) & 0xff] & 0x00ff0000) 81#define TE443(i) (Te0[(i) & 0xff] & 0x0000ff00) 82#define TE414(i) ((Te0[((i) >> 24) & 0xff] >> 8) & 0x000000ff) 83#define TE4(i) ((Te0[(i)] >> 8) & 0x000000ff) 84 85#define TD0(i) Td0[((i) >> 24) & 0xff] 86#define TD1(i) rotr(Td0[((i) >> 16) & 0xff], 8) 87#define TD2(i) rotr(Td0[((i) >> 8) & 0xff], 16) 88#define TD3(i) rotr(Td0[(i) & 0xff], 24) 89#define TD41(i) (Td4s[((i) >> 24) & 0xff] << 24) 90#define TD42(i) (Td4s[((i) >> 16) & 0xff] << 16) 91#define TD43(i) (Td4s[((i) >> 8) & 0xff] << 8) 92#define TD44(i) (Td4s[(i) & 0xff]) 93#define TD0_(i) Td0[(i) & 0xff] 94#define TD1_(i) rotr(Td0[(i) & 0xff], 8) 95#define TD2_(i) rotr(Td0[(i) & 0xff], 16) 96#define TD3_(i) rotr(Td0[(i) & 0xff], 24) 97 98#endif /* AES_SMALL_TABLES */ 99 100#ifdef _MSC_VER 101#define SWAP(x) (_lrotl(x, 8) & 0x00ff00ff | _lrotr(x, 8) & 0xff00ff00) 102#define GETU32(p) SWAP(*((u32 *)(p))) 103#define PUTU32(ct, st) { *((u32 *)(ct)) = SWAP((st)); } 104#else 105#define GETU32(pt) (((u32)(pt)[0] << 24) ^ ((u32)(pt)[1] << 16) ^ \ 106((u32)(pt)[2] << 8) ^ ((u32)(pt)[3])) 107#define PUTU32(ct, st) { \ 108(ct)[0] = (u8)((st) >> 24); (ct)[1] = (u8)((st) >> 16); \ 109(ct)[2] = (u8)((st) >> 8); (ct)[3] = (u8)(st); } 110#endif 111 112#define AES_PRIV_SIZE (4 * 44) 113 114void rijndaelKeySetupEnc(u32 rk[/*44*/], const u8 cipherKey[]); 115 116#endif /* AES_I_H */ 117