phantom/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2025-04-29 14:59:39 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

fuzz: move ConsumeNetGroupManager to util

Browse source
This commit is contained in:
brunoerg 2024-02-05 15:41:26 -03:00
parent fe624631ae
commit 18c8a0945b
2 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/test/fuzz/addrman.cpp
View file

@ -39,13 +39,6 @@ void initialize_addrman()
g_setup = testing_setup.get(); g_setup = testing_setup.get();
} }
[[nodiscard]] inline NetGroupManager ConsumeNetGroupManager(FuzzedDataProvider& fuzzed_data_provider) noexcept
{
std::vector<bool> asmap = ConsumeRandomLengthBitVector(fuzzed_data_provider);
if (!SanityCheckASMap(asmap, 128)) asmap.clear();
return NetGroupManager(asmap);
}
FUZZ_TARGET(data_stream_addr_man, .init = initialize_addrman) FUZZ_TARGET(data_stream_addr_man, .init = initialize_addrman)
{ {
FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()}; FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()};

8
src/test/fuzz/util/net.h
View file

@ -17,6 +17,7 @@
#include <test/fuzz/util.h> #include <test/fuzz/util.h>
#include <test/util/net.h> #include <test/util/net.h>
#include <threadsafety.h> #include <threadsafety.h>
#include <util/asmap.h>
#include <util/sock.h> #include <util/sock.h>
#include <chrono> #include <chrono>
@ -197,6 +198,13 @@ public:
return FuzzedSock{fuzzed_data_provider}; return FuzzedSock{fuzzed_data_provider};
} }
[[nodiscard]] inline NetGroupManager ConsumeNetGroupManager(FuzzedDataProvider& fuzzed_data_provider) noexcept
{
std::vector<bool> asmap = ConsumeRandomLengthBitVector(fuzzed_data_provider);
if (!SanityCheckASMap(asmap, 128)) asmap.clear();
return NetGroupManager(asmap);
}
inline CSubNet ConsumeSubNet(FuzzedDataProvider& fuzzed_data_provider) noexcept inline CSubNet ConsumeSubNet(FuzzedDataProvider& fuzzed_data_provider) noexcept
{ {
return {ConsumeNetAddr(fuzzed_data_provider), fuzzed_data_provider.ConsumeIntegral<uint8_t>()}; return {ConsumeNetAddr(fuzzed_data_provider), fuzzed_data_provider.ConsumeIntegral<uint8_t>()};