X-Git-Url: https://codewiz.org/gitweb?a=blobdiff_plain;f=algos%2Fmd2.c;h=4eee4ecd5e60ea82ce1905ab9bebafe0e22f39ff;hb=48e9459a1f736a5c256abff5ee4cd11ffad38602;hp=b8caeade9b14675d84e59ce6b9b486cebb3b31dd;hpb=a040d02cb3a6e8b59d72f8b2a1a4b095df42717c;p=bertos.git

diff --git a/algos/md2.c b/algos/md2.c
index b8caeade..4eee4ecd 100755
--- a/algos/md2.c
+++ b/algos/md2.c
@@ -13,8 +13,20 @@
 
 /*#*
  *#* $Log$
- *#* Revision 1.5  2007/01/31 16:42:43  asterix
- *#* Write md2_update function.
+ *#* Revision 1.12  2007/02/05 16:52:44  asterix
+ *#* Add define for harvard architecture.
+ *#*
+ *#* Revision 1.11  2007/02/02 18:15:31  asterix
+ *#* Add function MD2_test. Fix bug in md2_update function.
+ *#*
+ *#* Revision 1.9  2007/02/02 13:10:01  asterix
+ *#* Fix some bugs in md2_pad and md2_update fuction.
+ *#*
+ *#* Revision 1.8  2007/02/01 14:45:56  asterix
+ *#* Rewrite md2_update function and fix some bug.
+ *#*
+ *#* Revision 1.7  2007/01/31 18:04:15  asterix
+ *#* Write md2_end function
  *#*
  *#* Revision 1.4  2007/01/31 13:51:57  asterix
  *#* Write md2_compute function.
@@ -29,14 +41,18 @@
 
 #include "md2.h"
 
-#include <string.h> //memset();
+#include <string.h>           //memset(), memcpy();
 #include <cfg/compiler.h>
+#include <cfg/debug.h>        //ASSERT()
+#include <cfg/macros.h>       //MIN(), countof();
+#include <mware/pgm.h>   
+
 
 /*
  * Official array of 256 byte pemutation contructed from digits of pi, defined
  * in the RFC 1319.
  */
-static uint8_t md2_perm[256] =
+static const uint8_t PROGMEM md2_perm[256] = 
 {
   41, 46, 67, 201, 162, 216, 124, 1, 61, 54, 84, 161, 236, 240, 6,
   19, 98, 167, 5, 243, 192, 199, 115, 140, 152, 147, 43, 217, 188,
@@ -57,65 +73,83 @@ static uint8_t md2_perm[256] =
   166, 119, 114, 248, 235, 117, 75, 10, 49, 68, 80, 180, 143, 237,
   31, 26, 219, 153, 141, 51, 159, 17, 131, 20
 };
+
+#if CPU_HARVARD
+	#define	MD2_PERM(x) pgm_read_char(&md2_perm[x]) //Read from program memory, if CPU is harvard
+#else
+	#define	MD2_PERM(x) md2_perm[x]                 //Normaly from a const array.
+#endif
+
 /**
- * Pad function. Fill input array with unsigned char until 
- * lenght of block is equal to CONFIG_MD2_BLOCK_LEN.
- *
+ * Pad function. Put len_pad unsigned char in
+ * input block.
  */
-static void md2_pad(void *block, size_t len_pad)
+static void md2_pad(void *_block, size_t len_pad)
 {
-	if (len_pad <= CONFIG_MD2_BLOCK_LEN)
-	{
-		for(int i=(CONFIG_MD2_BLOCK_LEN-len_pad);i<len_pad; i++)
-		{
-			block[i]=(uint8_t)len_pad;
-		}	
-	}
+	uint8_t *block;
+
+	block = (uint8_t *)_block;
+
+	ASSERT(len_pad <= CONFIG_MD2_BLOCK_LEN);
+
+	/*
+	 * Fill input block with len_pad char.
+	 */
+	memset(block, len_pad, len_pad);
+
 }
 
