X-Git-Url: https://git.cryptolib.org/?a=blobdiff_plain;f=test_src%2Fmain-grain-test.c;h=99ab49380e895e0928ab3cd581f657fb90fa174e;hb=a45c328791dc391e6432051ee0c7972ef7fd8d9a;hp=036faad94490a06930f813491e2da0f7a57eaaf1;hpb=572b35bb7409fb362441e0812cb62eab9f2411f0;p=avr-crypto-lib.git

diff --git a/test_src/main-grain-test.c b/test_src/main-grain-test.c
index 036faad..99ab493 100644
--- a/test_src/main-grain-test.c
+++ b/test_src/main-grain-test.c
@@ -21,20 +21,14 @@
  * 
 */
 
-#include "config.h"
-#include "serial-tools.h"
-#include "uart.h"
-#include "debug.h"
-#include "cli.h"
+#include "main-test-common.h"
 
 #include "grain.h"
-#include "nessie_stream_test.h"
+#include "scal_grain.h"
+#include "scal-basic.h"
+#include "scal-nessie.h"
 #include "performance_test.h"
 
-#include <stdlib.h>
-#include <stdint.h>
-#include <string.h>
-
 char* algo_name = "Grain";
 
 /*****************************************************************************
@@ -64,15 +58,8 @@ uint8_t grain_getbyte_dummy_rev(grain_ctx_t* ctx){
 }
 
 void testrun_nessie_grain(void){
-	nessie_stream_ctx.outsize_b =   8; /* actually unused */
-	nessie_stream_ctx.keysize_b =  80; /* this is the one we have refrence vectors for */
-	nessie_stream_ctx.ivsize_b  =  64;
-	nessie_stream_ctx.name = algo_name;
-	nessie_stream_ctx.ctx_size_B = sizeof(grain_ctx_t);
-	nessie_stream_ctx.cipher_genctx = (nessie_stream_genctx_fpt)grain_genctx_dummy;
-	nessie_stream_ctx.cipher_enc = (nessie_stream_genenc_fpt)grain_getbyte_dummy_rev;
-	
-	nessie_stream_run();	
+	scal_nessie_set_estream(1);
+	scal_nessie_run(&grain_desc);
 }
 
 
@@ -83,17 +70,17 @@ void testrun_std_grain(void){
 	/* 1 */
 	memset(key, 0, 10);
 	memset(iv, 0, 8);
-	uart_putstr_P(PSTR("\r\n=== std test ==="));
-	uart_putstr_P(PSTR("\r\n key: "));
-	uart_hexdump(key, 10);
-	uart_putstr_P(PSTR("\r\n iv:  "));
-	uart_hexdump(key, 8);
+	cli_putstr_P(PSTR("\r\n=== std test ==="));
+	cli_putstr_P(PSTR("\r\n key: "));
+	cli_hexdump(key, 10);
+	cli_putstr_P(PSTR("\r\n iv:  "));
+	cli_hexdump(key, 8);
 	grain_init(key, iv, &ctx);
 	for(i=0; i<10; ++i){
 		out[i] = grain_getbyte_dummy(&ctx);
 	}
-	uart_putstr_P(PSTR("\r\n out: "));
-	uart_hexdump(out, 10);
+	cli_putstr_P(PSTR("\r\n out: "));
+	cli_hexdump(out, 10);
 	
 	/* 2 */
 	for(i=0; i<8; ++i){
@@ -105,19 +92,19 @@ void testrun_std_grain(void){
 	for(i=0; i<8; ++i){
 		iv[i] = i*0x22+1;
 	}
-	uart_putstr_P(PSTR("\r\n\r\n key: "));
-	uart_hexdump(key, 10);
-	uart_putstr_P(PSTR("\r\n iv:  "));
-	uart_hexdump(key, 8);
+	cli_putstr_P(PSTR("\r\n\r\n key: "));
+	cli_hexdump(key, 10);
+	cli_putstr_P(PSTR("\r\n iv:  "));
+	cli_hexdump(key, 8);
 	grain_init(key, iv, &ctx);
 	for(i=0; i<10; ++i){
 		out[i] = grain_getbyte_dummy(&ctx);
 	}
-	uart_putstr_P(PSTR("\r\n out: "));
-	uart_hexdump(out, 10);
+	cli_putstr_P(PSTR("\r\n out: "));
+	cli_hexdump(out, 10);
 	
 	
-	uart_putstr_P(PSTR("\r\n\r\n"));
+	cli_putstr_P(PSTR("\r\n\r\n"));
 }
 
 void testrun_performance_grain(void){
@@ -135,18 +122,18 @@ void testrun_performance_grain(void){
 	startTimer(1);
 	grain_init(key, iv, &ctx);
 	t = stopTimer();
-	uart_putstr_P(PSTR("\r\n\tctx-gen time: "));
+	cli_putstr_P(PSTR("\r\n\tctx-gen time: "));
 	ultoa((unsigned long)t, str, 10);
-	uart_putstr(str);	
+	cli_putstr(str);	
 	
 	startTimer(1);
 	grain_enc(&ctx);
 	t = stopTimer();
-	uart_putstr_P(PSTR("\r\n\tencrypt time: "));
+	cli_putstr_P(PSTR("\r\n\tencrypt time: "));
 	ultoa((unsigned long)t, str, 10);
-	uart_putstr(str);	
+	cli_putstr(str);	
 	
-	uart_putstr_P(PSTR("\r\n"));
+	cli_putstr_P(PSTR("\r\n"));
 }
 
 /*****************************************************************************
@@ -158,7 +145,7 @@ const char test_str[]        PROGMEM = "test";
 const char performance_str[] PROGMEM = "performance";
 const char echo_str[]        PROGMEM = "echo";
 
-cmdlist_entry_t cmdlist[] PROGMEM = {
+const cmdlist_entry_t cmdlist[] PROGMEM = {
 	{ nessie_str,      NULL, testrun_nessie_grain },
 	{ test_str,        NULL, testrun_std_grain},
 	{ performance_str, NULL, testrun_performance_grain},
@@ -167,14 +154,10 @@ cmdlist_entry_t cmdlist[] PROGMEM = {
 };
 
 int main (void){
-	DEBUG_INIT();
-	uart_putstr("\r\n");
-	cli_rx = uart_getc;
-	cli_tx = uart_putc;	 	
-	for(;;){
-		uart_putstr_P(PSTR("\r\n\r\nCrypto-VS ("));
-		uart_putstr(algo_name);
-		uart_putstr_P(PSTR(")\r\nloaded and running\r\n"));
-		cmd_interface(cmdlist);
+    main_setup();
+
+    for(;;){
+        welcome_msg(algo_name);
+        cmd_interface(cmdlist);
 	}
 }