Inside Bitcoin

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
16static 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
33BENCHMARK(FindByte);
34