int main(void)
{
int i;
unsigned char key1[16], key2[16], tinit[16];
unsigned char *pt, *ct;
printf("aes_xts_128_enc_perf:\n");
pt = malloc(TEST_LEN);
ct = malloc(TEST_LEN);
if (NULL == pt || NULL == ct) {
printf("malloc of testsize failed\n");
return -1;
}
mk_rand_data(key1, key2, tinit, pt, TEST_LEN);
XTS_AES_128_enc(key2, key1, tinit, TEST_LEN, pt, ct);
struct perf start, stop;
perf_start(&start);
for (i = 0; i < TEST_LOOPS; i++) {
XTS_AES_128_enc(key2, key1, tinit, TEST_LEN, pt, ct);
}
perf_stop(&stop);
printf("aes_xts_128_enc" TEST_TYPE_STR ": ");
perf_print(stop, start, (long long)TEST_LEN * i);
return 0;
}
int main(void)
{
int t, n;
unsigned char key1[16], key2[16], tinit[16];
unsigned char *pt, *ct, *dt;
int align, size, min_size;
unsigned char *efence_pt;
unsigned char *efence_ct;
unsigned char *efence_dt;
unsigned char *origin_pt;
unsigned char *origin_ct;
unsigned char *origin_dt;
unsigned char key1_exp_enc[16 * 11], key1_exp_dec[16 * 11];
unsigned char key2_exp_tw[16 * 11];
int i;
printf("aes_xts_128 enc/dec rand test, %d sets of %d max: ", RANDOMS, TEST_LEN);
pt = malloc(TEST_LEN);
ct = malloc(TEST_LEN);
dt = malloc(TEST_LEN);
if (NULL == pt || NULL == ct || NULL == dt) {
printf("malloc of testsize failed\n");
return -1;
}
mk_rand_data(key1, key2, tinit, pt, TEST_LEN);
XTS_AES_128_enc(key2, key1, tinit, TEST_LEN, pt, ct);
XTS_AES_128_dec(key2, key1, tinit, TEST_LEN, ct, dt);
if (memcmp(pt, dt, TEST_LEN)) {
printf("fail\n");
return -1;
}
putchar('.');
// Do tests with random data, keys and message size
for (t = 0; t < RANDOMS; t++) {
n = rand() % (TEST_LEN);
if (n < 17)
continue;
mk_rand_data(key1, key2, tinit, pt, n);
XTS_AES_128_enc(key2, key1, tinit, n, pt, ct);
XTS_AES_128_dec(key2, key1, tinit, n, ct, dt);
if (memcmp(pt, dt, n)) {
printf("fail rand %d, size %d\n", t, n);
return -1;
}
putchar('.');
fflush(0);
}
// Run tests at end of buffer for Electric Fence
align = 1;
min_size = 16;
for (size = 0; size <= TEST_SIZE - min_size; size += align) {
// Line up TEST_SIZE from end
efence_pt = pt + TEST_LEN - TEST_SIZE + size;
efence_ct = ct + TEST_LEN - TEST_SIZE + size;
efence_dt = dt + TEST_LEN - TEST_SIZE + size;
mk_rand_data(key1, key2, tinit, efence_pt, TEST_SIZE - size);
XTS_AES_128_enc(key2, key1, tinit, TEST_SIZE - size, efence_pt, efence_ct);
XTS_AES_128_dec(key2, key1, tinit, TEST_SIZE - size, efence_ct, efence_dt);
if (memcmp(efence_pt, efence_dt, TEST_SIZE - size)) {
printf("efence: fail size %d\n", TEST_SIZE - size);
return -1;
}
putchar('.');
fflush(0);
}
origin_pt = malloc(TEST_LEN);
origin_ct = malloc(TEST_LEN);
origin_dt = malloc(TEST_LEN);
if (NULL == origin_pt || NULL == origin_ct || NULL == origin_dt) {
printf("malloc of testsize failed\n");
return -1;
}
// For data lengths from 0 to 15 bytes, the functions return without any error
// codes, without reading or writing any data.
for (size = TEST_SIZE - min_size + align; size <= TEST_SIZE; size += align) {
// Line up TEST_SIZE from end
efence_pt = pt + TEST_LEN - TEST_SIZE + size;
efence_ct = ct + TEST_LEN - TEST_SIZE + size;
efence_dt = dt + TEST_LEN - TEST_SIZE + size;
mk_rand_data(key1, key2, tinit, efence_pt, TEST_SIZE - size);
memcpy(efence_ct, efence_pt, TEST_SIZE - size);
memcpy(efence_dt, efence_pt, TEST_SIZE - size);
memcpy(origin_pt, efence_pt, TEST_SIZE - size);
memcpy(origin_ct, efence_ct, TEST_SIZE - size);
memcpy(origin_dt, efence_dt, TEST_SIZE - size);
XTS_AES_128_enc(key2, key1, tinit, TEST_SIZE - size, efence_pt, efence_ct);
XTS_AES_128_dec(key2, key1, tinit, TEST_SIZE - size, efence_ct, efence_dt);
if (memcmp(efence_pt, origin_pt, TEST_SIZE - size)) {
printf("efence_pt: fail size %d\n", TEST_SIZE - size);
return -1;
}
if (memcmp(efence_ct, origin_ct, TEST_SIZE - size)) {
printf("efence_ct: fail size %d\n", TEST_SIZE - size);
return -1;
}
if (memcmp(efence_dt, origin_dt, TEST_SIZE - size)) {
printf("efence_dt: fail size %d\n", TEST_SIZE - size);
return -1;
}
putchar('.');
fflush(0);
}
for (i = 0; i < 16 * 11; i++) {
key2_exp_tw[i] = rand();
}
for (size = 0; size <= TEST_SIZE - min_size; size += align) {
// Line up TEST_SIZE from end
efence_pt = pt + TEST_LEN - TEST_SIZE + size;
efence_ct = ct + TEST_LEN - TEST_SIZE + size;
efence_dt = dt + TEST_LEN - TEST_SIZE + size;
mk_rand_data(key1, key2, tinit, efence_pt, TEST_SIZE - size);
aes_keyexp_128(key1, key1_exp_enc, key1_exp_dec);
XTS_AES_128_enc_expanded_key(key2_exp_tw, key1_exp_enc, tinit,
TEST_SIZE - size, efence_pt, efence_ct);
XTS_AES_128_dec_expanded_key(key2_exp_tw, key1_exp_dec, tinit,
TEST_SIZE - size, efence_ct, efence_dt);
if (memcmp(efence_pt, efence_dt, TEST_SIZE - size)) {
printf("efence_expanded_key: fail size %d\n", TEST_SIZE - size);
return -1;
}
putchar('.');
fflush(0);
}
// For data lengths from 0 to 15 bytes, the functions return without any error
// codes, without reading or writing any data.
for (size = TEST_SIZE - min_size + align; size <= TEST_SIZE; size += align) {
// Line up TEST_SIZE from end
efence_pt = pt + TEST_LEN - TEST_SIZE + size;
efence_ct = ct + TEST_LEN - TEST_SIZE + size;
efence_dt = dt + TEST_LEN - TEST_SIZE + size;
mk_rand_data(key1, key2, tinit, efence_pt, TEST_SIZE - size);
memcpy(efence_ct, efence_pt, TEST_SIZE - size);
memcpy(efence_dt, efence_pt, TEST_SIZE - size);
memcpy(origin_pt, efence_pt, TEST_SIZE - size);
memcpy(origin_ct, efence_ct, TEST_SIZE - size);
memcpy(origin_dt, efence_dt, TEST_SIZE - size);
aes_keyexp_128(key1, key1_exp_enc, key1_exp_dec);
XTS_AES_128_enc_expanded_key(key2_exp_tw, key1_exp_enc, tinit,
TEST_SIZE - size, efence_pt, efence_ct);
XTS_AES_128_dec_expanded_key(key2_exp_tw, key1_exp_dec, tinit,
TEST_SIZE - size, efence_ct, efence_dt);
if (memcmp(efence_pt, origin_pt, TEST_SIZE - size)) {
printf("efence_expanded_key for pt: fail size %d\n", TEST_SIZE - size);
return -1;
}
if (memcmp(efence_ct, origin_ct, TEST_SIZE - size)) {
printf("efence_expanded_key for ct: fail size %d\n", TEST_SIZE - size);
return -1;
}
if (memcmp(efence_dt, origin_dt, TEST_SIZE - size)) {
printf("efence_expanded_key for dt: fail size %d\n", TEST_SIZE - size);
return -1;
}
putchar('.');
fflush(0);
}
printf("Pass\n");
return 0;
}
