/* main-bmw-test.c */
/*
    This file is part of the AVR-Crypto-Lib.
    Copyright (C) 2006-2015 Daniel Otte (bg@nerilex.org)

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/
/*
 * BlueMidnightWish test-suit
 *
*/

#include "main-test-common.h"

#include "bmw_small.h"
#include "bmw_large.h"
#include "hfal_bmw_small.h"
#include "hfal_bmw_large.h"

#include "shavs.h"
#include "nessie_hash_test.h"
#include "hfal-nessie.h"
#include "hfal-performance.h"
#include "hfal-test.h"
#include "performance_test.h"

char *algo_name = "BlueMidnightWish";


const hfdesc_t *const algolist[] PROGMEM = {
	(hfdesc_t*)&bmw224_desc,
	(hfdesc_t*)&bmw256_desc,
	(hfdesc_t*)&bmw384_desc,
	(hfdesc_t*)&bmw512_desc,
	NULL
};

/*****************************************************************************
 *  additional validation-functions											 *
 *****************************************************************************/

void performance_bmw(void){
	hfal_performance_multiple(algolist);
}

void testrun_nessie_bmw(void){
	hfal_nessie_multiple(algolist);
}

void bmw224_test(void *msg, uint32_t length_b){
	hfal_test(&bmw224_desc, msg, length_b);
}

void bmw256_test(void *msg, uint32_t length_b){
	hfal_test(&bmw256_desc, msg, length_b);
}
void bmw384_test(void *msg, uint32_t length_b){
	hfal_test(&bmw384_desc, msg, length_b);
}

void bmw512_test(void *msg, uint32_t length_b){
	hfal_test(&bmw512_desc, msg, length_b);
}

void test506(void){
	/* Testvector of length = 506 from short KAT */
	uint8_t v[] = {
		0xB9, 0xE3, 0xE2, 0x75, 0x5E, 0xD1, 0x21, 0x53,
		0x81, 0xB2, 0x45, 0x57, 0xEE, 0x14, 0xF8, 0xCD,
		0x26, 0x87, 0xA0, 0x71, 0xAE, 0xB3, 0xC7, 0x4F,
		0x42, 0x1C, 0xFE, 0xA3, 0xCF, 0xF8, 0xA6, 0xEA,
		0x0D, 0x4B, 0xDA, 0x2A, 0xD4, 0xBD, 0x82, 0x43,
		0xCB, 0xA7, 0x2B, 0x48, 0x1C, 0xBD, 0x52, 0x6E,
		0x1E, 0xFA, 0x1D, 0x9F, 0xD4, 0x32, 0xB9, 0x87,
		0xE8, 0x12, 0x65, 0x82, 0x44, 0x29, 0xBA, 0xC0 };
	bmw512_test(v, 506);
}

void test507(void){
	/* Testvector of length = 506 from short KAT */
	uint8_t v[] = {
		0xE8, 0x33, 0x25, 0x35, 0xE9, 0xA2, 0x05, 0x24,
		0x67, 0xA1, 0x02, 0x05, 0xC8, 0x70, 0x00, 0xBF,
		0xC6, 0xA6, 0x8D, 0x73, 0x96, 0x2E, 0x69, 0xE4,
		0xE5, 0x6F, 0x10, 0xBC, 0x79, 0xC6, 0x33, 0xC5,
		0x2F, 0x4D, 0x00, 0x74, 0xD6, 0x07, 0x75, 0x95,
		0xB6, 0x60, 0x07, 0x2E, 0x10, 0x74, 0x0D, 0xFA,
		0x66, 0xBC, 0x13, 0x20, 0x46, 0x9A, 0x31, 0x96,
		0xE0, 0x21, 0xE1, 0x32, 0x7C, 0xC0, 0x3A, 0xE0 };
	bmw512_test(v, 507);
}

void testrun_stdtest_bmw(void){
	char *msg0 = "abc";
	char *msg1 = "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq";
	bmw224_test(msg0, strlen(msg0)*8);
	bmw224_test(msg1, strlen(msg1)*8);
	bmw256_test(msg0, strlen(msg0)*8);
	bmw256_test(msg1, strlen(msg1)*8);
	bmw384_test(msg0, strlen(msg0)*8);
	bmw384_test(msg1, strlen(msg1)*8);
	bmw512_test(msg0, strlen(msg0)*8);
	bmw512_test(msg1, strlen(msg1)*8);
}

void bmw256_short_test(void *msg, uint32_t length_b){
	bmw256_test("abc", 3*8);
}


void testshort(void){
	char *msg0 = "abc";
	bmw224_test(msg0, strlen(msg0)*8);
}

void testlshort(void){
	char *msg0 = "abc";
	bmw384_test(msg0, strlen(msg0)*8);
}

/*****************************************************************************
 *  main																	 *
 *****************************************************************************/

const char nessie_str[]      PROGMEM = "nessie";
const char test_str[]        PROGMEM = "test";
const char test224_str[]     PROGMEM = "test224";
const char test256_str[]     PROGMEM = "test256";
const char test384_str[]     PROGMEM = "test384";
const char test512_str[]     PROGMEM = "test512";
const char test506_str[]     PROGMEM = "506";
const char test507_str[]     PROGMEM = "507";
const char testshort_str[]   PROGMEM = "short";
const char testlshort_str[]  PROGMEM = "lshort";
const char performance_str[] PROGMEM = "performance";
const char echo_str[]        PROGMEM = "echo";
const char shavs_list_str[]  PROGMEM = "shavs_list";
const char shavs_set_str[]   PROGMEM = "shavs_set";
const char shavs_test1_str[] PROGMEM = "shavs_test1";
const char shavs_test3_str[] PROGMEM = "shavs_test3";

const cmdlist_entry_t cmdlist[] PROGMEM = {
	{ nessie_str,                  NULL, testrun_nessie_bmw            },
	{ test_str,                    NULL, testrun_stdtest_bmw           },
	{ testshort_str,               NULL, testshort                     },
	{ testlshort_str,              NULL, testlshort                    },
	{ test506_str,                 NULL, test506                       },
	{ test507_str,                 NULL, test507                       },
	{ performance_str,             NULL, performance_bmw               },
	{ shavs_list_str,              NULL, shavs_listalgos               },
	{ shavs_set_str,           (void*)1, (void_fpt)shavs_setalgo       },
	{ shavs_test1_str,             NULL, shavs_test1                   },
	{ shavs_test3_str,             NULL, shavs_test3                   },
	{ echo_str,                (void*)1, (void_fpt)echo_ctrl           },
	{ NULL,                        NULL, NULL                          }
};

int main (void){
    main_setup();

    shavs_algolist=(hfdesc_t**)algolist;
	shavs_algo=(hfdesc_t*)&bmw256_desc;
	for(;;){
	    welcome_msg(algo_name);
		cmd_interface(cmdlist);
	}
}