Bitcoin Core  24.99.0
P2P Digital Currency
policy_estimator.cpp
Go to the documentation of this file.
1 // Copyright (c) 2020-2022 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 <kernel/mempool_entry.h>
6 #include <policy/fees.h>
7 #include <policy/fees_args.h>
10 #include <test/fuzz/fuzz.h>
11 #include <test/fuzz/util.h>
12 #include <test/fuzz/util/mempool.h>
13 #include <test/util/setup_common.h>
14 #include <txmempool.h>
15 
16 #include <cstdint>
17 #include <optional>
18 #include <string>
19 #include <vector>
20 
21 namespace {
22 const BasicTestingSetup* g_setup;
23 } // namespace
24 
26 {
27  static const auto testing_setup = MakeNoLogFileContext<>();
28  g_setup = testing_setup.get();
29 }
30 
32 {
33  FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
34  CBlockPolicyEstimator block_policy_estimator{FeeestPath(*g_setup->m_node.args)};
35  LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 10000) {
36  CallOneOf(
37  fuzzed_data_provider,
38  [&] {
39  const std::optional<CMutableTransaction> mtx = ConsumeDeserializable<CMutableTransaction>(fuzzed_data_provider);
40  if (!mtx) {
41  return;
42  }
43  const CTransaction tx{*mtx};
44  block_policy_estimator.processTransaction(ConsumeTxMemPoolEntry(fuzzed_data_provider, tx), fuzzed_data_provider.ConsumeBool());
45  if (fuzzed_data_provider.ConsumeBool()) {
46  (void)block_policy_estimator.removeTx(tx.GetHash(), /*inBlock=*/fuzzed_data_provider.ConsumeBool());
47  }
48  },
49  [&] {
50  std::vector<CTxMemPoolEntry> mempool_entries;
51  LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 10000) {
52  const std::optional<CMutableTransaction> mtx = ConsumeDeserializable<CMutableTransaction>(fuzzed_data_provider);
53  if (!mtx) {
54  break;
55  }
56  const CTransaction tx{*mtx};
57  mempool_entries.push_back(ConsumeTxMemPoolEntry(fuzzed_data_provider, tx));
58  }
59  std::vector<const CTxMemPoolEntry*> ptrs;
60  ptrs.reserve(mempool_entries.size());
61  for (const CTxMemPoolEntry& mempool_entry : mempool_entries) {
62  ptrs.push_back(&mempool_entry);
63  }
64  block_policy_estimator.processBlock(fuzzed_data_provider.ConsumeIntegral<unsigned int>(), ptrs);
65  },
66  [&] {
67  (void)block_policy_estimator.removeTx(ConsumeUInt256(fuzzed_data_provider), /*inBlock=*/fuzzed_data_provider.ConsumeBool());
68  },
69  [&] {
70  block_policy_estimator.FlushUnconfirmed();
71  });
72  (void)block_policy_estimator.estimateFee(fuzzed_data_provider.ConsumeIntegral<int>());
73  EstimationResult result;
74  (void)block_policy_estimator.estimateRawFee(fuzzed_data_provider.ConsumeIntegral<int>(), fuzzed_data_provider.ConsumeFloatingPoint<double>(), fuzzed_data_provider.PickValueInArray(ALL_FEE_ESTIMATE_HORIZONS), fuzzed_data_provider.ConsumeBool() ? &result : nullptr);
75  FeeCalculation fee_calculation;
76  (void)block_policy_estimator.estimateSmartFee(fuzzed_data_provider.ConsumeIntegral<int>(), fuzzed_data_provider.ConsumeBool() ? &fee_calculation : nullptr, fuzzed_data_provider.ConsumeBool());
77  (void)block_policy_estimator.HighestTargetTracked(fuzzed_data_provider.PickValueInArray(ALL_FEE_ESTIMATE_HORIZONS));
78  }
79  {
80  FuzzedAutoFileProvider fuzzed_auto_file_provider = ConsumeAutoFile(fuzzed_data_provider);
81  AutoFile fuzzed_auto_file{fuzzed_auto_file_provider.open()};
82  block_policy_estimator.Write(fuzzed_auto_file);
83  block_policy_estimator.Read(fuzzed_auto_file);
84  }
85 }
Non-refcounted RAII wrapper for FILE*.
Definition: streams.h:488
The BlockPolicyEstimator is used for estimating the feerate needed for a transaction to be included i...
Definition: fees.h:133
The basic transaction that is broadcasted on the network and contained in blocks.
Definition: transaction.h:295
CTxMemPoolEntry stores data about the corresponding transaction, as well as data about all in-mempool...
Definition: mempool_entry.h:66
AutoFile open()
Definition: util.h:258
T PickValueInArray(const T(&array)[size])
fs::path FeeestPath(const ArgsManager &argsman)
Definition: fees_args.cpp:13
#define LIMITED_WHILE(condition, limit)
Can be used to limit a theoretically unbounded loop.
Definition: fuzz.h:18
static constexpr auto ALL_FEE_ESTIMATE_HORIZONS
Definition: fees.h:35
void initialize_policy_estimator()
FUZZ_TARGET_INIT(policy_estimator, initialize_policy_estimator)
Basic testing setup.
Definition: setup_common.h:85
CTxMemPoolEntry ConsumeTxMemPoolEntry(FuzzedDataProvider &fuzzed_data_provider, const CTransaction &tx) noexcept
Definition: mempool.cpp:17
FuzzedAutoFileProvider ConsumeAutoFile(FuzzedDataProvider &fuzzed_data_provider) noexcept
Definition: util.h:264
uint256 ConsumeUInt256(FuzzedDataProvider &fuzzed_data_provider) noexcept
Definition: util.h:149
size_t CallOneOf(FuzzedDataProvider &fuzzed_data_provider, Callables... callables)
Definition: util.h:36