Changeset View
Changeset View
Standalone View
Standalone View
src/bench/chacha20.cpp
// Copyright (c) 2019 The Bitcoin Core developers | // Copyright (c) 2019 The Bitcoin Core developers | ||||
// Distributed under the MIT software license, see the accompanying | // Distributed under the MIT software license, see the accompanying | ||||
// file COPYING or http://www.opensource.org/licenses/mit-license.php. | // file COPYING or http://www.opensource.org/licenses/mit-license.php. | ||||
#include <iostream> | #include <iostream> | ||||
#include <bench/bench.h> | #include <bench/bench.h> | ||||
#include <crypto/chacha20.h> | #include <crypto/chacha20.h> | ||||
#include <hash.h> | #include <hash.h> | ||||
/* Number of bytes to process per iteration */ | /* Number of bytes to process per iteration */ | ||||
static const uint64_t BUFFER_SIZE_TINY = 64; | static const uint64_t BUFFER_SIZE_TINY = 64; | ||||
static const uint64_t BUFFER_SIZE_SMALL = 256; | static const uint64_t BUFFER_SIZE_SMALL = 256; | ||||
static const uint64_t BUFFER_SIZE_LARGE = 1024 * 1024; | static const uint64_t BUFFER_SIZE_LARGE = 1024 * 1024; | ||||
static void CHACHA20(benchmark::State &state, size_t buffersize) { | static void CHACHA20(benchmark::Bench &bench, size_t buffersize) { | ||||
std::vector<uint8_t> key(32, 0); | std::vector<uint8_t> key(32, 0); | ||||
ChaCha20 ctx(key.data(), key.size()); | ChaCha20 ctx(key.data(), key.size()); | ||||
ctx.SetIV(0); | ctx.SetIV(0); | ||||
ctx.Seek(0); | ctx.Seek(0); | ||||
std::vector<uint8_t> in(buffersize, 0); | std::vector<uint8_t> in(buffersize, 0); | ||||
std::vector<uint8_t> out(buffersize, 0); | std::vector<uint8_t> out(buffersize, 0); | ||||
while (state.KeepRunning()) { | bench.batch(in.size()).unit("byte").run( | ||||
ctx.Crypt(in.data(), out.data(), in.size()); | [&] { ctx.Crypt(in.data(), out.data(), in.size()); }); | ||||
} | |||||
} | } | ||||
static void CHACHA20_64BYTES(benchmark::State &state) { | static void CHACHA20_64BYTES(benchmark::Bench &bench) { | ||||
CHACHA20(state, BUFFER_SIZE_TINY); | CHACHA20(bench, BUFFER_SIZE_TINY); | ||||
} | } | ||||
static void CHACHA20_256BYTES(benchmark::State &state) { | static void CHACHA20_256BYTES(benchmark::Bench &bench) { | ||||
CHACHA20(state, BUFFER_SIZE_SMALL); | CHACHA20(bench, BUFFER_SIZE_SMALL); | ||||
} | } | ||||
static void CHACHA20_1MB(benchmark::State &state) { | static void CHACHA20_1MB(benchmark::Bench &bench) { | ||||
CHACHA20(state, BUFFER_SIZE_LARGE); | CHACHA20(bench, BUFFER_SIZE_LARGE); | ||||
} | } | ||||
BENCHMARK(CHACHA20_64BYTES, 500000); | BENCHMARK(CHACHA20_64BYTES); | ||||
BENCHMARK(CHACHA20_256BYTES, 250000); | BENCHMARK(CHACHA20_256BYTES); | ||||
BENCHMARK(CHACHA20_1MB, 340); | BENCHMARK(CHACHA20_1MB); |