3 This file is part of the AVR-Crypto-Lib.
4 Copyright (C) 2010 Daniel Otte (daniel.otte@rub.de)
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.
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.
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/>.
23 #include <avr/pgmspace.h>
38 void keccak_dump_state(uint64_t a[5][5]){
41 cli_putstr_P(PSTR("\r\n"));
43 cli_putstr_P(PSTR(": "));
45 cli_hexdump_rev(&(a[i][j]), 8);
51 void keccak_dump_ctx(keccak_ctx_t* ctx){
52 keccak_dump_state(ctx->a);
53 cli_putstr_P(PSTR("\r\nDBG: r: "));
54 cli_hexdump_rev(&(ctx->r), 2);
55 cli_putstr_P(PSTR("\t c: "));
56 cli_hexdump_rev(&(ctx->c), 2);
57 cli_putstr_P(PSTR("\t d: "));
58 cli_hexdump(&(ctx->d), 1);
59 cli_putstr_P(PSTR("\t bs: "));
60 cli_hexdump(&(ctx->bs), 1);
66 const uint64_t rc[] PROGMEM = {
67 0x0000000000000001LL, 0x0000000000008082LL,
68 0x800000000000808ALL, 0x8000000080008000LL,
69 0x000000000000808BLL, 0x0000000080000001LL,
70 0x8000000080008081LL, 0x8000000000008009LL,
71 0x000000000000008ALL, 0x0000000000000088LL,
72 0x0000000080008009LL, 0x000000008000000ALL,
73 0x000000008000808BLL, 0x800000000000008BLL,
74 0x8000000000008089LL, 0x8000000000008003LL,
75 0x8000000000008002LL, 0x8000000000000080LL,
76 0x000000000000800ALL, 0x800000008000000ALL,
77 0x8000000080008081LL, 0x8000000000008080LL,
78 0x0000000080000001LL, 0x8000000080008008LL
82 const static uint8_t rc_comp[] PROGMEM = {
83 0x01, 0x92, 0xda, 0x70,
84 0x9b, 0x21, 0xf1, 0x59,
85 0x8a, 0x88, 0x39, 0x2a,
86 0xbb, 0xcb, 0xd9, 0x53,
87 0x52, 0xc0, 0x1a, 0x6a,
88 0xf1, 0xd0, 0x21, 0x78,
91 const uint8_t keccak_rotate_codes[5][5] PROGMEM = {
92 { ROT_CODE( 0), ROT_CODE(36), ROT_CODE( 3), ROT_CODE(41), ROT_CODE(18) },
93 { ROT_CODE( 1), ROT_CODE(44), ROT_CODE(10), ROT_CODE(45), ROT_CODE( 2) },
94 { ROT_CODE(62), ROT_CODE( 6), ROT_CODE(43), ROT_CODE(15), ROT_CODE(61) },
95 { ROT_CODE(28), ROT_CODE(55), ROT_CODE(25), ROT_CODE(21), ROT_CODE(56) },
96 { ROT_CODE(27), ROT_CODE(20), ROT_CODE(39), ROT_CODE( 8), ROT_CODE(14) }
99 const uint8_t keccak_rotate_codes[5][5] PROGMEM = {
100 { ROT_CODE( 0), ROT_CODE( 1), ROT_CODE(62), ROT_CODE(28), ROT_CODE(27) },
101 { ROT_CODE(36), ROT_CODE(44), ROT_CODE( 6), ROT_CODE(55), ROT_CODE(20) },
102 { ROT_CODE( 3), ROT_CODE(10), ROT_CODE(43), ROT_CODE(25), ROT_CODE(39) },
103 { ROT_CODE(41), ROT_CODE(45), ROT_CODE(15), ROT_CODE(21), ROT_CODE( 8) },
104 { ROT_CODE(18), ROT_CODE( 2), ROT_CODE(61), ROT_CODE(56), ROT_CODE(14) }
107 void keccak_theta(uint64_t *a, uint64_t *b);
108 extern const uint8_t rho_pi_idx_table[25] PROGMEM;
111 void keccak_round(uint64_t a[5][5], uint8_t rci){
119 keccak_theta((uint64_t*)a, (uint64_t*)b);
121 cli_putstr_P(PSTR("\r\nAfter theta:"));
122 keccak_dump_state(a);
126 const uint8_t* rot_code = (const uint8_t*)keccak_rotate_codes;
127 const uint8_t* idx_idx = (const uint8_t*)rho_pi_idx_table;
128 uint64_t *a_tmp = (uint64_t*)a;
129 for(i = 0; i < 25; ++i){
130 *((uint64_t*)(((uint8_t*)b) + pgm_read_byte(idx_idx++))) =
131 rotate64left_code(*a_tmp++, pgm_read_byte(rot_code++));
136 cli_putstr_P(PSTR("\r\n--- after rho & pi ---"));
137 keccak_dump_state(a);
140 // memcpy(a, b, 5 * 5 * 8);
141 // for(i = 1; i < 5; ++i){
143 for(j = 0; j < 5; ++j){
144 a[i][j] = b[i][j] ^ ((~(b[i][(j + 1) % 5])) & (b[i][(j + 2) % 5]));
147 // a[i][0] ^= ((~(b[i][1])) & (b[i][2]));
148 // a[i][1] ^= ((~(b[i][2])) & (b[i][3]));
149 // a[i][2] ^= ((~(b[i][3])) & (b[i][4]));
150 for(j = 0; j < 3 * 8; ++j){
151 ((uint8_t*)a)[i * 5 * 8 + j] ^=
152 (~((uint8_t*)b)[i * 5 * 8 + j + 8]) & ((uint8_t*)a)[i * 5 * 8 + j + 16];
154 a[i][3] ^= ((~(b[i][4])) & (b[i][0]));
155 a[i][4] ^= ((~(b[i][0])) & (b[i][1]));
160 cli_putstr_P(PSTR("\r\nAfter chi:"));
161 keccak_dump_state(a);
165 // memcpy_P(&t, &(rc_comp[rci]), 8);
167 t.v8[0] = pgm_read_byte(&(rc_comp[rci]));
181 cli_putstr_P(PSTR("\r\nAfter iota:"));
182 keccak_dump_state(a);
186 void keccak_f1600(uint64_t a[5][5]){
190 cli_putstr_P(PSTR("\r\n\r\n--- Round "));
192 cli_putstr_P(PSTR(" ---"));
198 void keccak_nextBlock(keccak_ctx_t* ctx, const void* block){
199 memxor(ctx->a, block, ctx->bs);
200 keccak_f1600(ctx->a);
203 void keccak_lastBlock(keccak_ctx_t* ctx, const void* block, uint16_t length_b){
204 while(length_b>=ctx->r){
205 keccak_nextBlock(ctx, block);
206 block = (uint8_t*)block + ctx->bs;
209 uint8_t tmp[ctx->bs];
211 memset(tmp, 0x00, ctx->bs);
212 memcpy(tmp, block, (length_b+7)/8);
215 /* we have some single bits */
217 t = tmp[length_b / 8] >> (8 - (length_b & 7));
218 t |= 0x01 << (length_b & 7);
219 tmp[length_b / 8] = t;
221 tmp[length_b / 8] = 0x01;
226 if(length_b / 8 + 1 + 3 <= ctx->bs){
227 memcpy(tmp + length_b / 8 + 1, pad, 3);
229 if(length_b / 8 + 1 + 2 <= ctx->bs){
230 memcpy(tmp+length_b/8+1, pad, 2);
231 keccak_nextBlock(ctx, tmp);
232 memset(tmp, 0x00, ctx->bs);
235 if(length_b/8+1+1 <= ctx->bs){
236 memcpy(tmp + length_b / 8 + 1, pad, 1);
237 keccak_nextBlock(ctx, tmp);
238 memset(tmp, 0x00, ctx->bs);
242 keccak_nextBlock(ctx, tmp);
243 memset(tmp, 0x00, ctx->bs);
250 keccak_nextBlock(ctx, tmp);
253 void keccak_ctx2hash(void* dest, uint16_t length_b, keccak_ctx_t* ctx){
254 while(length_b>=ctx->r){
255 memcpy(dest, ctx->a, ctx->bs);
256 dest = (uint8_t*)dest + ctx->bs;
258 keccak_f1600(ctx->a);
260 memcpy(dest, ctx->a, (length_b+7)/8);
263 void keccak224_ctx2hash(void* dest, keccak_ctx_t* ctx){
264 keccak_ctx2hash(dest, 224, ctx);
267 void keccak256_ctx2hash(void* dest, keccak_ctx_t* ctx){
268 keccak_ctx2hash(dest, 256, ctx);
271 void keccak384_ctx2hash(void* dest, keccak_ctx_t* ctx){
272 keccak_ctx2hash(dest, 384, ctx);
275 void keccak512_ctx2hash(void* dest, keccak_ctx_t* ctx){
276 keccak_ctx2hash(dest, 512, ctx);
280 1. SHA3-224: ⌊Keccak[r = 1152, c = 448, d = 28]⌋224
281 2. SHA3-256: ⌊Keccak[r = 1088, c = 512, d = 32]⌋256
282 3. SHA3-384: ⌊Keccak[r = 832, c = 768, d = 48]⌋384
283 4. SHA3-512: ⌊Keccak[r = 576, c = 1024, d = 64]⌋512
285 void keccak_init(uint16_t r, uint16_t c, uint8_t d, keccak_ctx_t* ctx){
286 memset(ctx->a, 0x00, 5 * 5 * 8);
290 ctx->bs = (uint8_t)(r / 8);
293 void keccak224_init(keccak_ctx_t* ctx){
294 keccak_init(1152, 448, 28, ctx);
297 void keccak256_init(keccak_ctx_t* ctx){
298 keccak_init(1088, 512, 32, ctx);
301 void keccak384_init(keccak_ctx_t* ctx){
302 keccak_init( 832, 768, 48, ctx);
305 void keccak512_init(keccak_ctx_t* ctx){
306 keccak_init( 576, 1024, 64, ctx);