Bitcoin Core Fuzz Coverage Report

Coverage Report

Created: 2026-06-01 16:00

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