Inside Bitcoin

1// Copyright (c) 2019-present The Bitcoin Core developers
2// Distributed under the MIT software license, see the accompanying
3// file COPYING or http://www.opensource.org/licenses/mit-license.php.
4
5#include <bench/bench.h>
6#include <crypto/poly1305.h>
7#include <util/byte_units.h>
8
9#include <cstddef>
10#include <cstdint>
11#include <span>
12#include <vector>
13
14/* Number of bytes to process per iteration */
15static constexpr uint64_t BUFFER_SIZE_TINY  = 64;
16static constexpr uint64_t BUFFER_SIZE_SMALL = 256;
17static constexpr uint64_t BUFFER_SIZE_LARGE{1_MiB};
18
19static void POLY1305(benchmark::Bench& bench, size_t buffersize)
20{
21    std::vector<std::byte> tag(Poly1305::TAGLEN, {});
22    std::vector<std::byte> key(Poly1305::KEYLEN, {});
23    std::vector<std::byte> in(buffersize, {});
24    bench.batch(in.size()).unit("byte").run([&] {
25        Poly1305{key}.Update(in).Finalize(tag);
26    });
27}
28
29static void POLY1305_64BYTES(benchmark::Bench& bench)
30{
31    POLY1305(bench, BUFFER_SIZE_TINY);
32}
33
34static void POLY1305_256BYTES(benchmark::Bench& bench)
35{
36    POLY1305(bench, BUFFER_SIZE_SMALL);
37}
38
39static void POLY1305_1MB(benchmark::Bench& bench)
40{
41    POLY1305(bench, BUFFER_SIZE_LARGE);
42}
43
44BENCHMARK(POLY1305_64BYTES);
45BENCHMARK(POLY1305_256BYTES);
46BENCHMARK(POLY1305_1MB);
47