Bitcoin Core  0.19.99
P2P Digital Currency
psbt_wallet_tests.cpp
Go to the documentation of this file.
1 // Copyright (c) 2017-2019 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 <util/bip32.h>
7 #include <util/strencodings.h>
8 #include <wallet/psbtwallet.h>
9 #include <wallet/wallet.h>
10 
11 #include <boost/test/unit_test.hpp>
12 #include <test/util/setup_common.h>
14 
16 
17 BOOST_AUTO_TEST_CASE(psbt_updater_test)
18 {
19  auto spk_man = m_wallet.GetOrCreateLegacyScriptPubKeyMan();
20  LOCK2(m_wallet.cs_wallet, spk_man->cs_KeyStore);
21 
22  // Create prevtxs and add to wallet
23  CDataStream s_prev_tx1(ParseHex("0200000000010158e87a21b56daf0c23be8e7070456c336f7cbaa5c8757924f545887bb2abdd7501000000171600145f275f436b09a8cc9a2eb2a2f528485c68a56323feffffff02d8231f1b0100000017a914aed962d6654f9a2b36608eb9d64d2b260db4f1118700c2eb0b0000000017a914b7f5faf40e3d40a5a459b1db3535f2b72fa921e88702483045022100a22edcc6e5bc511af4cc4ae0de0fcd75c7e04d8c1c3a8aa9d820ed4b967384ec02200642963597b9b1bc22c75e9f3e117284a962188bf5e8a74c895089046a20ad770121035509a48eb623e10aace8bfd0212fdb8a8e5af3c94b0b133b95e114cab89e4f7965000000"), SER_NETWORK, PROTOCOL_VERSION);
24  CTransactionRef prev_tx1;
25  s_prev_tx1 >> prev_tx1;
26  CWalletTx prev_wtx1(&m_wallet, prev_tx1);
27  m_wallet.mapWallet.emplace(prev_wtx1.GetHash(), std::move(prev_wtx1));
28 
29  CDataStream s_prev_tx2(ParseHex("0200000001aad73931018bd25f84ae400b68848be09db706eac2ac18298babee71ab656f8b0000000048473044022058f6fc7c6a33e1b31548d481c826c015bd30135aad42cd67790dab66d2ad243b02204a1ced2604c6735b6393e5b41691dd78b00f0c5942fb9f751856faa938157dba01feffffff0280f0fa020000000017a9140fb9463421696b82c833af241c78c17ddbde493487d0f20a270100000017a91429ca74f8a08f81999428185c97b5d852e4063f618765000000"), SER_NETWORK, PROTOCOL_VERSION);
30  CTransactionRef prev_tx2;
31  s_prev_tx2 >> prev_tx2;
32  CWalletTx prev_wtx2(&m_wallet, prev_tx2);
33  m_wallet.mapWallet.emplace(prev_wtx2.GetHash(), std::move(prev_wtx2));
34 
35  // Add scripts
36  CScript rs1;
37  CDataStream s_rs1(ParseHex("475221029583bf39ae0a609747ad199addd634fa6108559d6c5cd39b4c2183f1ab96e07f2102dab61ff49a14db6a7d02b0cd1fbb78fc4b18312b5b4e54dae4dba2fbfef536d752ae"), SER_NETWORK, PROTOCOL_VERSION);
38  s_rs1 >> rs1;
39  spk_man->AddCScript(rs1);
40 
41  CScript rs2;
42  CDataStream s_rs2(ParseHex("2200208c2353173743b595dfb4a07b72ba8e42e3797da74e87fe7d9d7497e3b2028903"), SER_NETWORK, PROTOCOL_VERSION);
43  s_rs2 >> rs2;
44  spk_man->AddCScript(rs2);
45 
46  CScript ws1;
47  CDataStream s_ws1(ParseHex("47522103089dc10c7ac6db54f91329af617333db388cead0c231f723379d1b99030b02dc21023add904f3d6dcf59ddb906b0dee23529b7ffb9ed50e5e86151926860221f0e7352ae"), SER_NETWORK, PROTOCOL_VERSION);
48  s_ws1 >> ws1;
49  spk_man->AddCScript(ws1);
50 
51  // Add hd seed
52  CKey key = DecodeSecret("5KSSJQ7UNfFGwVgpCZDSHm5rVNhMFcFtvWM3zQ8mW4qNDEN7LFd"); // Mainnet and uncompressed form of cUkG8i1RFfWGWy5ziR11zJ5V4U4W3viSFCfyJmZnvQaUsd1xuF3T
53  CPubKey master_pub_key = spk_man->DeriveNewSeed(key);
54  spk_man->SetHDSeed(master_pub_key);
55  spk_man->NewKeyPool();
56 
57  // Call FillPSBT
59  CDataStream ssData(ParseHex("70736274ff01009a020000000258e87a21b56daf0c23be8e7070456c336f7cbaa5c8757924f545887bb2abdd750000000000ffffffff838d0427d0ec650a68aa46bb0b098aea4422c071b2ca78352a077959d07cea1d0100000000ffffffff0270aaf00800000000160014d85c2b71d0060b09c9886aeb815e50991dda124d00e1f5050000000016001400aea9a2e5f0f876a588df5546e8742d1d87008f000000000000000000"), SER_NETWORK, PROTOCOL_VERSION);
60  ssData >> psbtx;
61 
62  // Fill transaction with our data
63  bool complete = true;
64  BOOST_REQUIRE_EQUAL(TransactionError::OK, FillPSBT(&m_wallet, psbtx, complete, SIGHASH_ALL, false, true));
65 
66  // Get the final tx
68  ssTx << psbtx;
69  std::string final_hex = HexStr(ssTx.begin(), ssTx.end());
70  BOOST_CHECK_EQUAL(final_hex, "70736274ff01009a020000000258e87a21b56daf0c23be8e7070456c336f7cbaa5c8757924f545887bb2abdd750000000000ffffffff838d0427d0ec650a68aa46bb0b098aea4422c071b2ca78352a077959d07cea1d0100000000ffffffff0270aaf00800000000160014d85c2b71d0060b09c9886aeb815e50991dda124d00e1f5050000000016001400aea9a2e5f0f876a588df5546e8742d1d87008f00000000000100bb0200000001aad73931018bd25f84ae400b68848be09db706eac2ac18298babee71ab656f8b0000000048473044022058f6fc7c6a33e1b31548d481c826c015bd30135aad42cd67790dab66d2ad243b02204a1ced2604c6735b6393e5b41691dd78b00f0c5942fb9f751856faa938157dba01feffffff0280f0fa020000000017a9140fb9463421696b82c833af241c78c17ddbde493487d0f20a270100000017a91429ca74f8a08f81999428185c97b5d852e4063f6187650000000104475221029583bf39ae0a609747ad199addd634fa6108559d6c5cd39b4c2183f1ab96e07f2102dab61ff49a14db6a7d02b0cd1fbb78fc4b18312b5b4e54dae4dba2fbfef536d752ae2206029583bf39ae0a609747ad199addd634fa6108559d6c5cd39b4c2183f1ab96e07f10d90c6a4f000000800000008000000080220602dab61ff49a14db6a7d02b0cd1fbb78fc4b18312b5b4e54dae4dba2fbfef536d710d90c6a4f0000008000000080010000800001012000c2eb0b0000000017a914b7f5faf40e3d40a5a459b1db3535f2b72fa921e88701042200208c2353173743b595dfb4a07b72ba8e42e3797da74e87fe7d9d7497e3b2028903010547522103089dc10c7ac6db54f91329af617333db388cead0c231f723379d1b99030b02dc21023add904f3d6dcf59ddb906b0dee23529b7ffb9ed50e5e86151926860221f0e7352ae2206023add904f3d6dcf59ddb906b0dee23529b7ffb9ed50e5e86151926860221f0e7310d90c6a4f000000800000008003000080220603089dc10c7ac6db54f91329af617333db388cead0c231f723379d1b99030b02dc10d90c6a4f00000080000000800200008000220203a9a4c37f5996d3aa25dbac6b570af0650394492942460b354753ed9eeca5877110d90c6a4f000000800000008004000080002202027f6399757d2eff55a136ad02c684b1838b6556e5f1b6b34282a94b6b5005109610d90c6a4f00000080000000800500008000");
71 
72  // Mutate the transaction so that one of the inputs is invalid
73  psbtx.tx->vin[0].prevout.n = 2;
74 
75  // Try to sign the mutated input
76  SignatureData sigdata;
77  psbtx.inputs[0].FillSignatureData(sigdata);
78  const std::unique_ptr<SigningProvider> provider = m_wallet.GetSigningProvider(ws1, sigdata);
79  BOOST_CHECK(!SignPSBTInput(*provider, psbtx, 0, SIGHASH_ALL));
80 }
81 
82 BOOST_AUTO_TEST_CASE(parse_hd_keypath)
83 {
84  std::vector<uint32_t> keypath;
85 
86  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));
87  BOOST_CHECK(!ParseHDKeypath("///////////////////////////", keypath));
88 
89  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));
90  BOOST_CHECK(!ParseHDKeypath("//////////////////////////'/", keypath));
91 
92  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));
93  BOOST_CHECK(!ParseHDKeypath("1///////////////////////////", keypath));
94 
95  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));
96  BOOST_CHECK(!ParseHDKeypath("1/'//////////////////////////", keypath));
97 
98  BOOST_CHECK(ParseHDKeypath("", keypath));
99  BOOST_CHECK(!ParseHDKeypath(" ", keypath));
100 
101  BOOST_CHECK(ParseHDKeypath("0", keypath));
102  BOOST_CHECK(!ParseHDKeypath("O", keypath));
103 
104  BOOST_CHECK(ParseHDKeypath("0000'/0000'/0000'", keypath));
105  BOOST_CHECK(!ParseHDKeypath("0000,/0000,/0000,", keypath));
106 
107  BOOST_CHECK(ParseHDKeypath("01234", keypath));
108  BOOST_CHECK(!ParseHDKeypath("0x1234", keypath));
109 
110  BOOST_CHECK(ParseHDKeypath("1", keypath));
111  BOOST_CHECK(!ParseHDKeypath(" 1", keypath));
112 
113  BOOST_CHECK(ParseHDKeypath("42", keypath));
114  BOOST_CHECK(!ParseHDKeypath("m42", keypath));
115 
116  BOOST_CHECK(ParseHDKeypath("4294967295", keypath)); // 4294967295 == 0xFFFFFFFF (uint32_t max)
117  BOOST_CHECK(!ParseHDKeypath("4294967296", keypath)); // 4294967296 == 0xFFFFFFFF (uint32_t max) + 1
118 
119  BOOST_CHECK(ParseHDKeypath("m", keypath));
120  BOOST_CHECK(!ParseHDKeypath("n", keypath));
121 
122  BOOST_CHECK(ParseHDKeypath("m/", keypath));
123  BOOST_CHECK(!ParseHDKeypath("n/", keypath));
124 
125  BOOST_CHECK(ParseHDKeypath("m/0", keypath));
126  BOOST_CHECK(!ParseHDKeypath("n/0", keypath));
127 
128  BOOST_CHECK(ParseHDKeypath("m/0'", keypath));
129  BOOST_CHECK(!ParseHDKeypath("m/0''", keypath));
130 
131  BOOST_CHECK(ParseHDKeypath("m/0'/0'", keypath));
132  BOOST_CHECK(!ParseHDKeypath("m/'0/0'", keypath));
133 
134  BOOST_CHECK(ParseHDKeypath("m/0/0", keypath));
135  BOOST_CHECK(!ParseHDKeypath("n/0/0", keypath));
136 
137  BOOST_CHECK(ParseHDKeypath("m/0/0/00", keypath));
138  BOOST_CHECK(!ParseHDKeypath("m/0/0/f00", keypath));
139 
140  BOOST_CHECK(ParseHDKeypath("m/0/0/000000000000000000000000000000000000000000000000000000000000000000000000000000000000", keypath));
141  BOOST_CHECK(!ParseHDKeypath("m/1/1/111111111111111111111111111111111111111111111111111111111111111111111111111111111111", keypath));
142 
143  BOOST_CHECK(ParseHDKeypath("m/0/00/0", keypath));
144  BOOST_CHECK(!ParseHDKeypath("m/0'/00/'0", keypath));
145 
146  BOOST_CHECK(ParseHDKeypath("m/1/", keypath));
147  BOOST_CHECK(!ParseHDKeypath("m/1//", keypath));
148 
149  BOOST_CHECK(ParseHDKeypath("m/0/4294967295", keypath)); // 4294967295 == 0xFFFFFFFF (uint32_t max)
150  BOOST_CHECK(!ParseHDKeypath("m/0/4294967296", keypath)); // 4294967296 == 0xFFFFFFFF (uint32_t max) + 1
151 
152  BOOST_CHECK(ParseHDKeypath("m/4294967295", keypath)); // 4294967295 == 0xFFFFFFFF (uint32_t max)
153  BOOST_CHECK(!ParseHDKeypath("m/4294967296", keypath)); // 4294967296 == 0xFFFFFFFF (uint32_t max) + 1
154 }
155 
std::shared_ptr< const CTransaction > CTransactionRef
Definition: transaction.h:408
std::shared_ptr< CWallet > m_wallet
Definition: wallet.cpp:512
std::vector< unsigned char > ParseHex(const char *psz)
bool SignPSBTInput(const SigningProvider &provider, PartiallySignedTransaction &psbt, int index, int sighash, SignatureData *out_sigdata, bool use_dummy)
Signs a PSBTInput, verifying that all provided data matches what is being signed. ...
Definition: psbt.cpp:236
A version of CTransaction with the PSBT format.
Definition: psbt.h:388
Double ended buffer combining vector and stream-like interfaces.
Definition: streams.h:201
TransactionError FillPSBT(const CWallet *pwallet, PartiallySignedTransaction &psbtx, bool &complete, int sighash_type, bool sign, bool bip32derivs)
Fills out a PSBT with information from the wallet.
Definition: psbtwallet.cpp:7
#define LOCK2(cs1, cs2)
Definition: sync.h:180
BOOST_AUTO_TEST_CASE(psbt_updater_test)
An encapsulated public key.
Definition: pubkey.h:30
const uint256 & GetHash() const
Definition: wallet.h:539
A transaction with a bunch of additional info that only the owner cares about.
Definition: wallet.h:253
const_iterator end() const
Definition: streams.h:290
const_iterator begin() const
Definition: streams.h:288
#define BOOST_FIXTURE_TEST_SUITE(a, b)
Definition: object.cpp:14
Testing setup and teardown for wallet.
bool ParseHDKeypath(const std::string &keypath_str, std::vector< uint32_t > &keypath)
Parse an HD keypaths like "m/7/0&#39;/2000".
Definition: bip32.cpp:12
#define BOOST_CHECK_EQUAL(v1, v2)
Definition: object.cpp:18
Serialized script, used inside transaction inputs and outputs.
Definition: script.h:390
static const int PROTOCOL_VERSION
network protocol versioning
Definition: version.h:12
#define BOOST_AUTO_TEST_SUITE_END()
Definition: object.cpp:16
std::string HexStr(const T itbegin, const T itend)
Definition: strencodings.h:125
An encapsulated private key.
Definition: key.h:27
CKey DecodeSecret(const std::string &str)
Definition: key_io.cpp:132
#define BOOST_CHECK(expr)
Definition: object.cpp:17