contrib/hostapd/src/crypto/aes_i.h

   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
  17 extern const u32 Te0[256];
  18 extern const u32 Te1[256];
  19 extern const u32 Te2[256];
  20 extern const u32 Te3[256];
  21 extern const u32 Te4[256];
  22 extern const u32 Td0[256];
  23 extern const u32 Td1[256];
  24 extern const u32 Td2[256];
  25 extern const u32 Td3[256];
  26 extern const u32 Td4[256];
  27 extern const u32 rcon[10];
  28 extern const u8 Td4s[256];
  29 extern 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
  70 static 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
 123 int rijndaelKeySetupEnc(u32 rk[], const u8 cipherKey[], int keyBits);
 124
 125 #endif /* AES_I_H */