]> git.cryptolib.org Git - arm-crypto-lib.git/blob - aes/aes_keyschedule.c
bigint looks good but needs more testing (gcdext fixed)
[arm-crypto-lib.git] / aes / aes_keyschedule.c
1 /* aes_keyschedule.c */
2 /*
3     This file is part of the ARM-Crypto-Lib.
4     Copyright (C) 2006-2010  Daniel Otte (daniel.otte@rub.de)
5
6     This program is free software: you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation, either version 3 of the License, or
9     (at your option) any later version.
10
11     This program is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with this program.  If not, see <http://www.gnu.org/licenses/>.
18 */
19 /**
20  * \file     aes_keyschedule.c
21  * \email    daniel.otte@rub.de
22  * \author   Daniel Otte
23  * \date     2008-12-30
24  * \license  GPLv3 or later
25  *
26  */
27
28 #include <stdint.h>
29 #include "aes.h"
30 #include "aes_keyschedule.h"
31 #include "aes_sbox.h"
32 #include <string.h>
33
34 static
35 void aes_rotword(void* a){
36         uint8_t t;
37         t=((uint8_t*)a)[0];
38         ((uint8_t*)a)[0] = ((uint8_t*)a)[1];
39         ((uint8_t*)a)[1] = ((uint8_t*)a)[2];
40         ((uint8_t*)a)[2] = ((uint8_t*)a)[3];
41         ((uint8_t*)a)[3] = t;
42 }
43
44 uint8_t rc_tab[] = { 0x01, 0x02, 0x04, 0x08,
45                      0x10, 0x20, 0x40, 0x80,
46                      0x1b, 0x36 };
47
48 void aes_init(const void* key, uint16_t keysize_b, aes_genctx_t* ctx){
49         uint8_t hi,i,nk, next_nk;
50         uint8_t rc=0;
51         union {
52                 uint32_t v32;
53                 uint8_t  v8[4];
54         } tmp;
55         nk=keysize_b>>5; /* 4, 6, 8 */
56         hi=4*(nk+6+1);
57         memcpy(ctx, key, keysize_b/8);
58         next_nk = nk;
59         for(i=nk;i<hi;++i){
60                 tmp.v32 = ((uint32_t*)(ctx->key[0].ks))[i-1];
61                 if(i!=next_nk){
62                         if(nk==8 && i%8==4){
63                                 tmp.v8[0] = aes_sbox[tmp.v8[0]];
64                                 tmp.v8[1] = aes_sbox[tmp.v8[1]];
65                                 tmp.v8[2] = aes_sbox[tmp.v8[2]];
66                                 tmp.v8[3] = aes_sbox[tmp.v8[3]];
67                         }
68                 } else {
69                         next_nk += nk;
70                         aes_rotword(&(tmp.v32));
71                         tmp.v8[0] = aes_sbox[tmp.v8[0]];
72                         tmp.v8[1] = aes_sbox[tmp.v8[1]];
73                         tmp.v8[2] = aes_sbox[tmp.v8[2]];
74                         tmp.v8[3] = aes_sbox[tmp.v8[3]];
75                         tmp.v8[0] ^= rc_tab[rc];
76                         rc++;
77                 }
78                 ((uint32_t*)(ctx->key[0].ks))[i] = ((uint32_t*)(ctx->key[0].ks))[i-nk]
79                                                    ^ tmp.v32;
80         }
81 }
82
83 void aes128_init(const void* key, aes128_ctx_t* ctx){
84         aes_init(key, 128, (aes_genctx_t*)ctx);
85 }
86
87 void aes192_init(const void* key, aes192_ctx_t* ctx){
88         aes_init(key, 192, (aes_genctx_t*)ctx);
89 }
90
91 void aes256_init(const void* key, aes256_ctx_t* ctx){
92         aes_init(key, 256, (aes_genctx_t*)ctx);
93 }