-static void md2_compute(void *state, void *checksum, void *block)
+static void md2_compute(void *_state, void *_checksum, void *_block)
 {
-	int t=0;	
-	uint8_t compute_array[CONFIG_MD2_BLOCK_LEN * 3];
-	
+	int i = 0;
+	uint16_t t = 0;
+	uint8_t compute_array[COMPUTE_ARRAY_LEN];
+	uint8_t *state;
+	uint8_t *checksum;
+	uint8_t *block;
+
+	state = (uint8_t *)_state;
+	checksum  = (uint8_t *)_checksum;
+	block = (uint8_t *)_block;
+
 	/*
- 	 * Copy state and checksum context in compute array. 
- 	 */
+	 * Copy state and checksum context in compute array.
+	 */
 	memcpy(compute_array, state, CONFIG_MD2_BLOCK_LEN);
-	memcpy(compute_array + CONFIG_MD2_BLOCK_LEN, state, CONFIG_MD2_BLOCK_LEN);
-	
+	memcpy(compute_array + CONFIG_MD2_BLOCK_LEN, block, CONFIG_MD2_BLOCK_LEN);
+
 	/*
- 	 * Fill compute array with state XOR block
- 	 */
-	for(int i=0; i< CONFIG_MD2_BLOCK_LEN; i++)
+	 * Fill compute array with state XOR block
+	 */
+	for(i = 0; i < CONFIG_MD2_BLOCK_LEN; i++)
 		compute_array[i + (CONFIG_MD2_BLOCK_LEN * 2)] = state[i] ^ block[i];
-	
+
 	/*
 	 * Encryt block.
 	 */
-	for(i=0; i<NUM_COMPUTE_ROUNDS; i++)
+	for(i = 0; i < NUM_COMPUTE_ROUNDS; i++)
 	{
-		for(int j=0; j<COMPUTE_ARRAY_LEN; j++)
+		for(int j = 0; j < COMPUTE_ARRAY_LEN; j++)
 		{
-			compute_array[j] ^= md2_perm [t];
+			compute_array[j] ^= MD2_PERM(t);
 			t = compute_array[j];
 		}
-		
+
 		t = (t + i) & 0xff; //modulo 256.
 	}
 	/*
-	 * Update checksum.	
+	 * Update checksum.
 	 */
 	t = checksum[CONFIG_MD2_BLOCK_LEN - 1];
 
-	for(i=0; i< CONFIG_MD2_BLOCK_LEN; i++)
+	for(i = 0; i < CONFIG_MD2_BLOCK_LEN; i++)
 	{
-		checksum[i]  ^= md2_perm [block[i] ^ t];
+		checksum[i]  ^= MD2_PERM(block[i] ^ t);
 		t = checksum[i];
 	}
-	
+
 	/*
-	 * Update state and clean compute array. 
+	 * Update state and clean compute array.
 	 */
 	memcpy(state, compute_array, CONFIG_MD2_BLOCK_LEN);
 	memset(compute_array, 0, sizeof(compute_array));
@@ -128,62 +162,142 @@ static void md2_compute(void *state, void *checksum, void *block)
  */
 void md2_init(Md2Context *context)
 {
-	context->counter = 0;
-	memset(context->state, 0, sizeof(context->state));
-	memset(context->checksum, 0, sizeof(context->checksum));
+
+	memset(context, 0, sizeof(Md2Context));
 
 }
+
 /**
- * Update bock.
- *
- *
- *
+ * Update block.
  */
-void md2_update(Md2Context *context, void *block_in, size_t block_len)
+void md2_update(Md2Context *context, const void *_block_in, size_t block_len)
 {
-	uint8_t store_block_len;    //Lenght of block store in context buffer
-	uint8_t empty_block_len;    //Lenght of block empty in context buffer
-	
-	store_block_len = context->counter;
-	empty_block_len = CONFIG_MD2_BLOCK_LEN - store_block_len;
-	
-	/*
-	 * Fill context buffer with input block and compute it.
-	 */
-	if(block_len > empty_block_len)
-	{
-		memcpy(context->buffer[store_block_len], block_in, empty_block_len);
-		md2_compute(context->state, context->checksum, context->buffer);
-		block_len -= empty_block_len;
-	}
-	else
+
+	const uint8_t *block_in;
+	size_t cpy_len;
+
+
+	block_in = (const uint8_t *)_block_in;
+
+	while(block_len > 0)
 	{
+		/*
+		 * Choose a number of block that fill input context buffer.
+		 */
+		cpy_len = MIN(block_len, CONFIG_MD2_BLOCK_LEN - context->counter);
 
-		memcpy(context->buffer[store_block_len], block_in, block_len);
-		context->counter += block_len;
+		
+		/*
+		 * Copy in the buffer input block.
+		 */
+		memcpy(&context->buffer[context->counter], block_in, cpy_len);
 
+		/*
+		 * Update a context counter, input block length and remaning 
+		 * context buffer block lenght.
+		 */
+		context->counter += cpy_len;
+		block_len -= cpy_len;
+		block_in += cpy_len;
+
+		/*
+		 * If buffer is full, compute it.
+		 */
+		if (context->counter >= CONFIG_MD2_BLOCK_LEN)
+		{
+			md2_compute(context->state, context->checksum, context->buffer);
+			context->counter = 0;
+		}
 	}
-	
+
+
+}
+/**
+ * Ends an MD2 message digest operation.
+ * This fuction take an context and return a pointer
+ * to context state.
+ *
+ * \param context in input.
+ * \return a pointer to context state (message digest).
+ */
+uint8_t  *md2_end(Md2Context *context)
+{
+
+	uint8_t buf[CONFIG_MD2_BLOCK_LEN];
+
 	/*
-	 * Split input in block long CONFIG_MD2_BLOCK_LEN and compute it.
+	 * Fill remaning empty context buffer.
 	 */
-	for(int i = empty_block_len + 1;  i + CONFIG_MD2_BLOCK_LEN < block_len; i += CONFIG_MD2_BLOCK_LEN)
-	{
-		memcpy(context->buffer, block_in[i], CONFIG_MD2_BLOCK_LEN);
-		md2_compute(context->state, context->checksum, context->buffer);
-	}
+	md2_pad(buf, CONFIG_MD2_BLOCK_LEN - context->counter);
+
+	/*
+	 * Update context buffer and compute it.
+	 */
+	md2_update(context, buf, CONFIG_MD2_BLOCK_LEN - context->counter);
 
 	/*
-	 * Copy remaining block in context buffer and update context counter.
+	 * Add context checksum to message input.
 	 */
-	memcpy(context->buffer, block_in[i], block_len - i);
-	context->counter = block_len - i;
-	
+	md2_update(context, context->checksum, CONFIG_MD2_BLOCK_LEN);
+
+
+	return context->state; //return a pointer to message digest.
 }
 /**
- * 
+ * MD2 test fuction.
+ * This function test MD2 algorithm with a standard string specified
+ * in RFC 1319.
+ *
+ * \note This test work with official array of 256 byte pemutation
+ * contructed from digits of pi, defined in the RFC 1319.
+ *
  */
-void md2_end(Md2Context *context, void *msg_digest)
+bool md2_test(void)
+{
+
+	Md2Context context;
+
+	const char *test[] =
+	{
+		"",
+		"message digest",
+		"abcdefghijklmnopqrstuvwxyz",
+		"12345678901234567890123456789012345678901234567890123456789012345678901234567890"
+	};
+
+
+	const uint8_t *result[] = {
+		"\x83\x50\xe5\xa3\xe2\x4c\x15\x3d\xf2\x27\x5c\x9f\x80\x69\x27\x73",
+		"\xab\x4f\x49\x6b\xfb\x2a\x53\x0b\x21\x9f\xf3\x30\x31\xfe\x06\xb0",
+		"\x4e\x8d\xdf\xf3\x65\x02\x92\xab\x5a\x41\x08\xc3\xaa\x47\x94\x0b",
+		"\xd5\x97\x6f\x79\xd8\x3d\x3a\x0d\xc9\x80\x6c\x3c\x66\xf3\xef\xd8",
+	};
+
+
+	for (int i = 0; i < countof(test); i++)
+	{
+		md2_init(&context);
+		md2_update(&context, test[i], strlen(test[i]));
+
+		if(memcmp(result[i], md2_end(&context), CONFIG_MD2_BLOCK_LEN))
+			return false;
+	}
+
+	return true;
+}
+
+#if 0
+
+#include <stdio.h>
+int main(int argc, char * argv[])
 {
 
+	if(md2_test())
+		printf("MD2 algorithm work well!\n");
+	else
+		printf("MD2 algorithm doesn't work well.\n");
+
 }
+
+#endif
+