|
| 1 | +/* |
| 2 | +--------------------------------------------------------------------------- |
| 3 | +Copyright (c) 1998-2013, Brian Gladman, Worcester, UK. All rights reserved. |
| 4 | +
|
| 5 | +The redistribution and use of this software (with or without changes) |
| 6 | +is allowed without the payment of fees or royalties provided that: |
| 7 | +
|
| 8 | + source code distributions include the above copyright notice, this |
| 9 | + list of conditions and the following disclaimer; |
| 10 | +
|
| 11 | + binary distributions include the above copyright notice, this list |
| 12 | + of conditions and the following disclaimer in their documentation. |
| 13 | +
|
| 14 | +This software is provided 'as is' with no explicit or implied warranties |
| 15 | +in respect of its operation, including, but not limited to, correctness |
| 16 | +and fitness for purpose. |
| 17 | +--------------------------------------------------------------------------- |
| 18 | +Issue Date: 20/12/2007 |
| 19 | +*/ |
| 20 | + |
| 21 | +#include <stdint.h> |
| 22 | + |
| 23 | +#if defined(__cplusplus) |
| 24 | +extern "C" |
| 25 | +{ |
| 26 | +#endif |
| 27 | + |
| 28 | +#define TABLE_ALIGN 32 |
| 29 | +#define WPOLY 0x011b |
| 30 | +#define N_COLS 4 |
| 31 | +#define AES_BLOCK_SIZE 16 |
| 32 | +#define RC_LENGTH (5 * (AES_BLOCK_SIZE / 4 - 2)) |
| 33 | + |
| 34 | +#if defined(_MSC_VER) |
| 35 | +#define ALIGN __declspec(align(TABLE_ALIGN)) |
| 36 | +#elif defined(__GNUC__) |
| 37 | +#define ALIGN __attribute__ ((aligned(16))) |
| 38 | +#else |
| 39 | +#define ALIGN |
| 40 | +#endif |
| 41 | + |
| 42 | +#define rf1(r,c) (r) |
| 43 | +#define word_in(x,c) (*((uint32_t*)(x)+(c))) |
| 44 | +#define word_out(x,c,v) (*((uint32_t*)(x)+(c)) = (v)) |
| 45 | + |
| 46 | +#define s(x,c) x[c] |
| 47 | +#define si(y,x,c) (s(y,c) = word_in(x, c)) |
| 48 | +#define so(y,x,c) word_out(y, c, s(x,c)) |
| 49 | +#define state_in(y,x) si(y,x,0); si(y,x,1); si(y,x,2); si(y,x,3) |
| 50 | +#define state_out(y,x) so(y,x,0); so(y,x,1); so(y,x,2); so(y,x,3) |
| 51 | +#define round(rm,y,x,k) rm(y,x,k,0); rm(y,x,k,1); rm(y,x,k,2); rm(y,x,k,3) |
| 52 | +#define to_byte(x) ((x) & 0xff) |
| 53 | +#define bval(x,n) to_byte((x) >> (8 * (n))) |
| 54 | + |
| 55 | +#define fwd_var(x,r,c)\ |
| 56 | + ( r == 0 ? ( c == 0 ? s(x,0) : c == 1 ? s(x,1) : c == 2 ? s(x,2) : s(x,3))\ |
| 57 | + : r == 1 ? ( c == 0 ? s(x,1) : c == 1 ? s(x,2) : c == 2 ? s(x,3) : s(x,0))\ |
| 58 | + : r == 2 ? ( c == 0 ? s(x,2) : c == 1 ? s(x,3) : c == 2 ? s(x,0) : s(x,1))\ |
| 59 | + : ( c == 0 ? s(x,3) : c == 1 ? s(x,0) : c == 2 ? s(x,1) : s(x,2))) |
| 60 | + |
| 61 | +#define fwd_rnd(y,x,k,c) (s(y,c) = (k)[c] ^ four_tables(x,t_use(f,n),fwd_var,rf1,c)) |
| 62 | + |
| 63 | +#define sb_data(w) {\ |
| 64 | + w(0x63), w(0x7c), w(0x77), w(0x7b), w(0xf2), w(0x6b), w(0x6f), w(0xc5),\ |
| 65 | + w(0x30), w(0x01), w(0x67), w(0x2b), w(0xfe), w(0xd7), w(0xab), w(0x76),\ |
| 66 | + w(0xca), w(0x82), w(0xc9), w(0x7d), w(0xfa), w(0x59), w(0x47), w(0xf0),\ |
| 67 | + w(0xad), w(0xd4), w(0xa2), w(0xaf), w(0x9c), w(0xa4), w(0x72), w(0xc0),\ |
| 68 | + w(0xb7), w(0xfd), w(0x93), w(0x26), w(0x36), w(0x3f), w(0xf7), w(0xcc),\ |
| 69 | + w(0x34), w(0xa5), w(0xe5), w(0xf1), w(0x71), w(0xd8), w(0x31), w(0x15),\ |
| 70 | + w(0x04), w(0xc7), w(0x23), w(0xc3), w(0x18), w(0x96), w(0x05), w(0x9a),\ |
| 71 | + w(0x07), w(0x12), w(0x80), w(0xe2), w(0xeb), w(0x27), w(0xb2), w(0x75),\ |
| 72 | + w(0x09), w(0x83), w(0x2c), w(0x1a), w(0x1b), w(0x6e), w(0x5a), w(0xa0),\ |
| 73 | + w(0x52), w(0x3b), w(0xd6), w(0xb3), w(0x29), w(0xe3), w(0x2f), w(0x84),\ |
| 74 | + w(0x53), w(0xd1), w(0x00), w(0xed), w(0x20), w(0xfc), w(0xb1), w(0x5b),\ |
| 75 | + w(0x6a), w(0xcb), w(0xbe), w(0x39), w(0x4a), w(0x4c), w(0x58), w(0xcf),\ |
| 76 | + w(0xd0), w(0xef), w(0xaa), w(0xfb), w(0x43), w(0x4d), w(0x33), w(0x85),\ |
| 77 | + w(0x45), w(0xf9), w(0x02), w(0x7f), w(0x50), w(0x3c), w(0x9f), w(0xa8),\ |
| 78 | + w(0x51), w(0xa3), w(0x40), w(0x8f), w(0x92), w(0x9d), w(0x38), w(0xf5),\ |
| 79 | + w(0xbc), w(0xb6), w(0xda), w(0x21), w(0x10), w(0xff), w(0xf3), w(0xd2),\ |
| 80 | + w(0xcd), w(0x0c), w(0x13), w(0xec), w(0x5f), w(0x97), w(0x44), w(0x17),\ |
| 81 | + w(0xc4), w(0xa7), w(0x7e), w(0x3d), w(0x64), w(0x5d), w(0x19), w(0x73),\ |
| 82 | + w(0x60), w(0x81), w(0x4f), w(0xdc), w(0x22), w(0x2a), w(0x90), w(0x88),\ |
| 83 | + w(0x46), w(0xee), w(0xb8), w(0x14), w(0xde), w(0x5e), w(0x0b), w(0xdb),\ |
| 84 | + w(0xe0), w(0x32), w(0x3a), w(0x0a), w(0x49), w(0x06), w(0x24), w(0x5c),\ |
| 85 | + w(0xc2), w(0xd3), w(0xac), w(0x62), w(0x91), w(0x95), w(0xe4), w(0x79),\ |
| 86 | + w(0xe7), w(0xc8), w(0x37), w(0x6d), w(0x8d), w(0xd5), w(0x4e), w(0xa9),\ |
| 87 | + w(0x6c), w(0x56), w(0xf4), w(0xea), w(0x65), w(0x7a), w(0xae), w(0x08),\ |
| 88 | + w(0xba), w(0x78), w(0x25), w(0x2e), w(0x1c), w(0xa6), w(0xb4), w(0xc6),\ |
| 89 | + w(0xe8), w(0xdd), w(0x74), w(0x1f), w(0x4b), w(0xbd), w(0x8b), w(0x8a),\ |
| 90 | + w(0x70), w(0x3e), w(0xb5), w(0x66), w(0x48), w(0x03), w(0xf6), w(0x0e),\ |
| 91 | + w(0x61), w(0x35), w(0x57), w(0xb9), w(0x86), w(0xc1), w(0x1d), w(0x9e),\ |
| 92 | + w(0xe1), w(0xf8), w(0x98), w(0x11), w(0x69), w(0xd9), w(0x8e), w(0x94),\ |
| 93 | + w(0x9b), w(0x1e), w(0x87), w(0xe9), w(0xce), w(0x55), w(0x28), w(0xdf),\ |
| 94 | + w(0x8c), w(0xa1), w(0x89), w(0x0d), w(0xbf), w(0xe6), w(0x42), w(0x68),\ |
| 95 | + w(0x41), w(0x99), w(0x2d), w(0x0f), w(0xb0), w(0x54), w(0xbb), w(0x16) } |
| 96 | + |
| 97 | +#define rc_data(w) {\ |
| 98 | + w(0x01), w(0x02), w(0x04), w(0x08), w(0x10),w(0x20), w(0x40), w(0x80),\ |
| 99 | + w(0x1b), w(0x36) } |
| 100 | + |
| 101 | +#define bytes2word(b0, b1, b2, b3) (((uint32_t)(b3) << 24) | \ |
| 102 | + ((uint32_t)(b2) << 16) | ((uint32_t)(b1) << 8) | (b0)) |
| 103 | + |
| 104 | +#define h0(x) (x) |
| 105 | +#define w0(p) bytes2word(p, 0, 0, 0) |
| 106 | +#define w1(p) bytes2word(0, p, 0, 0) |
| 107 | +#define w2(p) bytes2word(0, 0, p, 0) |
| 108 | +#define w3(p) bytes2word(0, 0, 0, p) |
| 109 | + |
| 110 | +#define u0(p) bytes2word(f2(p), p, p, f3(p)) |
| 111 | +#define u1(p) bytes2word(f3(p), f2(p), p, p) |
| 112 | +#define u2(p) bytes2word(p, f3(p), f2(p), p) |
| 113 | +#define u3(p) bytes2word(p, p, f3(p), f2(p)) |
| 114 | + |
| 115 | +#define v0(p) bytes2word(fe(p), f9(p), fd(p), fb(p)) |
| 116 | +#define v1(p) bytes2word(fb(p), fe(p), f9(p), fd(p)) |
| 117 | +#define v2(p) bytes2word(fd(p), fb(p), fe(p), f9(p)) |
| 118 | +#define v3(p) bytes2word(f9(p), fd(p), fb(p), fe(p)) |
| 119 | + |
| 120 | +#define f2(x) ((x<<1) ^ (((x>>7) & 1) * WPOLY)) |
| 121 | +#define f4(x) ((x<<2) ^ (((x>>6) & 1) * WPOLY) ^ (((x>>6) & 2) * WPOLY)) |
| 122 | +#define f8(x) ((x<<3) ^ (((x>>5) & 1) * WPOLY) ^ (((x>>5) & 2) * WPOLY) ^ (((x>>5) & 4) * WPOLY)) |
| 123 | +#define f3(x) (f2(x) ^ x) |
| 124 | +#define f9(x) (f8(x) ^ x) |
| 125 | +#define fb(x) (f8(x) ^ f2(x) ^ x) |
| 126 | +#define fd(x) (f8(x) ^ f4(x) ^ x) |
| 127 | +#define fe(x) (f8(x) ^ f4(x) ^ f2(x)) |
| 128 | + |
| 129 | +#define t_dec(m,n) t_##m##n |
| 130 | +#define t_set(m,n) t_##m##n |
| 131 | +#define t_use(m,n) t_##m##n |
| 132 | + |
| 133 | +#define d_4(t,n,b,e,f,g,h) ALIGN const t n[4][256] = { b(e), b(f), b(g), b(h) } |
| 134 | + |
| 135 | +#define four_tables(x,tab,vf,rf,c) \ |
| 136 | + (tab[0][bval(vf(x,0,c),rf(0,c))] \ |
| 137 | + ^ tab[1][bval(vf(x,1,c),rf(1,c))] \ |
| 138 | + ^ tab[2][bval(vf(x,2,c),rf(2,c))] \ |
| 139 | + ^ tab[3][bval(vf(x,3,c),rf(3,c))]) |
| 140 | + |
| 141 | +d_4(uint32_t, t_dec(f,n), sb_data, u0, u1, u2, u3); |
| 142 | + |
| 143 | +void aesb_single_round(const uint8_t *in, uint8_t *out, uint8_t *expandedKey) |
| 144 | +{ |
| 145 | + uint32_t b0[4], b1[4]; |
| 146 | + const uint32_t *kp = (uint32_t *) expandedKey; |
| 147 | + state_in(b0, in); |
| 148 | + |
| 149 | + round(fwd_rnd, b1, b0, kp); |
| 150 | + |
| 151 | + state_out(out, b1); |
| 152 | +} |
| 153 | + |
| 154 | +void aesb_pseudo_round(const uint8_t *in, uint8_t *out, uint8_t *expandedKey) |
| 155 | +{ |
| 156 | + uint32_t b0[4], b1[4]; |
| 157 | + const uint32_t *kp = (uint32_t *) expandedKey; |
| 158 | + state_in(b0, in); |
| 159 | + |
| 160 | + round(fwd_rnd, b1, b0, kp); |
| 161 | + round(fwd_rnd, b0, b1, kp + 1 * N_COLS); |
| 162 | + round(fwd_rnd, b1, b0, kp + 2 * N_COLS); |
| 163 | + round(fwd_rnd, b0, b1, kp + 3 * N_COLS); |
| 164 | + round(fwd_rnd, b1, b0, kp + 4 * N_COLS); |
| 165 | + round(fwd_rnd, b0, b1, kp + 5 * N_COLS); |
| 166 | + round(fwd_rnd, b1, b0, kp + 6 * N_COLS); |
| 167 | + round(fwd_rnd, b0, b1, kp + 7 * N_COLS); |
| 168 | + round(fwd_rnd, b1, b0, kp + 8 * N_COLS); |
| 169 | + round(fwd_rnd, b0, b1, kp + 9 * N_COLS); |
| 170 | + |
| 171 | + state_out(out, b0); |
| 172 | +} |
| 173 | + |
| 174 | + |
| 175 | +#if defined(__cplusplus) |
| 176 | +} |
| 177 | +#endif |
0 commit comments