X-Git-Url: https://git.cryptolib.org/?a=blobdiff_plain;f=keccak%2Fkeccak-stub.c;h=2ffbe4a7e426bb4fe5d7881931213f62a5919eed;hb=29a44972ae3749a6a273d936f2e15327ecae8a94;hp=ff198e7c3fe692b0af74880e4b8779f63881abc2;hpb=ddc66bc0543c50a0f36a3c26abf90dfef36dc3d9;p=avr-crypto-lib.git

diff --git a/keccak/keccak-stub.c b/keccak/keccak-stub.c
index ff198e7..2ffbe4a 100644
--- a/keccak/keccak-stub.c
+++ b/keccak/keccak-stub.c
@@ -20,11 +20,8 @@
 #include <stdint.h>
 #include <stdlib.h>
 #include <string.h>
-#include <avr/pgmspace.h>
 #include "memxor.h"
-#include "rotate64.h"
 #include "keccak.h"
-#include "stdio.h"
 
 #ifdef DEBUG
 #  undef DEBUG
@@ -34,6 +31,7 @@
 
 #if DEBUG
 #include "cli.h"
+#include "stdio.h"
 
 void keccak_dump_state(uint64_t a[5][5]){
 	uint8_t i,j;
@@ -62,79 +60,50 @@ void keccak_dump_ctx(keccak_ctx_t* ctx){
 
 #endif
 
-
-const uint8_t keccak_rc_comp[] PROGMEM = {
-		0x01, 0x92, 0xda, 0x70,
-		0x9b, 0x21, 0xf1, 0x59,
-		0x8a, 0x88, 0x39, 0x2a,
-		0xbb, 0xcb, 0xd9, 0x53,
-		0x52, 0xc0, 0x1a, 0x6a,
-		0xf1, 0xd0, 0x21, 0x78,
-};
-
-const uint8_t keccak_rotate_codes[5][5] PROGMEM = {
-        { ROT_CODE( 0), ROT_CODE( 1), ROT_CODE(62), ROT_CODE(28), ROT_CODE(27) },
-        { ROT_CODE(36), ROT_CODE(44), ROT_CODE( 6), ROT_CODE(55), ROT_CODE(20) },
-        { ROT_CODE( 3), ROT_CODE(10), ROT_CODE(43), ROT_CODE(25), ROT_CODE(39) },
-        { ROT_CODE(41), ROT_CODE(45), ROT_CODE(15), ROT_CODE(21), ROT_CODE( 8) },
-        { ROT_CODE(18), ROT_CODE( 2), ROT_CODE(61), ROT_CODE(56), ROT_CODE(14) }
-};
-
 void keccak_f1600(uint64_t a[5][5]);
 
-void keccak_nextBlock(keccak_ctx_t* ctx, const void* block){
-	memxor(ctx->a, block, ctx->bs);
-	keccak_f1600(ctx->a);
-}
-
 void keccak_lastBlock(keccak_ctx_t* ctx, const void* block, uint16_t length_b){
-	while(length_b>=ctx->r){
+	while(length_b >= ctx->r){
 		keccak_nextBlock(ctx, block);
 		block = (uint8_t*)block + ctx->bs;
 		length_b -=  ctx->r;
 	}
-	uint8_t tmp[ctx->bs];
-	uint8_t pad[3];
-	memset(tmp, 0x00, ctx->bs);
-	memcpy(tmp, block, (length_b+7)/8);
-	/* appand 1 */
+	memxor(ctx->a, block, (length_b)/8);
+	/* append 1 */
 	if(length_b & 7){
 		/* we have some single bits */
 		uint8_t t;
-		t = tmp[length_b / 8] >> (8 - (length_b & 7));
+		t = ((uint8_t*)block)[length_b / 8] >> (8 - (length_b & 7));
 		t |= 0x01 << (length_b & 7);
-		tmp[length_b / 8] = t;
+		((uint8_t*)ctx->a)[length_b / 8] ^= t;
 	}else{
-		tmp[length_b / 8] = 0x01;
+	    ((uint8_t*)ctx->a)[length_b / 8] ^= 0x01;
 	}
-	pad[0] = ctx->d;
-	pad[1] = ctx->bs;
-	pad[2] = 0x01;
 	if(length_b / 8 + 1 + 3 <= ctx->bs){
-		memcpy(tmp + length_b / 8 + 1, pad, 3);
+        *((uint8_t*)ctx->a + length_b / 8 + 1) ^= ctx->d;
+        *((uint8_t*)ctx->a + length_b / 8 + 2) ^= ctx->bs;
+        *((uint8_t*)ctx->a + length_b / 8 + 3) ^= 1;
 	}else{
 		if(length_b / 8 + 1 + 2 <= ctx->bs){
-			memcpy(tmp+length_b/8+1, pad, 2);
-			keccak_nextBlock(ctx, tmp);
-			memset(tmp, 0x00, ctx->bs);
-			tmp[0] = 0x01;
+            *((uint8_t*)ctx->a + length_b / 8 + 1) ^= ctx->d;
+            *((uint8_t*)ctx->a + length_b / 8 + 2) ^= ctx->bs;
+			keccak_f1600(ctx->a);
+			((uint8_t*)ctx->a)[0] ^= 0x01;
 		}else{
 			if(length_b/8+1+1 <= ctx->bs){
-				memcpy(tmp + length_b / 8 + 1, pad, 1);
-				keccak_nextBlock(ctx, tmp);
-				memset(tmp, 0x00, ctx->bs);
-				tmp[0] = ctx->bs;
-				tmp[1] = 0x01;
+				*((uint8_t*)ctx->a + length_b / 8 + 1) ^= ctx->d;
+				keccak_f1600(ctx->a);
+				((uint8_t*)ctx->a)[0] ^= ctx->bs;
+				((uint8_t*)ctx->a)[1] ^= 0x01;
 			}else{
-				keccak_nextBlock(ctx, tmp);
-				memset(tmp, 0x00, ctx->bs);
-				tmp[0] = ctx->d;
-				tmp[1] = ctx->bs;
-				tmp[2] = 0x01;
+				keccak_f1600(ctx->a);
+				((uint8_t*)ctx->a)[0] ^= ctx->d;
+				((uint8_t*)ctx->a)[1] ^= ctx->bs;
+				((uint8_t*)ctx->a)[2] ^= 0x01;
 			}
 		}
 	}
-	keccak_nextBlock(ctx, tmp);
+	keccak_f1600(ctx->a);
 }
 
 void keccak_ctx2hash(void* dest, uint16_t length_b, keccak_ctx_t* ctx){