Branch data Line data Source code
1 : : // Copyright (c) 2021 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 <minisketch.h>
6 : : #include <node/minisketchwrapper.h>
7 : : #include <test/fuzz/FuzzedDataProvider.h>
8 : : #include <test/fuzz/fuzz.h>
9 : : #include <test/fuzz/util.h>
10 : : #include <util/check.h>
11 : :
12 : : #include <map>
13 : : #include <numeric>
14 : :
15 : : namespace {
16 : :
17 : 1564 : Minisketch MakeFuzzMinisketch32(size_t capacity, uint32_t impl)
18 : : {
19 [ + - - + ]: 1564 : return Assert(Minisketch(32, impl, capacity));
20 : : }
21 : :
22 : : } // namespace
23 : :
24 [ + - ]: 848 : FUZZ_TARGET(minisketch)
25 : : {
26 : 392 : FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()};
27 : :
28 : 392 : const auto capacity{fuzzed_data_provider.ConsumeIntegralInRange<size_t>(1, 200)};
29 : 392 : const uint32_t impl{fuzzed_data_provider.ConsumeIntegralInRange<uint32_t>(0, Minisketch::MaxImplementation())};
30 [ + + ]: 392 : if (!Minisketch::ImplementationSupported(32, impl)) return;
31 : :
32 : 391 : Minisketch sketch_a{MakeFuzzMinisketch32(capacity, impl)};
33 [ + - ]: 391 : Minisketch sketch_b{MakeFuzzMinisketch32(capacity, impl)};
34 : 391 : sketch_a.SetSeed(fuzzed_data_provider.ConsumeIntegral<uint64_t>());
35 : 391 : sketch_b.SetSeed(fuzzed_data_provider.ConsumeIntegral<uint64_t>());
36 : :
37 : : // Fill two sets and keep the difference in a map
38 : 391 : std::map<uint32_t, bool> diff;
39 [ + + + + ]: 27179 : LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 10000)
40 : : {
41 : 26788 : const auto entry{fuzzed_data_provider.ConsumeIntegralInRange<uint32_t>(1, std::numeric_limits<uint32_t>::max() - 1)};
42 : 52447 : const auto KeepDiff{[&] {
43 : 25659 : bool& mut{diff[entry]};
44 : 25659 : mut = !mut;
45 : 52447 : }};
46 [ + - ]: 26788 : CallOneOf(
47 : : fuzzed_data_provider,
48 : 15968 : [&] {
49 : 15968 : sketch_a.Add(entry);
50 : 15968 : KeepDiff();
51 : 15968 : },
52 : 9691 : [&] {
53 : 9691 : sketch_b.Add(entry);
54 : 9691 : KeepDiff();
55 : 9691 : },
56 : 1129 : [&] {
57 : 1129 : sketch_a.Add(entry);
58 : 1129 : sketch_b.Add(entry);
59 : 1129 : });
60 : : }
61 : 12785 : const auto num_diff{std::accumulate(diff.begin(), diff.end(), size_t{0}, [](auto n, const auto& e) { return n + e.second; })};
62 : :
63 [ + - ]: 391 : Minisketch sketch_ar{MakeFuzzMinisketch32(capacity, impl)};
64 [ + - ]: 391 : Minisketch sketch_br{MakeFuzzMinisketch32(capacity, impl)};
65 : 391 : sketch_ar.SetSeed(fuzzed_data_provider.ConsumeIntegral<uint64_t>());
66 : 391 : sketch_br.SetSeed(fuzzed_data_provider.ConsumeIntegral<uint64_t>());
67 : :
68 [ + - ]: 391 : sketch_ar.Deserialize(sketch_a.Serialize());
69 [ + - ]: 391 : sketch_br.Deserialize(sketch_b.Serialize());
70 : :
71 [ + + ]: 391 : Minisketch sketch_diff{std::move(fuzzed_data_provider.ConsumeBool() ? sketch_a : sketch_ar)};
72 [ + + ]: 391 : sketch_diff.Merge(fuzzed_data_provider.ConsumeBool() ? sketch_b : sketch_br);
73 : :
74 [ + + ]: 391 : if (capacity >= num_diff) {
75 : 380 : const auto max_elements{fuzzed_data_provider.ConsumeIntegralInRange<size_t>(num_diff, capacity)};
76 [ + - + - ]: 380 : const auto dec{*Assert(sketch_diff.Decode(max_elements))};
77 [ - + + - ]: 380 : Assert(dec.size() == num_diff);
78 [ + + ]: 10956 : for (auto d : dec) {
79 [ + - + - ]: 10576 : Assert(diff.at(d));
80 : : }
81 : 380 : }
82 [ + - + + : 1564 : }
+ - + + ]
|