1  // Copyright (c) 2020-2021 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  }