Line data    Source code

       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           4 : FUZZ_TARGET(script_descriptor_cache)
      17             : {
      18           0 :     FuzzedDataProvider fuzzed_data_provider(buffer.data(), buffer.size());
      19           0 :     DescriptorCache descriptor_cache;
      20           0 :     LIMITED_WHILE(fuzzed_data_provider.ConsumeBool(), 10000) {
      21           0 :         const std::vector<uint8_t> code = fuzzed_data_provider.ConsumeBytes<uint8_t>(BIP32_EXTKEY_SIZE);
      22           0 :         if (code.size() == BIP32_EXTKEY_SIZE) {
      23           0 :             CExtPubKey xpub;
      24           0 :             xpub.Decode(code.data());
      25           2 :             const uint32_t key_exp_pos = fuzzed_data_provider.ConsumeIntegral<uint32_t>();
      26           0 :             CExtPubKey xpub_fetched;
      27           0 :             if (fuzzed_data_provider.ConsumeBool()) {
      28           0 :                 (void)descriptor_cache.GetCachedParentExtPubKey(key_exp_pos, xpub_fetched);
      29           0 :                 descriptor_cache.CacheParentExtPubKey(key_exp_pos, xpub);
      30           0 :                 assert(descriptor_cache.GetCachedParentExtPubKey(key_exp_pos, xpub_fetched));
      31           0 :             } else {
      32           0 :                 const uint32_t der_index = fuzzed_data_provider.ConsumeIntegral<uint32_t>();
      33           0 :                 (void)descriptor_cache.GetCachedDerivedExtPubKey(key_exp_pos, der_index, xpub_fetched);
      34           0 :                 descriptor_cache.CacheDerivedExtPubKey(key_exp_pos, der_index, xpub);
      35           0 :                 assert(descriptor_cache.GetCachedDerivedExtPubKey(key_exp_pos, der_index, xpub_fetched));
      36             :             }
      37           0 :             assert(xpub == xpub_fetched);
      38           0 :         }
      39           0 :         (void)descriptor_cache.GetCachedParentExtPubKeys();
      40           0 :         (void)descriptor_cache.GetCachedDerivedExtPubKeys();
      41           0 :     }
      42           0 : }