/root/bitcoin/src/test/fuzz/buffered_file.cpp
Line | Count | Source |
1 | | // Copyright (c) 2020-2022 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 <span.h> |
6 | | #include <streams.h> |
7 | | #include <test/fuzz/FuzzedDataProvider.h> |
8 | | #include <test/fuzz/fuzz.h> |
9 | | #include <test/fuzz/util.h> |
10 | | |
11 | | #include <array> |
12 | | #include <cstddef> |
13 | | #include <cstdint> |
14 | | #include <iostream> |
15 | | #include <optional> |
16 | | #include <vector> |
17 | | |
18 | | FUZZ_TARGET(buffered_file) |
19 | 0 | { |
20 | 0 | FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()}; |
21 | 0 | FuzzedFileProvider fuzzed_file_provider{fuzzed_data_provider}; |
22 | 0 | std::optional<BufferedFile> opt_buffered_file; |
23 | 0 | AutoFile fuzzed_file{ |
24 | 0 | fuzzed_file_provider.open(), |
25 | 0 | ConsumeRandomLengthByteVector<std::byte>(fuzzed_data_provider), |
26 | 0 | }; |
27 | 0 | try { |
28 | 0 | auto n_buf_size = fuzzed_data_provider.ConsumeIntegralInRange<uint64_t>(0, 4096); |
29 | 0 | auto n_rewind_in = fuzzed_data_provider.ConsumeIntegralInRange<uint64_t>(0, 4096); |
30 | 0 | opt_buffered_file.emplace(fuzzed_file, n_buf_size, n_rewind_in); |
31 | 0 | } catch (const std::ios_base::failure&) { |
32 | 0 | } |
33 | 0 | if (opt_buffered_file && !fuzzed_file.IsNull()) { |
34 | 0 | bool setpos_fail = false; |
35 | 0 | LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 100) |
36 | 0 | { |
37 | 0 | CallOneOf( |
38 | 0 | fuzzed_data_provider, |
39 | 0 | [&] { |
40 | 0 | std::array<std::byte, 4096> arr{}; |
41 | 0 | try { |
42 | 0 | opt_buffered_file->read({arr.data(), fuzzed_data_provider.ConsumeIntegralInRange<size_t>(0, 4096)}); |
43 | 0 | } catch (const std::ios_base::failure&) { |
44 | 0 | } |
45 | 0 | }, |
46 | 0 | [&] { |
47 | 0 | opt_buffered_file->SetLimit(fuzzed_data_provider.ConsumeIntegralInRange<uint64_t>(0, 4096)); |
48 | 0 | }, |
49 | 0 | [&] { |
50 | 0 | if (!opt_buffered_file->SetPos(fuzzed_data_provider.ConsumeIntegralInRange<uint64_t>(0, 4096))) { |
51 | 0 | setpos_fail = true; |
52 | 0 | } |
53 | 0 | }, |
54 | 0 | [&] { |
55 | 0 | if (setpos_fail) { |
56 | | // Calling FindByte(...) after a failed SetPos(...) call may result in an infinite loop. |
57 | 0 | return; |
58 | 0 | } |
59 | 0 | try { |
60 | 0 | opt_buffered_file->FindByte(std::byte(fuzzed_data_provider.ConsumeIntegral<uint8_t>())); |
61 | 0 | } catch (const std::ios_base::failure&) { |
62 | 0 | } |
63 | 0 | }, |
64 | 0 | [&] { |
65 | 0 | ReadFromStream(fuzzed_data_provider, *opt_buffered_file); |
66 | 0 | }); |
67 | 0 | } |
68 | 0 | opt_buffered_file->GetPos(); |
69 | 0 | } |
70 | 0 | } |