bench.c

#include "sys/time.h"
#include <math.h>
#include <stdio.h>

#include "chachapoly_aead.h"
#include "poly1305.h"

static const uint8_t testkey[32] = {
    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};

static const uint8_t testnonce[32] = {0x00, 0x01, 0x02, 0x03,
    0x04, 0x05, 0x06, 0x07};

static const uint8_t testdata[12] = {0x48, 0x65, 0x6c, 0x6c, 0x6f, 0x20,
    0x77, 0x6f, 0x72, 0x6c, 0x64, 0x21};

static const uint64_t BUFFER_SIZE = 1000 * 1000;

static const uint8_t aead_keys[64] = {
    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, 0x00,
    0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
    0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
    0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00};

static double gettimedouble(void)
{
    struct timeval tv;
    gettimeofday(&tv, NULL);
    return tv.tv_usec * 0.000001 + tv.tv_sec;
}

static void print_number(double x)
{
    double y = x;
    int c = 0;
    if (y < 0.0) {
        y = -y;
    }
    while (y < 100.0) {
        y *= 10.0;
        c++;
    }
    printf("%.*f", c, x);
}

static void run_benchmark(char* name, void (*benchmark)(void*), void (*setup)(void*), void (*teardown)(void*), void* data, int count, int iter)
{
    int i;
    double min = HUGE_VAL;
    double sum = 0.0;
    double max = 0.0;
    for (i = 0; i < count; i++) {
        double begin, total;
        if (setup != NULL) {
            setup(data);
        }
        begin = gettimedouble();
        benchmark(data);
        total = gettimedouble() - begin;
        if (teardown != NULL) {
            teardown(data);
        }
        if (total < min) {
            min = total;
        }
        if (total > max) {
            max = total;
        }
        sum += total;
    }
    printf("%s: min ", name);
    print_number(min * 1000000000.0 / iter);
    printf("ns / avg ");
    print_number((sum / count) * 1000000000.0 / iter);
    printf("ns / max ");
    print_number(max * 1000000000.0 / iter);
    printf("ns\n");
}

static void bench_chacha_ivsetup(void* data)
{
    struct chacha_ctx* ctx = (struct chacha_ctx*)data;
    int i;
    for (i = 0; i < 50000; i++) {
        chacha_ivsetup(ctx, testnonce, NULL);
    }
}

static void bench_chacha_keysetup(void* data)
{
    struct chacha_ctx* ctx = (struct chacha_ctx*)data;
    int i;
    for (i = 0; i < 50000; i++) {
        chacha_keysetup(ctx, testkey, 256);
    }
}

static void bench_chacha_encrypt(void* data)
{
    struct chacha_ctx* ctx = (struct chacha_ctx*)data;
    uint8_t scratch[16] = {0};
    int i;
    for (i = 0; i < 4000000 / 16; i++) {
        chacha_encrypt_bytes(ctx, scratch, scratch, 16);
    }
}

static void bench_poly1305_auth(void* data)
{
    struct chacha_ctx* ctx = (struct chacha_ctx*)data;
    uint8_t poly1305_tag[16] = {0};
    int i;
    for (i = 0; i < 4000000 / 12; i++) {
        poly1305_auth(poly1305_tag, testdata, 12, testkey);
    }
}

static void bench_chacha20poly1305_init(void* data)
{
    struct chachapolyaead_ctx* ctx = (struct chachapolyaead_ctx*)data;
    int i;
    for (i = 0; i < 50000; i++) {
        chacha20poly1305_init(ctx, aead_keys, 32, aead_keys + 32, 32);
    }
}

static void bench_chacha20poly1305_crypt(void* data)
{
    struct chachapolyaead_ctx* ctx = (struct chachapolyaead_ctx*)data;
    int i;
    uint32_t seqnr = 0;

    uint8_t buffer[BUFFER_SIZE + 16];
    for (i = 0; i < 30; i++) {
        chacha20poly1305_crypt(ctx, seqnr, seqnr, 0, buffer, BUFFER_SIZE + 16, buffer, BUFFER_SIZE, 1);
    }
}

int main(void)
{
    struct chacha_ctx ctx_chacha;
    struct chachapolyaead_ctx aead_ctx;
    run_benchmark("chacha_ivsetup", bench_chacha_ivsetup, NULL, NULL, &ctx_chacha,
        20, 50000);
    run_benchmark("chacha_keysetup", bench_chacha_keysetup, NULL, NULL,
        &ctx_chacha, 20, 50000);
    run_benchmark("chacha_encrypt", bench_chacha_encrypt, NULL, NULL, &ctx_chacha,
        20, 4000000);
    run_benchmark("poly1305_auth", bench_poly1305_auth, NULL, NULL, &ctx_chacha,
        20, 4000000);
    run_benchmark("chacha20poly1305_init", bench_chacha20poly1305_init, NULL,
        NULL, &aead_ctx, 20, 4000000);
    run_benchmark("chacha20poly1305_crypt 1MB", bench_chacha20poly1305_crypt,
        NULL, NULL, &aead_ctx, 20, 30);
    return 0;
}