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 <chainparams.h>
6 : : #include <net.h>
7 : : #include <net_permissions.h>
8 : : #include <netaddress.h>
9 : : #include <protocol.h>
10 : : #include <random.h>
11 : : #include <test/fuzz/FuzzedDataProvider.h>
12 : : #include <test/fuzz/fuzz.h>
13 : : #include <test/fuzz/util.h>
14 : : #include <test/fuzz/util/net.h>
15 : : #include <test/util/net.h>
16 : : #include <test/util/setup_common.h>
17 : : #include <util/asmap.h>
18 : : #include <util/chaintype.h>
19 : : #include <util/time.h>
20 : :
21 : : #include <cstdint>
22 : : #include <optional>
23 : : #include <string>
24 : : #include <vector>
25 : :
26 : 2 : void initialize_net()
27 : : {
28 [ + - + - ]: 4 : static const auto testing_setup = MakeNoLogFileContext<>(ChainType::MAIN);
29 [ + - ]: 4 : }
30 : :
31 [ + - ]: 1607 : FUZZ_TARGET(net, .init = initialize_net)
32 : : {
33 : 1165 : FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
34 : 1165 : SetMockTime(ConsumeTime(fuzzed_data_provider));
35 : 1165 : CNode node{ConsumeNode(fuzzed_data_provider)};
36 [ + - ]: 1165 : node.SetCommonVersion(fuzzed_data_provider.ConsumeIntegral<int>());
37 [ + + ]: 2330 : if (const auto service_opt =
38 [ + + ]: 1165 : ConsumeDeserializable<CService>(fuzzed_data_provider, ConsumeDeserializationParams<CNetAddr::SerParams>(fuzzed_data_provider)))
39 : : {
40 [ + - ]: 363 : node.SetAddrLocal(*service_opt);
41 : 0 : }
42 [ + + + + ]: 195249 : LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 10000) {
43 [ + - ]: 194084 : CallOneOf(
44 : : fuzzed_data_provider,
45 : 9010 : [&] {
46 : 9010 : node.CloseSocketDisconnect();
47 : 9010 : },
48 : 112568 : [&] {
49 : 112568 : CNodeStats stats;
50 [ + - ]: 112568 : node.CopyStats(stats);
51 : 112568 : },
52 : 10877 : [&] {
53 [ - + ]: 10877 : const CNode* add_ref_node = node.AddRef();
54 [ - + ]: 10877 : assert(add_ref_node == &node);
55 : 10877 : },
56 : 21754 : [&] {
57 [ + + ]: 21754 : if (node.GetRefCount() > 0) {
58 : 1049 : node.Release();
59 : : }
60 : 21754 : },
61 : 39875 : [&] {
62 : 39875 : const std::vector<uint8_t> b = ConsumeRandomLengthByteVector(fuzzed_data_provider);
63 : 39875 : bool complete;
64 [ + - ]: 39875 : node.ReceiveMsgBytes(b, complete);
65 : 39875 : });
66 : : }
67 : :
68 [ + - ]: 1165 : (void)node.GetAddrLocal();
69 : 1165 : (void)node.GetId();
70 : 1165 : (void)node.GetLocalNonce();
71 : 1165 : const int ref_count = node.GetRefCount();
72 [ - + ]: 1165 : assert(ref_count >= 0);
73 : 1165 : (void)node.GetCommonVersion();
74 : :
75 : 1165 : const NetPermissionFlags net_permission_flags = ConsumeWeakEnum(fuzzed_data_provider, ALL_NET_PERMISSION_FLAGS);
76 [ + - ]: 1165 : (void)node.HasPermission(net_permission_flags);
77 [ + - ]: 1165 : (void)node.ConnectedThroughNetwork();
78 : 1165 : }
79 : :
80 [ + - ]: 1393 : FUZZ_TARGET(local_address, .init = initialize_net)
81 : : {
82 : 951 : FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
83 : 951 : SetMockTime(ConsumeTime(fuzzed_data_provider));
84 : 951 : CService service{ConsumeService(fuzzed_data_provider)};
85 : 951 : CNode node{ConsumeNode(fuzzed_data_provider)};
86 : 951 : {
87 [ + - ]: 951 : LOCK(g_maplocalhost_mutex);
88 [ + - ]: 951 : mapLocalHost.clear();
89 : 951 : }
90 [ + + + + ]: 403179 : LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 10000) {
91 [ + - ]: 402228 : CallOneOf(
92 : : fuzzed_data_provider,
93 : 75062 : [&] {
94 : 75062 : service = ConsumeService(fuzzed_data_provider);
95 : 75062 : },
96 : 149110 : [&] {
97 : 149110 : const bool added{AddLocal(service, fuzzed_data_provider.ConsumeIntegralInRange<int>(0, LOCAL_MAX - 1))};
98 [ + + ]: 149110 : if (!added) return;
99 [ - + ]: 113679 : assert(service.IsRoutable());
100 [ - + ]: 113679 : assert(IsLocal(service));
101 [ - + ]: 113679 : assert(SeenLocal(service));
102 : : },
103 : 40558 : [&] {
104 : 40558 : (void)RemoveLocal(service);
105 : 40558 : },
106 : 25766 : [&] {
107 : 25766 : (void)SeenLocal(service);
108 : 25766 : },
109 : 13066 : [&] {
110 : 13066 : (void)IsLocal(service);
111 : 13066 : },
112 : 98666 : [&] {
113 : 98666 : (void)GetLocalAddress(node);
114 : 98666 : });
115 : : }
116 : 951 : }
|