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/chacha20.h>
7#include <crypto/chacha20poly1305.h>
8#include <util/byte_units.h>
9
10#include <cstddef>
11#include <cstdint>
12#include <span>
13#include <vector>
14
15/* Number of bytes to process per iteration */
16static const uint64_t BUFFER_SIZE_TINY  = 64;
17static const uint64_t BUFFER_SIZE_SMALL = 256;
18static const uint64_t BUFFER_SIZE_LARGE{1_MiB};
19
20static void CHACHA20(benchmark::Bench& bench, size_t buffersize)
21{
22    std::vector<std::byte> key(32, {});
23    ChaCha20 ctx(key);
24    ctx.Seek({0, 0}, 0);
25    std::vector<std::byte> in(buffersize, {});
26    std::vector<std::byte> out(buffersize, {});
27    bench.batch(in.size()).unit("byte").run([&] {
28        ctx.Crypt(in, out);
29    });
30}
31
32static void FSCHACHA20POLY1305(benchmark::Bench& bench, size_t buffersize)
33{
34    std::vector<std::byte> key(32);
35    FSChaCha20Poly1305 ctx(key, 224);
36    std::vector<std::byte> in(buffersize);
37    std::vector<std::byte> aad;
38    std::vector<std::byte> out(buffersize + FSChaCha20Poly1305::EXPANSION);
39    bench.batch(in.size()).unit("byte").run([&] {
40        ctx.Encrypt(in, aad, out);
41    });
42}
43
44static void CHACHA20_64BYTES(benchmark::Bench& bench)
45{
46    CHACHA20(bench, BUFFER_SIZE_TINY);
47}
48
49static void CHACHA20_256BYTES(benchmark::Bench& bench)
50{
51    CHACHA20(bench, BUFFER_SIZE_SMALL);
52}
53
54static void CHACHA20_1MB(benchmark::Bench& bench)
55{
56    CHACHA20(bench, BUFFER_SIZE_LARGE);
57}
58
59static void FSCHACHA20POLY1305_64BYTES(benchmark::Bench& bench)
60{
61    FSCHACHA20POLY1305(bench, BUFFER_SIZE_TINY);
62}
63
64static void FSCHACHA20POLY1305_256BYTES(benchmark::Bench& bench)
65{
66    FSCHACHA20POLY1305(bench, BUFFER_SIZE_SMALL);
67}
68
69static void FSCHACHA20POLY1305_1MB(benchmark::Bench& bench)
70{
71    FSCHACHA20POLY1305(bench, BUFFER_SIZE_LARGE);
72}
73
74BENCHMARK(CHACHA20_64BYTES);
75BENCHMARK(CHACHA20_256BYTES);
76BENCHMARK(CHACHA20_1MB);
77BENCHMARK(FSCHACHA20POLY1305_64BYTES);
78BENCHMARK(FSCHACHA20POLY1305_256BYTES);
79BENCHMARK(FSCHACHA20POLY1305_1MB);
80