Branch data Line data Source code
1 : : // Copyright (c) 2018-2022 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 <script/descriptor.h>
6 : :
7 : : #include <hash.h>
8 : : #include <key_io.h>
9 : : #include <pubkey.h>
10 : : #include <script/miniscript.h>
11 : : #include <script/parsing.h>
12 : : #include <script/script.h>
13 : : #include <script/signingprovider.h>
14 : : #include <script/solver.h>
15 : : #include <uint256.h>
16 : :
17 : : #include <common/args.h>
18 : : #include <span.h>
19 : : #include <util/bip32.h>
20 : : #include <util/check.h>
21 : : #include <util/strencodings.h>
22 : : #include <util/vector.h>
23 : :
24 : : #include <algorithm>
25 : : #include <memory>
26 : : #include <numeric>
27 : : #include <optional>
28 : : #include <string>
29 : : #include <vector>
30 : :
31 : : using util::Split;
32 : :
33 : : namespace {
34 : :
35 : : ////////////////////////////////////////////////////////////////////////////
36 : : // Checksum //
37 : : ////////////////////////////////////////////////////////////////////////////
38 : :
39 : : // This section implements a checksum algorithm for descriptors with the
40 : : // following properties:
41 : : // * Mistakes in a descriptor string are measured in "symbol errors". The higher
42 : : // the number of symbol errors, the harder it is to detect:
43 : : // * An error substituting a character from 0123456789()[],'/*abcdefgh@:$%{} for
44 : : // another in that set always counts as 1 symbol error.
45 : : // * Note that hex encoded keys are covered by these characters. Xprvs and
46 : : // xpubs use other characters too, but already have their own checksum
47 : : // mechanism.
48 : : // * Function names like "multi()" use other characters, but mistakes in
49 : : // these would generally result in an unparsable descriptor.
50 : : // * A case error always counts as 1 symbol error.
51 : : // * Any other 1 character substitution error counts as 1 or 2 symbol errors.
52 : : // * Any 1 symbol error is always detected.
53 : : // * Any 2 or 3 symbol error in a descriptor of up to 49154 characters is always detected.
54 : : // * Any 4 symbol error in a descriptor of up to 507 characters is always detected.
55 : : // * Any 5 symbol error in a descriptor of up to 77 characters is always detected.
56 : : // * Is optimized to minimize the chance a 5 symbol error in a descriptor up to 387 characters is undetected
57 : : // * Random errors have a chance of 1 in 2**40 of being undetected.
58 : : //
59 : : // These properties are achieved by expanding every group of 3 (non checksum) characters into
60 : : // 4 GF(32) symbols, over which a cyclic code is defined.
61 : :
62 : : /*
63 : : * Interprets c as 8 groups of 5 bits which are the coefficients of a degree 8 polynomial over GF(32),
64 : : * multiplies that polynomial by x, computes its remainder modulo a generator, and adds the constant term val.
65 : : *
66 : : * This generator is G(x) = x^8 + {30}x^7 + {23}x^6 + {15}x^5 + {14}x^4 + {10}x^3 + {6}x^2 + {12}x + {9}.
67 : : * It is chosen to define an cyclic error detecting code which is selected by:
68 : : * - Starting from all BCH codes over GF(32) of degree 8 and below, which by construction guarantee detecting
69 : : * 3 errors in windows up to 19000 symbols.
70 : : * - Taking all those generators, and for degree 7 ones, extend them to degree 8 by adding all degree-1 factors.
71 : : * - Selecting just the set of generators that guarantee detecting 4 errors in a window of length 512.
72 : : * - Selecting one of those with best worst-case behavior for 5 errors in windows of length up to 512.
73 : : *
74 : : * The generator and the constants to implement it can be verified using this Sage code:
75 : : * B = GF(2) # Binary field
76 : : * BP.<b> = B[] # Polynomials over the binary field
77 : : * F_mod = b**5 + b**3 + 1
78 : : * F.<f> = GF(32, modulus=F_mod, repr='int') # GF(32) definition
79 : : * FP.<x> = F[] # Polynomials over GF(32)
80 : : * E_mod = x**3 + x + F.fetch_int(8)
81 : : * E.<e> = F.extension(E_mod) # Extension field definition
82 : : * alpha = e**2743 # Choice of an element in extension field
83 : : * for p in divisors(E.order() - 1): # Verify alpha has order 32767.
84 : : * assert((alpha**p == 1) == (p % 32767 == 0))
85 : : * G = lcm([(alpha**i).minpoly() for i in [1056,1057,1058]] + [x + 1])
86 : : * print(G) # Print out the generator
87 : : * for i in [1,2,4,8,16]: # Print out {1,2,4,8,16}*(G mod x^8), packed in hex integers.
88 : : * v = 0
89 : : * for coef in reversed((F.fetch_int(i)*(G % x**8)).coefficients(sparse=True)):
90 : : * v = v*32 + coef.integer_representation()
91 : : * print("0x%x" % v)
92 : : */
93 : 2897742 : uint64_t PolyMod(uint64_t c, int val)
94 : : {
95 : 2897742 : uint8_t c0 = c >> 35;
96 : 2897742 : c = ((c & 0x7ffffffff) << 5) ^ val;
97 [ + + ]: 2897742 : if (c0 & 1) c ^= 0xf5dee51989;
98 [ + + ]: 2897742 : if (c0 & 2) c ^= 0xa9fdca3312;
99 [ + + ]: 2897742 : if (c0 & 4) c ^= 0x1bab10e32d;
100 [ + + ]: 2897742 : if (c0 & 8) c ^= 0x3706b1677a;
101 [ + + ]: 2897742 : if (c0 & 16) c ^= 0x644d626ffd;
102 : 2897742 : return c;
103 : : }
104 : :
105 : 15268 : std::string DescriptorChecksum(const Span<const char>& span)
106 : : {
107 : : /** A character set designed such that:
108 : : * - The most common 'unprotected' descriptor characters (hex, keypaths) are in the first group of 32.
109 : : * - Case errors cause an offset that's a multiple of 32.
110 : : * - As many alphabetic characters are in the same group (while following the above restrictions).
111 : : *
112 : : * If p(x) gives the position of a character c in this character set, every group of 3 characters
113 : : * (a,b,c) is encoded as the 4 symbols (p(a) & 31, p(b) & 31, p(c) & 31, (p(a) / 32) + 3 * (p(b) / 32) + 9 * (p(c) / 32).
114 : : * This means that changes that only affect the lower 5 bits of the position, or only the higher 2 bits, will just
115 : : * affect a single symbol.
116 : : *
117 : : * As a result, within-group-of-32 errors count as 1 symbol, as do cross-group errors that don't affect
118 : : * the position within the groups.
119 : : */
120 : 15268 : static const std::string INPUT_CHARSET =
121 : : "0123456789()[],'/*abcdefgh@:$%{}"
122 : : "IJKLMNOPQRSTUVWXYZ&+-.;<=>?!^_|~"
123 [ + + + - : 15300 : "ijklmnopqrstuvwxyzABCDEFGH`#\"\\ ";
+ - ]
124 : :
125 : : /** The character set for the checksum itself (same as bech32). */
126 [ + + + - : 15300 : static const std::string CHECKSUM_CHARSET = "qpzry9x8gf2tvdw0s3jn54khce6mua7l";
+ - ]
127 : :
128 : 15268 : uint64_t c = 1;
129 : 15268 : int cls = 0;
130 : 15268 : int clscount = 0;
131 [ + + ]: 2093082 : for (auto ch : span) {
132 : 2077815 : auto pos = INPUT_CHARSET.find(ch);
133 [ + + ]: 2077815 : if (pos == std::string::npos) return "";
134 : 2077814 : c = PolyMod(c, pos & 31); // Emit a symbol for the position inside the group, for every character.
135 : 2077814 : cls = cls * 3 + (pos >> 5); // Accumulate the group numbers
136 [ + + ]: 2077814 : if (++clscount == 3) {
137 : : // Emit an extra symbol representing the group numbers, for every 3 characters.
138 : 683447 : c = PolyMod(c, cls);
139 : 683447 : cls = 0;
140 : 683447 : clscount = 0;
141 : : }
142 : : }
143 [ + + ]: 15267 : if (clscount > 0) c = PolyMod(c, cls);
144 [ + + ]: 137403 : for (int j = 0; j < 8; ++j) c = PolyMod(c, 0); // Shift further to determine the checksum.
145 : 15267 : c ^= 1; // Prevent appending zeroes from not affecting the checksum.
146 : :
147 : 15267 : std::string ret(8, ' ');
148 [ + + ]: 137403 : for (int j = 0; j < 8; ++j) ret[j] = CHECKSUM_CHARSET[(c >> (5 * (7 - j))) & 31];
149 : 15267 : return ret;
150 : 15267 : }
151 : :
152 [ + - + - ]: 29192 : std::string AddChecksum(const std::string& str) { return str + "#" + DescriptorChecksum(str); }
153 : :
154 : : ////////////////////////////////////////////////////////////////////////////
155 : : // Internal representation //
156 : : ////////////////////////////////////////////////////////////////////////////
157 : :
158 : : typedef std::vector<uint32_t> KeyPath;
159 : :
160 : : /** Interface for public key objects in descriptors. */
161 : : struct PubkeyProvider
162 : : {
163 : : protected:
164 : : //! Index of this key expression in the descriptor
165 : : //! E.g. If this PubkeyProvider is key1 in multi(2, key1, key2, key3), then m_expr_index = 0
166 : : uint32_t m_expr_index;
167 : :
168 : : public:
169 : 18179 : explicit PubkeyProvider(uint32_t exp_index) : m_expr_index(exp_index) {}
170 : :
171 : 0 : virtual ~PubkeyProvider() = default;
172 : :
173 : : /** Compare two public keys represented by this provider.
174 : : * Used by the Miniscript descriptors to check for duplicate keys in the script.
175 : : */
176 : 465 : bool operator<(PubkeyProvider& other) const {
177 [ + - ]: 465 : CPubKey a, b;
178 : 465 : SigningProvider dummy;
179 [ + - ]: 465 : KeyOriginInfo dummy_info;
180 : :
181 [ + - ]: 465 : GetPubKey(0, dummy, a, dummy_info);
182 [ + - ]: 465 : other.GetPubKey(0, dummy, b, dummy_info);
183 : :
184 : 465 : return a < b;
185 : 465 : }
186 : :
187 : : /** Derive a public key.
188 : : * read_cache is the cache to read keys from (if not nullptr)
189 : : * write_cache is the cache to write keys to (if not nullptr)
190 : : * Caches are not exclusive but this is not tested. Currently we use them exclusively
191 : : */
192 : : virtual bool GetPubKey(int pos, const SigningProvider& arg, CPubKey& key, KeyOriginInfo& info, const DescriptorCache* read_cache = nullptr, DescriptorCache* write_cache = nullptr) const = 0;
193 : :
194 : : /** Whether this represent multiple public keys at different positions. */
195 : : virtual bool IsRange() const = 0;
196 : :
197 : : /** Get the size of the generated public key(s) in bytes (33 or 65). */
198 : : virtual size_t GetSize() const = 0;
199 : :
200 : : enum class StringType {
201 : : PUBLIC,
202 : : COMPAT // string calculation that mustn't change over time to stay compatible with previous software versions
203 : : };
204 : :
205 : : /** Get the descriptor string form. */
206 : : virtual std::string ToString(StringType type=StringType::PUBLIC) const = 0;
207 : :
208 : : /** Get the descriptor string form including private data (if available in arg). */
209 : : virtual bool ToPrivateString(const SigningProvider& arg, std::string& out) const = 0;
210 : :
211 : : /** Get the descriptor string form with the xpub at the last hardened derivation,
212 : : * and always use h for hardened derivation.
213 : : */
214 : : virtual bool ToNormalizedString(const SigningProvider& arg, std::string& out, const DescriptorCache* cache = nullptr) const = 0;
215 : :
216 : : /** Derive a private key, if private data is available in arg. */
217 : : virtual bool GetPrivKey(int pos, const SigningProvider& arg, CKey& key) const = 0;
218 : :
219 : : /** Return the non-extended public key for this PubkeyProvider, if it has one. */
220 : : virtual std::optional<CPubKey> GetRootPubKey() const = 0;
221 : : /** Return the extended public key for this PubkeyProvider, if it has one. */
222 : : virtual std::optional<CExtPubKey> GetRootExtPubKey() const = 0;
223 : :
224 : : /** Make a deep copy of this PubkeyProvider */
225 : : virtual std::unique_ptr<PubkeyProvider> Clone() const = 0;
226 : : };
227 : :
228 : : class OriginPubkeyProvider final : public PubkeyProvider
229 : : {
230 : : KeyOriginInfo m_origin;
231 : : std::unique_ptr<PubkeyProvider> m_provider;
232 : : bool m_apostrophe;
233 : :
234 : 154 : std::string OriginString(StringType type, bool normalized=false) const
235 : : {
236 : : // If StringType==COMPAT, always use the apostrophe to stay compatible with previous versions
237 [ + + + + : 154 : bool use_apostrophe = (!normalized && m_apostrophe) || type == StringType::COMPAT;
+ + ]
238 [ + - + - ]: 308 : return HexStr(m_origin.fingerprint) + FormatHDKeypath(m_origin.path, use_apostrophe);
239 : : }
240 : :
241 : : public:
242 : 8552 : OriginPubkeyProvider(uint32_t exp_index, KeyOriginInfo info, std::unique_ptr<PubkeyProvider> provider, bool apostrophe) : PubkeyProvider(exp_index), m_origin(std::move(info)), m_provider(std::move(provider)), m_apostrophe(apostrophe) {}
243 : 2471 : bool GetPubKey(int pos, const SigningProvider& arg, CPubKey& key, KeyOriginInfo& info, const DescriptorCache* read_cache = nullptr, DescriptorCache* write_cache = nullptr) const override
244 : : {
245 [ + - ]: 2471 : if (!m_provider->GetPubKey(pos, arg, key, info, read_cache, write_cache)) return false;
246 : 2471 : std::copy(std::begin(m_origin.fingerprint), std::end(m_origin.fingerprint), info.fingerprint);
247 : 2471 : info.path.insert(info.path.begin(), m_origin.path.begin(), m_origin.path.end());
248 : 2471 : return true;
249 : : }
250 : 34 : bool IsRange() const override { return m_provider->IsRange(); }
251 : 49 : size_t GetSize() const override { return m_provider->GetSize(); }
252 [ + - + - : 234 : std::string ToString(StringType type) const override { return "[" + OriginString(type) + "]" + m_provider->ToString(type); }
+ - ]
253 : 80 : bool ToPrivateString(const SigningProvider& arg, std::string& ret) const override
254 : : {
255 [ + - ]: 80 : std::string sub;
256 [ + - + + ]: 80 : if (!m_provider->ToPrivateString(arg, sub)) return false;
257 [ + - + - : 92 : ret = "[" + OriginString(StringType::PUBLIC) + "]" + std::move(sub);
+ - ]
258 : 46 : return true;
259 : 80 : }
260 : 30 : bool ToNormalizedString(const SigningProvider& arg, std::string& ret, const DescriptorCache* cache) const override
261 : : {
262 [ + - ]: 30 : std::string sub;
263 [ + - + - ]: 30 : if (!m_provider->ToNormalizedString(arg, sub, cache)) return false;
264 : : // If m_provider is a BIP32PubkeyProvider, we may get a string formatted like a OriginPubkeyProvider
265 : : // In that case, we need to strip out the leading square bracket and fingerprint from the substring,
266 : : // and append that to our own origin string.
267 [ + + ]: 30 : if (sub[0] == '[') {
268 [ + - ]: 4 : sub = sub.substr(9);
269 [ + - + - : 4 : ret = "[" + OriginString(StringType::PUBLIC, /*normalized=*/true) + std::move(sub);
+ - ]
270 : : } else {
271 [ + - + - : 52 : ret = "[" + OriginString(StringType::PUBLIC, /*normalized=*/true) + "]" + std::move(sub);
+ - ]
272 : : }
273 : : return true;
274 : 30 : }
275 : 0 : bool GetPrivKey(int pos, const SigningProvider& arg, CKey& key) const override
276 : : {
277 : 0 : return m_provider->GetPrivKey(pos, arg, key);
278 : : }
279 : 0 : std::optional<CPubKey> GetRootPubKey() const override
280 : : {
281 : 0 : return m_provider->GetRootPubKey();
282 : : }
283 : 0 : std::optional<CExtPubKey> GetRootExtPubKey() const override
284 : : {
285 : 0 : return m_provider->GetRootExtPubKey();
286 : : }
287 : 0 : std::unique_ptr<PubkeyProvider> Clone() const override
288 : : {
289 [ # # # # ]: 0 : return std::make_unique<OriginPubkeyProvider>(m_expr_index, m_origin, m_provider->Clone(), m_apostrophe);
290 : : }
291 : : };
292 : :
293 : : /** An object representing a parsed constant public key in a descriptor. */
294 : 0 : class ConstPubkeyProvider final : public PubkeyProvider
295 : : {
296 : : CPubKey m_pubkey;
297 : : bool m_xonly;
298 : :
299 : : public:
300 : 8893 : ConstPubkeyProvider(uint32_t exp_index, const CPubKey& pubkey, bool xonly) : PubkeyProvider(exp_index), m_pubkey(pubkey), m_xonly(xonly) {}
301 : 4816 : bool GetPubKey(int pos, const SigningProvider& arg, CPubKey& key, KeyOriginInfo& info, const DescriptorCache* read_cache = nullptr, DescriptorCache* write_cache = nullptr) const override
302 : : {
303 : 4816 : key = m_pubkey;
304 [ + + ]: 4816 : info.path.clear();
305 : 4816 : CKeyID keyid = m_pubkey.GetID();
306 : 4816 : std::copy(keyid.begin(), keyid.begin() + sizeof(info.fingerprint), info.fingerprint);
307 : 4816 : return true;
308 : : }
309 : 690 : bool IsRange() const override { return false; }
310 : 386 : size_t GetSize() const override { return m_pubkey.size(); }
311 [ + + + - ]: 1307 : std::string ToString(StringType type) const override { return m_xonly ? HexStr(m_pubkey).substr(2) : HexStr(m_pubkey); }
312 : 226 : bool ToPrivateString(const SigningProvider& arg, std::string& ret) const override
313 : : {
314 : 226 : CKey key;
315 [ + + ]: 226 : if (m_xonly) {
316 [ + - + + ]: 48 : for (const auto& keyid : XOnlyPubKey(m_pubkey).GetKeyIDs()) {
317 [ + - ]: 40 : arg.GetKey(keyid, key);
318 [ + + ]: 40 : if (key.IsValid()) break;
319 : 20 : }
320 : : } else {
321 [ + - + - ]: 206 : arg.GetKey(m_pubkey.GetID(), key);
322 : : }
323 [ + + ]: 226 : if (!key.IsValid()) return false;
324 [ + - ]: 174 : ret = EncodeSecret(key);
325 : 174 : return true;
326 : 226 : }
327 : 200 : bool ToNormalizedString(const SigningProvider& arg, std::string& ret, const DescriptorCache* cache) const override
328 : : {
329 : 200 : ret = ToString(StringType::PUBLIC);
330 : 200 : return true;
331 : : }
332 : 13 : bool GetPrivKey(int pos, const SigningProvider& arg, CKey& key) const override
333 : : {
334 : 13 : return arg.GetKey(m_pubkey.GetID(), key);
335 : : }
336 : 0 : std::optional<CPubKey> GetRootPubKey() const override
337 : : {
338 : 0 : return m_pubkey;
339 : : }
340 : 0 : std::optional<CExtPubKey> GetRootExtPubKey() const override
341 : : {
342 : 0 : return std::nullopt;
343 : : }
344 : 0 : std::unique_ptr<PubkeyProvider> Clone() const override
345 : : {
346 : 0 : return std::make_unique<ConstPubkeyProvider>(m_expr_index, m_pubkey, m_xonly);
347 : : }
348 : : };
349 : :
350 : : enum class DeriveType {
351 : : NO,
352 : : UNHARDENED,
353 : : HARDENED,
354 : : };
355 : :
356 : : /** An object representing a parsed extended public key in a descriptor. */
357 : : class BIP32PubkeyProvider final : public PubkeyProvider
358 : : {
359 : : // Root xpub, path, and final derivation step type being used, if any
360 : : CExtPubKey m_root_extkey;
361 : : KeyPath m_path;
362 : : DeriveType m_derive;
363 : : // Whether ' or h is used in harded derivation
364 : : bool m_apostrophe;
365 : :
366 : 1792 : bool GetExtKey(const SigningProvider& arg, CExtKey& ret) const
367 : : {
368 : 1792 : CKey key;
369 [ + - + - : 1792 : if (!arg.GetKey(m_root_extkey.pubkey.GetID(), key)) return false;
+ + ]
370 : 1664 : ret.nDepth = m_root_extkey.nDepth;
371 : 1664 : std::copy(m_root_extkey.vchFingerprint, m_root_extkey.vchFingerprint + sizeof(ret.vchFingerprint), ret.vchFingerprint);
372 : 1664 : ret.nChild = m_root_extkey.nChild;
373 : 1664 : ret.chaincode = m_root_extkey.chaincode;
374 [ + - ]: 1664 : ret.key = key;
375 : : return true;
376 : 1792 : }
377 : :
378 : : // Derives the last xprv
379 : 1434 : bool GetDerivedExtKey(const SigningProvider& arg, CExtKey& xprv, CExtKey& last_hardened) const
380 : : {
381 [ + + ]: 1434 : if (!GetExtKey(arg, xprv)) return false;
382 [ + + ]: 4966 : for (auto entry : m_path) {
383 [ + - ]: 3536 : if (!xprv.Derive(xprv, entry)) return false;
384 [ + + ]: 3536 : if (entry >> 31) {
385 : 3012 : last_hardened = xprv;
386 : : }
387 : : }
388 : : return true;
389 : : }
390 : :
391 : 2552 : bool IsHardened() const
392 : : {
393 [ + + ]: 2552 : if (m_derive == DeriveType::HARDENED) return true;
394 [ + + ]: 2760 : for (auto entry : m_path) {
395 [ + + ]: 1620 : if (entry >> 31) return true;
396 : : }
397 : : return false;
398 : : }
399 : :
400 : : public:
401 : 734 : BIP32PubkeyProvider(uint32_t exp_index, const CExtPubKey& extkey, KeyPath path, DeriveType derive, bool apostrophe) : PubkeyProvider(exp_index), m_root_extkey(extkey), m_path(std::move(path)), m_derive(derive), m_apostrophe(apostrophe) {}
402 : 26910 : bool IsRange() const override { return m_derive != DeriveType::NO; }
403 : 164 : size_t GetSize() const override { return 33; }
404 : 310741 : bool GetPubKey(int pos, const SigningProvider& arg, CPubKey& key_out, KeyOriginInfo& final_info_out, const DescriptorCache* read_cache = nullptr, DescriptorCache* write_cache = nullptr) const override
405 : : {
406 : : // Info of parent of the to be derived pubkey
407 [ + - ]: 310741 : KeyOriginInfo parent_info;
408 [ + - ]: 310741 : CKeyID keyid = m_root_extkey.pubkey.GetID();
409 : 310741 : std::copy(keyid.begin(), keyid.begin() + sizeof(parent_info.fingerprint), parent_info.fingerprint);
410 [ + - ]: 310741 : parent_info.path = m_path;
411 : :
412 : : // Info of the derived key itself which is copied out upon successful completion
413 [ + - ]: 310741 : KeyOriginInfo final_info_out_tmp = parent_info;
414 [ + + + - ]: 310741 : if (m_derive == DeriveType::UNHARDENED) final_info_out_tmp.path.push_back((uint32_t)pos);
415 [ + + + - ]: 310741 : if (m_derive == DeriveType::HARDENED) final_info_out_tmp.path.push_back(((uint32_t)pos) | 0x80000000L);
416 : :
417 : : // Derive keys or fetch them from cache
418 : 310741 : CExtPubKey final_extkey = m_root_extkey;
419 : 310741 : CExtPubKey parent_extkey = m_root_extkey;
420 [ + + ]: 310741 : CExtPubKey last_hardened_extkey;
421 : 310741 : bool der = true;
422 [ + + ]: 310741 : if (read_cache) {
423 [ + - + + ]: 308189 : if (!read_cache->GetCachedDerivedExtPubKey(m_expr_index, pos, final_extkey)) {
424 [ + + ]: 308147 : if (m_derive == DeriveType::HARDENED) return false;
425 : : // Try to get the derivation parent
426 [ + - + + ]: 307147 : if (!read_cache->GetCachedParentExtPubKey(m_expr_index, parent_extkey)) return false;
427 : 306873 : final_extkey = parent_extkey;
428 [ + + + - ]: 306873 : if (m_derive == DeriveType::UNHARDENED) der = parent_extkey.Derive(final_extkey, pos);
429 : : }
430 [ + + ]: 2552 : } else if (IsHardened()) {
431 [ + - ]: 1412 : CExtKey xprv;
432 : 1412 : CExtKey lh_xprv;
433 [ + - + + ]: 1412 : if (!GetDerivedExtKey(arg, xprv, lh_xprv)) return false;
434 [ + - ]: 1408 : parent_extkey = xprv.Neuter();
435 [ + + + - ]: 1408 : if (m_derive == DeriveType::UNHARDENED) der = xprv.Derive(xprv, pos);
436 [ + + + - ]: 1408 : if (m_derive == DeriveType::HARDENED) der = xprv.Derive(xprv, pos | 0x80000000UL);
437 [ + - ]: 1408 : final_extkey = xprv.Neuter();
438 [ + + ]: 1408 : if (lh_xprv.key.IsValid()) {
439 [ + - ]: 1378 : last_hardened_extkey = lh_xprv.Neuter();
440 : : }
441 : 1412 : } else {
442 [ + + ]: 2384 : for (auto entry : m_path) {
443 [ + - + - ]: 1244 : if (!parent_extkey.Derive(parent_extkey, entry)) return false;
444 : : }
445 : 1140 : final_extkey = parent_extkey;
446 [ + + + - ]: 1140 : if (m_derive == DeriveType::UNHARDENED) der = parent_extkey.Derive(final_extkey, pos);
447 [ - + ]: 1140 : assert(m_derive != DeriveType::HARDENED);
448 : : }
449 [ + - ]: 308853 : if (!der) return false;
450 : :
451 [ + - ]: 309463 : final_info_out = final_info_out_tmp;
452 : 309463 : key_out = final_extkey.pubkey;
453 : :
454 [ + + ]: 309463 : if (write_cache) {
455 : : // Only cache parent if there is any unhardened derivation
456 [ + + ]: 1896 : if (m_derive != DeriveType::HARDENED) {
457 [ + - ]: 860 : write_cache->CacheParentExtPubKey(m_expr_index, parent_extkey);
458 : : // Cache last hardened xpub if we have it
459 [ + + ]: 860 : if (last_hardened_extkey.pubkey.IsValid()) {
460 [ + - ]: 330 : write_cache->CacheLastHardenedExtPubKey(m_expr_index, last_hardened_extkey);
461 : : }
462 [ + - ]: 1036 : } else if (final_info_out.path.size() > 0) {
463 [ + - ]: 1036 : write_cache->CacheDerivedExtPubKey(m_expr_index, pos, final_extkey);
464 : : }
465 : : }
466 : :
467 : : return true;
468 : 310741 : }
469 : 13809 : std::string ToString(StringType type, bool normalized) const
470 : : {
471 : : // If StringType==COMPAT, always use the apostrophe to stay compatible with previous versions
472 [ + + + + : 13809 : const bool use_apostrophe = (!normalized && m_apostrophe) || type == StringType::COMPAT;
+ + ]
473 [ + - + - ]: 27618 : std::string ret = EncodeExtPubKey(m_root_extkey) + FormatHDKeypath(m_path, /*apostrophe=*/use_apostrophe);
474 [ + + ]: 13809 : if (IsRange()) {
475 [ + - ]: 13501 : ret += "/*";
476 [ + + + + ]: 13501 : if (m_derive == DeriveType::HARDENED) ret += use_apostrophe ? '\'' : 'h';
477 : : }
478 : 13809 : return ret;
479 : 0 : }
480 : 13797 : std::string ToString(StringType type=StringType::PUBLIC) const override
481 : : {
482 : 13669 : return ToString(type, /*normalized=*/false);
483 : : }
484 : 358 : bool ToPrivateString(const SigningProvider& arg, std::string& out) const override
485 : : {
486 [ + - ]: 358 : CExtKey key;
487 [ + - + + ]: 358 : if (!GetExtKey(arg, key)) return false;
488 [ + - + - : 234 : out = EncodeExtKey(key) + FormatHDKeypath(m_path, /*apostrophe=*/m_apostrophe);
+ - ]
489 [ + + ]: 234 : if (IsRange()) {
490 [ + - ]: 118 : out += "/*";
491 [ + + + + ]: 370 : if (m_derive == DeriveType::HARDENED) out += m_apostrophe ? '\'' : 'h';
492 : : }
493 : : return true;
494 : 358 : }
495 : 158 : bool ToNormalizedString(const SigningProvider& arg, std::string& out, const DescriptorCache* cache) const override
496 : : {
497 [ + + ]: 158 : if (m_derive == DeriveType::HARDENED) {
498 : 12 : out = ToString(StringType::PUBLIC, /*normalized=*/true);
499 : :
500 : 12 : return true;
501 : : }
502 : : // Step backwards to find the last hardened step in the path
503 : 146 : int i = (int)m_path.size() - 1;
504 [ + + ]: 346 : for (; i >= 0; --i) {
505 [ + + ]: 218 : if (m_path.at(i) >> 31) {
506 : : break;
507 : : }
508 : : }
509 : : // Either no derivation or all unhardened derivation
510 [ + + ]: 146 : if (i == -1) {
511 : 128 : out = ToString();
512 : 128 : return true;
513 : : }
514 : : // Get the path to the last hardened stup
515 : 18 : KeyOriginInfo origin;
516 : 18 : int k = 0;
517 [ + + ]: 44 : for (; k <= i; ++k) {
518 : : // Add to the path
519 [ + - + - ]: 26 : origin.path.push_back(m_path.at(k));
520 : : }
521 : : // Build the remaining path
522 : 18 : KeyPath end_path;
523 [ + + ]: 36 : for (; k < (int)m_path.size(); ++k) {
524 [ + - + - ]: 18 : end_path.push_back(m_path.at(k));
525 : : }
526 : : // Get the fingerprint
527 [ + - ]: 18 : CKeyID id = m_root_extkey.pubkey.GetID();
528 : 18 : std::copy(id.begin(), id.begin() + 4, origin.fingerprint);
529 : :
530 [ - + ]: 18 : CExtPubKey xpub;
531 [ - + ]: 18 : CExtKey lh_xprv;
532 : : // If we have the cache, just get the parent xpub
533 [ - + ]: 18 : if (cache != nullptr) {
534 [ # # ]: 0 : cache->GetCachedLastHardenedExtPubKey(m_expr_index, xpub);
535 : : }
536 [ + - ]: 18 : if (!xpub.pubkey.IsValid()) {
537 : : // Cache miss, or nor cache, or need privkey
538 [ + - ]: 18 : CExtKey xprv;
539 [ + - - + ]: 18 : if (!GetDerivedExtKey(arg, xprv, lh_xprv)) return false;
540 [ + - ]: 18 : xpub = lh_xprv.Neuter();
541 : 18 : }
542 [ - + ]: 18 : assert(xpub.pubkey.IsValid());
543 : :
544 : : // Build the string
545 [ + - + - : 36 : std::string origin_str = HexStr(origin.fingerprint) + FormatHDKeypath(origin.path);
+ - ]
546 [ + - + - : 36 : out = "[" + origin_str + "]" + EncodeExtPubKey(xpub) + FormatHDKeypath(end_path);
+ - + - +
- ]
547 [ + + ]: 18 : if (IsRange()) {
548 [ + - ]: 4 : out += "/*";
549 [ - + ]: 4 : assert(m_derive == DeriveType::UNHARDENED);
550 : : }
551 : 18 : return true;
552 : 36 : }
553 : 4 : bool GetPrivKey(int pos, const SigningProvider& arg, CKey& key) const override
554 : : {
555 [ + - ]: 4 : CExtKey extkey;
556 : 4 : CExtKey dummy;
557 [ + - + - ]: 4 : if (!GetDerivedExtKey(arg, extkey, dummy)) return false;
558 [ - + - - : 4 : if (m_derive == DeriveType::UNHARDENED && !extkey.Derive(extkey, pos)) return false;
- - ]
559 [ + - + - : 4 : if (m_derive == DeriveType::HARDENED && !extkey.Derive(extkey, pos | 0x80000000UL)) return false;
+ - ]
560 [ + - ]: 4 : key = extkey.key;
561 : : return true;
562 : 4 : }
563 : 0 : std::optional<CPubKey> GetRootPubKey() const override
564 : : {
565 : 0 : return std::nullopt;
566 : : }
567 : 0 : std::optional<CExtPubKey> GetRootExtPubKey() const override
568 : : {
569 : 0 : return m_root_extkey;
570 : : }
571 : 28 : std::unique_ptr<PubkeyProvider> Clone() const override
572 : : {
573 [ - + ]: 28 : return std::make_unique<BIP32PubkeyProvider>(m_expr_index, m_root_extkey, m_path, m_derive, m_apostrophe);
574 : : }
575 : : };
576 : :
577 : : /** Base class for all Descriptor implementations. */
578 : : class DescriptorImpl : public Descriptor
579 : : {
580 : : protected:
581 : : //! Public key arguments for this descriptor (size 1 for PK, PKH, WPKH; any size for WSH and Multisig).
582 : : const std::vector<std::unique_ptr<PubkeyProvider>> m_pubkey_args;
583 : : //! The string name of the descriptor function.
584 : : const std::string m_name;
585 : :
586 : : //! The sub-descriptor arguments (empty for everything but SH and WSH).
587 : : //! In doc/descriptors.m this is referred to as SCRIPT expressions sh(SCRIPT)
588 : : //! and wsh(SCRIPT), and distinct from KEY expressions and ADDR expressions.
589 : : //! Subdescriptors can only ever generate a single script.
590 : : const std::vector<std::unique_ptr<DescriptorImpl>> m_subdescriptor_args;
591 : :
592 : : //! Return a serialization of anything except pubkey and script arguments, to be prepended to those.
593 : 14997 : virtual std::string ToStringExtra() const { return ""; }
594 : :
595 : : /** A helper function to construct the scripts for this descriptor.
596 : : *
597 : : * This function is invoked once by ExpandHelper.
598 : : *
599 : : * @param pubkeys The evaluations of the m_pubkey_args field.
600 : : * @param scripts The evaluations of m_subdescriptor_args (one for each m_subdescriptor_args element).
601 : : * @param out A FlatSigningProvider to put scripts or public keys in that are necessary to the solver.
602 : : * The origin info of the provided pubkeys is automatically added.
603 : : * @return A vector with scriptPubKeys for this descriptor.
604 : : */
605 : : virtual std::vector<CScript> MakeScripts(const std::vector<CPubKey>& pubkeys, Span<const CScript> scripts, FlatSigningProvider& out) const = 0;
606 : :
607 : : public:
608 [ + - ]: 7850 : DescriptorImpl(std::vector<std::unique_ptr<PubkeyProvider>> pubkeys, const std::string& name) : m_pubkey_args(std::move(pubkeys)), m_name(name), m_subdescriptor_args() {}
609 [ + - + - ]: 840 : DescriptorImpl(std::vector<std::unique_ptr<PubkeyProvider>> pubkeys, std::unique_ptr<DescriptorImpl> script, const std::string& name) : m_pubkey_args(std::move(pubkeys)), m_name(name), m_subdescriptor_args(Vector(std::move(script))) {}
610 [ + - ]: 334 : DescriptorImpl(std::vector<std::unique_ptr<PubkeyProvider>> pubkeys, std::vector<std::unique_ptr<DescriptorImpl>> scripts, const std::string& name) : m_pubkey_args(std::move(pubkeys)), m_name(name), m_subdescriptor_args(std::move(scripts)) {}
611 : :
612 : : enum class StringType
613 : : {
614 : : PUBLIC,
615 : : PRIVATE,
616 : : NORMALIZED,
617 : : COMPAT, // string calculation that mustn't change over time to stay compatible with previous software versions
618 : : };
619 : :
620 : : // NOLINTNEXTLINE(misc-no-recursion)
621 : 2078 : bool IsSolvable() const override
622 : : {
623 [ + + ]: 3121 : for (const auto& arg : m_subdescriptor_args) {
624 [ + - ]: 1043 : if (!arg->IsSolvable()) return false;
625 : : }
626 : : return true;
627 : : }
628 : :
629 : : // NOLINTNEXTLINE(misc-no-recursion)
630 : 13573 : bool IsRange() const final
631 : : {
632 [ + + ]: 14367 : for (const auto& pubkey : m_pubkey_args) {
633 [ + + ]: 13539 : if (pubkey->IsRange()) return true;
634 : : }
635 [ + + ]: 950 : for (const auto& arg : m_subdescriptor_args) {
636 [ + + ]: 228 : if (arg->IsRange()) return true;
637 : : }
638 : : return false;
639 : : }
640 : :
641 : : // NOLINTNEXTLINE(misc-no-recursion)
642 : 14791 : virtual bool ToStringSubScriptHelper(const SigningProvider* arg, std::string& ret, const StringType type, const DescriptorCache* cache = nullptr) const
643 : : {
644 : 14791 : size_t pos = 0;
645 [ + + ]: 15390 : for (const auto& scriptarg : m_subdescriptor_args) {
646 [ - + ]: 699 : if (pos++) ret += ",";
647 [ + - ]: 699 : std::string tmp;
648 [ + - + + ]: 699 : if (!scriptarg->ToStringHelper(arg, tmp, type, cache)) return false;
649 [ + - ]: 1198 : ret += tmp;
650 : 699 : }
651 : : return true;
652 : : }
653 : :
654 : : // NOLINTNEXTLINE(misc-no-recursion)
655 : 15257 : virtual bool ToStringHelper(const SigningProvider* arg, std::string& out, const StringType type, const DescriptorCache* cache = nullptr) const
656 : : {
657 : 15257 : std::string extra = ToStringExtra();
658 [ + + ]: 15257 : size_t pos = extra.size() > 0 ? 1 : 0;
659 [ + - + - ]: 15257 : std::string ret = m_name + "(" + extra;
660 [ + + ]: 30218 : for (const auto& pubkey : m_pubkey_args) {
661 [ + + + - ]: 15105 : if (pos++) ret += ",";
662 [ + + + + : 15105 : std::string tmp;
- ]
663 [ + + + + : 15105 : switch (type) {
- ]
664 : 358 : case StringType::NORMALIZED:
665 [ + - + - ]: 358 : if (!pubkey->ToNormalizedString(*arg, tmp, cache)) return false;
666 : : break;
667 : 472 : case StringType::PRIVATE:
668 [ + - + + ]: 472 : if (!pubkey->ToPrivateString(*arg, tmp)) return false;
669 : : break;
670 : 13806 : case StringType::PUBLIC:
671 [ + - ]: 13806 : tmp = pubkey->ToString();
672 : 13806 : break;
673 : 469 : case StringType::COMPAT:
674 [ + - ]: 469 : tmp = pubkey->ToString(PubkeyProvider::StringType::COMPAT);
675 : 469 : break;
676 : : }
677 [ + - ]: 29922 : ret += tmp;
678 : 15105 : }
679 [ + - ]: 30226 : std::string subscript;
680 [ + - + + ]: 15113 : if (!ToStringSubScriptHelper(arg, subscript, type, cache)) return false;
681 [ + + + + : 15013 : if (pos && subscript.size()) ret += ',';
+ - ]
682 [ + - ]: 30026 : out = std::move(ret) + std::move(subscript) + ")";
683 : 15013 : return true;
684 : 15257 : }
685 : :
686 : 14054 : std::string ToString(bool compat_format) const final
687 : : {
688 [ + + ]: 14054 : std::string ret;
689 [ + + + - ]: 27711 : ToStringHelper(nullptr, ret, compat_format ? StringType::COMPAT : StringType::PUBLIC);
690 [ + - ]: 14054 : return AddChecksum(ret);
691 : 14054 : }
692 : :
693 : 352 : bool ToPrivateString(const SigningProvider& arg, std::string& out) const override
694 : : {
695 : 352 : bool ret = ToStringHelper(&arg, out, StringType::PRIVATE);
696 : 352 : out = AddChecksum(out);
697 : 352 : return ret;
698 : : }
699 : :
700 : 190 : bool ToNormalizedString(const SigningProvider& arg, std::string& out, const DescriptorCache* cache) const override final
701 : : {
702 : 190 : bool ret = ToStringHelper(&arg, out, StringType::NORMALIZED, cache);
703 : 190 : out = AddChecksum(out);
704 : 190 : return ret;
705 : : }
706 : :
707 : : // NOLINTNEXTLINE(misc-no-recursion)
708 : 385477 : bool ExpandHelper(int pos, const SigningProvider& arg, const DescriptorCache* read_cache, std::vector<CScript>& output_scripts, FlatSigningProvider& out, DescriptorCache* write_cache) const
709 : : {
710 : 385477 : std::vector<std::pair<CPubKey, KeyOriginInfo>> entries;
711 [ + - ]: 385477 : entries.reserve(m_pubkey_args.size());
712 : :
713 : : // Construct temporary data in `entries`, `subscripts`, and `subprovider` to avoid producing output in case of failure.
714 [ + + ]: 698830 : for (const auto& p : m_pubkey_args) {
715 [ + - ]: 314627 : entries.emplace_back();
716 [ + - + + ]: 314627 : if (!p->GetPubKey(pos, arg, entries.back().first, entries.back().second, read_cache, write_cache)) return false;
717 : : }
718 : 384203 : std::vector<CScript> subscripts;
719 : 384203 : FlatSigningProvider subprovider;
720 [ + + ]: 458492 : for (const auto& subarg : m_subdescriptor_args) {
721 : 74360 : std::vector<CScript> outscripts;
722 [ + - + + ]: 74360 : if (!subarg->ExpandHelper(pos, arg, read_cache, outscripts, subprovider, write_cache)) return false;
723 [ - + ]: 74289 : assert(outscripts.size() == 1);
724 [ + - ]: 74289 : subscripts.emplace_back(std::move(outscripts[0]));
725 : 74360 : }
726 [ + - ]: 384132 : out.Merge(std::move(subprovider));
727 : :
728 : 384132 : std::vector<CPubKey> pubkeys;
729 [ + - ]: 384132 : pubkeys.reserve(entries.size());
730 [ + + ]: 697485 : for (auto& entry : entries) {
731 [ + - ]: 313353 : pubkeys.push_back(entry.first);
732 [ + - + - ]: 626706 : out.origins.emplace(entry.first.GetID(), std::make_pair<CPubKey, KeyOriginInfo>(CPubKey(entry.first), std::move(entry.second)));
733 : : }
734 : :
735 [ + - ]: 768264 : output_scripts = MakeScripts(pubkeys, Span{subscripts}, out);
736 : 384132 : return true;
737 : 385477 : }
738 : :
739 : 2989 : bool Expand(int pos, const SigningProvider& provider, std::vector<CScript>& output_scripts, FlatSigningProvider& out, DescriptorCache* write_cache = nullptr) const final
740 : : {
741 : 2989 : return ExpandHelper(pos, provider, nullptr, output_scripts, out, write_cache);
742 : : }
743 : :
744 : 308128 : bool ExpandFromCache(int pos, const DescriptorCache& read_cache, std::vector<CScript>& output_scripts, FlatSigningProvider& out) const final
745 : : {
746 : 308128 : return ExpandHelper(pos, DUMMY_SIGNING_PROVIDER, &read_cache, output_scripts, out, nullptr);
747 : : }
748 : :
749 : : // NOLINTNEXTLINE(misc-no-recursion)
750 : 17 : void ExpandPrivate(int pos, const SigningProvider& provider, FlatSigningProvider& out) const final
751 : : {
752 [ + + ]: 34 : for (const auto& p : m_pubkey_args) {
753 : 17 : CKey key;
754 [ + - - + ]: 17 : if (!p->GetPrivKey(pos, provider, key)) continue;
755 [ + - + - : 17 : out.keys.emplace(key.GetPubKey().GetID(), key);
+ - ]
756 : 17 : }
757 [ - + ]: 17 : for (const auto& arg : m_subdescriptor_args) {
758 : 0 : arg->ExpandPrivate(pos, provider, out);
759 : : }
760 : 17 : }
761 : :
762 : 159 : std::optional<OutputType> GetOutputType() const override { return std::nullopt; }
763 : :
764 : 0 : std::optional<int64_t> ScriptSize() const override { return {}; }
765 : :
766 : : /** A helper for MaxSatisfactionWeight.
767 : : *
768 : : * @param use_max_sig Whether to assume ECDSA signatures will have a high-r.
769 : : * @return The maximum size of the satisfaction in raw bytes (with no witness meaning).
770 : : */
771 : 0 : virtual std::optional<int64_t> MaxSatSize(bool use_max_sig) const { return {}; }
772 : :
773 : 8 : std::optional<int64_t> MaxSatisfactionWeight(bool) const override { return {}; }
774 : :
775 : 4 : std::optional<int64_t> MaxSatisfactionElems() const override { return {}; }
776 : :
777 : : // NOLINTNEXTLINE(misc-no-recursion)
778 : 0 : void GetPubKeys(std::set<CPubKey>& pubkeys, std::set<CExtPubKey>& ext_pubs) const override
779 : : {
780 [ # # ]: 0 : for (const auto& p : m_pubkey_args) {
781 : 0 : std::optional<CPubKey> pub = p->GetRootPubKey();
782 [ # # ]: 0 : if (pub) pubkeys.insert(*pub);
783 : 0 : std::optional<CExtPubKey> ext_pub = p->GetRootExtPubKey();
784 [ # # ]: 0 : if (ext_pub) ext_pubs.insert(*ext_pub);
785 : : }
786 [ # # ]: 0 : for (const auto& arg : m_subdescriptor_args) {
787 : 0 : arg->GetPubKeys(pubkeys, ext_pubs);
788 : : }
789 : 0 : }
790 : :
791 : : virtual std::unique_ptr<DescriptorImpl> Clone() const = 0;
792 : : };
793 : :
794 : : /** A parsed addr(A) descriptor. */
795 : : class AddressDescriptor final : public DescriptorImpl
796 : : {
797 : : const CTxDestination m_destination;
798 : : protected:
799 : 25 : std::string ToStringExtra() const override { return EncodeDestination(m_destination); }
800 [ # # ]: 0 : std::vector<CScript> MakeScripts(const std::vector<CPubKey>&, Span<const CScript>, FlatSigningProvider&) const override { return Vector(GetScriptForDestination(m_destination)); }
801 : : public:
802 [ + - ]: 25 : AddressDescriptor(CTxDestination destination) : DescriptorImpl({}, "addr"), m_destination(std::move(destination)) {}
803 : 0 : bool IsSolvable() const final { return false; }
804 : :
805 : 0 : std::optional<OutputType> GetOutputType() const override
806 : : {
807 : 0 : return OutputTypeFromDestination(m_destination);
808 : : }
809 : 0 : bool IsSingleType() const final { return true; }
810 : 0 : bool ToPrivateString(const SigningProvider& arg, std::string& out) const final { return false; }
811 : :
812 [ # # ]: 0 : std::optional<int64_t> ScriptSize() const override { return GetScriptForDestination(m_destination).size(); }
813 : 0 : std::unique_ptr<DescriptorImpl> Clone() const override
814 : : {
815 [ # # ]: 0 : return std::make_unique<AddressDescriptor>(m_destination);
816 : : }
817 : : };
818 : :
819 : : /** A parsed raw(H) descriptor. */
820 : : class RawDescriptor final : public DescriptorImpl
821 : : {
822 : : const CScript m_script;
823 : : protected:
824 [ + + ]: 16 : std::string ToStringExtra() const override { return HexStr(m_script); }
825 : 0 : std::vector<CScript> MakeScripts(const std::vector<CPubKey>&, Span<const CScript>, FlatSigningProvider&) const override { return Vector(m_script); }
826 : : public:
827 [ + - ]: 8 : RawDescriptor(CScript script) : DescriptorImpl({}, "raw"), m_script(std::move(script)) {}
828 : 0 : bool IsSolvable() const final { return false; }
829 : :
830 : 0 : std::optional<OutputType> GetOutputType() const override
831 : : {
832 : 0 : CTxDestination dest;
833 [ # # ]: 0 : ExtractDestination(m_script, dest);
834 [ # # ]: 0 : return OutputTypeFromDestination(dest);
835 : 0 : }
836 : 0 : bool IsSingleType() const final { return true; }
837 : 0 : bool ToPrivateString(const SigningProvider& arg, std::string& out) const final { return false; }
838 : :
839 [ # # ]: 0 : std::optional<int64_t> ScriptSize() const override { return m_script.size(); }
840 : :
841 : 0 : std::unique_ptr<DescriptorImpl> Clone() const override
842 : : {
843 [ # # ]: 0 : return std::make_unique<RawDescriptor>(m_script);
844 : : }
845 : : };
846 : :
847 : : /** A parsed pk(P) descriptor. */
848 : : class PKDescriptor final : public DescriptorImpl
849 : : {
850 : : private:
851 : : const bool m_xonly;
852 : : protected:
853 : 808 : std::vector<CScript> MakeScripts(const std::vector<CPubKey>& keys, Span<const CScript>, FlatSigningProvider&) const override
854 : : {
855 [ + + ]: 808 : if (m_xonly) {
856 [ + - + - ]: 540 : CScript script = CScript() << ToByteVector(XOnlyPubKey(keys[0])) << OP_CHECKSIG;
857 [ + - ]: 540 : return Vector(std::move(script));
858 : 540 : } else {
859 [ + - ]: 536 : return Vector(GetScriptForRawPubKey(keys[0]));
860 : : }
861 : : }
862 : : public:
863 [ + - + - ]: 563 : PKDescriptor(std::unique_ptr<PubkeyProvider> prov, bool xonly = false) : DescriptorImpl(Vector(std::move(prov)), "pk"), m_xonly(xonly) {}
864 : 5 : bool IsSingleType() const final { return true; }
865 : :
866 : 11 : std::optional<int64_t> ScriptSize() const override {
867 [ + - ]: 11 : return 1 + (m_xonly ? 32 : m_pubkey_args[0]->GetSize()) + 1;
868 : : }
869 : :
870 : 64 : std::optional<int64_t> MaxSatSize(bool use_max_sig) const override {
871 [ - + ]: 6 : const auto ecdsa_sig_size = use_max_sig ? 72 : 71;
872 [ + - + - ]: 64 : return 1 + (m_xonly ? 65 : ecdsa_sig_size);
873 : : }
874 : :
875 : 58 : std::optional<int64_t> MaxSatisfactionWeight(bool use_max_sig) const override {
876 [ + + ]: 58 : return *MaxSatSize(use_max_sig) * WITNESS_SCALE_FACTOR;
877 : : }
878 : :
879 : 56 : std::optional<int64_t> MaxSatisfactionElems() const override { return 1; }
880 : :
881 : 0 : std::unique_ptr<DescriptorImpl> Clone() const override
882 : : {
883 [ # # # # ]: 0 : return std::make_unique<PKDescriptor>(m_pubkey_args.at(0)->Clone(), m_xonly);
884 : : }
885 : : };
886 : :
887 : : /** A parsed pkh(P) descriptor. */
888 : : class PKHDescriptor final : public DescriptorImpl
889 : : {
890 : : protected:
891 : 72502 : std::vector<CScript> MakeScripts(const std::vector<CPubKey>& keys, Span<const CScript>, FlatSigningProvider& out) const override
892 : : {
893 : 72502 : CKeyID id = keys[0].GetID();
894 : 72502 : out.pubkeys.emplace(id, keys[0]);
895 [ + - + - ]: 145004 : return Vector(GetScriptForDestination(PKHash(id)));
896 : : }
897 : : public:
898 [ + - + - ]: 364 : PKHDescriptor(std::unique_ptr<PubkeyProvider> prov) : DescriptorImpl(Vector(std::move(prov)), "pkh") {}
899 : 68 : std::optional<OutputType> GetOutputType() const override { return OutputType::LEGACY; }
900 : 32 : bool IsSingleType() const final { return true; }
901 : :
902 : 18 : std::optional<int64_t> ScriptSize() const override { return 1 + 1 + 1 + 20 + 1 + 1; }
903 : :
904 : 47 : std::optional<int64_t> MaxSatSize(bool use_max_sig) const override {
905 [ + + ]: 47 : const auto sig_size = use_max_sig ? 72 : 71;
906 : 47 : return 1 + sig_size + 1 + m_pubkey_args[0]->GetSize();
907 : : }
908 : :
909 : 41 : std::optional<int64_t> MaxSatisfactionWeight(bool use_max_sig) const override {
910 : 41 : return *MaxSatSize(use_max_sig) * WITNESS_SCALE_FACTOR;
911 : : }
912 : :
913 : 32 : std::optional<int64_t> MaxSatisfactionElems() const override { return 2; }
914 : :
915 : 0 : std::unique_ptr<DescriptorImpl> Clone() const override
916 : : {
917 [ # # # # ]: 0 : return std::make_unique<PKHDescriptor>(m_pubkey_args.at(0)->Clone());
918 : : }
919 : : };
920 : :
921 : : /** A parsed wpkh(P) descriptor. */
922 : : class WPKHDescriptor final : public DescriptorImpl
923 : : {
924 : : protected:
925 : 163253 : std::vector<CScript> MakeScripts(const std::vector<CPubKey>& keys, Span<const CScript>, FlatSigningProvider& out) const override
926 : : {
927 : 163253 : CKeyID id = keys[0].GetID();
928 : 163253 : out.pubkeys.emplace(id, keys[0]);
929 [ + - + - ]: 326506 : return Vector(GetScriptForDestination(WitnessV0KeyHash(id)));
930 : : }
931 : : public:
932 [ + - + - ]: 6129 : WPKHDescriptor(std::unique_ptr<PubkeyProvider> prov) : DescriptorImpl(Vector(std::move(prov)), "wpkh") {}
933 : 11988 : std::optional<OutputType> GetOutputType() const override { return OutputType::BECH32; }
934 : 12343 : bool IsSingleType() const final { return true; }
935 : :
936 : 22 : std::optional<int64_t> ScriptSize() const override { return 1 + 1 + 20; }
937 : :
938 : 5790 : std::optional<int64_t> MaxSatSize(bool use_max_sig) const override {
939 [ + + ]: 17 : const auto sig_size = use_max_sig ? 72 : 71;
940 : 5790 : return (1 + sig_size + 1 + 33);
941 : : }
942 : :
943 : 5773 : std::optional<int64_t> MaxSatisfactionWeight(bool use_max_sig) const override {
944 [ + + ]: 5773 : return MaxSatSize(use_max_sig);
945 : : }
946 : :
947 : 5780 : std::optional<int64_t> MaxSatisfactionElems() const override { return 2; }
948 : :
949 : 0 : std::unique_ptr<DescriptorImpl> Clone() const override
950 : : {
951 [ # # # # ]: 0 : return std::make_unique<WPKHDescriptor>(m_pubkey_args.at(0)->Clone());
952 : : }
953 : : };
954 : :
955 : : /** A parsed combo(P) descriptor. */
956 : : class ComboDescriptor final : public DescriptorImpl
957 : : {
958 : : protected:
959 : 107 : std::vector<CScript> MakeScripts(const std::vector<CPubKey>& keys, Span<const CScript>, FlatSigningProvider& out) const override
960 : : {
961 : 107 : std::vector<CScript> ret;
962 [ + - ]: 107 : CKeyID id = keys[0].GetID();
963 [ + - ]: 107 : out.pubkeys.emplace(id, keys[0]);
964 [ + - + - ]: 107 : ret.emplace_back(GetScriptForRawPubKey(keys[0])); // P2PK
965 [ + - + - : 214 : ret.emplace_back(GetScriptForDestination(PKHash(id))); // P2PKH
+ - ]
966 [ + + ]: 107 : if (keys[0].IsCompressed()) {
967 [ + - ]: 83 : CScript p2wpkh = GetScriptForDestination(WitnessV0KeyHash(id));
968 [ + - + - ]: 83 : out.scripts.emplace(CScriptID(p2wpkh), p2wpkh);
969 [ + - ]: 83 : ret.emplace_back(p2wpkh);
970 [ + - + - : 166 : ret.emplace_back(GetScriptForDestination(ScriptHash(p2wpkh))); // P2SH-P2WPKH
+ - ]
971 : 83 : }
972 : 107 : return ret;
973 : 0 : }
974 : : public:
975 [ + - + - ]: 22 : ComboDescriptor(std::unique_ptr<PubkeyProvider> prov) : DescriptorImpl(Vector(std::move(prov)), "combo") {}
976 : 4 : bool IsSingleType() const final { return false; }
977 : 0 : std::unique_ptr<DescriptorImpl> Clone() const override
978 : : {
979 [ # # # # ]: 0 : return std::make_unique<ComboDescriptor>(m_pubkey_args.at(0)->Clone());
980 : : }
981 : : };
982 : :
983 : : /** A parsed multi(...) or sortedmulti(...) descriptor */
984 : : class MultisigDescriptor final : public DescriptorImpl
985 : : {
986 : : const int m_threshold;
987 : : const bool m_sorted;
988 : : protected:
989 : 213 : std::string ToStringExtra() const override { return strprintf("%i", m_threshold); }
990 : 644 : std::vector<CScript> MakeScripts(const std::vector<CPubKey>& keys, Span<const CScript>, FlatSigningProvider&) const override {
991 [ + + ]: 644 : if (m_sorted) {
992 : 72 : std::vector<CPubKey> sorted_keys(keys);
993 : 72 : std::sort(sorted_keys.begin(), sorted_keys.end());
994 [ + - + - ]: 144 : return Vector(GetScriptForMultisig(m_threshold, sorted_keys));
995 : 72 : }
996 [ + - ]: 1144 : return Vector(GetScriptForMultisig(m_threshold, keys));
997 : : }
998 : : public:
999 [ + + + - ]: 656 : MultisigDescriptor(int threshold, std::vector<std::unique_ptr<PubkeyProvider>> providers, bool sorted = false) : DescriptorImpl(std::move(providers), sorted ? "sortedmulti" : "multi"), m_threshold(threshold), m_sorted(sorted) {}
1000 : 7 : bool IsSingleType() const final { return true; }
1001 : :
1002 : 37 : std::optional<int64_t> ScriptSize() const override {
1003 : 37 : const auto n_keys = m_pubkey_args.size();
1004 : 184 : auto op = [](int64_t acc, const std::unique_ptr<PubkeyProvider>& pk) { return acc + 1 + pk->GetSize();};
1005 : 37 : const auto pubkeys_size{std::accumulate(m_pubkey_args.begin(), m_pubkey_args.end(), int64_t{0}, op)};
1006 [ - + + - ]: 74 : return 1 + BuildScript(n_keys).size() + BuildScript(m_threshold).size() + pubkeys_size;
1007 : : }
1008 : :
1009 : 44 : std::optional<int64_t> MaxSatSize(bool use_max_sig) const override {
1010 [ - + ]: 30 : const auto sig_size = use_max_sig ? 72 : 71;
1011 : 44 : return (1 + (1 + sig_size) * m_threshold);
1012 : : }
1013 : :
1014 : 14 : std::optional<int64_t> MaxSatisfactionWeight(bool use_max_sig) const override {
1015 [ - + ]: 14 : return *MaxSatSize(use_max_sig) * WITNESS_SCALE_FACTOR;
1016 : : }
1017 : :
1018 : 22 : std::optional<int64_t> MaxSatisfactionElems() const override { return 1 + m_threshold; }
1019 : :
1020 : 0 : std::unique_ptr<DescriptorImpl> Clone() const override
1021 : : {
1022 : 0 : std::vector<std::unique_ptr<PubkeyProvider>> providers;
1023 [ # # ]: 0 : providers.reserve(m_pubkey_args.size());
1024 [ # # ]: 0 : std::transform(m_pubkey_args.begin(), m_pubkey_args.end(), providers.begin(), [](const std::unique_ptr<PubkeyProvider>& p) { return p->Clone(); });
1025 [ # # # # ]: 0 : return std::make_unique<MultisigDescriptor>(m_threshold, std::move(providers), m_sorted);
1026 : 0 : }
1027 : : };
1028 : :
1029 : : /** A parsed (sorted)multi_a(...) descriptor. Always uses x-only pubkeys. */
1030 : : class MultiADescriptor final : public DescriptorImpl
1031 : : {
1032 : : const int m_threshold;
1033 : : const bool m_sorted;
1034 : : protected:
1035 : 14 : std::string ToStringExtra() const override { return strprintf("%i", m_threshold); }
1036 : 90 : std::vector<CScript> MakeScripts(const std::vector<CPubKey>& keys, Span<const CScript>, FlatSigningProvider&) const override {
1037 : 90 : CScript ret;
1038 : 90 : std::vector<XOnlyPubKey> xkeys;
1039 [ + - ]: 90 : xkeys.reserve(keys.size());
1040 [ + - + + ]: 240 : for (const auto& key : keys) xkeys.emplace_back(key);
1041 [ - + ]: 90 : if (m_sorted) std::sort(xkeys.begin(), xkeys.end());
1042 [ + - + - ]: 90 : ret << ToByteVector(xkeys[0]) << OP_CHECKSIG;
1043 [ + + ]: 150 : for (size_t i = 1; i < keys.size(); ++i) {
1044 [ + - + - ]: 120 : ret << ToByteVector(xkeys[i]) << OP_CHECKSIGADD;
1045 : : }
1046 [ + - + - ]: 90 : ret << m_threshold << OP_NUMEQUAL;
1047 [ + - ]: 90 : return Vector(std::move(ret));
1048 : 90 : }
1049 : : public:
1050 [ + - + - ]: 84 : MultiADescriptor(int threshold, std::vector<std::unique_ptr<PubkeyProvider>> providers, bool sorted = false) : DescriptorImpl(std::move(providers), sorted ? "sortedmulti_a" : "multi_a"), m_threshold(threshold), m_sorted(sorted) {}
1051 : 0 : bool IsSingleType() const final { return true; }
1052 : :
1053 : 0 : std::optional<int64_t> ScriptSize() const override {
1054 : 0 : const auto n_keys = m_pubkey_args.size();
1055 [ # # ]: 0 : return (1 + 32 + 1) * n_keys + BuildScript(m_threshold).size() + 1;
1056 : : }
1057 : :
1058 : 0 : std::optional<int64_t> MaxSatSize(bool use_max_sig) const override {
1059 : 0 : return (1 + 65) * m_threshold + (m_pubkey_args.size() - m_threshold);
1060 : : }
1061 : :
1062 : 0 : std::optional<int64_t> MaxSatisfactionElems() const override { return m_pubkey_args.size(); }
1063 : :
1064 : 0 : std::unique_ptr<DescriptorImpl> Clone() const override
1065 : : {
1066 : 0 : std::vector<std::unique_ptr<PubkeyProvider>> providers;
1067 [ # # ]: 0 : providers.reserve(m_pubkey_args.size());
1068 [ # # ]: 0 : for (const auto& arg : m_pubkey_args) {
1069 [ # # ]: 0 : providers.push_back(arg->Clone());
1070 : : }
1071 [ # # # # ]: 0 : return std::make_unique<MultiADescriptor>(m_threshold, std::move(providers), m_sorted);
1072 : 0 : }
1073 : : };
1074 : :
1075 : : /** A parsed sh(...) descriptor. */
1076 : : class SHDescriptor final : public DescriptorImpl
1077 : : {
1078 : : protected:
1079 : 72699 : std::vector<CScript> MakeScripts(const std::vector<CPubKey>&, Span<const CScript> scripts, FlatSigningProvider& out) const override
1080 : : {
1081 [ + - + - ]: 145398 : auto ret = Vector(GetScriptForDestination(ScriptHash(scripts[0])));
1082 [ + - + - : 72699 : if (ret.size()) out.scripts.emplace(CScriptID(scripts[0]), scripts[0]);
+ - ]
1083 : 72699 : return ret;
1084 : 0 : }
1085 : :
1086 [ + + ]: 136 : bool IsSegwit() const { return m_subdescriptor_args[0]->GetOutputType() == OutputType::BECH32; }
1087 : :
1088 : : public:
1089 [ + - ]: 469 : SHDescriptor(std::unique_ptr<DescriptorImpl> desc) : DescriptorImpl({}, std::move(desc), "sh") {}
1090 : :
1091 : 81 : std::optional<OutputType> GetOutputType() const override
1092 : : {
1093 [ - + ]: 81 : assert(m_subdescriptor_args.size() == 1);
1094 [ + + ]: 81 : if (IsSegwit()) return OutputType::P2SH_SEGWIT;
1095 : 33 : return OutputType::LEGACY;
1096 : : }
1097 : 28 : bool IsSingleType() const final { return true; }
1098 : :
1099 : 24 : std::optional<int64_t> ScriptSize() const override { return 1 + 1 + 20 + 1; }
1100 : :
1101 : 55 : std::optional<int64_t> MaxSatisfactionWeight(bool use_max_sig) const override {
1102 [ + - ]: 55 : if (const auto sat_size = m_subdescriptor_args[0]->MaxSatSize(use_max_sig)) {
1103 [ + - ]: 55 : if (const auto subscript_size = m_subdescriptor_args[0]->ScriptSize()) {
1104 : : // The subscript is never witness data.
1105 : 55 : const auto subscript_weight = (1 + *subscript_size) * WITNESS_SCALE_FACTOR;
1106 : : // The weight depends on whether the inner descriptor is satisfied using the witness stack.
1107 [ + + ]: 55 : if (IsSegwit()) return subscript_weight + *sat_size;
1108 : 22 : return subscript_weight + *sat_size * WITNESS_SCALE_FACTOR;
1109 : : }
1110 : : }
1111 : 0 : return {};
1112 : : }
1113 : :
1114 : 31 : std::optional<int64_t> MaxSatisfactionElems() const override {
1115 [ + - ]: 31 : if (const auto sub_elems = m_subdescriptor_args[0]->MaxSatisfactionElems()) return 1 + *sub_elems;
1116 : 0 : return {};
1117 : : }
1118 : :
1119 : 0 : std::unique_ptr<DescriptorImpl> Clone() const override
1120 : : {
1121 [ # # # # ]: 0 : return std::make_unique<SHDescriptor>(m_subdescriptor_args.at(0)->Clone());
1122 : : }
1123 : : };
1124 : :
1125 : : /** A parsed wsh(...) descriptor. */
1126 : : class WSHDescriptor final : public DescriptorImpl
1127 : : {
1128 : : protected:
1129 : 760 : std::vector<CScript> MakeScripts(const std::vector<CPubKey>&, Span<const CScript> scripts, FlatSigningProvider& out) const override
1130 : : {
1131 [ + - + - ]: 1520 : auto ret = Vector(GetScriptForDestination(WitnessV0ScriptHash(scripts[0])));
1132 [ + - + - : 760 : if (ret.size()) out.scripts.emplace(CScriptID(scripts[0]), scripts[0]);
+ - ]
1133 : 760 : return ret;
1134 : 0 : }
1135 : : public:
1136 [ + - ]: 371 : WSHDescriptor(std::unique_ptr<DescriptorImpl> desc) : DescriptorImpl({}, std::move(desc), "wsh") {}
1137 : 94 : std::optional<OutputType> GetOutputType() const override { return OutputType::BECH32; }
1138 : 18 : bool IsSingleType() const final { return true; }
1139 : :
1140 : 34 : std::optional<int64_t> ScriptSize() const override { return 1 + 1 + 32; }
1141 : :
1142 : 52 : std::optional<int64_t> MaxSatSize(bool use_max_sig) const override {
1143 [ + - ]: 52 : if (const auto sat_size = m_subdescriptor_args[0]->MaxSatSize(use_max_sig)) {
1144 [ + - ]: 52 : if (const auto subscript_size = m_subdescriptor_args[0]->ScriptSize()) {
1145 [ + + ]: 58 : return GetSizeOfCompactSize(*subscript_size) + *subscript_size + *sat_size;
1146 : : }
1147 : : }
1148 : 0 : return {};
1149 : : }
1150 : :
1151 : 36 : std::optional<int64_t> MaxSatisfactionWeight(bool use_max_sig) const override {
1152 : 36 : return MaxSatSize(use_max_sig);
1153 : : }
1154 : :
1155 : 26 : std::optional<int64_t> MaxSatisfactionElems() const override {
1156 [ + - ]: 26 : if (const auto sub_elems = m_subdescriptor_args[0]->MaxSatisfactionElems()) return 1 + *sub_elems;
1157 : 0 : return {};
1158 : : }
1159 : :
1160 : 0 : std::unique_ptr<DescriptorImpl> Clone() const override
1161 : : {
1162 [ # # # # ]: 0 : return std::make_unique<WSHDescriptor>(m_subdescriptor_args.at(0)->Clone());
1163 : : }
1164 : : };
1165 : :
1166 : : /** A parsed tr(...) descriptor. */
1167 : : class TRDescriptor final : public DescriptorImpl
1168 : : {
1169 : : std::vector<int> m_depths;
1170 : : protected:
1171 : 72499 : std::vector<CScript> MakeScripts(const std::vector<CPubKey>& keys, Span<const CScript> scripts, FlatSigningProvider& out) const override
1172 : : {
1173 [ - + ]: 72499 : TaprootBuilder builder;
1174 [ - + ]: 72499 : assert(m_depths.size() == scripts.size());
1175 [ + + ]: 73329 : for (size_t pos = 0; pos < m_depths.size(); ++pos) {
1176 [ + + + - ]: 1660 : builder.Add(m_depths[pos], scripts[pos], TAPROOT_LEAF_TAPSCRIPT);
1177 : : }
1178 [ - + ]: 72499 : if (!builder.IsComplete()) return {};
1179 [ - + ]: 72499 : assert(keys.size() == 1);
1180 : 72499 : XOnlyPubKey xpk(keys[0]);
1181 [ + - - + ]: 72499 : if (!xpk.IsFullyValid()) return {};
1182 [ + - ]: 72499 : builder.Finalize(xpk);
1183 [ + - ]: 72499 : WitnessV1Taproot output = builder.GetOutput();
1184 [ + - + - ]: 72499 : out.tr_trees[output] = builder;
1185 [ + - + - ]: 72499 : out.pubkeys.emplace(keys[0].GetID(), keys[0]);
1186 [ + - + - ]: 144998 : return Vector(GetScriptForDestination(output));
1187 : 72499 : }
1188 : 322 : bool ToStringSubScriptHelper(const SigningProvider* arg, std::string& ret, const StringType type, const DescriptorCache* cache = nullptr) const override
1189 : : {
1190 [ + + ]: 322 : if (m_depths.empty()) return true;
1191 : 84 : std::vector<bool> path;
1192 [ + + ]: 232 : for (size_t pos = 0; pos < m_depths.size(); ++pos) {
1193 [ + + + - ]: 148 : if (pos) ret += ',';
1194 [ + + ]: 296 : while ((int)path.size() <= m_depths[pos]) {
1195 [ + + + - ]: 148 : if (path.size()) ret += '{';
1196 [ + - ]: 148 : path.push_back(false);
1197 : : }
1198 [ + - ]: 148 : std::string tmp;
1199 [ + - - + ]: 148 : if (!m_subdescriptor_args[pos]->ToStringHelper(arg, tmp, type, cache)) return false;
1200 [ + - ]: 148 : ret += tmp;
1201 [ + - + + ]: 212 : while (!path.empty() && path.back()) {
1202 [ + - + - ]: 64 : if (path.size() > 1) ret += '}';
1203 : 64 : path.pop_back();
1204 : : }
1205 [ + - ]: 148 : if (!path.empty()) path.back() = true;
1206 : 148 : }
1207 : : return true;
1208 : 84 : }
1209 : : public:
1210 : 334 : TRDescriptor(std::unique_ptr<PubkeyProvider> internal_key, std::vector<std::unique_ptr<DescriptorImpl>> descs, std::vector<int> depths) :
1211 [ + - + - : 334 : DescriptorImpl(Vector(std::move(internal_key)), std::move(descs), "tr"), m_depths(std::move(depths))
- + ]
1212 : : {
1213 [ - + ]: 334 : assert(m_subdescriptor_args.size() == m_depths.size());
1214 : 334 : }
1215 : 62 : std::optional<OutputType> GetOutputType() const override { return OutputType::BECH32M; }
1216 : 25 : bool IsSingleType() const final { return true; }
1217 : :
1218 : 17 : std::optional<int64_t> ScriptSize() const override { return 1 + 1 + 32; }
1219 : :
1220 : 41 : std::optional<int64_t> MaxSatisfactionWeight(bool) const override {
1221 : : // FIXME: We assume keypath spend, which can lead to very large underestimations.
1222 : 41 : return 1 + 65;
1223 : : }
1224 : :
1225 : 24 : std::optional<int64_t> MaxSatisfactionElems() const override {
1226 : : // FIXME: See above, we assume keypath spend.
1227 : 24 : return 1;
1228 : : }
1229 : :
1230 : 0 : std::unique_ptr<DescriptorImpl> Clone() const override
1231 : : {
1232 : 0 : std::vector<std::unique_ptr<DescriptorImpl>> subdescs;
1233 [ # # ]: 0 : subdescs.reserve(m_subdescriptor_args.size());
1234 [ # # ]: 0 : std::transform(m_subdescriptor_args.begin(), m_subdescriptor_args.end(), subdescs.begin(), [](const std::unique_ptr<DescriptorImpl>& d) { return d->Clone(); });
1235 [ # # # # : 0 : return std::make_unique<TRDescriptor>(m_pubkey_args.at(0)->Clone(), std::move(subdescs), m_depths);
# # # # ]
1236 : 0 : }
1237 : : };
1238 : :
1239 : : /* We instantiate Miniscript here with a simple integer as key type.
1240 : : * The value of these key integers are an index in the
1241 : : * DescriptorImpl::m_pubkey_args vector.
1242 : : */
1243 : :
1244 : : /**
1245 : : * The context for converting a Miniscript descriptor into a Script.
1246 : : */
1247 : : class ScriptMaker {
1248 : : //! Keys contained in the Miniscript (the evaluation of DescriptorImpl::m_pubkey_args).
1249 : : const std::vector<CPubKey>& m_keys;
1250 : : //! The script context we're operating within (Tapscript or P2WSH).
1251 : : const miniscript::MiniscriptContext m_script_ctx;
1252 : :
1253 : : //! Get the ripemd160(sha256()) hash of this key.
1254 : : //! Any key that is valid in a descriptor serializes as 32 bytes within a Tapscript context. So we
1255 : : //! must not hash the sign-bit byte in this case.
1256 : 180 : uint160 GetHash160(uint32_t key) const {
1257 [ + + ]: 180 : if (miniscript::IsTapscript(m_script_ctx)) {
1258 : 60 : return Hash160(XOnlyPubKey{m_keys[key]});
1259 : : }
1260 : 120 : return m_keys[key].GetID();
1261 : : }
1262 : :
1263 : : public:
1264 : 680 : ScriptMaker(const std::vector<CPubKey>& keys LIFETIMEBOUND, const miniscript::MiniscriptContext script_ctx) : m_keys(keys), m_script_ctx{script_ctx} {}
1265 : :
1266 : 850 : std::vector<unsigned char> ToPKBytes(uint32_t key) const {
1267 : : // In Tapscript keys always serialize as x-only, whether an x-only key was used in the descriptor or not.
1268 [ + + ]: 850 : if (!miniscript::IsTapscript(m_script_ctx)) {
1269 : 630 : return {m_keys[key].begin(), m_keys[key].end()};
1270 : : }
1271 : 220 : const XOnlyPubKey xonly_pubkey{m_keys[key]};
1272 : 220 : return {xonly_pubkey.begin(), xonly_pubkey.end()};
1273 : : }
1274 : :
1275 : 180 : std::vector<unsigned char> ToPKHBytes(uint32_t key) const {
1276 : 180 : auto id = GetHash160(key);
1277 : 180 : return {id.begin(), id.end()};
1278 : : }
1279 : : };
1280 : :
1281 : : /**
1282 : : * The context for converting a Miniscript descriptor to its textual form.
1283 : : */
1284 : : class StringMaker {
1285 : : //! To convert private keys for private descriptors.
1286 : : const SigningProvider* m_arg;
1287 : : //! Keys contained in the Miniscript (a reference to DescriptorImpl::m_pubkey_args).
1288 : : const std::vector<std::unique_ptr<PubkeyProvider>>& m_pubkeys;
1289 : : //! Whether to serialize keys as private or public.
1290 : : bool m_private;
1291 : :
1292 : : public:
1293 : 186 : StringMaker(const SigningProvider* arg LIFETIMEBOUND, const std::vector<std::unique_ptr<PubkeyProvider>>& pubkeys LIFETIMEBOUND, bool priv)
1294 : 186 : : m_arg(arg), m_pubkeys(pubkeys), m_private(priv) {}
1295 : :
1296 : 424 : std::optional<std::string> ToString(uint32_t key) const
1297 : : {
1298 [ + + ]: 424 : std::string ret;
1299 [ + + ]: 424 : if (m_private) {
1300 [ + - + + ]: 112 : if (!m_pubkeys[key]->ToPrivateString(*m_arg, ret)) return {};
1301 : : } else {
1302 [ + - ]: 312 : ret = m_pubkeys[key]->ToString();
1303 : : }
1304 : 392 : return ret;
1305 : 424 : }
1306 : : };
1307 : :
1308 : : class MiniscriptDescriptor final : public DescriptorImpl
1309 : : {
1310 : : private:
1311 : : miniscript::NodeRef<uint32_t> m_node;
1312 : :
1313 : : protected:
1314 : 680 : std::vector<CScript> MakeScripts(const std::vector<CPubKey>& keys, Span<const CScript> scripts,
1315 : : FlatSigningProvider& provider) const override
1316 : : {
1317 : 680 : const auto script_ctx{m_node->GetMsCtx()};
1318 [ + + ]: 1710 : for (const auto& key : keys) {
1319 [ + + ]: 1030 : if (miniscript::IsTapscript(script_ctx)) {
1320 : 280 : provider.pubkeys.emplace(Hash160(XOnlyPubKey{key}), key);
1321 : : } else {
1322 : 750 : provider.pubkeys.emplace(key.GetID(), key);
1323 : : }
1324 : : }
1325 [ + - ]: 1360 : return Vector(m_node->ToScript(ScriptMaker(keys, script_ctx)));
1326 : : }
1327 : :
1328 : : public:
1329 : 324 : MiniscriptDescriptor(std::vector<std::unique_ptr<PubkeyProvider>> providers, miniscript::NodeRef<uint32_t> node)
1330 [ + - ]: 324 : : DescriptorImpl(std::move(providers), "?"), m_node(std::move(node)) {}
1331 : :
1332 : 186 : bool ToStringHelper(const SigningProvider* arg, std::string& out, const StringType type,
1333 : : const DescriptorCache* cache = nullptr) const override
1334 : : {
1335 [ + + ]: 186 : if (const auto res = m_node->ToString(StringMaker(arg, m_pubkey_args, type == StringType::PRIVATE))) {
1336 [ + - ]: 154 : out = *res;
1337 : 154 : return true;
1338 : 154 : }
1339 : 32 : return false;
1340 : : }
1341 : :
1342 : 296 : bool IsSolvable() const override { return true; }
1343 : 0 : bool IsSingleType() const final { return true; }
1344 : :
1345 : 32 : std::optional<int64_t> ScriptSize() const override { return m_node->ScriptSize(); }
1346 : :
1347 : 32 : std::optional<int64_t> MaxSatSize(bool) const override {
1348 : : // For Miniscript we always assume high-R ECDSA signatures.
1349 [ + - ]: 32 : return m_node->GetWitnessSize();
1350 : : }
1351 : :
1352 : 16 : std::optional<int64_t> MaxSatisfactionElems() const override {
1353 [ + - ]: 16 : return m_node->GetStackSize();
1354 : : }
1355 : :
1356 : 0 : std::unique_ptr<DescriptorImpl> Clone() const override
1357 : : {
1358 : 0 : std::vector<std::unique_ptr<PubkeyProvider>> providers;
1359 [ # # ]: 0 : providers.reserve(m_pubkey_args.size());
1360 [ # # ]: 0 : for (const auto& arg : m_pubkey_args) {
1361 [ # # ]: 0 : providers.push_back(arg->Clone());
1362 : : }
1363 [ # # # # : 0 : return std::make_unique<MiniscriptDescriptor>(std::move(providers), miniscript::MakeNodeRef<uint32_t>(*m_node));
# # # # ]
1364 : 0 : }
1365 : : };
1366 : :
1367 : : /** A parsed rawtr(...) descriptor. */
1368 : : class RawTRDescriptor final : public DescriptorImpl
1369 : : {
1370 : : protected:
1371 : 90 : std::vector<CScript> MakeScripts(const std::vector<CPubKey>& keys, Span<const CScript> scripts, FlatSigningProvider& out) const override
1372 : : {
1373 [ - + ]: 90 : assert(keys.size() == 1);
1374 : 90 : XOnlyPubKey xpk(keys[0]);
1375 [ - + ]: 90 : if (!xpk.IsFullyValid()) return {};
1376 [ + - ]: 90 : WitnessV1Taproot output{xpk};
1377 [ + - + - ]: 180 : return Vector(GetScriptForDestination(output));
1378 : : }
1379 : : public:
1380 [ + - + - ]: 42 : RawTRDescriptor(std::unique_ptr<PubkeyProvider> output_key) : DescriptorImpl(Vector(std::move(output_key)), "rawtr") {}
1381 : 9 : std::optional<OutputType> GetOutputType() const override { return OutputType::BECH32M; }
1382 : 3 : bool IsSingleType() const final { return true; }
1383 : :
1384 : 3 : std::optional<int64_t> ScriptSize() const override { return 1 + 1 + 32; }
1385 : :
1386 : 6 : std::optional<int64_t> MaxSatisfactionWeight(bool) const override {
1387 : : // We can't know whether there is a script path, so assume key path spend.
1388 : 6 : return 1 + 65;
1389 : : }
1390 : :
1391 : 3 : std::optional<int64_t> MaxSatisfactionElems() const override {
1392 : : // See above, we assume keypath spend.
1393 : 3 : return 1;
1394 : : }
1395 : :
1396 : 0 : std::unique_ptr<DescriptorImpl> Clone() const override
1397 : : {
1398 [ # # # # ]: 0 : return std::make_unique<RawTRDescriptor>(m_pubkey_args.at(0)->Clone());
1399 : : }
1400 : : };
1401 : :
1402 : : ////////////////////////////////////////////////////////////////////////////
1403 : : // Parser //
1404 : : ////////////////////////////////////////////////////////////////////////////
1405 : :
1406 : : enum class ParseScriptContext {
1407 : : TOP, //!< Top-level context (script goes directly in scriptPubKey)
1408 : : P2SH, //!< Inside sh() (script becomes P2SH redeemScript)
1409 : : P2WPKH, //!< Inside wpkh() (no script, pubkey only)
1410 : : P2WSH, //!< Inside wsh() (script becomes v0 witness script)
1411 : : P2TR, //!< Inside tr() (either internal key, or BIP342 script leaf)
1412 : : };
1413 : :
1414 : 1810 : std::optional<uint32_t> ParseKeyPathNum(Span<const char> elem, bool& apostrophe, std::string& error)
1415 : : {
1416 : 1810 : bool hardened = false;
1417 [ + + ]: 1810 : if (elem.size() > 0) {
1418 [ + + ]: 1804 : const char last = elem[elem.size() - 1];
1419 [ + + ]: 1804 : if (last == '\'' || last == 'h') {
1420 : 975 : elem = elem.first(elem.size() - 1);
1421 : 975 : hardened = true;
1422 : 975 : apostrophe = last == '\'';
1423 : : }
1424 : : }
1425 : 1810 : uint32_t p;
1426 [ + - + + ]: 1810 : if (!ParseUInt32(std::string(elem.begin(), elem.end()), &p)) {
1427 [ + - ]: 12 : error = strprintf("Key path value '%s' is not a valid uint32", std::string(elem.begin(), elem.end()));
1428 : 12 : return std::nullopt;
1429 [ + + ]: 1798 : } else if (p > 0x7FFFFFFFUL) {
1430 : 2 : error = strprintf("Key path value %u is out of range", p);
1431 : 2 : return std::nullopt;
1432 : : }
1433 : :
1434 : 1796 : return std::make_optional<uint32_t>(p | (((uint32_t)hardened) << 31));
1435 : : }
1436 : :
1437 : : /**
1438 : : * Parse a key path, being passed a split list of elements (the first element is ignored because it is always the key).
1439 : : *
1440 : : * @param[in] split BIP32 path string, using either ' or h for hardened derivation
1441 : : * @param[out] out Vector of parsed key paths
1442 : : * @param[out] apostrophe only updated if hardened derivation is found
1443 : : * @param[out] error parsing error message
1444 : : * @param[in] allow_multipath Allows the parsed path to use the multipath specifier
1445 : : * @returns false if parsing failed
1446 : : **/
1447 : 597 : [[nodiscard]] bool ParseKeyPath(const std::vector<Span<const char>>& split, std::vector<KeyPath>& out, bool& apostrophe, std::string& error, bool allow_multipath)
1448 : : {
1449 : 597 : KeyPath path;
1450 : 597 : std::optional<size_t> multipath_segment_index;
1451 : 597 : std::vector<uint32_t> multipath_values;
1452 : 597 : std::unordered_set<uint32_t> seen_multipath;
1453 : :
1454 [ + + ]: 2207 : for (size_t i = 1; i < split.size(); ++i) {
1455 [ + - ]: 1634 : const Span<const char>& elem = split[i];
1456 : :
1457 : : // Check if element contain multipath specifier
1458 [ + - + + : 1634 : if (!elem.empty() && elem.front() == '<' && elem.back() == '>') {
+ + ]
1459 [ + + ]: 122 : if (!allow_multipath) {
1460 [ + - + - ]: 4 : error = strprintf("Key path value '%s' specifies multipath in a section where multipath is not allowed", std::string(elem.begin(), elem.end()));
1461 : 2 : return false;
1462 : : }
1463 [ + + ]: 120 : if (multipath_segment_index) {
1464 [ + - ]: 597 : error = "Multiple multipath key path specifiers found";
1465 : : return false;
1466 : : }
1467 : :
1468 : : // Parse each possible value
1469 [ + - ]: 118 : std::vector<Span<const char>> nums = Split(Span(elem.begin()+1, elem.end()-1), ";");
1470 [ + + ]: 118 : if (nums.size() < 2) {
1471 [ + - ]: 12 : error = "Multipath key path specifiers must have at least two items";
1472 : : return false;
1473 : : }
1474 : :
1475 [ + + ]: 404 : for (const auto& num : nums) {
1476 [ + - ]: 298 : const auto& op_num = ParseKeyPathNum(num, apostrophe, error);
1477 [ + + ]: 298 : if (!op_num) return false;
1478 [ + - + + ]: 292 : auto [_, inserted] = seen_multipath.insert(*op_num);
1479 [ + + ]: 292 : if (!inserted) {
1480 [ + - ]: 2 : error = strprintf("Duplicated key path value %u in multipath specifier", *op_num);
1481 : 2 : return false;
1482 : : }
1483 [ + - ]: 290 : multipath_values.emplace_back(*op_num);
1484 : : }
1485 : :
1486 [ + - ]: 106 : path.emplace_back(); // Placeholder for multipath segment
1487 : 106 : multipath_segment_index = path.size()-1;
1488 : 118 : } else {
1489 [ + - ]: 1512 : const auto& op_num = ParseKeyPathNum(elem, apostrophe, error);
1490 [ + + ]: 1512 : if (!op_num) return false;
1491 [ + - ]: 1504 : path.emplace_back(*op_num);
1492 : : }
1493 : : }
1494 : :
1495 [ + + ]: 573 : if (!multipath_segment_index) {
1496 [ + - ]: 469 : out.emplace_back(std::move(path));
1497 : : } else {
1498 : : // Replace the multipath placeholder with each value while generating paths
1499 [ + + ]: 382 : for (size_t i = 0; i < multipath_values.size(); i++) {
1500 [ + - ]: 278 : KeyPath branch_path = path;
1501 [ + - ]: 278 : branch_path[*multipath_segment_index] = multipath_values[i];
1502 [ + - ]: 278 : out.emplace_back(std::move(branch_path));
1503 : 278 : }
1504 : : }
1505 : : return true;
1506 : 597 : }
1507 : :
1508 : : /** Parse a public key that excludes origin information. */
1509 : 944 : std::vector<std::unique_ptr<PubkeyProvider>> ParsePubkeyInner(uint32_t key_exp_index, const Span<const char>& sp, ParseScriptContext ctx, FlatSigningProvider& out, bool& apostrophe, std::string& error)
1510 : : {
1511 : 944 : std::vector<std::unique_ptr<PubkeyProvider>> ret;
1512 : 944 : bool permit_uncompressed = ctx == ParseScriptContext::TOP || ctx == ParseScriptContext::P2SH;
1513 [ + - ]: 944 : auto split = Split(sp, '/');
1514 [ + - + + ]: 1888 : std::string str(split[0].begin(), split[0].end());
1515 [ + + ]: 944 : if (str.size() == 0) {
1516 [ + - ]: 2 : error = "No key provided";
1517 : 2 : return {};
1518 : : }
1519 [ + + ]: 942 : if (split.size() == 1) {
1520 [ + - + + ]: 430 : if (IsHex(str)) {
1521 [ + - ]: 200 : std::vector<unsigned char> data = ParseHex(str);
1522 : 200 : CPubKey pubkey(data);
1523 [ + + + + ]: 200 : if (pubkey.IsValid() && !pubkey.IsValidNonHybrid()) {
1524 [ + - ]: 4 : error = "Hybrid public keys are not allowed";
1525 : 4 : return {};
1526 : : }
1527 [ + - + + ]: 196 : if (pubkey.IsFullyValid()) {
1528 [ + + + + ]: 154 : if (permit_uncompressed || pubkey.IsCompressed()) {
1529 [ + - + - ]: 150 : ret.emplace_back(std::make_unique<ConstPubkeyProvider>(key_exp_index, pubkey, false));
1530 : 150 : return ret;
1531 : : } else {
1532 [ + - ]: 4 : error = "Uncompressed keys are not allowed";
1533 : 4 : return {};
1534 : : }
1535 [ + - + + ]: 42 : } else if (data.size() == 32 && ctx == ParseScriptContext::P2TR) {
1536 : 41 : unsigned char fullkey[33] = {0x02};
1537 : 41 : std::copy(data.begin(), data.end(), fullkey + 1);
1538 : 41 : pubkey.Set(std::begin(fullkey), std::end(fullkey));
1539 [ + - + - ]: 41 : if (pubkey.IsFullyValid()) {
1540 [ + - + - ]: 41 : ret.emplace_back(std::make_unique<ConstPubkeyProvider>(key_exp_index, pubkey, true));
1541 : 41 : return ret;
1542 : : }
1543 : : }
1544 [ + - ]: 1 : error = strprintf("Pubkey '%s' is invalid", str);
1545 : 1 : return {};
1546 : 200 : }
1547 [ + - ]: 230 : CKey key = DecodeSecret(str);
1548 [ + + ]: 230 : if (key.IsValid()) {
1549 [ + + + + ]: 185 : if (permit_uncompressed || key.IsCompressed()) {
1550 [ + - ]: 180 : CPubKey pubkey = key.GetPubKey();
1551 [ + - + - ]: 180 : out.keys.emplace(pubkey.GetID(), key);
1552 [ + - + - ]: 180 : ret.emplace_back(std::make_unique<ConstPubkeyProvider>(key_exp_index, pubkey, ctx == ParseScriptContext::P2TR));
1553 : 180 : return ret;
1554 : : } else {
1555 [ + - ]: 5 : error = "Uncompressed keys are not allowed";
1556 : 5 : return {};
1557 : : }
1558 : : }
1559 : 230 : }
1560 [ + - ]: 557 : CExtKey extkey = DecodeExtKey(str);
1561 [ + - ]: 557 : CExtPubKey extpubkey = DecodeExtPubKey(str);
1562 [ + + + + ]: 557 : if (!extkey.key.IsValid() && !extpubkey.pubkey.IsValid()) {
1563 [ + - ]: 3 : error = strprintf("key '%s' is not valid", str);
1564 : 3 : return {};
1565 : : }
1566 : 554 : std::vector<KeyPath> paths;
1567 : 554 : DeriveType type = DeriveType::NO;
1568 [ + + ]: 554 : if (std::ranges::equal(split.back(), Span{"*"}.first(1))) {
1569 : 435 : split.pop_back();
1570 : 435 : type = DeriveType::UNHARDENED;
1571 [ + + + + ]: 119 : } else if (std::ranges::equal(split.back(), Span{"*'"}.first(2)) || std::ranges::equal(split.back(), Span{"*h"}.first(2))) {
1572 : 13 : apostrophe = std::ranges::equal(split.back(), Span{"*'"}.first(2));
1573 : 13 : split.pop_back();
1574 : 13 : type = DeriveType::HARDENED;
1575 : : }
1576 [ + - + + ]: 554 : if (!ParseKeyPath(split, paths, apostrophe, error, /*allow_multipath=*/true)) return {};
1577 [ + + ]: 532 : if (extkey.key.IsValid()) {
1578 [ + - ]: 123 : extpubkey = extkey.Neuter();
1579 [ + - + - ]: 123 : out.keys.emplace(extpubkey.pubkey.GetID(), extkey.key);
1580 : : }
1581 [ + + ]: 1238 : for (auto& path : paths) {
1582 [ + - + - ]: 1412 : ret.emplace_back(std::make_unique<BIP32PubkeyProvider>(key_exp_index, extpubkey, std::move(path), type, apostrophe));
1583 : : }
1584 : 532 : return ret;
1585 : 2055 : }
1586 : :
1587 : : /** Parse a public key including origin information (if enabled). */
1588 : 960 : std::vector<std::unique_ptr<PubkeyProvider>> ParsePubkey(uint32_t key_exp_index, const Span<const char>& sp, ParseScriptContext ctx, FlatSigningProvider& out, std::string& error)
1589 : : {
1590 : 960 : std::vector<std::unique_ptr<PubkeyProvider>> ret;
1591 [ + - ]: 960 : auto origin_split = Split(sp, ']');
1592 [ + + ]: 960 : if (origin_split.size() > 2) {
1593 [ + - ]: 4 : error = "Multiple ']' characters found for a single pubkey";
1594 : 4 : return {};
1595 : : }
1596 : : // This is set if either the origin or path suffix contains a hardened derivation.
1597 : 956 : bool apostrophe = false;
1598 [ + + ]: 956 : if (origin_split.size() == 1) {
1599 [ + - ]: 903 : return ParsePubkeyInner(key_exp_index, origin_split[0], ctx, out, apostrophe, error);
1600 : : }
1601 [ + - + + ]: 53 : if (origin_split[0].empty() || origin_split[0][0] != '[') {
1602 : 6 : error = strprintf("Key origin start '[ character expected but not found, got '%c' instead",
1603 [ + - + - ]: 2 : origin_split[0].empty() ? /** empty, implies split char */ ']' : origin_split[0][0]);
1604 : 2 : return {};
1605 : : }
1606 [ + - ]: 51 : auto slash_split = Split(origin_split[0].subspan(1), '/');
1607 [ + + ]: 51 : if (slash_split[0].size() != 8) {
1608 [ + - ]: 6 : error = strprintf("Fingerprint is not 4 bytes (%u characters instead of 8 characters)", slash_split[0].size());
1609 : 6 : return {};
1610 : : }
1611 [ + - + - ]: 90 : std::string fpr_hex = std::string(slash_split[0].begin(), slash_split[0].end());
1612 [ + - + + ]: 45 : if (!IsHex(fpr_hex)) {
1613 [ + - ]: 2 : error = strprintf("Fingerprint '%s' is not hex", fpr_hex);
1614 : 2 : return {};
1615 : : }
1616 [ + - ]: 43 : auto fpr_bytes = ParseHex(fpr_hex);
1617 [ - + ]: 43 : KeyOriginInfo info;
1618 : 43 : static_assert(sizeof(info.fingerprint) == 4, "Fingerprint must be 4 bytes");
1619 [ - + ]: 43 : assert(fpr_bytes.size() == 4);
1620 : 43 : std::copy(fpr_bytes.begin(), fpr_bytes.end(), info.fingerprint);
1621 : 43 : std::vector<KeyPath> path;
1622 [ + - + + ]: 43 : if (!ParseKeyPath(slash_split, path, apostrophe, error, /*allow_multipath=*/false)) return {};
1623 [ + - + - ]: 41 : info.path = path.at(0);
1624 [ + - ]: 41 : auto providers = ParsePubkeyInner(key_exp_index, origin_split[1], ctx, out, apostrophe, error);
1625 [ + + ]: 41 : if (providers.empty()) return {};
1626 [ + - ]: 39 : ret.reserve(providers.size());
1627 [ + + ]: 90 : for (auto& prov : providers) {
1628 [ + - + - ]: 102 : ret.emplace_back(std::make_unique<OriginPubkeyProvider>(key_exp_index, info, std::move(prov), apostrophe));
1629 : : }
1630 : 39 : return ret;
1631 : 1099 : }
1632 : :
1633 : 8010 : std::unique_ptr<PubkeyProvider> InferPubkey(const CPubKey& pubkey, ParseScriptContext ctx, const SigningProvider& provider)
1634 : : {
1635 : : // Key cannot be hybrid
1636 [ + + ]: 8010 : if (!pubkey.IsValidNonHybrid()) {
1637 : 2 : return nullptr;
1638 : : }
1639 : : // Uncompressed is only allowed in TOP and P2SH contexts
1640 [ + + + + ]: 8008 : if (ctx != ParseScriptContext::TOP && ctx != ParseScriptContext::P2SH && !pubkey.IsCompressed()) {
1641 : 1 : return nullptr;
1642 : : }
1643 : 8007 : std::unique_ptr<PubkeyProvider> key_provider = std::make_unique<ConstPubkeyProvider>(0, pubkey, false);
1644 [ + - ]: 8007 : KeyOriginInfo info;
1645 [ + - + - : 8007 : if (provider.GetKeyOrigin(pubkey.GetID(), info)) {
+ + ]
1646 [ + - - + ]: 7986 : return std::make_unique<OriginPubkeyProvider>(0, std::move(info), std::move(key_provider), /*apostrophe=*/false);
1647 : : }
1648 : 21 : return key_provider;
1649 : 8007 : }
1650 : :
1651 : 515 : std::unique_ptr<PubkeyProvider> InferXOnlyPubkey(const XOnlyPubKey& xkey, ParseScriptContext ctx, const SigningProvider& provider)
1652 : : {
1653 : 515 : CPubKey pubkey{xkey.GetEvenCorrespondingCPubKey()};
1654 : 515 : std::unique_ptr<PubkeyProvider> key_provider = std::make_unique<ConstPubkeyProvider>(0, pubkey, true);
1655 [ + - ]: 515 : KeyOriginInfo info;
1656 [ + - + - ]: 515 : if (provider.GetKeyOriginByXOnly(xkey, info)) {
1657 [ + - - + ]: 515 : return std::make_unique<OriginPubkeyProvider>(0, std::move(info), std::move(key_provider), /*apostrophe=*/false);
1658 : : }
1659 : 0 : return key_provider;
1660 : 515 : }
1661 : :
1662 : : /**
1663 : : * The context for parsing a Miniscript descriptor (either from Script or from its textual representation).
1664 : : */
1665 : 355 : struct KeyParser {
1666 : : //! The Key type is an index in DescriptorImpl::m_pubkey_args
1667 : : using Key = uint32_t;
1668 : : //! Must not be nullptr if parsing from string.
1669 : : FlatSigningProvider* m_out;
1670 : : //! Must not be nullptr if parsing from Script.
1671 : : const SigningProvider* m_in;
1672 : : //! List of multipath expanded keys contained in the Miniscript.
1673 : : mutable std::vector<std::vector<std::unique_ptr<PubkeyProvider>>> m_keys;
1674 : : //! Used to detect key parsing errors within a Miniscript.
1675 : : mutable std::string m_key_parsing_error;
1676 : : //! The script context we're operating within (Tapscript or P2WSH).
1677 : : const miniscript::MiniscriptContext m_script_ctx;
1678 : : //! The number of keys that were parsed before starting to parse this Miniscript descriptor.
1679 : : uint32_t m_offset;
1680 : :
1681 : 355 : KeyParser(FlatSigningProvider* out LIFETIMEBOUND, const SigningProvider* in LIFETIMEBOUND,
1682 : : miniscript::MiniscriptContext ctx, uint32_t offset = 0)
1683 : 355 : : m_out(out), m_in(in), m_script_ctx(ctx), m_offset(offset) {}
1684 : :
1685 : 465 : bool KeyCompare(const Key& a, const Key& b) const {
1686 : 465 : return *m_keys.at(a).at(0) < *m_keys.at(b).at(0);
1687 : : }
1688 : :
1689 : 529 : ParseScriptContext ParseContext() const {
1690 [ + - + ]: 529 : switch (m_script_ctx) {
1691 : : case miniscript::MiniscriptContext::P2WSH: return ParseScriptContext::P2WSH;
1692 : 145 : case miniscript::MiniscriptContext::TAPSCRIPT: return ParseScriptContext::P2TR;
1693 : : }
1694 : 0 : assert(false);
1695 : : }
1696 : :
1697 : 117 : template<typename I> std::optional<Key> FromString(I begin, I end) const
1698 : : {
1699 [ - + ]: 117 : assert(m_out);
1700 : 117 : Key key = m_keys.size();
1701 : 117 : auto pk = ParsePubkey(m_offset + key, {&*begin, &*end}, ParseContext(), *m_out, m_key_parsing_error);
1702 [ + + ]: 117 : if (pk.empty()) return {};
1703 [ + - ]: 115 : m_keys.emplace_back(std::move(pk));
1704 : 115 : return key;
1705 : 117 : }
1706 : :
1707 : 30 : std::optional<std::string> ToString(const Key& key) const
1708 : : {
1709 : 30 : return m_keys.at(key).at(0)->ToString();
1710 : : }
1711 : :
1712 : 340 : template<typename I> std::optional<Key> FromPKBytes(I begin, I end) const
1713 : : {
1714 [ - + ]: 340 : assert(m_in);
1715 : 340 : Key key = m_keys.size();
1716 [ + + + - ]: 340 : if (miniscript::IsTapscript(m_script_ctx) && end - begin == 32) {
1717 : 88 : XOnlyPubKey pubkey;
1718 : 88 : std::copy(begin, end, pubkey.begin());
1719 [ + - ]: 88 : if (auto pubkey_provider = InferPubkey(pubkey.GetEvenCorrespondingCPubKey(), ParseContext(), *m_in)) {
1720 [ + - ]: 88 : m_keys.emplace_back();
1721 [ + - ]: 88 : m_keys.back().push_back(std::move(pubkey_provider));
1722 : 88 : return key;
1723 : : }
1724 [ + - ]: 252 : } else if (!miniscript::IsTapscript(m_script_ctx)) {
1725 : 252 : CPubKey pubkey(begin, end);
1726 [ + - ]: 252 : if (auto pubkey_provider = InferPubkey(pubkey, ParseContext(), *m_in)) {
1727 [ + - ]: 252 : m_keys.emplace_back();
1728 [ + - ]: 252 : m_keys.back().push_back(std::move(pubkey_provider));
1729 : 252 : return key;
1730 : : }
1731 : : }
1732 : 0 : return {};
1733 : : }
1734 : :
1735 [ - + ]: 72 : template<typename I> std::optional<Key> FromPKHBytes(I begin, I end) const
1736 : : {
1737 [ - + ]: 72 : assert(end - begin == 20);
1738 [ - + ]: 72 : assert(m_in);
1739 : 72 : uint160 hash;
1740 : 72 : std::copy(begin, end, hash.begin());
1741 : 72 : CKeyID keyid(hash);
1742 : 72 : CPubKey pubkey;
1743 [ + - ]: 72 : if (m_in->GetPubKey(keyid, pubkey)) {
1744 [ + - ]: 72 : if (auto pubkey_provider = InferPubkey(pubkey, ParseContext(), *m_in)) {
1745 [ + - ]: 72 : Key key = m_keys.size();
1746 [ + - ]: 72 : m_keys.emplace_back();
1747 [ + - ]: 72 : m_keys.back().push_back(std::move(pubkey_provider));
1748 : 72 : return key;
1749 : : }
1750 : : }
1751 : 0 : return {};
1752 : : }
1753 : :
1754 : 2757 : miniscript::MiniscriptContext MsContext() const {
1755 [ - - + - : 2757 : return m_script_ctx;
+ - + - +
- + - + -
+ - + - +
- + - - -
+ - + - +
- + - + -
- - + - +
- - - - -
- - + - +
- + - - -
- - + - -
- + - + -
+ - + - +
- - - + -
+ - + - +
- + - + -
+ - + - +
- + - - -
+ - + - -
- - - - -
- - + - +
- + - + -
+ - - - +
- + - + -
+ - + + -
- + - ]
1756 : : }
1757 : : };
1758 : :
1759 : : /** Parse a script in a particular context. */
1760 : : // NOLINTNEXTLINE(misc-no-recursion)
1761 : 946 : std::vector<std::unique_ptr<DescriptorImpl>> ParseScript(uint32_t& key_exp_index, Span<const char>& sp, ParseScriptContext ctx, FlatSigningProvider& out, std::string& error)
1762 : : {
1763 : 946 : using namespace script;
1764 : :
1765 : 946 : std::vector<std::unique_ptr<DescriptorImpl>> ret;
1766 [ + - ]: 946 : auto expr = Expr(sp);
1767 [ + - + - : 946 : if (Func("pk", expr)) {
+ + ]
1768 [ + - ]: 69 : auto pubkeys = ParsePubkey(key_exp_index, expr, ctx, out, error);
1769 [ + + ]: 69 : if (pubkeys.empty()) {
1770 [ + - ]: 3 : error = strprintf("pk(): %s", error);
1771 : 3 : return {};
1772 : : }
1773 : 66 : ++key_exp_index;
1774 [ + + ]: 206 : for (auto& pubkey : pubkeys) {
1775 [ + - + - ]: 280 : ret.emplace_back(std::make_unique<PKDescriptor>(std::move(pubkey), ctx == ParseScriptContext::P2TR));
1776 : : }
1777 : 66 : return ret;
1778 : 69 : }
1779 [ + + + - : 1731 : if ((ctx == ParseScriptContext::TOP || ctx == ParseScriptContext::P2SH || ctx == ParseScriptContext::P2WSH) && Func("pkh", expr)) {
+ - + + +
+ ]
1780 [ + - ]: 118 : auto pubkeys = ParsePubkey(key_exp_index, expr, ctx, out, error);
1781 [ + + ]: 118 : if (pubkeys.empty()) {
1782 [ + - ]: 13 : error = strprintf("pkh(): %s", error);
1783 : 13 : return {};
1784 : : }
1785 : 105 : ++key_exp_index;
1786 [ + + ]: 226 : for (auto& pubkey : pubkeys) {
1787 [ + - + - ]: 242 : ret.emplace_back(std::make_unique<PKHDescriptor>(std::move(pubkey)));
1788 : : }
1789 : 105 : return ret;
1790 [ + + + - : 1613 : } else if (ctx != ParseScriptContext::P2TR && Func("pkh", expr)) {
+ - - + -
+ ]
1791 : : // Under Taproot, always the Miniscript parser deal with it.
1792 [ # # ]: 0 : error = "Can only have pkh at top level, in sh(), wsh(), or in tr()";
1793 : 0 : return {};
1794 : : }
1795 [ + + + - : 1270 : if (ctx == ParseScriptContext::TOP && Func("combo", expr)) {
+ - + + +
+ ]
1796 [ + - ]: 25 : auto pubkeys = ParsePubkey(key_exp_index, expr, ctx, out, error);
1797 [ + + ]: 25 : if (pubkeys.empty()) {
1798 [ + - ]: 3 : error = strprintf("combo(): %s", error);
1799 : 3 : return {};
1800 : : }
1801 : 22 : ++key_exp_index;
1802 [ + + ]: 44 : for (auto& pubkey : pubkeys) {
1803 [ + - + - ]: 44 : ret.emplace_back(std::make_unique<ComboDescriptor>(std::move(pubkey)));
1804 : : }
1805 : 22 : return ret;
1806 [ + - + - : 759 : } else if (Func("combo", expr)) {
+ + ]
1807 [ + - ]: 2 : error = "Can only have combo() at top level";
1808 : 2 : return {};
1809 : : }
1810 [ + - + - ]: 732 : const bool multi = Func("multi", expr);
1811 [ + + + - : 1395 : const bool sortedmulti = !multi && Func("sortedmulti", expr);
+ - + + ]
1812 [ + + + - : 1389 : const bool multi_a = !(multi || sortedmulti) && Func("multi_a", expr);
+ - + + ]
1813 [ + + + + : 1383 : const bool sortedmulti_a = !(multi || sortedmulti || multi_a) && Func("sortedmulti_a", expr);
+ - + - -
+ ]
1814 [ + + + + : 732 : if (((ctx == ParseScriptContext::TOP || ctx == ParseScriptContext::P2SH || ctx == ParseScriptContext::P2WSH) && (multi || sortedmulti)) ||
+ + + + ]
1815 [ + + - + ]: 23 : (ctx == ParseScriptContext::P2TR && (multi_a || sortedmulti_a))) {
1816 [ + - ]: 81 : auto threshold = Expr(expr);
1817 : 81 : uint32_t thres;
1818 : 81 : std::vector<std::vector<std::unique_ptr<PubkeyProvider>>> providers; // List of multipath expanded pubkeys
1819 [ + - + - : 162 : if (!ParseUInt32(std::string(threshold.begin(), threshold.end()), &thres)) {
+ + ]
1820 [ + - + - ]: 4 : error = strprintf("Multi threshold '%s' is not valid", std::string(threshold.begin(), threshold.end()));
1821 : 2 : return {};
1822 : : }
1823 : 79 : size_t script_size = 0;
1824 : 79 : size_t max_providers_len = 0;
1825 [ + + ]: 387 : while (expr.size()) {
1826 [ + - + - : 319 : if (!Const(",", expr)) {
- + ]
1827 [ # # ]: 0 : error = strprintf("Multi: expected ',', got '%c'", expr[0]);
1828 : 0 : return {};
1829 : : }
1830 [ + - ]: 319 : auto arg = Expr(expr);
1831 [ + - ]: 319 : auto pks = ParsePubkey(key_exp_index, arg, ctx, out, error);
1832 [ + + ]: 319 : if (pks.empty()) {
1833 [ + - ]: 11 : error = strprintf("Multi: %s", error);
1834 : 11 : return {};
1835 : : }
1836 [ + - + - ]: 308 : script_size += pks.at(0)->GetSize() + 1;
1837 [ + + ]: 308 : max_providers_len = std::max(max_providers_len, pks.size());
1838 [ + - ]: 308 : providers.emplace_back(std::move(pks));
1839 : 308 : key_exp_index++;
1840 : 319 : }
1841 [ + + + + : 68 : if ((multi || sortedmulti) && (providers.empty() || providers.size() > MAX_PUBKEYS_PER_MULTISIG)) {
+ - - + ]
1842 [ # # ]: 0 : error = strprintf("Cannot have %u keys in multisig; must have between 1 and %d keys, inclusive", providers.size(), MAX_PUBKEYS_PER_MULTISIG);
1843 : 0 : return {};
1844 [ + + - + : 68 : } else if ((multi_a || sortedmulti_a) && (providers.empty() || providers.size() > MAX_PUBKEYS_PER_MULTI_A)) {
+ - - + ]
1845 [ # # ]: 0 : error = strprintf("Cannot have %u keys in multi_a; must have between 1 and %d keys, inclusive", providers.size(), MAX_PUBKEYS_PER_MULTI_A);
1846 : 0 : return {};
1847 [ + + ]: 68 : } else if (thres < 1) {
1848 [ + - ]: 2 : error = strprintf("Multisig threshold cannot be %d, must be at least 1", thres);
1849 : 2 : return {};
1850 [ + + ]: 66 : } else if (thres > providers.size()) {
1851 [ + - ]: 2 : error = strprintf("Multisig threshold cannot be larger than the number of keys; threshold is %d but only %u keys specified", thres, providers.size());
1852 : 2 : return {};
1853 : : }
1854 [ + + ]: 64 : if (ctx == ParseScriptContext::TOP) {
1855 [ + + ]: 17 : if (providers.size() > 3) {
1856 [ + - ]: 2 : error = strprintf("Cannot have %u pubkeys in bare multisig; only at most 3 pubkeys", providers.size());
1857 : 2 : return {};
1858 : : }
1859 : : }
1860 [ + + ]: 62 : if (ctx == ParseScriptContext::P2SH) {
1861 : : // This limits the maximum number of compressed pubkeys to 15.
1862 [ + + ]: 26 : if (script_size + 3 > MAX_SCRIPT_ELEMENT_SIZE) {
1863 [ + - ]: 4 : error = strprintf("P2SH script is too large, %d bytes is larger than %d bytes", script_size + 3, MAX_SCRIPT_ELEMENT_SIZE);
1864 : 4 : return {};
1865 : : }
1866 : : }
1867 : :
1868 : : // Make sure all vecs are of the same length, or exactly length 1
1869 : : // For length 1 vectors, clone key providers until vector is the same length
1870 [ + + ]: 282 : for (auto& vec : providers) {
1871 [ + + ]: 226 : if (vec.size() == 1) {
1872 [ + + ]: 218 : for (size_t i = 1; i < max_providers_len; ++i) {
1873 [ + - + - : 16 : vec.emplace_back(vec.at(0)->Clone());
+ - ]
1874 : : }
1875 [ + + ]: 24 : } else if (vec.size() != max_providers_len) {
1876 [ + - ]: 2 : error = strprintf("multi(): Multipath derivation paths have mismatched lengths");
1877 : 2 : return {};
1878 : : }
1879 : : }
1880 : :
1881 : : // Build the final descriptors vector
1882 [ + + ]: 128 : for (size_t i = 0; i < max_providers_len; ++i) {
1883 : : // Build final pubkeys vectors by retrieving the i'th subscript for each vector in subscripts
1884 : 72 : std::vector<std::unique_ptr<PubkeyProvider>> pubs;
1885 [ + - ]: 72 : pubs.reserve(providers.size());
1886 [ + + ]: 336 : for (auto& pub : providers) {
1887 [ + - + - ]: 264 : pubs.emplace_back(std::move(pub.at(i)));
1888 : : }
1889 [ + + + + ]: 72 : if (multi || sortedmulti) {
1890 [ + - + - ]: 132 : ret.emplace_back(std::make_unique<MultisigDescriptor>(thres, std::move(pubs), sortedmulti));
1891 : : } else {
1892 [ + - + - ]: 12 : ret.emplace_back(std::make_unique<MultiADescriptor>(thres, std::move(pubs), sortedmulti_a));
1893 : : }
1894 : 72 : }
1895 : 56 : return ret;
1896 [ + - - + ]: 732 : } else if (multi || sortedmulti) {
1897 [ # # ]: 0 : error = "Can only have multi/sortedmulti at top level, in sh(), or in wsh()";
1898 : 0 : return {};
1899 [ + - - + ]: 651 : } else if (multi_a || sortedmulti_a) {
1900 [ # # ]: 0 : error = "Can only have multi_a/sortedmulti_a inside tr()";
1901 : 0 : return {};
1902 : : }
1903 [ + + + - : 1226 : if ((ctx == ParseScriptContext::TOP || ctx == ParseScriptContext::P2SH) && Func("wpkh", expr)) {
+ - + + +
+ ]
1904 [ + - ]: 190 : auto pubkeys = ParsePubkey(key_exp_index, expr, ParseScriptContext::P2WPKH, out, error);
1905 [ + + ]: 190 : if (pubkeys.empty()) {
1906 [ + - ]: 23 : error = strprintf("wpkh(): %s", error);
1907 : 23 : return {};
1908 : : }
1909 : 167 : key_exp_index++;
1910 [ + + ]: 342 : for (auto& pubkey : pubkeys) {
1911 [ + - + - ]: 350 : ret.emplace_back(std::make_unique<WPKHDescriptor>(std::move(pubkey)));
1912 : : }
1913 : 167 : return ret;
1914 [ + - + - : 651 : } else if (Func("wpkh", expr)) {
+ + ]
1915 [ + - ]: 3 : error = "Can only have wpkh() at top level or inside sh()";
1916 : 3 : return {};
1917 : : }
1918 [ + + + - : 817 : if (ctx == ParseScriptContext::TOP && Func("sh", expr)) {
+ - + + +
+ ]
1919 [ + - ]: 145 : auto descs = ParseScript(key_exp_index, expr, ParseScriptContext::P2SH, out, error);
1920 [ + + + - ]: 145 : if (descs.empty() || expr.size()) return {};
1921 : 125 : std::vector<std::unique_ptr<DescriptorImpl>> ret;
1922 [ + - ]: 125 : ret.reserve(descs.size());
1923 [ + + ]: 266 : for (auto& desc : descs) {
1924 [ + - + - : 141 : ret.push_back(std::make_unique<SHDescriptor>(std::move(desc)));
- + ]
1925 : : }
1926 : 125 : return ret;
1927 [ + - + - : 458 : } else if (Func("sh", expr)) {
+ + ]
1928 [ + - ]: 6 : error = "Can only have sh() at top level";
1929 : 6 : return {};
1930 : : }
1931 [ + + + - : 544 : if ((ctx == ParseScriptContext::TOP || ctx == ParseScriptContext::P2SH) && Func("wsh", expr)) {
+ - + + +
+ ]
1932 [ + - ]: 95 : auto descs = ParseScript(key_exp_index, expr, ParseScriptContext::P2WSH, out, error);
1933 [ + + + - ]: 95 : if (descs.empty() || expr.size()) return {};
1934 [ + + ]: 114 : for (auto& desc : descs) {
1935 [ + - + - ]: 118 : ret.emplace_back(std::make_unique<WSHDescriptor>(std::move(desc)));
1936 : : }
1937 : 55 : return ret;
1938 [ + - + - : 307 : } else if (Func("wsh", expr)) {
+ + ]
1939 [ + - ]: 3 : error = "Can only have wsh() at top level or inside sh()";
1940 : 3 : return {};
1941 : : }
1942 [ + + + - : 346 : if (ctx == ParseScriptContext::TOP && Func("addr", expr)) {
+ - + + +
+ ]
1943 [ + - + - ]: 2 : CTxDestination dest = DecodeDestination(std::string(expr.begin(), expr.end()));
1944 [ + - + - ]: 1 : if (!IsValidDestination(dest)) {
1945 [ + - ]: 1 : error = "Address is not valid";
1946 : 1 : return {};
1947 : : }
1948 [ # # # # ]: 0 : ret.emplace_back(std::make_unique<AddressDescriptor>(std::move(dest)));
1949 : 0 : return ret;
1950 [ + - + - : 209 : } else if (Func("addr", expr)) {
- + ]
1951 [ # # ]: 0 : error = "Can only have addr() at top level";
1952 : 0 : return {};
1953 : : }
1954 [ + + + - : 344 : if (ctx == ParseScriptContext::TOP && Func("tr", expr)) {
+ - + + +
+ ]
1955 [ + - ]: 116 : auto arg = Expr(expr);
1956 [ + - ]: 116 : auto internal_keys = ParsePubkey(key_exp_index, arg, ParseScriptContext::P2TR, out, error);
1957 [ + + ]: 116 : if (internal_keys.empty()) {
1958 [ + - ]: 2 : error = strprintf("tr(): %s", error);
1959 : 2 : return {};
1960 : : }
1961 [ + + ]: 114 : size_t max_providers_len = internal_keys.size();
1962 : 114 : ++key_exp_index;
1963 : 114 : std::vector<std::vector<std::unique_ptr<DescriptorImpl>>> subscripts; //!< list of multipath expanded script subexpressions
1964 : 114 : std::vector<int> depths; //!< depth in the tree of each subexpression (same length subscripts)
1965 [ + + ]: 114 : if (expr.size()) {
1966 [ + - + - : 39 : if (!Const(",", expr)) {
- + ]
1967 [ # # ]: 0 : error = strprintf("tr: expected ',', got '%c'", expr[0]);
1968 : 0 : return {};
1969 : : }
1970 : : /** The path from the top of the tree to what we're currently processing.
1971 : : * branches[i] == false: left branch in the i'th step from the top; true: right branch.
1972 : : */
1973 : 39 : std::vector<bool> branches;
1974 : : // Loop over all provided scripts. In every iteration exactly one script will be processed.
1975 : : // Use a do-loop because inside this if-branch we expect at least one script.
1976 : : do {
1977 : : // First process all open braces.
1978 [ + - + - : 125 : while (Const("{", expr)) {
+ + ]
1979 [ + - ]: 54 : branches.push_back(false); // new left branch
1980 [ - + ]: 54 : if (branches.size() > TAPROOT_CONTROL_MAX_NODE_COUNT) {
1981 [ # # ]: 0 : error = strprintf("tr() supports at most %i nesting levels", TAPROOT_CONTROL_MAX_NODE_COUNT);
1982 : 0 : return {};
1983 : : }
1984 : : }
1985 : : // Process the actual script expression.
1986 [ + - ]: 71 : auto sarg = Expr(expr);
1987 [ + - + - ]: 71 : subscripts.emplace_back(ParseScript(key_exp_index, sarg, ParseScriptContext::P2TR, out, error));
1988 [ + + ]: 71 : if (subscripts.back().empty()) return {};
1989 [ + + ]: 70 : max_providers_len = std::max(max_providers_len, subscripts.back().size());
1990 [ + - ]: 70 : depths.push_back(branches.size());
1991 : : // Process closing braces; one is expected for every right branch we were in.
1992 [ + + + + ]: 102 : while (branches.size() && branches.back()) {
1993 [ + - - + : 32 : if (!Const("}", expr)) {
+ - ]
1994 [ # # ]: 0 : error = strprintf("tr(): expected '}' after script expression");
1995 : 0 : return {};
1996 : : }
1997 : 32 : branches.pop_back(); // move up one level after encountering '}'
1998 : : }
1999 : : // If after that, we're at the end of a left branch, expect a comma.
2000 [ + + + - ]: 70 : if (branches.size() && !branches.back()) {
2001 [ + - + - : 32 : if (!Const(",", expr)) {
- + ]
2002 [ # # ]: 0 : error = strprintf("tr(): expected ',' after script expression");
2003 : 0 : return {};
2004 : : }
2005 : 32 : branches.back() = true; // And now we're in a right branch.
2006 : : }
2007 [ + + ]: 70 : } while (branches.size());
2008 : : // After we've explored a whole tree, we must be at the end of the expression.
2009 [ - + ]: 38 : if (expr.size()) {
2010 [ # # ]: 0 : error = strprintf("tr(): expected ')' after script expression");
2011 : 0 : return {};
2012 : : }
2013 : 39 : }
2014 [ + - - + ]: 113 : assert(TaprootBuilder::ValidDepths(depths));
2015 : :
2016 : : // Make sure all vecs are of the same length, or exactly length 1
2017 : : // For length 1 vectors, clone subdescs until vector is the same length
2018 [ + + ]: 177 : for (auto& vec : subscripts) {
2019 [ + + ]: 68 : if (vec.size() == 1) {
2020 [ - + ]: 34 : for (size_t i = 1; i < max_providers_len; ++i) {
2021 [ # # # # : 0 : vec.emplace_back(vec.at(0)->Clone());
# # ]
2022 : : }
2023 [ + + ]: 34 : } else if (vec.size() != max_providers_len) {
2024 [ + - ]: 4 : error = strprintf("tr(): Multipath subscripts have mismatched lengths");
2025 : 4 : return {};
2026 : : }
2027 : : }
2028 : :
2029 [ + + + + ]: 109 : if (internal_keys.size() > 1 && internal_keys.size() != max_providers_len) {
2030 [ + - ]: 2 : error = strprintf("tr(): Multipath internal key mismatches multipath subscripts lengths");
2031 : 2 : return {};
2032 : : }
2033 : :
2034 [ + + ]: 119 : while (internal_keys.size() < max_providers_len) {
2035 [ + - + - : 12 : internal_keys.emplace_back(internal_keys.at(0)->Clone());
+ - ]
2036 : : }
2037 : :
2038 : : // Build the final descriptors vector
2039 [ + + ]: 238 : for (size_t i = 0; i < max_providers_len; ++i) {
2040 : : // Build final subscripts vectors by retrieving the i'th subscript for each vector in subscripts
2041 : 131 : std::vector<std::unique_ptr<DescriptorImpl>> this_subs;
2042 [ + - ]: 131 : this_subs.reserve(subscripts.size());
2043 [ + + ]: 237 : for (auto& subs : subscripts) {
2044 [ + - + - ]: 106 : this_subs.emplace_back(std::move(subs.at(i)));
2045 : : }
2046 [ + - + - : 131 : ret.emplace_back(std::make_unique<TRDescriptor>(std::move(internal_keys.at(i)), std::move(this_subs), depths));
+ - ]
2047 : 131 : }
2048 : 107 : return ret;
2049 : :
2050 : :
2051 [ + - + - : 208 : } else if (Func("tr", expr)) {
- + ]
2052 [ # # ]: 0 : error = "Can only have tr at top level";
2053 : 0 : return {};
2054 : : }
2055 [ + + + - : 112 : if (ctx == ParseScriptContext::TOP && Func("rawtr", expr)) {
+ - + + +
+ ]
2056 [ + - ]: 7 : auto arg = Expr(expr);
2057 [ + + ]: 7 : if (expr.size()) {
2058 [ + - ]: 1 : error = strprintf("rawtr(): only one key expected.");
2059 : 1 : return {};
2060 : : }
2061 [ + - ]: 6 : auto output_keys = ParsePubkey(key_exp_index, arg, ParseScriptContext::P2TR, out, error);
2062 [ - + ]: 6 : if (output_keys.empty()) {
2063 [ # # ]: 0 : error = strprintf("rawtr(): %s", error);
2064 : 0 : return {};
2065 : : }
2066 : 6 : ++key_exp_index;
2067 [ + + ]: 12 : for (auto& pubkey : output_keys) {
2068 [ + - + - ]: 12 : ret.emplace_back(std::make_unique<RawTRDescriptor>(std::move(pubkey)));
2069 : : }
2070 : 6 : return ret;
2071 [ + - + - : 91 : } else if (Func("rawtr", expr)) {
- + ]
2072 [ # # ]: 0 : error = "Can only have rawtr at top level";
2073 : 0 : return {};
2074 : : }
2075 [ + + + - : 98 : if (ctx == ParseScriptContext::TOP && Func("raw", expr)) {
+ - + + +
+ ]
2076 [ + - + - ]: 4 : std::string str(expr.begin(), expr.end());
2077 [ + - + - ]: 2 : if (!IsHex(str)) {
2078 [ + - ]: 2 : error = "Raw script is not hex";
2079 : 2 : return {};
2080 : : }
2081 [ # # ]: 0 : auto bytes = ParseHex(str);
2082 [ # # # # ]: 0 : ret.emplace_back(std::make_unique<RawDescriptor>(CScript(bytes.begin(), bytes.end())));
2083 : 0 : return ret;
2084 [ + - + - : 85 : } else if (Func("raw", expr)) {
- + ]
2085 [ # # ]: 0 : error = "Can only have raw() at top level";
2086 : 0 : return {};
2087 : : }
2088 : : // Process miniscript expressions.
2089 : 83 : {
2090 : 83 : const auto script_ctx{ctx == ParseScriptContext::P2WSH ? miniscript::MiniscriptContext::P2WSH : miniscript::MiniscriptContext::TAPSCRIPT};
2091 [ + - ]: 83 : KeyParser parser(/*out = */&out, /* in = */nullptr, /* ctx = */script_ctx, key_exp_index);
2092 [ + - + - ]: 166 : auto node = miniscript::FromString(std::string(expr.begin(), expr.end()), parser);
2093 [ + + ]: 83 : if (parser.m_key_parsing_error != "") {
2094 : 2 : error = std::move(parser.m_key_parsing_error);
2095 : 2 : return {};
2096 : : }
2097 [ + + ]: 81 : if (node) {
2098 [ + + ]: 65 : if (ctx != ParseScriptContext::P2WSH && ctx != ParseScriptContext::P2TR) {
2099 [ + - ]: 3 : error = "Miniscript expressions can only be used in wsh or tr.";
2100 : 3 : return {};
2101 : : }
2102 [ + + + - ]: 62 : if (!node->IsSane() || node->IsNotSatisfiable()) {
2103 : : // Try to find the first insane sub for better error reporting.
2104 : 12 : auto insane_node = node.get();
2105 [ + - + + ]: 12 : if (const auto sub = node->FindInsaneSub()) insane_node = sub;
2106 [ + - + - : 24 : if (const auto str = insane_node->ToString(parser)) error = *str;
+ - ]
2107 [ + + ]: 12 : if (!insane_node->IsValid()) {
2108 [ + - ]: 4 : error += " is invalid";
2109 [ + - ]: 8 : } else if (!node->IsSane()) {
2110 [ + - ]: 8 : error += " is not sane";
2111 [ + + ]: 8 : if (!insane_node->IsNonMalleable()) {
2112 [ + - ]: 2 : error += ": malleable witnesses exist";
2113 [ + + + + ]: 6 : } else if (insane_node == node.get() && !insane_node->NeedsSignature()) {
2114 [ + - ]: 2 : error += ": witnesses without signature exist";
2115 [ + + ]: 4 : } else if (!insane_node->CheckTimeLocksMix()) {
2116 [ + - ]: 2 : error += ": contains mixes of timelocks expressed in blocks and seconds";
2117 [ + - ]: 2 : } else if (!insane_node->CheckDuplicateKey()) {
2118 [ + - ]: 2 : error += ": contains duplicate public keys";
2119 [ # # ]: 0 : } else if (!insane_node->ValidSatisfactions()) {
2120 [ # # ]: 0 : error += ": needs witnesses that may exceed resource limits";
2121 : : }
2122 : : } else {
2123 [ # # ]: 0 : error += " is not satisfiable";
2124 : : }
2125 : 12 : return {};
2126 : : }
2127 : : // A signature check is required for a miniscript to be sane. Therefore no sane miniscript
2128 : : // may have an empty list of public keys.
2129 [ + - ]: 50 : CHECK_NONFATAL(!parser.m_keys.empty());
2130 : 50 : key_exp_index += parser.m_keys.size();
2131 : : // Make sure all vecs are of the same length, or exactly length 1
2132 : : // For length 1 vectors, clone subdescs until vector is the same length
2133 : 50 : size_t num_multipath = std::max_element(parser.m_keys.begin(), parser.m_keys.end(),
2134 : 28 : [](const std::vector<std::unique_ptr<PubkeyProvider>>& a, const std::vector<std::unique_ptr<PubkeyProvider>>& b) {
2135 [ + - ]: 28 : return a.size() < b.size();
2136 : 50 : })->size();
2137 : :
2138 [ + + ]: 126 : for (auto& vec : parser.m_keys) {
2139 [ + + ]: 78 : if (vec.size() == 1) {
2140 [ - + ]: 66 : for (size_t i = 1; i < num_multipath; ++i) {
2141 [ # # # # : 0 : vec.emplace_back(vec.at(0)->Clone());
# # ]
2142 : : }
2143 [ + + ]: 12 : } else if (vec.size() != num_multipath) {
2144 [ + - ]: 2 : error = strprintf("Miniscript: Multipath derivation paths have mismatched lengths");
2145 : 2 : return {};
2146 : : }
2147 : : }
2148 : :
2149 : : // Build the final descriptors vector
2150 [ + + ]: 100 : for (size_t i = 0; i < num_multipath; ++i) {
2151 : : // Build final pubkeys vectors by retrieving the i'th subscript for each vector in subscripts
2152 : 52 : std::vector<std::unique_ptr<PubkeyProvider>> pubs;
2153 [ + - ]: 52 : pubs.reserve(parser.m_keys.size());
2154 [ + + ]: 134 : for (auto& pub : parser.m_keys) {
2155 [ + - + - ]: 82 : pubs.emplace_back(std::move(pub.at(i)));
2156 : : }
2157 [ + - + - ]: 52 : ret.emplace_back(std::make_unique<MiniscriptDescriptor>(std::move(pubs), node));
2158 : 52 : }
2159 : 48 : return ret;
2160 : : }
2161 : 150 : }
2162 [ + + ]: 16 : if (ctx == ParseScriptContext::P2SH) {
2163 [ + - ]: 2 : error = "A function is needed within P2SH";
2164 : 2 : return {};
2165 [ + + ]: 14 : } else if (ctx == ParseScriptContext::P2WSH) {
2166 [ + - ]: 2 : error = "A function is needed within P2WSH";
2167 : 2 : return {};
2168 : : }
2169 [ + - + - ]: 24 : error = strprintf("'%s' is not a valid descriptor function", std::string(expr.begin(), expr.end()));
2170 : 12 : return {};
2171 : 946 : }
2172 : :
2173 : 116 : std::unique_ptr<DescriptorImpl> InferMultiA(const CScript& script, ParseScriptContext ctx, const SigningProvider& provider)
2174 : : {
2175 : 116 : auto match = MatchMultiA(script);
2176 [ + + ]: 116 : if (!match) return {};
2177 : 36 : std::vector<std::unique_ptr<PubkeyProvider>> keys;
2178 [ + - ]: 36 : keys.reserve(match->second.size());
2179 [ + + ]: 96 : for (const auto keyspan : match->second) {
2180 [ - + ]: 60 : if (keyspan.size() != 32) return {};
2181 [ + - ]: 60 : auto key = InferXOnlyPubkey(XOnlyPubKey{keyspan}, ctx, provider);
2182 [ - + ]: 60 : if (!key) return {};
2183 [ + - ]: 60 : keys.push_back(std::move(key));
2184 : 60 : }
2185 [ + - - + ]: 36 : return std::make_unique<MultiADescriptor>(match->first, std::move(keys));
2186 : 152 : }
2187 : :
2188 : : // NOLINTNEXTLINE(misc-no-recursion)
2189 : 8105 : std::unique_ptr<DescriptorImpl> InferScript(const CScript& script, ParseScriptContext ctx, const SigningProvider& provider)
2190 : : {
2191 [ + + + + : 8105 : if (ctx == ParseScriptContext::P2TR && script.size() == 34 && script[0] == 32 && script[33] == OP_CHECKSIG) {
+ + + - -
+ ]
2192 : 216 : XOnlyPubKey key{Span{script}.subspan(1, 32)};
2193 [ + - - + ]: 216 : return std::make_unique<PKDescriptor>(InferXOnlyPubkey(key, ctx, provider), true);
2194 : : }
2195 : :
2196 [ + + ]: 7889 : if (ctx == ParseScriptContext::P2TR) {
2197 : 116 : auto ret = InferMultiA(script, ctx, provider);
2198 [ + + ]: 116 : if (ret) return ret;
2199 : 116 : }
2200 : :
2201 : 7853 : std::vector<std::vector<unsigned char>> data;
2202 [ + - ]: 7853 : TxoutType txntype = Solver(script, data);
2203 : :
2204 [ + + + - ]: 7853 : if (txntype == TxoutType::PUBKEY && (ctx == ParseScriptContext::TOP || ctx == ParseScriptContext::P2SH || ctx == ParseScriptContext::P2WSH)) {
2205 : 208 : CPubKey pubkey(data[0]);
2206 [ + - + + ]: 208 : if (auto pubkey_provider = InferPubkey(pubkey, ctx, provider)) {
2207 [ + - - + ]: 207 : return std::make_unique<PKDescriptor>(std::move(pubkey_provider));
2208 : 208 : }
2209 : : }
2210 [ + + + + ]: 7646 : if (txntype == TxoutType::PUBKEYHASH && (ctx == ParseScriptContext::TOP || ctx == ParseScriptContext::P2SH || ctx == ParseScriptContext::P2WSH)) {
2211 : 258 : uint160 hash(data[0]);
2212 [ + - ]: 258 : CKeyID keyid(hash);
2213 [ + - ]: 258 : CPubKey pubkey;
2214 [ + - + + ]: 258 : if (provider.GetPubKey(keyid, pubkey)) {
2215 [ + - + + ]: 244 : if (auto pubkey_provider = InferPubkey(pubkey, ctx, provider)) {
2216 [ + - - + ]: 243 : return std::make_unique<PKHDescriptor>(std::move(pubkey_provider));
2217 : 244 : }
2218 : : }
2219 : : }
2220 [ + + ]: 7403 : if (txntype == TxoutType::WITNESS_V0_KEYHASH && (ctx == ParseScriptContext::TOP || ctx == ParseScriptContext::P2SH)) {
2221 : 5956 : uint160 hash(data[0]);
2222 [ + - ]: 5956 : CKeyID keyid(hash);
2223 [ + - ]: 5956 : CPubKey pubkey;
2224 [ + - + + ]: 5956 : if (provider.GetPubKey(keyid, pubkey)) {
2225 [ + - + + ]: 5955 : if (auto pubkey_provider = InferPubkey(pubkey, ParseScriptContext::P2WPKH, provider)) {
2226 [ + - - + ]: 5954 : return std::make_unique<WPKHDescriptor>(std::move(pubkey_provider));
2227 : 5955 : }
2228 : : }
2229 : : }
2230 [ + + + - ]: 1449 : if (txntype == TxoutType::MULTISIG && (ctx == ParseScriptContext::TOP || ctx == ParseScriptContext::P2SH || ctx == ParseScriptContext::P2WSH)) {
2231 : 265 : bool ok = true;
2232 : 265 : std::vector<std::unique_ptr<PubkeyProvider>> providers;
2233 [ + + ]: 1456 : for (size_t i = 1; i + 1 < data.size(); ++i) {
2234 : 1191 : CPubKey pubkey(data[i]);
2235 [ + - + - ]: 1191 : if (auto pubkey_provider = InferPubkey(pubkey, ctx, provider)) {
2236 [ + - ]: 1191 : providers.push_back(std::move(pubkey_provider));
2237 : : } else {
2238 : 0 : ok = false;
2239 : 0 : break;
2240 : 1191 : }
2241 : : }
2242 [ + - - + ]: 265 : if (ok) return std::make_unique<MultisigDescriptor>((int)data[0][0], std::move(providers));
2243 : 265 : }
2244 [ + + ]: 1184 : if (txntype == TxoutType::SCRIPTHASH && ctx == ParseScriptContext::TOP) {
2245 : 334 : uint160 hash(data[0]);
2246 [ + - ]: 334 : CScriptID scriptid(hash);
2247 : 334 : CScript subscript;
2248 [ + - + + ]: 334 : if (provider.GetCScript(scriptid, subscript)) {
2249 [ + - ]: 328 : auto sub = InferScript(subscript, ParseScriptContext::P2SH, provider);
2250 [ + - + - : 328 : if (sub) return std::make_unique<SHDescriptor>(std::move(sub));
- + ]
2251 : 328 : }
2252 : 334 : }
2253 [ + + ]: 856 : if (txntype == TxoutType::WITNESS_V0_SCRIPTHASH && (ctx == ParseScriptContext::TOP || ctx == ParseScriptContext::P2SH)) {
2254 [ + - ]: 314 : CScriptID scriptid{RIPEMD160(data[0])};
2255 : 314 : CScript subscript;
2256 [ + - + + ]: 314 : if (provider.GetCScript(scriptid, subscript)) {
2257 [ + - ]: 312 : auto sub = InferScript(subscript, ParseScriptContext::P2WSH, provider);
2258 [ + - + - : 312 : if (sub) return std::make_unique<WSHDescriptor>(std::move(sub));
- + ]
2259 : 312 : }
2260 : 314 : }
2261 [ + + ]: 544 : if (txntype == TxoutType::WITNESS_V1_TAPROOT && ctx == ParseScriptContext::TOP) {
2262 : : // Extract x-only pubkey from output.
2263 : 239 : XOnlyPubKey pubkey;
2264 : 239 : std::copy(data[0].begin(), data[0].end(), pubkey.begin());
2265 : : // Request spending data.
2266 [ + - ]: 239 : TaprootSpendData tap;
2267 [ + - + + ]: 239 : if (provider.GetTaprootSpendData(pubkey, tap)) {
2268 : : // If found, convert it back to tree form.
2269 [ + - ]: 203 : auto tree = InferTaprootTree(tap, pubkey);
2270 [ + - ]: 203 : if (tree) {
2271 : : // If that works, try to infer subdescriptors for all leaves.
2272 : 203 : bool ok = true;
2273 : 203 : std::vector<std::unique_ptr<DescriptorImpl>> subscripts; //!< list of script subexpressions
2274 : 203 : std::vector<int> depths; //!< depth in the tree of each subexpression (same length subscripts)
2275 [ + - + + ]: 535 : for (const auto& [depth, script, leaf_ver] : *tree) {
2276 : 332 : std::unique_ptr<DescriptorImpl> subdesc;
2277 [ + - ]: 332 : if (leaf_ver == TAPROOT_LEAF_TAPSCRIPT) {
2278 [ + - ]: 664 : subdesc = InferScript(CScript(script.begin(), script.end()), ParseScriptContext::P2TR, provider);
2279 : : }
2280 [ - + ]: 332 : if (!subdesc) {
2281 : 0 : ok = false;
2282 : 0 : break;
2283 : : } else {
2284 [ + - ]: 332 : subscripts.push_back(std::move(subdesc));
2285 [ + - ]: 332 : depths.push_back(depth);
2286 : : }
2287 : 332 : }
2288 : 0 : if (ok) {
2289 [ + - ]: 203 : auto key = InferXOnlyPubkey(tap.internal_key, ParseScriptContext::P2TR, provider);
2290 [ + - - + ]: 203 : return std::make_unique<TRDescriptor>(std::move(key), std::move(subscripts), std::move(depths));
2291 : 203 : }
2292 : 203 : }
2293 : 203 : }
2294 : : // If the above doesn't work, construct a rawtr() descriptor with just the encoded x-only pubkey.
2295 [ + - + - ]: 36 : if (pubkey.IsFullyValid()) {
2296 [ + - ]: 36 : auto key = InferXOnlyPubkey(pubkey, ParseScriptContext::P2TR, provider);
2297 [ + - ]: 36 : if (key) {
2298 [ + - - + ]: 36 : return std::make_unique<RawTRDescriptor>(std::move(key));
2299 : : }
2300 : 36 : }
2301 : 239 : }
2302 : :
2303 [ + + ]: 305 : if (ctx == ParseScriptContext::P2WSH || ctx == ParseScriptContext::P2TR) {
2304 : 272 : const auto script_ctx{ctx == ParseScriptContext::P2WSH ? miniscript::MiniscriptContext::P2WSH : miniscript::MiniscriptContext::TAPSCRIPT};
2305 [ + - ]: 272 : KeyParser parser(/* out = */nullptr, /* in = */&provider, /* ctx = */script_ctx);
2306 [ + - ]: 272 : auto node = miniscript::FromScript(script, parser);
2307 [ + - - + ]: 272 : if (node && node->IsSane()) {
2308 : 272 : std::vector<std::unique_ptr<PubkeyProvider>> keys;
2309 [ + - ]: 272 : keys.reserve(parser.m_keys.size());
2310 [ + + ]: 684 : for (auto& key : parser.m_keys) {
2311 [ + - + - ]: 412 : keys.emplace_back(std::move(key.at(0)));
2312 : : }
2313 [ + - - + ]: 272 : return std::make_unique<MiniscriptDescriptor>(std::move(keys), std::move(node));
2314 [ - + ]: 272 : }
2315 : 544 : }
2316 : :
2317 : : // The following descriptors are all top-level only descriptors.
2318 : : // So if we are not at the top level, return early.
2319 [ - + ]: 33 : if (ctx != ParseScriptContext::TOP) return nullptr;
2320 : :
2321 : 33 : CTxDestination dest;
2322 [ + - + + ]: 33 : if (ExtractDestination(script, dest)) {
2323 [ + - + - ]: 25 : if (GetScriptForDestination(dest) == script) {
2324 [ + - - + ]: 25 : return std::make_unique<AddressDescriptor>(std::move(dest));
2325 : : }
2326 : : }
2327 : :
2328 [ + - - + ]: 8 : return std::make_unique<RawDescriptor>(script);
2329 : 7853 : }
2330 : :
2331 : :
2332 : : } // namespace
2333 : :
2334 : : /** Check a descriptor checksum, and update desc to be the checksum-less part. */
2335 : 681 : bool CheckChecksum(Span<const char>& sp, bool require_checksum, std::string& error, std::string* out_checksum = nullptr)
2336 : : {
2337 : 681 : auto check_split = Split(sp, '#');
2338 [ + + ]: 681 : if (check_split.size() > 2) {
2339 [ + - ]: 2 : error = "Multiple '#' symbols";
2340 : : return false;
2341 : : }
2342 [ + + + + ]: 679 : if (check_split.size() == 1 && require_checksum){
2343 [ + - ]: 681 : error = "Missing checksum";
2344 : : return false;
2345 : : }
2346 [ + + ]: 678 : if (check_split.size() == 2) {
2347 [ + + ]: 45 : if (check_split[1].size() != 8) {
2348 [ + - ]: 6 : error = strprintf("Expected 8 character checksum, not %u characters", check_split[1].size());
2349 : 6 : return false;
2350 : : }
2351 : : }
2352 [ + - ]: 672 : auto checksum = DescriptorChecksum(check_split[0]);
2353 [ + + ]: 672 : if (checksum.empty()) {
2354 [ + - ]: 672 : error = "Invalid characters in payload";
2355 : : return false;
2356 : : }
2357 [ + + ]: 671 : if (check_split.size() == 2) {
2358 [ + + ]: 38 : if (!std::equal(checksum.begin(), checksum.end(), check_split[1].begin())) {
2359 [ + - + - ]: 20 : error = strprintf("Provided checksum '%s' does not match computed checksum '%s'", std::string(check_split[1].begin(), check_split[1].end()), checksum);
2360 : 10 : return false;
2361 : : }
2362 : : }
2363 [ + + ]: 661 : if (out_checksum) *out_checksum = std::move(checksum);
2364 : 661 : sp = check_split[0];
2365 : 661 : return true;
2366 : 1353 : }
2367 : :
2368 : 651 : std::vector<std::unique_ptr<Descriptor>> Parse(const std::string& descriptor, FlatSigningProvider& out, std::string& error, bool require_checksum)
2369 : : {
2370 : 651 : Span<const char> sp{descriptor};
2371 [ + + ]: 651 : if (!CheckChecksum(sp, require_checksum, error)) return {};
2372 : 635 : uint32_t key_exp_index = 0;
2373 : 635 : auto ret = ParseScript(key_exp_index, sp, ParseScriptContext::TOP, out, error);
2374 [ + - + + ]: 635 : if (sp.size() == 0 && !ret.empty()) {
2375 : 507 : std::vector<std::unique_ptr<Descriptor>> descs;
2376 [ + - ]: 507 : descs.reserve(ret.size());
2377 [ + + ]: 1086 : for (auto& r : ret) {
2378 [ + - ]: 579 : descs.emplace_back(std::unique_ptr<Descriptor>(std::move(r)));
2379 : : }
2380 : 507 : return descs;
2381 : 507 : }
2382 : 128 : return {};
2383 : 635 : }
2384 : :
2385 : 30 : std::string GetDescriptorChecksum(const std::string& descriptor)
2386 : : {
2387 [ + - ]: 30 : std::string ret;
2388 : 30 : std::string error;
2389 [ + - ]: 30 : Span<const char> sp{descriptor};
2390 [ + - + + : 30 : if (!CheckChecksum(sp, false, error, &ret)) return "";
+ - ]
2391 : 26 : return ret;
2392 : 30 : }
2393 : :
2394 : 7133 : std::unique_ptr<Descriptor> InferDescriptor(const CScript& script, const SigningProvider& provider)
2395 : : {
2396 : 7133 : return InferScript(script, ParseScriptContext::TOP, provider);
2397 : : }
2398 : :
2399 : 399 : uint256 DescriptorID(const Descriptor& desc)
2400 : : {
2401 : 399 : std::string desc_str = desc.ToString(/*compat_format=*/true);
2402 : 399 : uint256 id;
2403 [ + - + - : 399 : CSHA256().Write((unsigned char*)desc_str.data(), desc_str.size()).Finalize(id.begin());
+ - ]
2404 : 399 : return id;
2405 : 399 : }
2406 : :
2407 : 1428 : void DescriptorCache::CacheParentExtPubKey(uint32_t key_exp_pos, const CExtPubKey& xpub)
2408 : : {
2409 : 1428 : m_parent_xpubs[key_exp_pos] = xpub;
2410 : 1428 : }
2411 : :
2412 : 3036 : void DescriptorCache::CacheDerivedExtPubKey(uint32_t key_exp_pos, uint32_t der_index, const CExtPubKey& xpub)
2413 : : {
2414 : 3036 : auto& xpubs = m_derived_xpubs[key_exp_pos];
2415 : 3036 : xpubs[der_index] = xpub;
2416 : 3036 : }
2417 : :
2418 : 898 : void DescriptorCache::CacheLastHardenedExtPubKey(uint32_t key_exp_pos, const CExtPubKey& xpub)
2419 : : {
2420 : 898 : m_last_hardened_xpubs[key_exp_pos] = xpub;
2421 : 898 : }
2422 : :
2423 : 307423 : bool DescriptorCache::GetCachedParentExtPubKey(uint32_t key_exp_pos, CExtPubKey& xpub) const
2424 : : {
2425 : 307423 : const auto& it = m_parent_xpubs.find(key_exp_pos);
2426 [ + + ]: 307423 : if (it == m_parent_xpubs.end()) return false;
2427 : 306873 : xpub = it->second;
2428 : 306873 : return true;
2429 : : }
2430 : :
2431 : 309189 : bool DescriptorCache::GetCachedDerivedExtPubKey(uint32_t key_exp_pos, uint32_t der_index, CExtPubKey& xpub) const
2432 : : {
2433 : 309189 : const auto& key_exp_it = m_derived_xpubs.find(key_exp_pos);
2434 [ + + ]: 309189 : if (key_exp_it == m_derived_xpubs.end()) return false;
2435 : 2040 : const auto& der_it = key_exp_it->second.find(der_index);
2436 [ + + ]: 2040 : if (der_it == key_exp_it->second.end()) return false;
2437 : 42 : xpub = der_it->second;
2438 : 42 : return true;
2439 : : }
2440 : :
2441 : 276 : bool DescriptorCache::GetCachedLastHardenedExtPubKey(uint32_t key_exp_pos, CExtPubKey& xpub) const
2442 : : {
2443 : 276 : const auto& it = m_last_hardened_xpubs.find(key_exp_pos);
2444 [ - + ]: 276 : if (it == m_last_hardened_xpubs.end()) return false;
2445 : 0 : xpub = it->second;
2446 : 0 : return true;
2447 : : }
2448 : :
2449 : 279182 : DescriptorCache DescriptorCache::MergeAndDiff(const DescriptorCache& other)
2450 : : {
2451 : 279182 : DescriptorCache diff;
2452 [ + - + + ]: 279458 : for (const auto& parent_xpub_pair : other.GetCachedParentExtPubKeys()) {
2453 [ + - ]: 276 : CExtPubKey xpub;
2454 [ + - - + ]: 276 : if (GetCachedParentExtPubKey(parent_xpub_pair.first, xpub)) {
2455 [ # # ]: 0 : if (xpub != parent_xpub_pair.second) {
2456 [ # # # # ]: 0 : throw std::runtime_error(std::string(__func__) + ": New cached parent xpub does not match already cached parent xpub");
2457 : : }
2458 : 0 : continue;
2459 : : }
2460 [ + - ]: 276 : CacheParentExtPubKey(parent_xpub_pair.first, parent_xpub_pair.second);
2461 [ + - ]: 276 : diff.CacheParentExtPubKey(parent_xpub_pair.first, parent_xpub_pair.second);
2462 : 0 : }
2463 [ + - + + ]: 280182 : for (const auto& derived_xpub_map_pair : other.GetCachedDerivedExtPubKeys()) {
2464 [ + + ]: 2000 : for (const auto& derived_xpub_pair : derived_xpub_map_pair.second) {
2465 [ + - ]: 1000 : CExtPubKey xpub;
2466 [ + - - + ]: 1000 : if (GetCachedDerivedExtPubKey(derived_xpub_map_pair.first, derived_xpub_pair.first, xpub)) {
2467 [ # # ]: 0 : if (xpub != derived_xpub_pair.second) {
2468 [ # # # # ]: 0 : throw std::runtime_error(std::string(__func__) + ": New cached derived xpub does not match already cached derived xpub");
2469 : : }
2470 : 0 : continue;
2471 : : }
2472 [ + - ]: 1000 : CacheDerivedExtPubKey(derived_xpub_map_pair.first, derived_xpub_pair.first, derived_xpub_pair.second);
2473 [ + - ]: 1000 : diff.CacheDerivedExtPubKey(derived_xpub_map_pair.first, derived_xpub_pair.first, derived_xpub_pair.second);
2474 : : }
2475 : 0 : }
2476 [ + - + + ]: 279458 : for (const auto& lh_xpub_pair : other.GetCachedLastHardenedExtPubKeys()) {
2477 [ + - ]: 276 : CExtPubKey xpub;
2478 [ + - - + ]: 276 : if (GetCachedLastHardenedExtPubKey(lh_xpub_pair.first, xpub)) {
2479 [ # # ]: 0 : if (xpub != lh_xpub_pair.second) {
2480 [ # # # # ]: 0 : throw std::runtime_error(std::string(__func__) + ": New cached last hardened xpub does not match already cached last hardened xpub");
2481 : : }
2482 : 0 : continue;
2483 : : }
2484 [ + - ]: 276 : CacheLastHardenedExtPubKey(lh_xpub_pair.first, lh_xpub_pair.second);
2485 [ + - ]: 276 : diff.CacheLastHardenedExtPubKey(lh_xpub_pair.first, lh_xpub_pair.second);
2486 : 0 : }
2487 : 279182 : return diff;
2488 : 0 : }
2489 : :
2490 : 558928 : ExtPubKeyMap DescriptorCache::GetCachedParentExtPubKeys() const
2491 : : {
2492 : 558928 : return m_parent_xpubs;
2493 : : }
2494 : :
2495 : 558928 : std::unordered_map<uint32_t, ExtPubKeyMap> DescriptorCache::GetCachedDerivedExtPubKeys() const
2496 : : {
2497 : 558928 : return m_derived_xpubs;
2498 : : }
2499 : :
2500 : 558373 : ExtPubKeyMap DescriptorCache::GetCachedLastHardenedExtPubKeys() const
2501 : : {
2502 : 558373 : return m_last_hardened_xpubs;
2503 : : }
|