script_descriptor_cache.cpp raw

   1  // Copyright (c) 2020-present 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 <pubkey.h>
   6  #include <script/descriptor.h>
   7  #include <test/fuzz/FuzzedDataProvider.h>
   8  #include <test/fuzz/fuzz.h>
   9  #include <test/fuzz/util.h>
  10  
  11  #include <cstdint>
  12  #include <optional>
  13  #include <string>
  14  #include <vector>
  15  
  16  FUZZ_TARGET(script_descriptor_cache)
  17  {
  18      FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
  19      DescriptorCache descriptor_cache;
  20      LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 10000) {
  21          const std::vector<uint8_t> code = fuzzed_data_provider.ConsumeBytes<uint8_t>(BIP32_EXTKEY_SIZE);
  22          if (code.size() == BIP32_EXTKEY_SIZE) {
  23              CExtPubKey xpub;
  24              xpub.Decode(code.data());
  25              const uint32_t key_exp_pos = fuzzed_data_provider.ConsumeIntegral<uint32_t>();
  26              CExtPubKey xpub_fetched;
  27              if (fuzzed_data_provider.ConsumeBool()) {
  28                  (void)descriptor_cache.GetCachedParentExtPubKey(key_exp_pos, xpub_fetched);
  29                  descriptor_cache.CacheParentExtPubKey(key_exp_pos, xpub);
  30                  assert(descriptor_cache.GetCachedParentExtPubKey(key_exp_pos, xpub_fetched));
  31              } else {
  32                  const uint32_t der_index = fuzzed_data_provider.ConsumeIntegral<uint32_t>();
  33                  (void)descriptor_cache.GetCachedDerivedExtPubKey(key_exp_pos, der_index, xpub_fetched);
  34                  descriptor_cache.CacheDerivedExtPubKey(key_exp_pos, der_index, xpub);
  35                  assert(descriptor_cache.GetCachedDerivedExtPubKey(key_exp_pos, der_index, xpub_fetched));
  36              }
  37              assert(xpub == xpub_fetched);
  38          }
  39          (void)descriptor_cache.GetCachedParentExtPubKeys();
  40          (void)descriptor_cache.GetCachedDerivedExtPubKeys();
  41      }
  42  }
  43