Skip to content

Commit 7c8c140

Browse files
fuzz: Implement fuzzed_dns_lookup_function as lambda
1 parent eceb3f7 commit 7c8c140

File tree

1 file changed

+9
-15
lines changed

1 file changed

+9
-15
lines changed

src/test/fuzz/netbase_dns_lookup.cpp

Lines changed: 9 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -12,27 +12,22 @@
1212
#include <string>
1313
#include <vector>
1414

15-
namespace {
16-
FuzzedDataProvider* fuzzed_data_provider_ptr = nullptr;
17-
18-
std::vector<CNetAddr> fuzzed_dns_lookup_function(const std::string& name, bool allow_lookup)
19-
{
20-
std::vector<CNetAddr> resolved_addresses;
21-
while (fuzzed_data_provider_ptr->ConsumeBool()) {
22-
resolved_addresses.push_back(ConsumeNetAddr(*fuzzed_data_provider_ptr));
23-
}
24-
return resolved_addresses;
25-
}
26-
} // namespace
27-
2815
FUZZ_TARGET(netbase_dns_lookup)
2916
{
3017
FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()};
31-
fuzzed_data_provider_ptr = &fuzzed_data_provider;
3218
const std::string name = fuzzed_data_provider.ConsumeRandomLengthString(512);
3319
const unsigned int max_results = fuzzed_data_provider.ConsumeIntegral<unsigned int>();
3420
const bool allow_lookup = fuzzed_data_provider.ConsumeBool();
3521
const int default_port = fuzzed_data_provider.ConsumeIntegral<int>();
22+
23+
auto fuzzed_dns_lookup_function = [&](const std::string&, bool) {
24+
std::vector<CNetAddr> resolved_addresses;
25+
while (fuzzed_data_provider.ConsumeBool()) {
26+
resolved_addresses.push_back(ConsumeNetAddr(fuzzed_data_provider));
27+
}
28+
return resolved_addresses;
29+
};
30+
3631
{
3732
std::vector<CNetAddr> resolved_addresses;
3833
if (LookupHost(name, resolved_addresses, max_results, allow_lookup, fuzzed_dns_lookup_function)) {
@@ -73,5 +68,4 @@ FUZZ_TARGET(netbase_dns_lookup)
7368
assert(resolved_subnet.IsValid());
7469
}
7570
}
76-
fuzzed_data_provider_ptr = nullptr;
7771
}

0 commit comments

Comments
 (0)