#include <stdio.h>
#include <stdlib.h>
#include <strings.h>

#include <rijndael.h>

static void try(int Nk, int Nb, const char *k, const char *i, const char *o, void (*op)(const void *, const void *, void *))
{
 void *h;
 unsigned char ob[Nb*4];

 h = rijndael_init(Nk,Nb);
 rijndael_set_key(h,k);
 (*op)(h,i,&ob[0]);
 if (bcmp(&ob[0],o,Nb*4))
  { int j;
    printf("Nk=%d Nb=%d\n",Nk,Nb);
    printf("key=");
    for (j=0;j<Nk*4;j++) printf("%02x",((const unsigned char *)k)[j]);
    printf("\n");
    printf("  in=");
    for (j=0;j<Nb*4;j++) printf("%02x",((const unsigned char *)i)[j]);
    printf("\n");
    printf("want=");
    for (j=0;j<Nb*4;j++) printf("%02x",((const unsigned char *)o)[j]);
    printf("\n");
    printf(" out=");
    for (j=0;j<Nb*4;j++)
     { if (ob[j] == ((const unsigned char *)o)[j])
	{ printf("  ");
	}
       else
	{ printf("%02x",ob[j]);
	}
     }
    printf("\n");
    exit(1);
  }
 rijndael_done(h);
}

int main(void);
int main(void)
{
 try(4,4,
	"\x2b\x7e\x15\x16\x28\xae\xd2\xa6\xab\xf7\x15\x88\x09\xcf\x4f\x3c",
	"\x32\x43\xf6\xa8\x88\x5a\x30\x8d\x31\x31\x98\xa2\xe0\x37\x07\x34",
	"\x39\x25\x84\x1d\x02\xdc\x09\xfb\xdc\x11\x85\x97\x19\x6a\x0b\x32",
	&rijndael_encrypt);
 try(4,4,
	"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f",
	"\x00\x11\x22\x33\x44\x55\x66\x77\x88\x99\xaa\xbb\xcc\xdd\xee\xff",
	"\x69\xc4\xe0\xd8\x6a\x7b\x04\x30\xd8\xcd\xb7\x80\x70\xb4\xc5\x5a",
	&rijndael_encrypt);
 try(6,4,
	"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b"
		"\x0c\x0d\x0e\x0f\x10\x11\x12\x13\x14\x15\x16\x17",
	"\x00\x11\x22\x33\x44\x55\x66\x77\x88\x99\xaa\xbb\xcc\xdd\xee\xff",
	"\xdd\xa9\x7c\xa4\x86\x4c\xdf\xe0\x6e\xaf\x70\xa0\xec\x0d\x71\x91",
	&rijndael_encrypt);
 try(8,4,
	"\x00\x01\x02\x03\x04\x05\x06\x07"
		"\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f"
		"\x10\x11\x12\x13\x14\x15\x16\x17"
		"\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f",
	"\x00\x11\x22\x33\x44\x55\x66\x77\x88\x99\xaa\xbb\xcc\xdd\xee\xff",
	"\x8e\xa2\xb7\xca\x51\x67\x45\xbf\xea\xfc\x49\x90\x4b\x49\x60\x89",
	&rijndael_encrypt);
 try(4,4,
	"\x2b\x7e\x15\x16\x28\xae\xd2\xa6\xab\xf7\x15\x88\x09\xcf\x4f\x3c",
	"\x39\x25\x84\x1d\x02\xdc\x09\xfb\xdc\x11\x85\x97\x19\x6a\x0b\x32",
	"\x32\x43\xf6\xa8\x88\x5a\x30\x8d\x31\x31\x98\xa2\xe0\x37\x07\x34",
	&rijndael_decrypt);
 try(4,4,
	"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f",
	"\x69\xc4\xe0\xd8\x6a\x7b\x04\x30\xd8\xcd\xb7\x80\x70\xb4\xc5\x5a",
	"\x00\x11\x22\x33\x44\x55\x66\x77\x88\x99\xaa\xbb\xcc\xdd\xee\xff",
	&rijndael_decrypt);
 try(6,4,
	"\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b"
		"\x0c\x0d\x0e\x0f\x10\x11\x12\x13\x14\x15\x16\x17",
	"\xdd\xa9\x7c\xa4\x86\x4c\xdf\xe0\x6e\xaf\x70\xa0\xec\x0d\x71\x91",
	"\x00\x11\x22\x33\x44\x55\x66\x77\x88\x99\xaa\xbb\xcc\xdd\xee\xff",
	&rijndael_decrypt);
 try(8,4,
	"\x00\x01\x02\x03\x04\x05\x06\x07"
		"\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f"
		"\x10\x11\x12\x13\x14\x15\x16\x17"
		"\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f",
	"\x8e\xa2\xb7\xca\x51\x67\x45\xbf\xea\xfc\x49\x90\x4b\x49\x60\x89",
	"\x00\x11\x22\x33\x44\x55\x66\x77\x88\x99\xaa\xbb\xcc\xdd\xee\xff",
	&rijndael_decrypt);
 return(0);
}