#include <stdio.h>
#include <string.h>

#include "square.h"

static void squarePrint (const byte *block, const char *tag)
{
	printf ("%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x %s\n",
		block[ 0], block[ 1], block[ 2], block[ 3],
		block[ 4], block[ 5], block[ 6], block[ 7],
		block[ 8], block[ 9], block[10], block[11],
		block[12], block[13], block[14], block[15],
		tag);
} /* squarePrint */


int main (void)
{
	int i;
	squareBlock key, block;
	squareKeySchedule roundKeys_e;

	printf ("===========================================================================\n\n"
			"Validation data set for Square v1.0\n\n");

	printf ("===========================================================================\n\n"
			"Encryption of the null data block with every key where a single bit is set:\n\n");
	memset (block, 0, SQUARE_BLOCKSIZE);
	squarePrint (block, "plaintext\n");
	for (i = 0; i < 128; i++) {
		memset (key, 0, SQUARE_BLOCKSIZE);
		key[i/8] |= 1 << (7 - i%8); /* set only the i-th bit of the i-th test key */
		squarePrint (key, "key");
		memset (block, 0, SQUARE_BLOCKSIZE);
		squareExpandKey (key, roundKeys_e);
		squareEncrypt ((word32 *)block, roundKeys_e);
		squarePrint (block, "ciphertext\n\n");
	}

	printf ("===========================================================================\n\n"
			"Encryption of a sample data block with every key where a single bit is set:\n\n");
	memcpy (block, "\x0f\x1e\x2d\x3c\x4b\x5a\x69\x78\x87\x96\xa5\xb4\xc3\xd2\xe1\xf0", SQUARE_BLOCKSIZE);
	squarePrint (block, "plaintext\n");
	for (i = 0; i < 128; i++) {
		memset (key, 0, SQUARE_BLOCKSIZE);
		key[i/8] |= 1 << (7 - i%8); /* set only the i-th bit of the i-th test key */
		squarePrint (key, "key");
		memcpy (block, "\x0f\x1e\x2d\x3c\x4b\x5a\x69\x78\x87\x96\xa5\xb4\xc3\xd2\xe1\xf0", SQUARE_BLOCKSIZE);
		squareExpandKey (key, roundKeys_e);
		squareEncrypt ((word32 *)block, roundKeys_e);
		squarePrint (block, "ciphertext\n\n");
	}

	printf ("===========================================================================\n\n");

	return 0;
}