Branch data     Line data    Source code

       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 <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                 :           1 : void initialize_validation_load_mempool()
      34                 :             : {
      35   [ +  -  +  - ]:           2 :     static const auto testing_setup = MakeNoLogFileContext<const TestingSetup>();
      36                 :           1 :     g_setup = testing_setup.get();
      37         [ +  - ]:           2 : }
      38                 :             : 
      39         [ +  - ]:        1827 : FUZZ_TARGET(validation_load_mempool, .init = initialize_validation_load_mempool)
      40                 :             : {
      41                 :        1415 :     FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()};
      42                 :        1415 :     SetMockTime(ConsumeTime(fuzzed_data_provider));
      43         [ +  - ]:        1415 :     FuzzedFileProvider fuzzed_file_provider{fuzzed_data_provider};
      44                 :             : 
      45         [ +  - ]:        1415 :     bilingual_str error;
      46   [ +  -  +  - ]:        1415 :     CTxMemPool pool{MemPoolOptionsForTest(g_setup->m_node), error};
      47         [ +  - ]:        1415 :     Assert(error.empty());
      48                 :             : 
      49         [ +  - ]:        1415 :     auto& chainstate{static_cast<DummyChainState&>(g_setup->m_node.chainman->ActiveChainstate())};
      50         [ +  - ]:        1415 :     chainstate.SetMempool(&pool);
      51                 :             : 
      52                 :        4245 :     auto fuzzed_fopen = [&](const fs::path&, const char*) {
      53                 :        2830 :         return fuzzed_file_provider.open();
      54                 :        1415 :     };
      55         [ +  - ]:        2830 :     (void)LoadMempool(pool, MempoolPath(g_setup->m_args), chainstate,
      56                 :             :                       {
      57                 :             :                           .mockable_fopen_function = fuzzed_fopen,
      58                 :             :                       });
      59         [ +  - ]:        1415 :     pool.SetLoadTried(true);
      60   [ +  -  +  - ]:        2830 :     (void)DumpMempool(pool, MempoolPath(g_setup->m_args), fuzzed_fopen, true);
      61         [ +  - ]:        4245 : }