95 lines
2.7 KiB
C++
95 lines
2.7 KiB
C++
|
// SPDX-License-Identifier: AGPL-1.0-only
|
||
|
|
// Copyright (C) 2018 Ludvig Strigeus <info@tunsafe.com>. All Rights Reserved.
|
||
|
|
#include "stdafx.h"
|
||
|
|
#include "tunsafe_types.h"
|
||
|
|
#include "crypto/chacha20poly1305.h"
|
||
|
|
#include "crypto/aesgcm/aes.h"
|
||
|
|
#include "tunsafe_cpu.h"
|
||
|
|
|
||
|
|
#include <functional>
|
||
|
|
#include <string.h>
|
||
|
|
|
||
|
|
#if defined(OS_FREEBSD) || defined(OS_LINUX)
|
||
|
|
#include <time.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
typedef uint64 LARGE_INTEGER;
|
||
|
|
void QueryPerformanceCounter(LARGE_INTEGER *x) {
|
||
|
|
struct timespec ts;
|
||
|
|
if (clock_gettime(CLOCK_MONOTONIC, &ts) != 0) {
|
||
|
|
fprintf(stderr, "clock_gettime failed\n");
|
||
|
|
exit(1);
|
||
|
|
}
|
||
|
|
*x = (uint64)ts.tv_sec * 1000000000 + ts.tv_nsec;
|
||
|
|
}
|
||
|
|
|
||
|
|
void QueryPerformanceFrequency(LARGE_INTEGER *x) {
|
||
|
|
*x = 1000000000;
|
||
|
|
}
|
||
|
|
#elif defined(OS_MACOSX)
|
||
|
|
#include <mach/mach.h>
|
||
|
|
#include <mach/mach_time.h>
|
||
|
|
typedef uint64 LARGE_INTEGER;
|
||
|
|
|
||
|
|
void QueryPerformanceCounter(LARGE_INTEGER *x) {
|
||
|
|
*x = mach_absolute_time();
|
||
|
|
}
|
||
|
|
|
||
|
|
void QueryPerformanceFrequency(LARGE_INTEGER *x) {
|
||
|
|
mach_timebase_info_data_t timebase = { 0, 0 };
|
||
|
|
if (mach_timebase_info(&timebase) != 0)
|
||
|
|
abort();
|
||
|
|
printf("numer/denom: %d %d\n", timebase.numer, timebase.denom);
|
||
|
|
*x = timebase.denom * 1000000000;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|
||
|
|
|
||
|
|
int gcm_self_test();
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
void *fake_glb;
|
||
|
|
void Benchmark() {
|
||
|
|
int64 a, b, f, t1 = 0, t2 = 0;
|
||
|
|
|
||
|
|
#if WITH_AESGCM
|
||
|
|
gcm_self_test();
|
||
|
|
#endif // WITH_AESGCM
|
||
|
|
|
||
|
|
PrintCpuFeatures();
|
||
|
|
|
||
|
|
QueryPerformanceFrequency((LARGE_INTEGER*)&f);
|
||
|
|
|
||
|
|
uint8 dst[1500 + 16];
|
||
|
|
uint8 key[32] = {0, 1, 2, 3, 4, 5, 6};
|
||
|
|
uint8 mac[16];
|
||
|
|
|
||
|
|
fake_glb = dst;
|
||
|
|
|
||
|
|
auto RunOneBenchmark = [&](const char *name, const std::function<uint64(size_t)> &ff) {
|
||
|
|
uint64 bytes = 0;
|
||
|
|
QueryPerformanceCounter((LARGE_INTEGER*)&b);
|
||
|
|
size_t i;
|
||
|
|
for (i = 0; bytes < 1000000000; i++)
|
||
|
|
bytes += ff(i);
|
||
|
|
QueryPerformanceCounter((LARGE_INTEGER*)&a);
|
||
|
|
RINFO("%s: %f MB/s", name, (double)bytes * 0.000001 / (a - b) * f);
|
||
|
|
};
|
||
|
|
|
||
|
|
memset(dst, 0, 1500);
|
||
|
|
RunOneBenchmark("chacha20-encrypt", [&](size_t i) -> uint64 { chacha20poly1305_encrypt(dst, dst, 1460, NULL, 0, i, key); return 1460; });
|
||
|
|
RunOneBenchmark("chacha20-decrypt", [&](size_t i) -> uint64 { chacha20poly1305_decrypt_get_mac(dst, dst, 1460, NULL, 0, i, key, mac); return 1460; });
|
||
|
|
|
||
|
|
RunOneBenchmark("poly1305-only", [&](size_t i) -> uint64 { poly1305_get_mac(dst, 1460, NULL, 0, i, key, mac); return 1460; });
|
||
|
|
|
||
|
|
#if WITH_AESGCM
|
||
|
|
if (X86_PCAP_AES) {
|
||
|
|
AesGcm128StaticContext sctx;
|
||
|
|
CRYPTO_gcm128_init(&sctx, key, 128);
|
||
|
|
|
||
|
|
RunOneBenchmark("aes128-gcm-encrypt", [&](size_t i) -> uint64 { aesgcm_encrypt(dst, dst, 1460, NULL, 0, i, &sctx); return 1460; });
|
||
|
|
RunOneBenchmark("aes128-gcm-decrypt", [&](size_t i) -> uint64 { aesgcm_decrypt_get_mac(dst, dst, 1460, NULL, 0, i, &sctx, mac); return 1460; });
|
||
|
|
}
|
||
|
|
#endif // WITH_AESGCM
|
||
|
|
}
|