1/*
2 * AES (Rijndael) cipher
3 * Copyright (c) 2003-2012, 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 TE411(i) (Te4[((i) >> 24) & 0xff] & 0xff000000)
48#define TE422(i) (Te4[((i) >> 16) & 0xff] & 0x00ff0000)
49#define TE433(i) (Te4[((i) >> 8) & 0xff] & 0x0000ff00)
50#define TE444(i) (Te4[(i) & 0xff] & 0x000000ff)
51#define TE4(i) (Te4[(i)] & 0x000000ff)
52
53#define TD0(i) Td0[((i) >> 24) & 0xff]
54#define TD1(i) Td1[((i) >> 16) & 0xff]
55#define TD2(i) Td2[((i) >> 8) & 0xff]
56#define TD3(i) Td3[(i) & 0xff]
57#define TD41(i) (Td4[((i) >> 24) & 0xff] & 0xff000000)
58#define TD42(i) (Td4[((i) >> 16) & 0xff] & 0x00ff0000)
59#define TD43(i) (Td4[((i) >> 8) & 0xff] & 0x0000ff00)
60#define TD44(i) (Td4[(i) & 0xff] & 0x000000ff)
61#define TD0_(i) Td0[(i) & 0xff]
62#define TD1_(i) Td1[(i) & 0xff]
63#define TD2_(i) Td2[(i) & 0xff]
64#define TD3_(i) Td3[(i) & 0xff]
65
66#else /* AES_SMALL_TABLES */
67
68#define RCON(i) (rcons[(i)] << 24)
69
70static inline u32 rotr(u32 val, int bits)
71{
72	return (val >> bits) | (val << (32 - bits));
73}
74
75#define TE0(i) Te0[((i) >> 24) & 0xff]
76#define TE1(i) rotr(Te0[((i) >> 16) & 0xff], 8)
77#define TE2(i) rotr(Te0[((i) >> 8) & 0xff], 16)
78#define TE3(i) rotr(Te0[(i) & 0xff], 24)
79#define TE41(i) ((Te0[((i) >> 24) & 0xff] << 8) & 0xff000000)
80#define TE42(i) (Te0[((i) >> 16) & 0xff] & 0x00ff0000)
81#define TE43(i) (Te0[((i) >> 8) & 0xff] & 0x0000ff00)
82#define TE44(i) ((Te0[(i) & 0xff] >> 8) & 0x000000ff)
83#define TE421(i) ((Te0[((i) >> 16) & 0xff] << 8) & 0xff000000)
84#define TE432(i) (Te0[((i) >> 8) & 0xff] & 0x00ff0000)
85#define TE443(i) (Te0[(i) & 0xff] & 0x0000ff00)
86#define TE414(i) ((Te0[((i) >> 24) & 0xff] >> 8) & 0x000000ff)
87#define TE411(i) ((Te0[((i) >> 24) & 0xff] << 8) & 0xff000000)
88#define TE422(i) (Te0[((i) >> 16) & 0xff] & 0x00ff0000)
89#define TE433(i) (Te0[((i) >> 8) & 0xff] & 0x0000ff00)
90#define TE444(i) ((Te0[(i) & 0xff] >> 8) & 0x000000ff)
91#define TE4(i) ((Te0[(i)] >> 8) & 0x000000ff)
92
93#define TD0(i) Td0[((i) >> 24) & 0xff]
94#define TD1(i) rotr(Td0[((i) >> 16) & 0xff], 8)
95#define TD2(i) rotr(Td0[((i) >> 8) & 0xff], 16)
96#define TD3(i) rotr(Td0[(i) & 0xff], 24)
97#define TD41(i) (Td4s[((i) >> 24) & 0xff] << 24)
98#define TD42(i) (Td4s[((i) >> 16) & 0xff] << 16)
99#define TD43(i) (Td4s[((i) >> 8) & 0xff] << 8)
100#define TD44(i) (Td4s[(i) & 0xff])
101#define TD0_(i) Td0[(i) & 0xff]
102#define TD1_(i) rotr(Td0[(i) & 0xff], 8)
103#define TD2_(i) rotr(Td0[(i) & 0xff], 16)
104#define TD3_(i) rotr(Td0[(i) & 0xff], 24)
105
106#endif /* AES_SMALL_TABLES */
107
108#ifdef _MSC_VER
109#define SWAP(x) (_lrotl(x, 8) & 0x00ff00ff | _lrotr(x, 8) & 0xff00ff00)
110#define GETU32(p) SWAP(*((u32 *)(p)))
111#define PUTU32(ct, st) { *((u32 *)(ct)) = SWAP((st)); }
112#else
113#define GETU32(pt) (((u32)(pt)[0] << 24) ^ ((u32)(pt)[1] << 16) ^ \
114((u32)(pt)[2] <<  8) ^ ((u32)(pt)[3]))
115#define PUTU32(ct, st) { \
116(ct)[0] = (u8)((st) >> 24); (ct)[1] = (u8)((st) >> 16); \
117(ct)[2] = (u8)((st) >>  8); (ct)[3] = (u8)(st); }
118#endif
119
120#define AES_PRIV_SIZE (4 * 4 * 15 + 4)
121#define AES_PRIV_NR_POS (4 * 15)
122
123int rijndaelKeySetupEnc(u32 rk[], const u8 cipherKey[], int keyBits);
124
125#endif /* AES_I_H */
126