Skip to content

Commit 4308aa6

Browse files
tests: Add fuzzing harness for functions in net_permissions.h
1 parent 43ff0d9 commit 4308aa6

File tree

2 files changed

+58
-0
lines changed

2 files changed

+58
-0
lines changed

src/Makefile.test.include

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ FUZZ_TARGETS = \
4444
test/fuzz/merkle_block_deserialize \
4545
test/fuzz/messageheader_deserialize \
4646
test/fuzz/multiplication_overflow \
47+
test/fuzz/net_permissions \
4748
test/fuzz/netaddr_deserialize \
4849
test/fuzz/netaddress \
4950
test/fuzz/out_point_deserialize \
@@ -530,6 +531,12 @@ test_fuzz_multiplication_overflow_LDADD = $(FUZZ_SUITE_LD_COMMON)
530531
test_fuzz_multiplication_overflow_LDFLAGS = $(RELDFLAGS) $(AM_LDFLAGS) $(LIBTOOL_APP_LDFLAGS)
531532
test_fuzz_multiplication_overflow_SOURCES = $(FUZZ_SUITE) test/fuzz/multiplication_overflow.cpp
532533

534+
test_fuzz_net_permissions_CPPFLAGS = $(AM_CPPFLAGS) $(BITCOIN_INCLUDES)
535+
test_fuzz_net_permissions_CXXFLAGS = $(AM_CXXFLAGS) $(PIE_FLAGS)
536+
test_fuzz_net_permissions_LDADD = $(FUZZ_SUITE_LD_COMMON)
537+
test_fuzz_net_permissions_LDFLAGS = $(RELDFLAGS) $(AM_LDFLAGS) $(LIBTOOL_APP_LDFLAGS)
538+
test_fuzz_net_permissions_SOURCES = $(FUZZ_SUITE) test/fuzz/net_permissions.cpp
539+
533540
test_fuzz_netaddr_deserialize_CPPFLAGS = $(AM_CPPFLAGS) $(BITCOIN_INCLUDES) -DNETADDR_DESERIALIZE=1
534541
test_fuzz_netaddr_deserialize_CXXFLAGS = $(AM_CXXFLAGS) $(PIE_FLAGS)
535542
test_fuzz_netaddr_deserialize_LDADD = $(FUZZ_SUITE_LD_COMMON)

src/test/fuzz/net_permissions.cpp

Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,51 @@
1+
// Copyright (c) 2020 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 <net_permissions.h>
6+
#include <optional.h>
7+
#include <test/fuzz/FuzzedDataProvider.h>
8+
#include <test/fuzz/fuzz.h>
9+
#include <test/fuzz/util.h>
10+
11+
#include <cassert>
12+
#include <cstdint>
13+
#include <string>
14+
#include <vector>
15+
16+
void test_one_input(const std::vector<uint8_t>& buffer)
17+
{
18+
FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
19+
const std::string s = fuzzed_data_provider.ConsumeRandomLengthString(32);
20+
const NetPermissionFlags net_permission_flags = fuzzed_data_provider.ConsumeBool() ? fuzzed_data_provider.PickValueInArray<NetPermissionFlags>({
21+
NetPermissionFlags::PF_NONE,
22+
NetPermissionFlags::PF_BLOOMFILTER,
23+
NetPermissionFlags::PF_RELAY,
24+
NetPermissionFlags::PF_FORCERELAY,
25+
NetPermissionFlags::PF_NOBAN,
26+
NetPermissionFlags::PF_MEMPOOL,
27+
NetPermissionFlags::PF_ISIMPLICIT,
28+
NetPermissionFlags::PF_ALL,
29+
}) :
30+
static_cast<NetPermissionFlags>(fuzzed_data_provider.ConsumeIntegral<uint32_t>());
31+
32+
NetWhitebindPermissions net_whitebind_permissions;
33+
std::string error_net_whitebind_permissions;
34+
if (NetWhitebindPermissions::TryParse(s, net_whitebind_permissions, error_net_whitebind_permissions)) {
35+
(void)NetPermissions::ToStrings(net_whitebind_permissions.m_flags);
36+
(void)NetPermissions::AddFlag(net_whitebind_permissions.m_flags, net_permission_flags);
37+
assert(NetPermissions::HasFlag(net_whitebind_permissions.m_flags, net_permission_flags));
38+
(void)NetPermissions::ClearFlag(net_whitebind_permissions.m_flags, net_permission_flags);
39+
(void)NetPermissions::ToStrings(net_whitebind_permissions.m_flags);
40+
}
41+
42+
NetWhitelistPermissions net_whitelist_permissions;
43+
std::string error_net_whitelist_permissions;
44+
if (NetWhitelistPermissions::TryParse(s, net_whitelist_permissions, error_net_whitelist_permissions)) {
45+
(void)NetPermissions::ToStrings(net_whitelist_permissions.m_flags);
46+
(void)NetPermissions::AddFlag(net_whitelist_permissions.m_flags, net_permission_flags);
47+
assert(NetPermissions::HasFlag(net_whitelist_permissions.m_flags, net_permission_flags));
48+
(void)NetPermissions::ClearFlag(net_whitelist_permissions.m_flags, net_permission_flags);
49+
(void)NetPermissions::ToStrings(net_whitelist_permissions.m_flags);
50+
}
51+
}

0 commit comments

Comments
 (0)