]> git.cryptolib.org Git - avr-crypto-lib.git/blob - aes/aes_keyschedule.c
syncing with bzr
[avr-crypto-lib.git] / aes / aes_keyschedule.c
1 /* aes_keyschedule.c */
2 /*
3     This file is part of the AVR-Crypto-Lib.
4     Copyright (C) 2008, 2009  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 #include <avr/pgmspace.h>
34
35 static
36 void aes_rotword(void* a){
37         uint8_t t;
38         t=((uint8_t*)a)[0];
39         ((uint8_t*)a)[0] = ((uint8_t*)a)[1];
40         ((uint8_t*)a)[1] = ((uint8_t*)a)[2];
41         ((uint8_t*)a)[2] = ((uint8_t*)a)[3];
42         ((uint8_t*)a)[3] = t;
43 }
44
45 const uint8_t rc_tab[] PROGMEM = { 0x01, 0x02, 0x04, 0x08,
46                              0x10, 0x20, 0x40, 0x80,
47                              0x1b, 0x36 };
48
49 void aes_init(const void* key, uint16_t keysize_b, aes_genctx_t* ctx){
50         uint8_t hi,i,nk, next_nk;
51         uint8_t rc=0;
52         union {
53                 uint32_t v32;
54                 uint8_t  v8[4];
55         } tmp;
56         nk=keysize_b>>5; /* 4, 6, 8 */
57         hi=4*(nk+6+1);
58         memcpy(ctx, key, keysize_b/8);
59         next_nk = nk;
60         for(i=nk;i<hi;++i){
61                 tmp.v32 = ((uint32_t*)(ctx->key[0].ks))[i-1];
62                 if(i!=next_nk){
63                         if(nk==8 && i%8==4){
64                                 tmp.v8[0] = pgm_read_byte(aes_sbox+tmp.v8[0]);
65                                 tmp.v8[1] = pgm_read_byte(aes_sbox+tmp.v8[1]);
66                                 tmp.v8[2] = pgm_read_byte(aes_sbox+tmp.v8[2]);
67                                 tmp.v8[3] = pgm_read_byte(aes_sbox+tmp.v8[3]);
68                         }
69                 } else {
70                         next_nk += nk;
71                         aes_rotword(&(tmp.v32));
72                         tmp.v8[0] = pgm_read_byte(aes_sbox+tmp.v8[0]);
73                         tmp.v8[1] = pgm_read_byte(aes_sbox+tmp.v8[1]);
74                         tmp.v8[2] = pgm_read_byte(aes_sbox+tmp.v8[2]);
75                         tmp.v8[3] = pgm_read_byte(aes_sbox+tmp.v8[3]);
76                         tmp.v8[0] ^= pgm_read_byte(rc_tab+rc);
77                         rc++;
78                 }
79                 ((uint32_t*)(ctx->key[0].ks))[i] = ((uint32_t*)(ctx->key[0].ks))[i-nk]
80                                                    ^ tmp.v32;
81         }
82 }
83
84 void aes128_init(const void* key, aes128_ctx_t* ctx){
85         aes_init(key, 128, (aes_genctx_t*)ctx);
86 }
87
88 void aes192_init(const void* key, aes192_ctx_t* ctx){
89         aes_init(key, 192, (aes_genctx_t*)ctx);
90 }
91
92 void aes256_init(const void* key, aes256_ctx_t* ctx){
93         aes_init(key, 256, (aes_genctx_t*)ctx);
94 }