Bitcoin Core  27.99.0
P2P Digital Currency
script_descriptor_cache.cpp
Go to the documentation of this file.
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>
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 }
Cache for single descriptor's derived extended pubkeys.
Definition: descriptor.h:19
bool GetCachedParentExtPubKey(uint32_t key_exp_pos, CExtPubKey &xpub) const
Retrieve a cached parent xpub.
std::unordered_map< uint32_t, ExtPubKeyMap > GetCachedDerivedExtPubKeys() const
Retrieve all cached derived xpubs.
void CacheDerivedExtPubKey(uint32_t key_exp_pos, uint32_t der_index, const CExtPubKey &xpub)
Cache an xpub derived at an index.
ExtPubKeyMap GetCachedParentExtPubKeys() const
Retrieve all cached parent xpubs.
void CacheParentExtPubKey(uint32_t key_exp_pos, const CExtPubKey &xpub)
Cache a parent xpub.
bool GetCachedDerivedExtPubKey(uint32_t key_exp_pos, uint32_t der_index, CExtPubKey &xpub) const
Retrieve a cached xpub derived at an index.
std::vector< T > ConsumeBytes(size_t num_bytes)
#define LIMITED_WHILE(condition, limit)
Can be used to limit a theoretically unbounded loop.
Definition: fuzz.h:23
const unsigned int BIP32_EXTKEY_SIZE
Definition: pubkey.h:19
FUZZ_TARGET(script_descriptor_cache)
void Decode(const unsigned char code[BIP32_EXTKEY_SIZE])
Definition: pubkey.cpp:375
assert(!tx.IsCoinBase())