validation_load_mempool.cpp raw

   1  // Copyright (c) 2020-2022 The Limenka 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 <node/mempool_persist.h>
   6  
   7  #include <node/mempool_args.h>
   8  #include <node/mempool_persist_args.h>
   9  #include <test/fuzz/FuzzedDataProvider.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 <test/util/txmempool.h>
  15  #include <txmempool.h>
  16  #include <util/check.h>
  17  #include <util/time.h>
  18  #include <util/translation.h>
  19  #include <validation.h>
  20  
  21  #include <cstdint>
  22  #include <vector>
  23  
  24  using node::DumpMempool;
  25  using node::LoadMempool;
  26  
  27  using node::MempoolPath;
  28  
  29  namespace {
  30  const TestingSetup* g_setup;
  31  } // namespace
  32  
  33  void initialize_validation_load_mempool()
  34  {
  35      static const auto testing_setup = MakeNoLogFileContext<const TestingSetup>();
  36      g_setup = testing_setup.get();
  37  }
  38  
  39  FUZZ_TARGET(validation_load_mempool, .init = initialize_validation_load_mempool)
  40  {
  41      SeedRandomStateForTest(SeedRand::ZEROS);
  42      FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()};
  43      SetMockTime(ConsumeTime(fuzzed_data_provider));
  44      FuzzedFileProvider fuzzed_file_provider{fuzzed_data_provider};
  45  
  46      bilingual_str error;
  47      CTxMemPool pool{MemPoolOptionsForTest(g_setup->m_node), error};
  48      Assert(error.empty());
  49  
  50      auto& chainstate{static_cast<DummyChainState&>(g_setup->m_node.chainman->ActiveChainstate())};
  51      chainstate.SetMempool(&pool);
  52  
  53      auto fuzzed_fopen = [&](const fs::path&, const char*) {
  54          return fuzzed_file_provider.open();
  55      };
  56      (void)LoadMempool(pool, MempoolPath(g_setup->m_args), chainstate,
  57                        {
  58                            .mockable_fopen_function = fuzzed_fopen,
  59                            .load_knots_data = true,
  60                        });
  61      pool.SetLoadTried(true);
  62      (void)DumpMempool(pool, MempoolPath(g_setup->m_args), fuzzed_fopen, true);
  63  }
  64