65816-crypto/aestest.c

193 lines
5.6 KiB
C

/*
* Copyright (c) 2017 Stephen Heumann
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <MiscTool.h>
#include <Memory.h>
#include <orca.h>
#include "aes.h"
void print_hexbytes(char *prefix, unsigned char *data, unsigned int n) {
int i;
printf("%s", prefix);
for (i = 0; i < n; i++) {
printf("%02x ", data[i]);
}
printf("\n");
}
void aes128_test(void) {
int i;
struct aes_context aes_context = {
{0x00,0x11,0x22,0x33,0x44,0x55,0x66,0x77,0x88,0x99,0xaa,0xbb,0xcc,0xdd,0xee,0xff},
{0},
{0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f}
};
print_hexbytes("Input: ", aes_context.data, 16);
print_hexbytes("Key: ", aes_context.key, 16);
aes128_expandkey(&aes_context);
#ifdef PRINT_ROUND_KEYS
for (i = 1; i <= 10; i++) {
printf("Round key %2i: ", i);
print_hexbytes("", aes_context.key + i*16, 16);
}
#endif
aes_encrypt(&aes_context);
print_hexbytes("Output: ", aes_context.data, 16);
aes128_decrypt(&aes_context);
print_hexbytes("Decrypted: ", aes_context.data, 16);
}
void aes192_test(void) {
int i;
struct aes_context aes_context = {
{0x00,0x11,0x22,0x33,0x44,0x55,0x66,0x77,0x88,0x99,0xaa,0xbb,0xcc,0xdd,0xee,0xff},
{0},
{0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f,
0x10,0x11,0x12,0x13,0x14,0x15,0x16,0x17}
};
print_hexbytes("Input: ", aes_context.data, 16);
print_hexbytes("Key: ", aes_context.key, 24);
aes192_expandkey(&aes_context);
#ifdef PRINT_ROUND_KEYS
for (i = 1; i <= 12; i++) {
printf("Round key %2i: ", i);
print_hexbytes("", aes_context.key + i*16, 16);
}
#endif
aes_encrypt(&aes_context);
print_hexbytes("Output: ", aes_context.data, 16);
aes192_decrypt(&aes_context);
print_hexbytes("Decrypted: ", aes_context.data, 16);
}
void aes256_test(void) {
int i;
struct aes_context aes_context = {
{0x00,0x11,0x22,0x33,0x44,0x55,0x66,0x77,0x88,0x99,0xaa,0xbb,0xcc,0xdd,0xee,0xff},
{0},
{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}
};
print_hexbytes("Input: ", aes_context.data, 16);
print_hexbytes("Key: ", aes_context.key, 32);
aes256_expandkey(&aes_context);
#ifdef PRINT_ROUND_KEYS
for (i = 1; i <= 14; i++) {
printf("Round key %2i: ", i);
print_hexbytes("", aes_context.key + i*16, 16);
}
#endif
aes_encrypt(&aes_context);
print_hexbytes("Output: ", aes_context.data, 16);
aes256_decrypt(&aes_context);
print_hexbytes("Decrypted: ", aes_context.data, 16);
}
unsigned long aes128_time_test(unsigned int iters) {
unsigned int i;
unsigned long tick_count;
long double bytes_per_sec;
struct aes_context *aes_context, **aes_context_hndl;
static struct aes_context aes_context_init = {
{0x00,0x11,0x22,0x33,0x44,0x55,0x66,0x77,0x88,0x99,0xaa,0xbb,0xcc,0xdd,0xee,0xff},
{0},
{0x00,0x01,0x02,0x03,0x04,0x05,0x06,0x07,0x08,0x09,0x0a,0x0b,0x0c,0x0d,0x0e,0x0f}
};
aes_context_hndl = (struct aes_context **)NewHandle(sizeof(struct aes_context),
userid(), attrFixed|attrPage|attrBank|attrNoCross, 0x000000);
if (toolerror())
return 0;
aes_context = *aes_context_hndl;
*aes_context = aes_context_init;
aes128_expandkey(aes_context);
tick_count = GetTick();
for (i = 0; i < iters; i++) {
aes_encrypt(aes_context);
}
tick_count = GetTick() - tick_count;
bytes_per_sec = (long double)iters * 16 * 60 / tick_count;
printf("Encryption: %u iterations takes %lu ticks (%lf bytes/sec)\n",
iters, tick_count, bytes_per_sec);
tick_count = GetTick();
for (i = 0; i < iters; i++) {
aes128_decrypt(aes_context);
}
tick_count = GetTick() - tick_count;
bytes_per_sec = (long double)iters * 16 * 60 / tick_count;
printf("Decryption: %u iterations takes %lu ticks (%lf bytes/sec)\n",
iters, tick_count, bytes_per_sec);
print_hexbytes("Decrypted: ", aes_context->data, 16);
}
int main(int argc, char **argv) {
unsigned int test_iters = 0;
if (argc > 1) {
errno = 0;
test_iters = strtoul(argv[1], NULL, 10);
if (argc > 2 || errno != 0) {
fprintf(stderr, "Usage: aestest [test_iters]\n");
exit(1);
}
}
printf("AES-128 test:\n");
aes128_test();
printf("AES-192 test:\n");
aes192_test();
printf("AES-256 test:\n");
aes256_test();
if (test_iters)
aes128_time_test(test_iters);
}