streams_findbyte.cpp raw

   1  // Copyright (c) 2023-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 <streams.h>
   7  #include <test/util/setup_common.h>
   8  #include <util/check.h>
   9  #include <util/fs.h>
  10  
  11  #include <cstddef>
  12  #include <cstdint>
  13  #include <cstdio>
  14  #include <memory>
  15  
  16  static void FindByte(benchmark::Bench& bench)
  17  {
  18      const auto testing_setup{MakeNoLogFileContext<const BasicTestingSetup>(ChainType::REGTEST)};
  19      AutoFile file{fsbridge::fopen(testing_setup->m_path_root / "streams_tmp", "w+b")};
  20      const size_t file_size = 200;
  21      uint8_t data[file_size] = {0};
  22      data[file_size - 1] = 1;
  23      file << data;
  24      file.seek(0, SEEK_SET);
  25      BufferedFile bf{file, /*nBufSize=*/file_size + 1, /*nRewindIn=*/file_size};
  26  
  27      bench.setup([&] { bf.SetPos(0); })
  28          .run([&] { bf.FindByte(std::byte(1)); });
  29  
  30      assert(file.fclose() == 0);
  31  }
  32  
  33  BENCHMARK(FindByte);
  34