/* See LICENSE file for copyright and license details. */
#ifdef SUPPORT_BLAKE2S
# define TEST
# include "../common.h"


static struct testcase testcases[] = {
	{1, 0, "", "69217a3079908094e11121d042354a7c1f55b6482ca1a51e1b250dfd1ed0eef9"}
};


#else
# define TEST_UNSUPPORTED
# include "../common.h"
#endif


static int
kat_test(void)
{
	TEST_MAIN("BLAKE2s", BLAKE2S, 0);
}


int
main(void)
{
#ifdef SUPPORT_BLAKE2S
	struct libhashsum_hasher hasher1, hasher2;
	char salt1[8], salt2[8], salt3[8], key[32], buf[1024];
	size_t i;
#endif

	int r = kat_test();
	if (r)
		return r;

#ifdef SUPPORT_BLAKE2S
	memset(salt1, 0, sizeof(salt1));
	memset(salt2, 0xAB, sizeof(salt2));
	memset(salt3, 0xCD, sizeof(salt3));

	ASSERT(!libhashsum_init_blake2s_hasher(&hasher1, 0, NULL, NULL, NULL, 0));
	ASSERT(!libhashsum_init_blake2s_hasher(&hasher2, 0, salt1, NULL, NULL, 0));
	ASSERT(!strcmp(hasher1.algorithm_string, "BLAKE2s"));
	ASSERT(!strcmp(hasher2.algorithm_string, "BLAKE2s"));
	ASSERT(!hasher1.finalise_const(&hasher1, NULL, 0, 0));
	ASSERT(!hasher2.finalise_const(&hasher2, NULL, 0, 0));
	ASSERT(!memcmp(hasher1.hash_output, hasher2.hash_output, hasher1.hash_size));

	ASSERT(!libhashsum_init_blake2s_hasher(&hasher1, 0, NULL, NULL, NULL, 0));
	ASSERT(!libhashsum_init_blake2s_hasher(&hasher2, 0, salt1, NULL, NULL, 0));
	ASSERT(!hasher1.finalise_const(&hasher1, "testdata", 8U, 0));
	ASSERT(!hasher2.finalise_const(&hasher2, "testdata", 8U, 0));
	ASSERT(!memcmp(hasher1.hash_output, hasher2.hash_output, hasher1.hash_size));

	ASSERT(!libhashsum_init_blake2s_hasher(&hasher1, 0, salt1, NULL, NULL, 0));
	ASSERT(!libhashsum_init_blake2s_hasher(&hasher2, 0, salt2, NULL, NULL, 0));
	ASSERT(!strcmp(hasher1.algorithm_string, "BLAKE2s"));
	ASSERT(!strcmp(hasher2.algorithm_string, "BLAKE2s[salt=abababababababab]"));
	ASSERT(!hasher1.finalise_const(&hasher1, NULL, 0, 0));
	ASSERT(!hasher2.finalise_const(&hasher2, NULL, 0, 0));
	ASSERT(memcmp(hasher1.hash_output, hasher2.hash_output, hasher1.hash_size));

	ASSERT(!libhashsum_init_blake2s_hasher(&hasher1, 0, salt1, NULL, NULL, 0));
	ASSERT(!libhashsum_init_blake2s_hasher(&hasher2, 0, salt2, NULL, NULL, 0));
	ASSERT(!hasher1.finalise_const(&hasher1, "testdata", 8U, 0));
	ASSERT(!hasher2.finalise_const(&hasher2, "testdata", 8U, 0));
	ASSERT(memcmp(hasher1.hash_output, hasher2.hash_output, hasher1.hash_size));

	ASSERT(!libhashsum_init_hasher_from_string(&hasher1, "BLAKE2s[n=64,salt=0000000000000000]"));
	ASSERT(!libhashsum_init_hasher_from_string(&hasher2, "BLAKE2s[n=64,salt=abababababababab]"));
	ASSERT(!strcmp(hasher1.algorithm_string, "BLAKE2s[n=64]"));
	ASSERT(!strcmp(hasher2.algorithm_string, "BLAKE2s[n=64,salt=abababababababab]"));
	ASSERT(!hasher1.finalise_const(&hasher1, NULL, 0, 0));
	ASSERT(!hasher2.finalise_const(&hasher2, NULL, 0, 0));
	ASSERT(memcmp(hasher1.hash_output, hasher2.hash_output, hasher1.hash_size));
	ASSERT(hasher1.hash_size == 64U / 8U);

	ASSERT(!libhashsum_init_blake2s_hasher(&hasher1, 0, NULL, salt1, NULL, 0));
	ASSERT(!libhashsum_init_blake2s_hasher(&hasher2, 0, NULL, salt2, NULL, 0));
	ASSERT(!strcmp(hasher1.algorithm_string, "BLAKE2s"));
	ASSERT(!strcmp(hasher2.algorithm_string, "BLAKE2s[pepper=abababababababab]"));
	ASSERT(!hasher1.finalise_const(&hasher1, NULL, 0, 0));
	ASSERT(!hasher2.finalise_const(&hasher2, NULL, 0, 0));
	ASSERT(memcmp(hasher1.hash_output, hasher2.hash_output, hasher1.hash_size));

	ASSERT(!libhashsum_init_blake2s_hasher(&hasher1, 0, NULL, NULL, NULL, 0));
	ASSERT(!libhashsum_init_blake2s_hasher(&hasher2, 0, salt2, salt3, NULL, 0));
	ASSERT(!strcmp(hasher1.algorithm_string, "BLAKE2s"));
	ASSERT(!strcmp(hasher2.algorithm_string, "BLAKE2s[salt=abababababababab,pepper=cdcdcdcdcdcdcdcd]"));
	ASSERT(!hasher1.finalise_const(&hasher1, NULL, 0, 0));
	ASSERT(!hasher2.finalise_const(&hasher2, NULL, 0, 0));
	ASSERT(memcmp(hasher1.hash_output, hasher2.hash_output, hasher1.hash_size));

	ASSERT(!libhashsum_init_hasher_from_string(&hasher1, "BLAKE2s[salt=0101010101010101,pepper=0202020202020202]"));
	ASSERT(!strcmp(hasher1.algorithm_string, "BLAKE2s[salt=0101010101010101,pepper=0202020202020202]"));

	ASSERT(!libhashsum_init_blake2s_hasher(&hasher1, 0, salt2, NULL, NULL, 0));
	ASSERT(!libhashsum_init_blake2s_hasher(&hasher2, 0, NULL, salt2, NULL, 0));
	ASSERT(!hasher1.finalise_const(&hasher1, "testdata", 8U, 0));
	ASSERT(!hasher2.finalise_const(&hasher2, "testdata", 8U, 0));
	ASSERT(memcmp(hasher1.hash_output, hasher2.hash_output, hasher1.hash_size));

	for (i = 0; i < sizeof(key); i++)
		key[i] = (char)i;
	ASSERT(!libhashsum_init_blake2s_hasher(&hasher1, 0, NULL, NULL, key, 16 * 8U));
	ASSERT(!libhashsum_init_blake2s_hasher(&hasher2, 0, NULL, NULL, key, 32 * 8U));
	ASSERT(!strcmp(hasher1.algorithm_string, "BLAKE2s[key=000102030405060708090a0b0c0d0e0f]"));
	ASSERT(!strcmp(hasher2.algorithm_string, "BLAKE2s[key=000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f]"));
	ASSERT(!hasher1.finalise_const(&hasher1, NULL, 0, 0));
	ASSERT(!hasher2.finalise_const(&hasher2, NULL, 0, 0));
	ASSERT(memcmp(hasher1.hash_output, hasher2.hash_output, hasher1.hash_size));
	hex(buf, hasher2.hash_output, hasher2.hash_size);
	ASSERT(!strcmp(buf, "48a8997da407876b3d79c0d92325ad3b89cbb754d86ab71aee047ad345fd2c49"));

	ASSERT(!libhashsum_init_hasher_from_string(&hasher1, "BLAKE2s[pepper=0202020202020202,key=0123456789]"));
	ASSERT(!strcmp(hasher1.algorithm_string, "BLAKE2s[pepper=0202020202020202,key=0123456789]"));
#endif

	return 0;
}