load_external_block_file.cpp raw

   1  // Copyright (c) 2020-2021 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 <chainparams.h>
   6  #include <clientversion.h>
   7  #include <flatfile.h>
   8  #include <test/fuzz/FuzzedDataProvider.h>
   9  #include <test/fuzz/fuzz.h>
  10  #include <test/fuzz/util.h>
  11  #include <test/util/setup_common.h>
  12  #include <util/time.h>
  13  #include <validation.h>
  14  
  15  #include <cstdint>
  16  #include <vector>
  17  
  18  namespace {
  19  const TestingSetup* g_setup;
  20  } // namespace
  21  
  22  void initialize_load_external_block_file()
  23  {
  24      static const auto testing_setup = MakeNoLogFileContext<const TestingSetup>();
  25      g_setup = testing_setup.get();
  26  }
  27  
  28  FUZZ_TARGET(load_external_block_file, .init = initialize_load_external_block_file)
  29  {
  30      FuzzedDataProvider fuzzed_data_provider{buffer.data(), buffer.size()};
  31      SetMockTime(ConsumeTime(fuzzed_data_provider));
  32      FuzzedFileProvider fuzzed_file_provider{fuzzed_data_provider};
  33      AutoFile fuzzed_block_file{fuzzed_file_provider.open()};
  34      if (fuzzed_block_file.IsNull()) {
  35          return;
  36      }
  37      if (fuzzed_data_provider.ConsumeBool()) {
  38          // Corresponds to the -reindex case (track orphan blocks across files).
  39          FlatFilePos flat_file_pos;
  40          std::multimap<uint256, FlatFilePos> blocks_with_unknown_parent;
  41          g_setup->m_node.chainman->LoadExternalBlockFile(fuzzed_block_file, &flat_file_pos, &blocks_with_unknown_parent);
  42      } else {
  43          // Corresponds to the -loadblock= case (orphan blocks aren't tracked across files).
  44          g_setup->m_node.chainman->LoadExternalBlockFile(fuzzed_block_file);
  45      }
  46  }
  47