#include <stdio.h>
#include "k5-int.h"
static char key[32];
static char plain[16], cipher[16], zero[16];
static krb5_keyblock enc_key;
static krb5_data ivec;
static krb5_data in, out;
static void init()
{
enc_key.contents = key;
enc_key.length = 16;
ivec.data = zero;
ivec.length = 16;
in.data = plain;
in.length = 16;
out.data = cipher;
out.length = 16;
}
static void enc()
{
krb5int_aes_encrypt(&enc_key, &ivec, &in, &out);
}
static void hexdump(const char *label, const char *cp, int len)
{
printf("%s=", label);
while (len--) printf("%02X", 0xff & *cp++);
printf("\n");
}
static void set_bit(char *ptr, int bitnum)
{
int bytenum;
bytenum = bitnum / 8;
bitnum %= 8;
ptr[bytenum] = 1 << (7 - bitnum);
}
static void vk_test_1(int len)
{
int i;
enc_key.length = len;
printf("\nKEYSIZE=%d\n\n", len * 8);
memset(plain, 0, sizeof(plain));
hexdump("PT", plain, 16);
for (i = 0; i < len * 8; i++) {
memset(key, 0, len);
set_bit(key, i);
printf("\nI=%d\n", i+1);
hexdump("KEY", key, len);
enc();
hexdump("CT", cipher, 16);
}
printf("\n==========\n");
}
static void vk_test()
{
vk_test_1(16);
vk_test_1(32);
}
static void vt_test_1(int len)
{
int i;
enc_key.length = len;
printf("\nKEYSIZE=%d\n\n", len * 8);
memset(key, 0, len);
hexdump("KEY", key, len);
for (i = 0; i < 16 * 8; i++) {
memset(plain, 0, sizeof(plain));
set_bit(plain, i);
printf("\nI=%d\n", i+1);
hexdump("PT", plain, 16);
enc();
hexdump("CT", cipher, 16);
}
printf("\n==========\n");
}
static void vt_test()
{
vt_test_1(16);
vt_test_1(32);
}
int main (int argc, char *argv[])
{
if (argc > 2 || (argc == 2 && strcmp(argv[1], "-k"))) {
fprintf(stderr,
"usage:\t%s -k\tfor variable-key tests\n"
" or:\t%s \tfor variable-plaintext tests\n",
argv[0], argv[0]);
return 1;
}
init();
if (argc == 2)
vk_test();
else
vt_test();
return 0;
}