/root/bitcoin/src/test/fuzz/autofile.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/fuzz.h> |
8 | | #include <test/fuzz/FuzzedDataProvider.h> |
9 | | #include <test/fuzz/util.h> |
10 | | #include <util/obfuscation.h> |
11 | | |
12 | | #include <array> |
13 | | #include <cstddef> |
14 | | #include <cstdio> |
15 | | #include <iostream> |
16 | | #include <vector> |
17 | | |
18 | | FUZZ_TARGET(autofile) |
19 | 0 | { |
20 | 0 | FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()}; |
21 | 0 | FuzzedFileProvider fuzzed_file_provider{fuzzed_data_provider}; |
22 | 0 | const auto key_bytes{ConsumeFixedLengthByteVector<std::byte>(fuzzed_data_provider, Obfuscation::KEY_SIZE)}; |
23 | 0 | AutoFile auto_file{ |
24 | 0 | fuzzed_file_provider.open(), |
25 | 0 | Obfuscation{std::span{key_bytes}.first<Obfuscation::KEY_SIZE>()}, |
26 | 0 | }; |
27 | 0 | LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 100) |
28 | 0 | { |
29 | 0 | CallOneOf( |
30 | 0 | fuzzed_data_provider, |
31 | 0 | [&] { |
32 | 0 | std::array<std::byte, 4096> arr{}; |
33 | 0 | try { |
34 | 0 | auto_file.read({arr.data(), fuzzed_data_provider.ConsumeIntegralInRange<size_t>(0, 4096)}); |
35 | 0 | } catch (const std::ios_base::failure&) { |
36 | 0 | } |
37 | 0 | }, |
38 | 0 | [&] { |
39 | 0 | const std::array<std::byte, 4096> arr{}; |
40 | 0 | try { |
41 | 0 | auto_file.write({arr.data(), fuzzed_data_provider.ConsumeIntegralInRange<size_t>(0, 4096)}); |
42 | 0 | } catch (const std::ios_base::failure&) { |
43 | 0 | } |
44 | 0 | }, |
45 | 0 | [&] { |
46 | 0 | try { |
47 | 0 | auto_file.ignore(fuzzed_data_provider.ConsumeIntegralInRange<size_t>(0, 4096)); |
48 | 0 | } catch (const std::ios_base::failure&) { |
49 | 0 | } |
50 | 0 | }, |
51 | 0 | [&] { |
52 | 0 | (void)auto_file.fclose(); |
53 | 0 | }, |
54 | 0 | [&] { |
55 | 0 | ReadFromStream(fuzzed_data_provider, auto_file); |
56 | 0 | }, |
57 | 0 | [&] { |
58 | 0 | WriteToStream(fuzzed_data_provider, auto_file); |
59 | 0 | }); |
60 | 0 | } |
61 | 0 | (void)auto_file.IsNull(); |
62 | 0 | if (fuzzed_data_provider.ConsumeBool()) { |
63 | 0 | FILE* f = auto_file.release(); |
64 | 0 | if (f != nullptr) { |
65 | 0 | fclose(f); |
66 | 0 | } |
67 | 0 | } else { |
68 | 0 | (void)auto_file.fclose(); |
69 | 0 | } |
70 | 0 | } |