Bitcoin Core 29.99.0
P2P Digital Currency
psbt_wallet_tests.cpp
Go to the documentation of this file.
1// Copyright (c) 2017-2022 The Bitcoin Core developers
2// Distributed under the MIT software license, see the accompanying
3// file COPYING or http://www.opensource.org/licenses/mit-license.php.
4
5#include <key_io.h>
6#include <node/types.h>
7#include <util/bip32.h>
8#include <util/strencodings.h>
9#include <wallet/wallet.h>
10
11#include <boost/test/unit_test.hpp>
14
15using namespace util::hex_literals;
16
17namespace wallet {
18BOOST_FIXTURE_TEST_SUITE(psbt_wallet_tests, WalletTestingSetup)
19
20static void import_descriptor(CWallet& wallet, const std::string& descriptor)
22{
23 AssertLockHeld(wallet.cs_wallet);
24 FlatSigningProvider provider;
25 std::string error;
26 auto descs = Parse(descriptor, provider, error, /* require_checksum=*/ false);
27 assert(descs.size() == 1);
28 auto& desc = descs.at(0);
29 WalletDescriptor w_desc(std::move(desc), 0, 0, 10, 0);
30 auto spk_manager = *Assert(wallet.AddWalletDescriptor(w_desc, provider, "", false));
31 assert(spk_manager);
32}
33
34BOOST_AUTO_TEST_CASE(psbt_updater_test)
35{
36 LOCK(m_wallet.cs_wallet);
37 m_wallet.SetWalletFlag(WALLET_FLAG_DESCRIPTORS);
38
39 // Create prevtxs and add to wallet
40 DataStream s_prev_tx1{
41 "0200000000010158e87a21b56daf0c23be8e7070456c336f7cbaa5c8757924f545887bb2abdd7501000000171600145f275f436b09a8cc9a2eb2a2f528485c68a56323feffffff02d8231f1b0100000017a914aed962d6654f9a2b36608eb9d64d2b260db4f1118700c2eb0b0000000017a914b7f5faf40e3d40a5a459b1db3535f2b72fa921e88702483045022100a22edcc6e5bc511af4cc4ae0de0fcd75c7e04d8c1c3a8aa9d820ed4b967384ec02200642963597b9b1bc22c75e9f3e117284a962188bf5e8a74c895089046a20ad770121035509a48eb623e10aace8bfd0212fdb8a8e5af3c94b0b133b95e114cab89e4f7965000000"_hex,
42 };
43 CTransactionRef prev_tx1;
44 s_prev_tx1 >> TX_WITH_WITNESS(prev_tx1);
45 m_wallet.mapWallet.emplace(std::piecewise_construct, std::forward_as_tuple(prev_tx1->GetHash()), std::forward_as_tuple(prev_tx1, TxStateInactive{}));
46
47 DataStream s_prev_tx2{
48 "0200000001aad73931018bd25f84ae400b68848be09db706eac2ac18298babee71ab656f8b0000000048473044022058f6fc7c6a33e1b31548d481c826c015bd30135aad42cd67790dab66d2ad243b02204a1ced2604c6735b6393e5b41691dd78b00f0c5942fb9f751856faa938157dba01feffffff0280f0fa020000000017a9140fb9463421696b82c833af241c78c17ddbde493487d0f20a270100000017a91429ca74f8a08f81999428185c97b5d852e4063f618765000000"_hex,
49 };
50 CTransactionRef prev_tx2;
51 s_prev_tx2 >> TX_WITH_WITNESS(prev_tx2);
52 m_wallet.mapWallet.emplace(std::piecewise_construct, std::forward_as_tuple(prev_tx2->GetHash()), std::forward_as_tuple(prev_tx2, TxStateInactive{}));
53
54 // Import descriptors for keys and scripts
55 import_descriptor(m_wallet, "sh(multi(2,xprv9s21ZrQH143K2LE7W4Xf3jATf9jECxSb7wj91ZnmY4qEJrS66Qru9RFqq8xbkgT32ya6HqYJweFdJUEDf5Q6JFV7jMiUws7kQfe6Tv4RbfN/0h/0h/0h,xprv9s21ZrQH143K2LE7W4Xf3jATf9jECxSb7wj91ZnmY4qEJrS66Qru9RFqq8xbkgT32ya6HqYJweFdJUEDf5Q6JFV7jMiUws7kQfe6Tv4RbfN/0h/0h/1h))");
56 import_descriptor(m_wallet, "sh(wsh(multi(2,xprv9s21ZrQH143K2LE7W4Xf3jATf9jECxSb7wj91ZnmY4qEJrS66Qru9RFqq8xbkgT32ya6HqYJweFdJUEDf5Q6JFV7jMiUws7kQfe6Tv4RbfN/0h/0h/2h,xprv9s21ZrQH143K2LE7W4Xf3jATf9jECxSb7wj91ZnmY4qEJrS66Qru9RFqq8xbkgT32ya6HqYJweFdJUEDf5Q6JFV7jMiUws7kQfe6Tv4RbfN/0h/0h/3h)))");
57 import_descriptor(m_wallet, "wpkh(xprv9s21ZrQH143K2LE7W4Xf3jATf9jECxSb7wj91ZnmY4qEJrS66Qru9RFqq8xbkgT32ya6HqYJweFdJUEDf5Q6JFV7jMiUws7kQfe6Tv4RbfN/0h/0h/*h)");
58
59 // Call FillPSBT
61 DataStream ssData{
62 "70736274ff01009a020000000258e87a21b56daf0c23be8e7070456c336f7cbaa5c8757924f545887bb2abdd750000000000ffffffff838d0427d0ec650a68aa46bb0b098aea4422c071b2ca78352a077959d07cea1d0100000000ffffffff0270aaf00800000000160014d85c2b71d0060b09c9886aeb815e50991dda124d00e1f5050000000016001400aea9a2e5f0f876a588df5546e8742d1d87008f000000000000000000"_hex,
63 };
64 ssData >> psbtx;
65
66 // Fill transaction with our data
67 bool complete = true;
68 BOOST_REQUIRE(!m_wallet.FillPSBT(psbtx, complete, SIGHASH_ALL, false, true));
69
70 // Get the final tx
71 DataStream ssTx{};
72 ssTx << psbtx;
73 std::string final_hex = HexStr(ssTx);
74 BOOST_CHECK_EQUAL(final_hex, "70736274ff01009a020000000258e87a21b56daf0c23be8e7070456c336f7cbaa5c8757924f545887bb2abdd750000000000ffffffff838d0427d0ec650a68aa46bb0b098aea4422c071b2ca78352a077959d07cea1d0100000000ffffffff0270aaf00800000000160014d85c2b71d0060b09c9886aeb815e50991dda124d00e1f5050000000016001400aea9a2e5f0f876a588df5546e8742d1d87008f00000000000100bb0200000001aad73931018bd25f84ae400b68848be09db706eac2ac18298babee71ab656f8b0000000048473044022058f6fc7c6a33e1b31548d481c826c015bd30135aad42cd67790dab66d2ad243b02204a1ced2604c6735b6393e5b41691dd78b00f0c5942fb9f751856faa938157dba01feffffff0280f0fa020000000017a9140fb9463421696b82c833af241c78c17ddbde493487d0f20a270100000017a91429ca74f8a08f81999428185c97b5d852e4063f6187650000000104475221029583bf39ae0a609747ad199addd634fa6108559d6c5cd39b4c2183f1ab96e07f2102dab61ff49a14db6a7d02b0cd1fbb78fc4b18312b5b4e54dae4dba2fbfef536d752ae2206029583bf39ae0a609747ad199addd634fa6108559d6c5cd39b4c2183f1ab96e07f10d90c6a4f000000800000008000000080220602dab61ff49a14db6a7d02b0cd1fbb78fc4b18312b5b4e54dae4dba2fbfef536d710d90c6a4f0000008000000080010000800001008a020000000158e87a21b56daf0c23be8e7070456c336f7cbaa5c8757924f545887bb2abdd7501000000171600145f275f436b09a8cc9a2eb2a2f528485c68a56323feffffff02d8231f1b0100000017a914aed962d6654f9a2b36608eb9d64d2b260db4f1118700c2eb0b0000000017a914b7f5faf40e3d40a5a459b1db3535f2b72fa921e8876500000001012000c2eb0b0000000017a914b7f5faf40e3d40a5a459b1db3535f2b72fa921e88701042200208c2353173743b595dfb4a07b72ba8e42e3797da74e87fe7d9d7497e3b2028903010547522103089dc10c7ac6db54f91329af617333db388cead0c231f723379d1b99030b02dc21023add904f3d6dcf59ddb906b0dee23529b7ffb9ed50e5e86151926860221f0e7352ae2206023add904f3d6dcf59ddb906b0dee23529b7ffb9ed50e5e86151926860221f0e7310d90c6a4f000000800000008003000080220603089dc10c7ac6db54f91329af617333db388cead0c231f723379d1b99030b02dc10d90c6a4f00000080000000800200008000220203a9a4c37f5996d3aa25dbac6b570af0650394492942460b354753ed9eeca5877110d90c6a4f000000800000008004000080002202027f6399757d2eff55a136ad02c684b1838b6556e5f1b6b34282a94b6b5005109610d90c6a4f00000080000000800500008000");
75
76 // Mutate the transaction so that one of the inputs is invalid
77 psbtx.tx->vin[0].prevout.n = 2;
78
79 // Try to sign the mutated input
80 SignatureData sigdata;
81 BOOST_CHECK(m_wallet.FillPSBT(psbtx, complete, SIGHASH_ALL, true, true));
82}
83
84BOOST_AUTO_TEST_CASE(parse_hd_keypath)
85{
86 std::vector<uint32_t> keypath;
87
88 BOOST_CHECK(ParseHDKeypath("1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1", keypath));
89 BOOST_CHECK(!ParseHDKeypath("///////////////////////////", keypath));
90
91 BOOST_CHECK(ParseHDKeypath("1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1'/1", keypath));
92 BOOST_CHECK(!ParseHDKeypath("//////////////////////////'/", keypath));
93
94 BOOST_CHECK(ParseHDKeypath("1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/", keypath));
95 BOOST_CHECK(!ParseHDKeypath("1///////////////////////////", keypath));
96
97 BOOST_CHECK(ParseHDKeypath("1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1/1'/", keypath));
98 BOOST_CHECK(!ParseHDKeypath("1/'//////////////////////////", keypath));
99
100 BOOST_CHECK(ParseHDKeypath("", keypath));
101 BOOST_CHECK(!ParseHDKeypath(" ", keypath));
102
103 BOOST_CHECK(ParseHDKeypath("0", keypath));
104 BOOST_CHECK(!ParseHDKeypath("O", keypath));
105
106 BOOST_CHECK(ParseHDKeypath("0000'/0000'/0000'", keypath));
107 BOOST_CHECK(!ParseHDKeypath("0000,/0000,/0000,", keypath));
108
109 BOOST_CHECK(ParseHDKeypath("01234", keypath));
110 BOOST_CHECK(!ParseHDKeypath("0x1234", keypath));
111
112 BOOST_CHECK(ParseHDKeypath("1", keypath));
113 BOOST_CHECK(!ParseHDKeypath(" 1", keypath));
114
115 BOOST_CHECK(ParseHDKeypath("42", keypath));
116 BOOST_CHECK(!ParseHDKeypath("m42", keypath));
117
118 BOOST_CHECK(ParseHDKeypath("4294967295", keypath)); // 4294967295 == 0xFFFFFFFF (uint32_t max)
119 BOOST_CHECK(!ParseHDKeypath("4294967296", keypath)); // 4294967296 == 0xFFFFFFFF (uint32_t max) + 1
120
121 BOOST_CHECK(ParseHDKeypath("m", keypath));
122 BOOST_CHECK(!ParseHDKeypath("n", keypath));
123
124 BOOST_CHECK(ParseHDKeypath("m/", keypath));
125 BOOST_CHECK(!ParseHDKeypath("n/", keypath));
126
127 BOOST_CHECK(ParseHDKeypath("m/0", keypath));
128 BOOST_CHECK(!ParseHDKeypath("n/0", keypath));
129
130 BOOST_CHECK(ParseHDKeypath("m/0'", keypath));
131 BOOST_CHECK(!ParseHDKeypath("m/0''", keypath));
132
133 BOOST_CHECK(ParseHDKeypath("m/0'/0'", keypath));
134 BOOST_CHECK(!ParseHDKeypath("m/'0/0'", keypath));
135
136 BOOST_CHECK(ParseHDKeypath("m/0/0", keypath));
137 BOOST_CHECK(!ParseHDKeypath("n/0/0", keypath));
138
139 BOOST_CHECK(ParseHDKeypath("m/0/0/00", keypath));
140 BOOST_CHECK(!ParseHDKeypath("m/0/0/f00", keypath));
141
142 BOOST_CHECK(ParseHDKeypath("m/0/0/000000000000000000000000000000000000000000000000000000000000000000000000000000000000", keypath));
143 BOOST_CHECK(!ParseHDKeypath("m/1/1/111111111111111111111111111111111111111111111111111111111111111111111111111111111111", keypath));
144
145 BOOST_CHECK(ParseHDKeypath("m/0/00/0", keypath));
146 BOOST_CHECK(!ParseHDKeypath("m/0'/00/'0", keypath));
147
148 BOOST_CHECK(ParseHDKeypath("m/1/", keypath));
149 BOOST_CHECK(!ParseHDKeypath("m/1//", keypath));
150
151 BOOST_CHECK(ParseHDKeypath("m/0/4294967295", keypath)); // 4294967295 == 0xFFFFFFFF (uint32_t max)
152 BOOST_CHECK(!ParseHDKeypath("m/0/4294967296", keypath)); // 4294967296 == 0xFFFFFFFF (uint32_t max) + 1
153
154 BOOST_CHECK(ParseHDKeypath("m/4294967295", keypath)); // 4294967295 == 0xFFFFFFFF (uint32_t max)
155 BOOST_CHECK(!ParseHDKeypath("m/4294967296", keypath)); // 4294967296 == 0xFFFFFFFF (uint32_t max) + 1
156}
157
159} // namespace wallet
bool ParseHDKeypath(const std::string &keypath_str, std::vector< uint32_t > &keypath)
Parse an HD keypaths like "m/7/0'/2000".
Definition: bip32.cpp:13
#define Assert(val)
Identity function.
Definition: check.h:106
Double ended buffer combining vector and stream-like interfaces.
Definition: streams.h:147
A CWallet maintains a set of transactions and balances, and provides the ability to create new transa...
Definition: wallet.h:300
Descriptor with some wallet metadata.
Definition: walletutil.h:85
static UniValue Parse(std::string_view raw)
Parse string to UniValue or throw runtime_error if string contains invalid JSON.
Definition: client.cpp:317
BOOST_FIXTURE_TEST_SUITE(cuckoocache_tests, BasicTestingSetup)
Test Suite for CuckooCache.
BOOST_AUTO_TEST_SUITE_END()
std::string HexStr(const std::span< const uint8_t > s)
Convert a span of bytes to a lower-case hexadecimal string.
Definition: hex_base.cpp:29
@ SIGHASH_ALL
Definition: interpreter.h:30
""_hex is a compile-time user-defined literal returning a std::array<std::byte>, equivalent to ParseH...
Definition: strencodings.h:427
BOOST_AUTO_TEST_CASE(bnb_test)
@ WALLET_FLAG_DESCRIPTORS
Indicate that this wallet supports DescriptorScriptPubKeyMan.
Definition: walletutil.h:74
static void import_descriptor(CWallet &wallet, const std::string &descriptor) EXCLUSIVE_LOCKS_REQUIRED(wallet.cs_wallet)
is a home for public enum and struct type definitions that are used internally by node code,...
#define BOOST_CHECK_EQUAL(v1, v2)
Definition: object.cpp:18
#define BOOST_CHECK(expr)
Definition: object.cpp:17
static constexpr TransactionSerParams TX_WITH_WITNESS
Definition: transaction.h:195
std::shared_ptr< const CTransaction > CTransactionRef
Definition: transaction.h:423
A version of CTransaction with the PSBT format.
Definition: psbt.h:1111
std::optional< CMutableTransaction > tx
Definition: psbt.h:1112
State of transaction not confirmed or conflicting with a known block and not in the mempool.
Definition: transaction.h:58
#define LOCK(cs)
Definition: sync.h:257
#define EXCLUSIVE_LOCKS_REQUIRED(...)
Definition: threadsafety.h:49
AssertLockHeld(pool.cs)
assert(!tx.IsCoinBase())
std::shared_ptr< CWallet > m_wallet
Definition: interfaces.cpp:545