#include <stdio.h>
#include "rijndael.h"

int main()
{ /* test driver */
    int i,nb,nk;
//    char key[32];
//    char block[32];

    gentables();

/*    for (i=0;i<32;i++) key[i]=0;
    key[0]=1;
    for (i=0;i<32;i++) block[i]=i;

    for (nb=4;nb<=8;nb+=2)
        for (nk=4;nk<=8;nk+=2)
    {  
        printf("\nBlock Size= %d bits, Key Size= %d bits\n",nb*32,nk*32);
        gkey(nb,nk,key);

		printf("key = \n");
		for(int tt=0;tt<nk*4; tt++)
		{
			printf("%02x ", (unsigned char)key[tt]);
		}
		printf("\n");

        printf("Plain=   ");
        for (i=0;i<nb*4;i++) printf("%02x",block[i]);
        printf("\n");
        encrypt(block);
        printf("Encrypt= ");
        for (i=0;i<nb*4;i++) printf("%02x",(unsigned char)block[i]);
        printf("\n");
        decrypt(block);
        printf("Decrypt= ");
        for (i=0;i<nb*4;i++) printf("%02x",block[i]);
        printf("\n");
    }
*/
    char block[]	= {
		0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
		0x88, 0x99, 0xaa, 0xbb, 0xcc, 0xdd, 0xee, 0xff,
	/*	0x0, 0x1, 0x2, 0x3, 0x4, 0x5, 0x6, 0x7, 
		0x8, 0x9, 0x10, 0x11, 0x12, 0x13, 0x14, 0x15*/
	};
	char key[] = {
		0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
		0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f,
		0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17,
		/*0x18, 0x19, 0x1a, 0x1b, 0x1c, 0x1d, 0x1e, 0x1f*/};
	gentables();
	gkey(sizeof(block)*8/32, sizeof(key)*8/32, key);

	printf("\nkey = \n");
	for (i=0;i<sizeof(key);i++) printf("%02x", (unsigned char )key[i]);
	
	printf("\nplaintext = \n");
	for (i=0;i<sizeof(block);i++) printf("%02x", (unsigned char )block[i]);

	encrypt(block);
    
	printf("\nciphertext = \n");
	for (i=0;i<sizeof(block);i++) printf("%02x", (unsigned char )block[i]);

	decrypt(block);

	printf("\nafter decrypt = \n");
	for (i=0;i<sizeof(block);i++) printf("%02x", (unsigned char )block[i]);

	return 0;
}