X-Git-Url: https://git.cryptolib.org/?a=blobdiff_plain;f=test_src%2Fmain-rabbit-test.c;h=4f66c66702dcb6c85ccdf29fed05c0ca2d421594;hb=ca591a22fb1e3500567c3d0b730009b44ba33c2c;hp=b60de60decc1d9941eb30d434f228eace58e614d;hpb=35dc9566e40c9f68fa216c70eaa6d5b0597448fe;p=avr-crypto-lib.git

diff --git a/test_src/main-rabbit-test.c b/test_src/main-rabbit-test.c
index b60de60..4f66c66 100644
--- a/test_src/main-rabbit-test.c
+++ b/test_src/main-rabbit-test.c
@@ -17,25 +17,16 @@
     along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
-
-#include "config.h"
-
-#include "uart_i.h"
-#include "debug.h"
+#include "main-test-common.h"
 
 #include "rabbit.h"
-#include "cli.h"
 #include "performance_test.h"
 
 #include "scal_rabbit.h"
 #include "scal-basic.h"
 #include "scal-nessie.h"
 
-#include <stdlib.h>
-#include <stdint.h>
-#include <string.h>
-
-char* algo_name = "Rabbit";
+const char* algo_name = "Rabbit";
 
 /*****************************************************************************
  *  additional validation-functions											 *
@@ -60,12 +51,14 @@ void test_vector(void* key, void* iv, const void* ref){
 		cli_hexdump(ctx.buffer, 16);
 	}
 	ctx.buffer_idx=16;
+	rabbit_gen(&ctx);
 	if(!ref || (memcmp_P(ctx.buffer, ref, 16))!=0){
 		fail = 1;
 		cli_putstr_P(PSTR("\r\n S[0]: "));
 		cli_hexdump(ctx.buffer, 16);
 	}
 	ctx.buffer_idx=16;
+	rabbit_gen(&ctx);
 	if(!ref || (memcmp_P(ctx.buffer, ref, 16))!=0){
 		fail = 1;
 		cli_putstr_P(PSTR("\r\n S[0]: "));
@@ -212,7 +205,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_rabbit},
 	{ performance_str, NULL, testrun_performance_rabbit},
 	{ first_str,       NULL, nessie_first},
@@ -222,15 +215,11 @@ cmdlist_entry_t cmdlist[] PROGMEM = {
 };
 
 int main (void){
-	DEBUG_INIT();
-
-	cli_rx = (cli_rx_fpt)uart0_getc;
-	cli_tx = (cli_tx_fpt)uart0_putc;
-	for(;;){
-		cli_putstr_P(PSTR("\r\n\r\nCrypto-VS ("));
-		cli_putstr(algo_name);
-		cli_putstr_P(PSTR(")\r\nloaded and running\r\n"));
-		cmd_interface(cmdlist);
+    main_setup();
+
+    for(;;){
+        welcome_msg(algo_name);
+        cmd_interface(cmdlist);
 	}
 }