mirror of
https://github.com/bitcoin/bitcoin.git
synced 2026-09-14 15:25:51 +02:00
af0ee28eb6
Replace hard-coded MiB byte conversions (e.g. `1024*1024`, `1<<20`, `1048576`) with the existing `_MiB` literal to improve readability and avoid repeating constants.
In the few spots where arithmetic involves signed values, the result is identical to the previous code assuming those quantities never turn negative.
Also switch to brace init on every declaration assigned from `_MiB`/`_GiB` literals so a future oversized value (e.g. `unsigned int x{4096_MiB}`) becomes a compile error through the C++11 narrowing check instead of silently truncating.
Extend unit tests to cover the 32-bit `size_t` overflow boundary and to assert equivalence for integer and floating-point conversions.
Co-authored-by: MarcoFalke <*~=`'#}+{/-|&$^_@721217.xyz>
Co-authored-by: w0xlt <94266259+w0xlt@users.noreply.github.com>
81 lines
2.1 KiB
C++
81 lines
2.1 KiB
C++
// Copyright (c) 2019-present The Bitcoin Core developers
|
|
// Distributed under the MIT software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
|
|
#include <bench/bench.h>
|
|
#include <crypto/chacha20.h>
|
|
#include <crypto/chacha20poly1305.h>
|
|
#include <span.h>
|
|
#include <util/byte_units.h>
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <vector>
|
|
|
|
/* Number of bytes to process per iteration */
|
|
static const uint64_t BUFFER_SIZE_TINY = 64;
|
|
static const uint64_t BUFFER_SIZE_SMALL = 256;
|
|
static const uint64_t BUFFER_SIZE_LARGE{1_MiB};
|
|
|
|
static void CHACHA20(benchmark::Bench& bench, size_t buffersize)
|
|
{
|
|
std::vector<std::byte> key(32, {});
|
|
ChaCha20 ctx(key);
|
|
ctx.Seek({0, 0}, 0);
|
|
std::vector<std::byte> in(buffersize, {});
|
|
std::vector<std::byte> out(buffersize, {});
|
|
bench.batch(in.size()).unit("byte").run([&] {
|
|
ctx.Crypt(in, out);
|
|
});
|
|
}
|
|
|
|
static void FSCHACHA20POLY1305(benchmark::Bench& bench, size_t buffersize)
|
|
{
|
|
std::vector<std::byte> key(32);
|
|
FSChaCha20Poly1305 ctx(key, 224);
|
|
std::vector<std::byte> in(buffersize);
|
|
std::vector<std::byte> aad;
|
|
std::vector<std::byte> out(buffersize + FSChaCha20Poly1305::EXPANSION);
|
|
bench.batch(in.size()).unit("byte").run([&] {
|
|
ctx.Encrypt(in, aad, out);
|
|
});
|
|
}
|
|
|
|
static void CHACHA20_64BYTES(benchmark::Bench& bench)
|
|
{
|
|
CHACHA20(bench, BUFFER_SIZE_TINY);
|
|
}
|
|
|
|
static void CHACHA20_256BYTES(benchmark::Bench& bench)
|
|
{
|
|
CHACHA20(bench, BUFFER_SIZE_SMALL);
|
|
}
|
|
|
|
static void CHACHA20_1MB(benchmark::Bench& bench)
|
|
{
|
|
CHACHA20(bench, BUFFER_SIZE_LARGE);
|
|
}
|
|
|
|
static void FSCHACHA20POLY1305_64BYTES(benchmark::Bench& bench)
|
|
{
|
|
FSCHACHA20POLY1305(bench, BUFFER_SIZE_TINY);
|
|
}
|
|
|
|
static void FSCHACHA20POLY1305_256BYTES(benchmark::Bench& bench)
|
|
{
|
|
FSCHACHA20POLY1305(bench, BUFFER_SIZE_SMALL);
|
|
}
|
|
|
|
static void FSCHACHA20POLY1305_1MB(benchmark::Bench& bench)
|
|
{
|
|
FSCHACHA20POLY1305(bench, BUFFER_SIZE_LARGE);
|
|
}
|
|
|
|
BENCHMARK(CHACHA20_64BYTES);
|
|
BENCHMARK(CHACHA20_256BYTES);
|
|
BENCHMARK(CHACHA20_1MB);
|
|
BENCHMARK(FSCHACHA20POLY1305_64BYTES);
|
|
BENCHMARK(FSCHACHA20POLY1305_256BYTES);
|
|
BENCHMARK(FSCHACHA20POLY1305_1MB);
|