litespeed-quic/tests/test_dec.c
2022-05-06 12:49:46 -04:00

82 lines
2.4 KiB
C

/* Copyright (c) 2017 - 2022 LiteSpeed Technologies Inc. See LICENSE. */
/*
* test_dec.c -- Benchmark decryption using aligned and non-aligned buffers.
*/
#include <assert.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#ifndef WIN32
#include <unistd.h>
#else
#include <getopt.h>
#endif
#include <openssl/aead.h>
#include <openssl/rand.h>
#define MAX_SIZE 1400
int
main (int argc, char **argv)
{
EVP_AEAD_CTX aead_ctx;
int opt, n = 1, r;
size_t sealed_len, opened_len;
unsigned char key[16];
unsigned char data[1328];
unsigned char sealed_buf[63 + MAX_SIZE], *sealed = sealed_buf;
unsigned char opened_buf[63 + MAX_SIZE], *opened = opened_buf;
while (-1 != (opt = getopt(argc, argv, "an:")))
{
switch (opt)
{
case 'a':
if ((uintptr_t) sealed & (64 - 1))
sealed += 64 - ((uintptr_t) sealed & (64 - 1));
if ((uintptr_t) opened & (64 - 1))
opened += 64 - ((uintptr_t) opened & (64 - 1));
break;
case 'n': /* Number of decrypt iterations */
n = atoi(optarg);
break;
default:
exit(EXIT_FAILURE);
}
}
RAND_bytes(key, sizeof(key));
RAND_bytes(data, sizeof(data));
EVP_AEAD_CTX_init(&aead_ctx, EVP_aead_aes_128_gcm(), key, sizeof(key),
12, NULL);
r = EVP_AEAD_CTX_seal(&aead_ctx, sealed, &sealed_len, MAX_SIZE,
key, sizeof(key), data, sizeof(data), NULL, 0);
if (!r)
{
fprintf(stderr, "cannot seal\n");
exit(EXIT_FAILURE);
}
printf("buffers are %saligned\n", ((uintptr_t) opened & (64 - 1)) ||
((uintptr_t) sealed & (64 - 1)) ? "not " : "");
/* Check that decryption works first time around */
r = EVP_AEAD_CTX_open(&aead_ctx, opened, &opened_len, MAX_SIZE,
key, sizeof(key), sealed, sealed_len, NULL, 0);
assert(r && opened_len == sizeof(data) &&
0 == memcmp(data, opened, sizeof(data)));
--n;
/* Do no bother checking return value in the loop */
while (n-- > 0)
{
EVP_AEAD_CTX_open(&aead_ctx, opened, &opened_len, MAX_SIZE,
key, sizeof(key), sealed, sealed_len, NULL, 0);
}
exit(EXIT_SUCCESS);
}