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  
 6  #include <bench/bench.h>
 7  #include <crypto/poly1305.h>
 8  #include <span.h>
 9  
10  #include <cstddef>
11  #include <cstdint>
12  #include <vector>
13  
14  /* Number of bytes to process per iteration */
15  static constexpr uint64_t BUFFER_SIZE_TINY  = 64;
16  static constexpr uint64_t BUFFER_SIZE_SMALL = 256;
17  static constexpr uint64_t BUFFER_SIZE_LARGE = 1024*1024;
18  
19  static 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  
29  static void POLY1305_64BYTES(benchmark::Bench& bench)
30  {
31      POLY1305(bench, BUFFER_SIZE_TINY);
32  }
33  
34  static void POLY1305_256BYTES(benchmark::Bench& bench)
35  {
36      POLY1305(bench, BUFFER_SIZE_SMALL);
37  }
38  
39  static void POLY1305_1MB(benchmark::Bench& bench)
40  {
41      POLY1305(bench, BUFFER_SIZE_LARGE);
42  }
43  
44  BENCHMARK(POLY1305_64BYTES);
45  BENCHMARK(POLY1305_256BYTES);
46  BENCHMARK(POLY1305_1MB);