Changeset View
Changeset View
Standalone View
Standalone View
src/test/fuzz/cuckoocache.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 <cuckoocache.h> | |||||
#include <script/sigcache.h> | |||||
#include <test/fuzz/FuzzedDataProvider.h> | |||||
#include <test/fuzz/fuzz.h> | |||||
#include <test/fuzz/util.h> | |||||
#include <test/util/setup_common.h> | |||||
#include <cstdint> | |||||
#include <string> | |||||
#include <vector> | |||||
namespace { | |||||
FuzzedDataProvider *fuzzed_data_provider_ptr = nullptr; | |||||
struct RandomHasher { | |||||
template <uint8_t> uint32_t operator()(const bool & /* unused */) const { | |||||
assert(fuzzed_data_provider_ptr != nullptr); | |||||
return fuzzed_data_provider_ptr->ConsumeIntegral<uint32_t>(); | |||||
} | |||||
}; | |||||
struct FuzzCuckooCacheKey { | |||||
bool b; | |||||
public: | |||||
FuzzCuckooCacheKey(bool _b = false) : b(_b){}; | |||||
operator bool() const { return b; } | |||||
}; | |||||
} // namespace | |||||
void test_one_input(const std::vector<uint8_t> &buffer) { | |||||
FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size()); | |||||
fuzzed_data_provider_ptr = &fuzzed_data_provider; | |||||
CuckooCache::cache<CuckooCache::KeyOnly<FuzzCuckooCacheKey>, RandomHasher> | |||||
cuckoo_cache{}; | |||||
if (fuzzed_data_provider.ConsumeBool()) { | |||||
const size_t megabytes = | |||||
fuzzed_data_provider.ConsumeIntegralInRange<size_t>(0, 16); | |||||
cuckoo_cache.setup_bytes(megabytes << 20); | |||||
} else { | |||||
cuckoo_cache.setup( | |||||
fuzzed_data_provider.ConsumeIntegralInRange<uint32_t>(0, 4096)); | |||||
} | |||||
while (fuzzed_data_provider.ConsumeBool()) { | |||||
if (fuzzed_data_provider.ConsumeBool()) { | |||||
cuckoo_cache.insert( | |||||
FuzzCuckooCacheKey(fuzzed_data_provider.ConsumeBool())); | |||||
} else { | |||||
cuckoo_cache.contains( | |||||
FuzzCuckooCacheKey(fuzzed_data_provider.ConsumeBool()), | |||||
fuzzed_data_provider.ConsumeBool()); | |||||
} | |||||
} | |||||
fuzzed_data_provider_ptr = nullptr; | |||||
} |