Changeset View
Changeset View
Standalone View
Standalone View
src/test/fuzz/bloom_filter.cpp
- This file was added.
// Copyright (c) 2020 The Bitcoin Core developers | |||||
// Distributed under the MIT software license, see the accompanying | |||||
// file COPYING or http://www.opensource.org/licenses/mit-license.php. | |||||
#include <bloom.h> | |||||
#include <optional.h> | |||||
#include <primitives/transaction.h> | |||||
#include <test/fuzz/FuzzedDataProvider.h> | |||||
#include <test/fuzz/fuzz.h> | |||||
#include <test/fuzz/util.h> | |||||
#include <uint256.h> | |||||
#include <cassert> | |||||
#include <cstdint> | |||||
#include <string> | |||||
#include <vector> | |||||
void test_one_input(const std::vector<uint8_t> &buffer) { | |||||
FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size()); | |||||
CBloomFilter bloom_filter{ | |||||
fuzzed_data_provider.ConsumeIntegralInRange<unsigned int>(1, 10000000), | |||||
1.0 / fuzzed_data_provider.ConsumeIntegralInRange<unsigned int>( | |||||
1, std::numeric_limits<unsigned int>::max()), | |||||
fuzzed_data_provider.ConsumeIntegral<unsigned int>(), | |||||
static_cast<uint8_t>(fuzzed_data_provider.PickValueInArray( | |||||
{BLOOM_UPDATE_NONE, BLOOM_UPDATE_ALL, BLOOM_UPDATE_P2PUBKEY_ONLY, | |||||
BLOOM_UPDATE_MASK}))}; | |||||
while (fuzzed_data_provider.remaining_bytes() > 0) { | |||||
switch (fuzzed_data_provider.ConsumeIntegralInRange(0, 6)) { | |||||
case 0: { | |||||
const std::vector<uint8_t> &b = | |||||
ConsumeRandomLengthByteVector(fuzzed_data_provider); | |||||
(void)bloom_filter.contains(b); | |||||
bloom_filter.insert(b); | |||||
const bool present = bloom_filter.contains(b); | |||||
assert(present); | |||||
break; | |||||
} | |||||
case 1: { | |||||
const Optional<COutPoint> out_point = | |||||
ConsumeDeserializable<COutPoint>(fuzzed_data_provider); | |||||
if (!out_point) { | |||||
break; | |||||
} | |||||
(void)bloom_filter.contains(*out_point); | |||||
bloom_filter.insert(*out_point); | |||||
const bool present = bloom_filter.contains(*out_point); | |||||
assert(present); | |||||
break; | |||||
} | |||||
case 2: { | |||||
const Optional<uint256> u256 = | |||||
ConsumeDeserializable<uint256>(fuzzed_data_provider); | |||||
if (!u256) { | |||||
break; | |||||
} | |||||
(void)bloom_filter.contains(*u256); | |||||
bloom_filter.insert(*u256); | |||||
const bool present = bloom_filter.contains(*u256); | |||||
assert(present); | |||||
break; | |||||
} | |||||
case 3: | |||||
bloom_filter.clear(); | |||||
break; | |||||
case 4: | |||||
bloom_filter.reset( | |||||
fuzzed_data_provider.ConsumeIntegral<unsigned int>()); | |||||
break; | |||||
case 5: { | |||||
const Optional<CMutableTransaction> mut_tx = | |||||
ConsumeDeserializable<CMutableTransaction>( | |||||
fuzzed_data_provider); | |||||
if (!mut_tx) { | |||||
break; | |||||
} | |||||
const CTransaction tx{*mut_tx}; | |||||
(void)bloom_filter.IsRelevantAndUpdate(tx); | |||||
break; | |||||
} | |||||
case 6: | |||||
bloom_filter.UpdateEmptyFull(); | |||||
break; | |||||
} | |||||
(void)bloom_filter.IsWithinSizeConstraints(); | |||||
} | |||||
} |