X-Git-Url: https://git.cryptolib.org/?a=blobdiff_plain;f=md5.c;h=003699e5ef4ae65486bece16d9cdfbe21a4cca2b;hb=8f855d283a31a468ea014774c4723a8b77b81644;hp=c995b2a07a82bfe1b2120f6d927dd627cbe47806;hpb=5ac75cfae217122b540c1a6d258054230dc534c3;p=avr-crypto-lib.git

diff --git a/md5.c b/md5.c
index c995b2a..003699e 100644
--- a/md5.c
+++ b/md5.c
@@ -1,6 +1,6 @@
 /* md5.c */
 /*
-    This file is part of the Crypto-avr-lib/microcrypt-lib.
+    This file is part of the AVR-Crypto-Lib.
     Copyright (C) 2008  Daniel Otte (daniel.otte@rub.de)
 
     This program is free software: you can redistribute it and/or modify
@@ -27,7 +27,7 @@
 
  #include "md5.h"
  #include "md5_sbox.h"
- #include "uart.h" 
+ #include "cli.h" 
  #include <stdint.h>
  #include <string.h>
  
@@ -73,27 +73,28 @@ void md5_core(uint32_t* a, void* block, uint8_t as, uint8_t s, uint8_t i, uint8_
 	/* a = b + ((a + F(b,c,d) + X[k] + T[i]) <<< s). */
 #ifdef DEBUG
 	char funcc[]={'*', '-', '+', '~'};
-	uart_putstr("\r\n DBG: md5_core [");
-	uart_putc(funcc[fi]);
-	uart_hexdump(&as, 1); uart_putc(' ');
-	uart_hexdump(&k, 1); uart_putc(' ');
-	uart_hexdump(&s, 1); uart_putc(' ');
-	uart_hexdump(&i, 1); uart_putc(']');
+	cli_putstr("\r\n DBG: md5_core [");
+	cli_putc(funcc[fi]);
+	cli_hexdump(&as, 1); cli_putc(' ');
+	cli_hexdump(&k, 1); cli_putc(' ');
+	cli_hexdump(&s, 1); cli_putc(' ');
+	cli_hexdump(&i, 1); cli_putc(']');
 #endif	
-	t = a[as] + funcs[fi](a[(as+1)&3], a[(as+2)&3], a[(as+3)&3]) + *((uint32_t*)block) + md5_T[i] ;
+	t = a[as] + funcs[fi](a[(as+1)&3], a[(as+2)&3], a[(as+3)&3]) 
+	    + *((uint32_t*)block) + pgm_read_dword(md5_T+i) ;
 	a[as]=a[(as+1)&3] + ROTL32(t, s);
 }
 
-void md5_nextBlock(md5_ctx_t *state, void* block){
+void md5_nextBlock(md5_ctx_t *state, const void* block){
 	uint32_t	a[4];
 	uint8_t		m,n,i=0;
 	/* this requires other mixed sboxes */
 #ifdef DEBUG
-	uart_putstr("\r\n DBG: md5_nextBlock: block:\r\n");
-	uart_hexdump(block, 16);	uart_putstr("\r\n");
-	uart_hexdump(block+16, 16);	uart_putstr("\r\n");
-	uart_hexdump(block+32, 16);	uart_putstr("\r\n");
-	uart_hexdump(block+48, 16);	uart_putstr("\r\n");
+	cli_putstr("\r\n DBG: md5_nextBlock: block:\r\n");
+	cli_hexdump(block, 16);	cli_putstr("\r\n");
+	cli_hexdump(block+16, 16);	cli_putstr("\r\n");
+	cli_hexdump(block+32, 16);	cli_putstr("\r\n");
+	cli_hexdump(block+48, 16);	cli_putstr("\r\n");
 #endif	
 	
 	a[0]=state->a[0];
@@ -136,7 +137,7 @@ void md5_nextBlock(md5_ctx_t *state, void* block){
 	state->counter++;
 }
 
-void md5_lastBlock(md5_ctx_t *state, void* block, uint16_t length_b){
+void md5_lastBlock(md5_ctx_t *state, const void* block, uint16_t length_b){
 	uint16_t l;
 	uint8_t b[64];
 	while (length_b >= 512){
@@ -160,8 +161,25 @@ void md5_lastBlock(md5_ctx_t *state, void* block, uint16_t length_b){
 	if(l+sizeof(uint64_t) >= 512/8){
 		md5_nextBlock(state, b);
 		state->counter--;
-		memset(b, 0, 64);
+		memset(b, 0, 64-8);
 	}
 	*((uint64_t*)&b[64-sizeof(uint64_t)]) = (state->counter * 512) + length_b;
 	md5_nextBlock(state, b);
 }
+
+void md5_ctx2hash(md5_hash_t* dest, const md5_ctx_t* state){
+	memcpy(dest, state->a, MD5_HASH_BYTES);
+}
+
+void md5(md5_hash_t* dest, const void* msg, uint32_t length_b){
+	md5_ctx_t ctx;
+	md5_init(&ctx);
+	while(length_b>=MD5_BLOCK_BITS){
+		md5_nextBlock(&ctx, msg);
+		msg = (uint8_t*)msg + MD5_BLOCK_BYTES;
+		length_b -= MD5_BLOCK_BITS;
+	}
+	md5_lastBlock(&ctx, msg, length_b);
+	md5_ctx2hash(dest, &ctx);
+}
+