Branch data     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         [ #  # ]:           0 : 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                 :           0 :             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                 :             :             } 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                 :             :         }
      39         [ #  # ]:           0 :         (void)descriptor_cache.GetCachedParentExtPubKeys();
      40         [ #  # ]:           0 :         (void)descriptor_cache.GetCachedDerivedExtPubKeys();
      41                 :           0 :     }
      42                 :           0 : }