Branch data Line data Source code
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 <addrdb.h>
6 : : #include <addrman.h>
7 : : #include <addrman_impl.h>
8 : : #include <chainparams.h>
9 : : #include <common/args.h>
10 : : #include <merkleblock.h>
11 : : #include <random.h>
12 : : #include <test/fuzz/FuzzedDataProvider.h>
13 : : #include <test/fuzz/fuzz.h>
14 : : #include <test/fuzz/util.h>
15 : : #include <test/fuzz/util/net.h>
16 : : #include <test/util/setup_common.h>
17 : : #include <time.h>
18 : : #include <util/asmap.h>
19 : : #include <util/chaintype.h>
20 : :
21 : : #include <cassert>
22 : : #include <cstdint>
23 : : #include <optional>
24 : : #include <string>
25 : : #include <vector>
26 : :
27 : : namespace {
28 : : const BasicTestingSetup* g_setup;
29 : :
30 : 7701 : int32_t GetCheckRatio()
31 : : {
32 [ + - + - ]: 15402 : return std::clamp<int32_t>(g_setup->m_node.args->GetIntArg("-checkaddrman", 0), 0, 1000000);
33 : : }
34 : : } // namespace
35 : :
36 : 3 : void initialize_addrman()
37 : : {
38 [ + - + - ]: 6 : static const auto testing_setup = MakeNoLogFileContext<>(ChainType::REGTEST);
39 : 3 : g_setup = testing_setup.get();
40 [ + - ]: 6 : }
41 : :
42 [ + - ]: 2205 : FUZZ_TARGET(data_stream_addr_man, .init = initialize_addrman)
43 : : {
44 : 1773 : SeedRandomStateForTest(SeedRand::ZEROS);
45 : 1773 : FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()};
46 : 1773 : DataStream data_stream = ConsumeDataStream(fuzzed_data_provider);
47 : 1773 : NetGroupManager netgroupman{ConsumeNetGroupManager(fuzzed_data_provider)};
48 [ + - + - ]: 1773 : AddrMan addr_man(netgroupman, /*deterministic=*/false, GetCheckRatio());
49 : 1773 : try {
50 [ + + ]: 1773 : ReadFromStream(addr_man, data_stream);
51 [ - + ]: 1066 : } catch (const std::exception&) {
52 : 1066 : }
53 : 1773 : }
54 : :
55 : : /**
56 : : * Generate a random address. Always returns a valid address.
57 : : */
58 : 10626891 : CNetAddr RandAddr(FuzzedDataProvider& fuzzed_data_provider, FastRandomContext& fast_random_context)
59 : : {
60 : 10626891 : CNetAddr addr;
61 [ + - + - ]: 10626891 : assert(!addr.IsValid());
62 [ + + + - : 23225140 : for (size_t i = 0; i < 8 && !addr.IsValid(); ++i) {
+ + ]
63 [ + + + + ]: 14870820 : if (fuzzed_data_provider.remaining_bytes() > 1 && fuzzed_data_provider.ConsumeBool()) {
64 : 1772479 : addr = ConsumeNetAddr(fuzzed_data_provider);
65 : : } else {
66 : 10825770 : addr = ConsumeNetAddr(fuzzed_data_provider, &fast_random_context);
67 : : }
68 : : }
69 : :
70 : : // Return a dummy IPv4 5.5.5.5 if we generated an invalid address.
71 [ + - + + ]: 10626891 : if (!addr.IsValid()) {
72 : 8830 : in_addr v4_addr = {};
73 : 8830 : v4_addr.s_addr = 0x05050505;
74 [ + - ]: 17660 : addr = CNetAddr{v4_addr};
75 : : }
76 : :
77 : 10626891 : return addr;
78 : 0 : }
79 : :
80 : : /** Fill addrman with lots of addresses from lots of sources. */
81 : 1420 : void FillAddrman(AddrMan& addrman, FuzzedDataProvider& fuzzed_data_provider)
82 : : {
83 : : // Add a fraction of the addresses to the "tried" table.
84 : : // 0, 1, 2, 3 corresponding to 0%, 100%, 50%, 33%
85 : 1420 : const size_t n = fuzzed_data_provider.ConsumeIntegralInRange<size_t>(0, 3);
86 : :
87 : 1420 : const size_t num_sources = fuzzed_data_provider.ConsumeIntegralInRange<size_t>(1, 50);
88 : 1420 : CNetAddr prev_source;
89 : : // Generate a FastRandomContext seed to use inside the loops instead of
90 : : // fuzzed_data_provider. When fuzzed_data_provider is exhausted it
91 : : // just returns 0.
92 : 1420 : FastRandomContext fast_random_context{ConsumeUInt256(fuzzed_data_provider)};
93 [ + + ]: 43938 : for (size_t i = 0; i < num_sources; ++i) {
94 [ + - ]: 42518 : const auto source = RandAddr(fuzzed_data_provider, fast_random_context);
95 : 42518 : const size_t num_addresses = fast_random_context.randrange(500) + 1; // [1..500]
96 : :
97 [ + + ]: 10626891 : for (size_t j = 0; j < num_addresses; ++j) {
98 [ + - + - ]: 10584373 : const auto addr = CAddress{CService{RandAddr(fuzzed_data_provider, fast_random_context), 8333}, NODE_NETWORK};
99 : 10584373 : const std::chrono::seconds time_penalty{fast_random_context.randrange(100000001)};
100 [ + - + - : 21168746 : addrman.Add({addr}, source, time_penalty);
+ + - - ]
101 : :
102 [ + + + - : 10584373 : if (n > 0 && addrman.Size() % n == 0) {
+ + ]
103 [ + - ]: 6356330 : addrman.Good(addr, Now<NodeSeconds>());
104 : : }
105 : :
106 : : // Add 10% of the addresses from more than one source.
107 [ + + + - : 10584373 : if (fast_random_context.randrange(10) == 0 && prev_source.IsValid()) {
+ + ]
108 [ + - + - : 2064590 : addrman.Add({addr}, prev_source, time_penalty);
+ + - - ]
109 : : }
110 : 10584373 : }
111 : 85036 : prev_source = source;
112 : 42518 : }
113 : 11618088 : }
114 : :
115 [ + - ]: 2537 : FUZZ_TARGET(addrman, .init = initialize_addrman)
116 : : {
117 : 2105 : SeedRandomStateForTest(SeedRand::ZEROS);
118 : 2105 : FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
119 : 2105 : SetMockTime(ConsumeTime(fuzzed_data_provider));
120 : 2105 : NetGroupManager netgroupman{ConsumeNetGroupManager(fuzzed_data_provider)};
121 [ + - + - ]: 2105 : auto addr_man_ptr = std::make_unique<AddrManDeterministic>(netgroupman, fuzzed_data_provider, GetCheckRatio());
122 [ + + ]: 2105 : if (fuzzed_data_provider.ConsumeBool()) {
123 : 1313 : const std::vector<uint8_t> serialized_data{ConsumeRandomLengthByteVector(fuzzed_data_provider)};
124 [ + - ]: 1313 : DataStream ds{serialized_data};
125 : 1313 : try {
126 [ + + ]: 2626 : ds >> *addr_man_ptr;
127 [ - + ]: 983 : } catch (const std::ios_base::failure&) {
128 [ + - + - ]: 1966 : addr_man_ptr = std::make_unique<AddrManDeterministic>(netgroupman, fuzzed_data_provider, GetCheckRatio());
129 : 983 : }
130 : 1313 : }
131 : 2105 : AddrManDeterministic& addr_man = *addr_man_ptr;
132 [ + + + + ]: 656487 : LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 10000) {
133 [ + - ]: 654382 : CallOneOf(
134 : : fuzzed_data_provider,
135 : 8892 : [&] {
136 : 8892 : addr_man.ResolveCollisions();
137 : 8892 : },
138 : 111696 : [&] {
139 : 111696 : (void)addr_man.SelectTriedCollision();
140 : 111696 : },
141 : 13678 : [&] {
142 : 13678 : std::vector<CAddress> addresses;
143 [ + + + + ]: 4064080 : LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 10000) {
144 [ + - ]: 4050402 : addresses.push_back(ConsumeAddress(fuzzed_data_provider));
145 : : }
146 : 13678 : auto net_addr = ConsumeNetAddr(fuzzed_data_provider);
147 [ + - ]: 13678 : auto time_penalty = std::chrono::seconds{ConsumeTime(fuzzed_data_provider, 0, 100000000)};
148 [ + - ]: 13678 : addr_man.Add(addresses, net_addr, time_penalty);
149 : 13678 : },
150 : 465075 : [&] {
151 : 465075 : auto addr = ConsumeService(fuzzed_data_provider);
152 [ + - ]: 465075 : auto time = NodeSeconds{std::chrono::seconds{ConsumeTime(fuzzed_data_provider)}};
153 [ + - ]: 465075 : addr_man.Good(addr, time);
154 : 465075 : },
155 : 8038 : [&] {
156 : 8038 : auto addr = ConsumeService(fuzzed_data_provider);
157 : 8038 : auto count_failure = fuzzed_data_provider.ConsumeBool();
158 [ + - ]: 8038 : auto time = NodeSeconds{std::chrono::seconds{ConsumeTime(fuzzed_data_provider)}};
159 [ + - ]: 8038 : addr_man.Attempt(addr, count_failure, time);
160 : 8038 : },
161 : 18937 : [&] {
162 : 18937 : auto addr = ConsumeService(fuzzed_data_provider);
163 [ + - ]: 18937 : auto time = NodeSeconds{std::chrono::seconds{ConsumeTime(fuzzed_data_provider)}};
164 [ + - ]: 18937 : addr_man.Connected(addr, time);
165 : 18937 : },
166 : 28066 : [&] {
167 : 28066 : auto addr = ConsumeService(fuzzed_data_provider);
168 : 28066 : auto n_services = ConsumeWeakEnum(fuzzed_data_provider, ALL_SERVICE_FLAGS);
169 [ + - ]: 28066 : addr_man.SetServices(addr, n_services);
170 : 28066 : });
171 : : }
172 : 2105 : const AddrMan& const_addr_man{addr_man};
173 : 2105 : std::optional<Network> network;
174 [ + + ]: 2105 : if (fuzzed_data_provider.ConsumeBool()) {
175 : 236 : network = fuzzed_data_provider.PickValueInArray(ALL_NETWORKS);
176 : : }
177 : 2105 : auto max_addresses = fuzzed_data_provider.ConsumeIntegralInRange<size_t>(0, 4096);
178 : 2105 : auto max_pct = fuzzed_data_provider.ConsumeIntegralInRange<size_t>(0, 100);
179 : 2105 : auto filtered = fuzzed_data_provider.ConsumeBool();
180 [ + - ]: 2105 : (void)const_addr_man.GetAddr(max_addresses, max_pct, network, filtered);
181 : :
182 : 2105 : std::unordered_set<Network> nets;
183 [ + + ]: 16840 : for (const auto& net : ALL_NETWORKS) {
184 [ + + ]: 14735 : if (fuzzed_data_provider.ConsumeBool()) {
185 [ + - ]: 1811 : nets.insert(net);
186 : : }
187 : : }
188 [ + - ]: 2105 : (void)const_addr_man.Select(fuzzed_data_provider.ConsumeBool(), nets);
189 : :
190 : 2105 : std::optional<bool> in_new;
191 [ + + ]: 2105 : if (fuzzed_data_provider.ConsumeBool()) {
192 : 246 : in_new = fuzzed_data_provider.ConsumeBool();
193 : : }
194 [ + - ]: 2105 : (void)const_addr_man.Size(network, in_new);
195 : 2105 : DataStream data_stream{};
196 [ + - ]: 4210 : data_stream << const_addr_man;
197 : 2105 : }
198 : :
199 : : // Check that serialize followed by unserialize produces the same addrman.
200 [ + - ]: 1852 : FUZZ_TARGET(addrman_serdeser, .init = initialize_addrman)
201 : : {
202 : 1420 : SeedRandomStateForTest(SeedRand::ZEROS);
203 : 1420 : FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
204 : 1420 : SetMockTime(ConsumeTime(fuzzed_data_provider));
205 : :
206 : 1420 : NetGroupManager netgroupman{ConsumeNetGroupManager(fuzzed_data_provider)};
207 [ + - + - ]: 1420 : AddrManDeterministic addr_man1{netgroupman, fuzzed_data_provider, GetCheckRatio()};
208 [ + - + - ]: 1420 : AddrManDeterministic addr_man2{netgroupman, fuzzed_data_provider, GetCheckRatio()};
209 : :
210 : 1420 : DataStream data_stream{};
211 : :
212 [ + - ]: 1420 : FillAddrman(addr_man1, fuzzed_data_provider);
213 [ + - ]: 1420 : data_stream << addr_man1;
214 [ + - ]: 1420 : data_stream >> addr_man2;
215 [ + - - + ]: 1420 : assert(addr_man1 == addr_man2);
216 : 1420 : }
|
