Skip to content

Commit 102f326

Browse files
tests: Add fuzzing harness for classes/functions in blockfilter.h
1 parent 87d24e6 commit 102f326

File tree

3 files changed

+61
-0
lines changed

3 files changed

+61
-0
lines changed

src/Makefile.test.include

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ FUZZ_TARGETS = \
1717
test/fuzz/block_filter_deserialize \
1818
test/fuzz/block_header \
1919
test/fuzz/block_header_and_short_txids_deserialize \
20+
test/fuzz/blockfilter \
2021
test/fuzz/blockheader_deserialize \
2122
test/fuzz/blocklocator_deserialize \
2223
test/fuzz/blockmerkleroot \
@@ -369,6 +370,12 @@ test_fuzz_block_header_and_short_txids_deserialize_LDADD = $(FUZZ_SUITE_LD_COMMO
369370
test_fuzz_block_header_and_short_txids_deserialize_LDFLAGS = $(RELDFLAGS) $(AM_LDFLAGS) $(LIBTOOL_APP_LDFLAGS)
370371
test_fuzz_block_header_and_short_txids_deserialize_SOURCES = $(FUZZ_SUITE) test/fuzz/deserialize.cpp
371372

373+
test_fuzz_blockfilter_CPPFLAGS = $(AM_CPPFLAGS) $(BITCOIN_INCLUDES)
374+
test_fuzz_blockfilter_CXXFLAGS = $(AM_CXXFLAGS) $(PIE_FLAGS)
375+
test_fuzz_blockfilter_LDADD = $(FUZZ_SUITE_LD_COMMON)
376+
test_fuzz_blockfilter_LDFLAGS = $(RELDFLAGS) $(AM_LDFLAGS) $(LIBTOOL_APP_LDFLAGS)
377+
test_fuzz_blockfilter_SOURCES = $(FUZZ_SUITE) test/fuzz/blockfilter.cpp
378+
372379
test_fuzz_blockheader_deserialize_CPPFLAGS = $(AM_CPPFLAGS) $(BITCOIN_INCLUDES) -DBLOCKHEADER_DESERIALIZE=1
373380
test_fuzz_blockheader_deserialize_CXXFLAGS = $(AM_CXXFLAGS) $(PIE_FLAGS)
374381
test_fuzz_blockheader_deserialize_LDADD = $(FUZZ_SUITE_LD_COMMON)

src/test/fuzz/blockfilter.cpp

Lines changed: 44 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,44 @@
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 <blockfilter.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 <cstdint>
12+
#include <string>
13+
#include <vector>
14+
15+
void test_one_input(const std::vector<uint8_t>& buffer)
16+
{
17+
FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
18+
const Optional<BlockFilter> block_filter = ConsumeDeserializable<BlockFilter>(fuzzed_data_provider);
19+
if (!block_filter) {
20+
return;
21+
}
22+
{
23+
(void)block_filter->ComputeHeader(ConsumeUInt256(fuzzed_data_provider));
24+
(void)block_filter->GetBlockHash();
25+
(void)block_filter->GetEncodedFilter();
26+
(void)block_filter->GetHash();
27+
}
28+
{
29+
const BlockFilterType block_filter_type = block_filter->GetFilterType();
30+
(void)BlockFilterTypeName(block_filter_type);
31+
}
32+
{
33+
const GCSFilter gcs_filter = block_filter->GetFilter();
34+
(void)gcs_filter.GetN();
35+
(void)gcs_filter.GetParams();
36+
(void)gcs_filter.GetEncoded();
37+
(void)gcs_filter.Match(ConsumeRandomLengthByteVector(fuzzed_data_provider));
38+
GCSFilter::ElementSet element_set;
39+
while (fuzzed_data_provider.ConsumeBool()) {
40+
element_set.insert(ConsumeRandomLengthByteVector(fuzzed_data_provider));
41+
gcs_filter.MatchAny(element_set);
42+
}
43+
}
44+
}

src/test/fuzz/util.h

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
#include <streams.h>
1414
#include <test/fuzz/FuzzedDataProvider.h>
1515
#include <test/fuzz/fuzz.h>
16+
#include <uint256.h>
1617
#include <version.h>
1718

1819
#include <cstdint>
@@ -70,6 +71,15 @@ NODISCARD inline CScriptNum ConsumeScriptNum(FuzzedDataProvider& fuzzed_data_pro
7071
return CScriptNum{fuzzed_data_provider.ConsumeIntegral<int64_t>()};
7172
}
7273

74+
NODISCARD inline uint256 ConsumeUInt256(FuzzedDataProvider& fuzzed_data_provider) noexcept
75+
{
76+
const std::vector<unsigned char> v256 = fuzzed_data_provider.ConsumeBytes<unsigned char>(sizeof(uint256));
77+
if (v256.size() != sizeof(uint256)) {
78+
return {};
79+
}
80+
return uint256{v256};
81+
}
82+
7383
template <typename T>
7484
bool MultiplicationOverflow(T i, T j)
7585
{

0 commit comments

Comments
 (0)