Line data Source code
1 : // Copyright (c) 2017-2021 The Bitcoin Core developers
2 : // Distributed under the MIT software license, see the accompanying
3 : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
4 :
5 : #include <key_io.h>
6 : #include <util/bip32.h>
7 : #include <util/strencodings.h>
8 : #include <wallet/wallet.h>
9 :
10 : #include <boost/test/unit_test.hpp>
11 : #include <wallet/test/wallet_test_fixture.h>
12 :
13 : namespace wallet {
14 146 : BOOST_FIXTURE_TEST_SUITE(psbt_wallet_tests, WalletTestingSetup)
15 :
16 3 : static void import_descriptor(CWallet& wallet, const std::string& descriptor)
17 : {
18 3 : LOCK(wallet.cs_wallet);
19 3 : FlatSigningProvider provider;
20 3 : std::string error;
21 3 : std::unique_ptr<Descriptor> desc = Parse(descriptor, provider, error, /* require_checksum=*/ false);
22 3 : assert(desc);
23 3 : WalletDescriptor w_desc(std::move(desc), 0, 0, 10, 0);
24 3 : wallet.AddWalletDescriptor(w_desc, provider, "", false);
25 3 : }
26 :
27 148 : BOOST_AUTO_TEST_CASE(psbt_updater_test)
28 : {
29 1 : LOCK(m_wallet.cs_wallet);
30 1 : m_wallet.SetWalletFlag(WALLET_FLAG_DESCRIPTORS);
31 :
32 : // Create prevtxs and add to wallet
33 1 : CDataStream s_prev_tx1(ParseHex("0200000000010158e87a21b56daf0c23be8e7070456c336f7cbaa5c8757924f545887bb2abdd7501000000171600145f275f436b09a8cc9a2eb2a2f528485c68a56323feffffff02d8231f1b0100000017a914aed962d6654f9a2b36608eb9d64d2b260db4f1118700c2eb0b0000000017a914b7f5faf40e3d40a5a459b1db3535f2b72fa921e88702483045022100a22edcc6e5bc511af4cc4ae0de0fcd75c7e04d8c1c3a8aa9d820ed4b967384ec02200642963597b9b1bc22c75e9f3e117284a962188bf5e8a74c895089046a20ad770121035509a48eb623e10aace8bfd0212fdb8a8e5af3c94b0b133b95e114cab89e4f7965000000"), SER_NETWORK, PROTOCOL_VERSION);
34 1 : CTransactionRef prev_tx1;
35 1 : s_prev_tx1 >> prev_tx1;
36 1 : m_wallet.mapWallet.emplace(std::piecewise_construct, std::forward_as_tuple(prev_tx1->GetHash()), std::forward_as_tuple(prev_tx1, TxStateInactive{}));
37 :
38 1 : CDataStream s_prev_tx2(ParseHex("0200000001aad73931018bd25f84ae400b68848be09db706eac2ac18298babee71ab656f8b0000000048473044022058f6fc7c6a33e1b31548d481c826c015bd30135aad42cd67790dab66d2ad243b02204a1ced2604c6735b6393e5b41691dd78b00f0c5942fb9f751856faa938157dba01feffffff0280f0fa020000000017a9140fb9463421696b82c833af241c78c17ddbde493487d0f20a270100000017a91429ca74f8a08f81999428185c97b5d852e4063f618765000000"), SER_NETWORK, PROTOCOL_VERSION);
39 1 : CTransactionRef prev_tx2;
40 1 : s_prev_tx2 >> prev_tx2;
41 1 : m_wallet.mapWallet.emplace(std::piecewise_construct, std::forward_as_tuple(prev_tx2->GetHash()), std::forward_as_tuple(prev_tx2, TxStateInactive{}));
42 :
43 : // Import descriptors for keys and scripts
44 1 : import_descriptor(m_wallet, "sh(multi(2,xprv9s21ZrQH143K2LE7W4Xf3jATf9jECxSb7wj91ZnmY4qEJrS66Qru9RFqq8xbkgT32ya6HqYJweFdJUEDf5Q6JFV7jMiUws7kQfe6Tv4RbfN/0h/0h/0h,xprv9s21ZrQH143K2LE7W4Xf3jATf9jECxSb7wj91ZnmY4qEJrS66Qru9RFqq8xbkgT32ya6HqYJweFdJUEDf5Q6JFV7jMiUws7kQfe6Tv4RbfN/0h/0h/1h))");
45 1 : import_descriptor(m_wallet, "sh(multi(2,xprv9s21ZrQH143K2LE7W4Xf3jATf9jECxSb7wj91ZnmY4qEJrS66Qru9RFqq8xbkgT32ya6HqYJweFdJUEDf5Q6JFV7jMiUws7kQfe6Tv4RbfN/0h/0h/2h,xprv9s21ZrQH143K2LE7W4Xf3jATf9jECxSb7wj91ZnmY4qEJrS66Qru9RFqq8xbkgT32ya6HqYJweFdJUEDf5Q6JFV7jMiUws7kQfe6Tv4RbfN/0h/0h/3h))");
46 1 : import_descriptor(m_wallet, "pkh(xprv9s21ZrQH143K2LE7W4Xf3jATf9jECxSb7wj91ZnmY4qEJrS66Qru9RFqq8xbkgT32ya6HqYJweFdJUEDf5Q6JFV7jMiUws7kQfe6Tv4RbfN/0h/0h/*h)");
47 :
48 : // Call FillPSBT
49 1 : PartiallySignedTransaction psbtx;
50 1 : CDataStream ssData(ParseHex("70736274ff01009a020000000258e87a21b56daf0c23be8e7070456c336f7cbaa5c8757924f545887bb2abdd750000000000ffffffff838d0427d0ec650a68aa46bb0b098aea4422c071b2ca78352a077959d07cea1d0100000000ffffffff0270aaf00800000000160014d85c2b71d0060b09c9886aeb815e50991dda124d00e1f5050000000016001400aea9a2e5f0f876a588df5546e8742d1d87008f000000000000000000"), SER_NETWORK, PROTOCOL_VERSION);
51 1 : ssData >> psbtx;
52 :
53 : // Use CTransaction for the constant parts of the
54 : // transaction to avoid rehashing.
55 1 : const CTransaction txConst(*psbtx.tx);
56 :
57 : // Fill transaction with our data
58 1 : bool complete = true;
59 1 : BOOST_REQUIRE_EQUAL(TransactionError::OK, m_wallet.FillPSBT(psbtx, complete, SIGHASH_ALL, false, true));
60 :
61 : // Get the final tx
62 1 : CDataStream ssTx(SER_NETWORK, PROTOCOL_VERSION);
63 1 : ssTx << psbtx;
64 1 : std::string final_hex = HexStr(ssTx);
65 :
66 1 : BOOST_CHECK_EQUAL(final_hex, "70736274ff01009a020000000258e87a21b56daf0c23be8e7070456c336f7cbaa5c8757924f545887bb2abdd750000000000ffffffff838d0427d0ec650a68aa46bb0b098aea4422c071b2ca78352a077959d07cea1d0100000000ffffffff0270aaf00800000000160014d85c2b71d0060b09c9886aeb815e50991dda124d00e1f5050000000016001400aea9a2e5f0f876a588df5546e8742d1d87008f00000000000100bb0200000001aad73931018bd25f84ae400b68848be09db706eac2ac18298babee71ab656f8b0000000048473044022058f6fc7c6a33e1b31548d481c826c015bd30135aad42cd67790dab66d2ad243b02204a1ced2604c6735b6393e5b41691dd78b00f0c5942fb9f751856faa938157dba01feffffff0280f0fa020000000017a9140fb9463421696b82c833af241c78c17ddbde493487d0f20a270100000017a91429ca74f8a08f81999428185c97b5d852e4063f6187650000000104475221029583bf39ae0a609747ad199addd634fa6108559d6c5cd39b4c2183f1ab96e07f2102dab61ff49a14db6a7d02b0cd1fbb78fc4b18312b5b4e54dae4dba2fbfef536d752ae2206029583bf39ae0a609747ad199addd634fa6108559d6c5cd39b4c2183f1ab96e07f10d90c6a4f000000800000008000000080220602dab61ff49a14db6a7d02b0cd1fbb78fc4b18312b5b4e54dae4dba2fbfef536d710d90c6a4f00000080000000800100008000000000");
67 :
68 : // Mutate the transaction so that one of the inputs is invalid
69 1 : psbtx.tx->vin[0].prevout.n = 2;
70 :
71 : // Try to sign the mutated input
72 1 : SignatureData sigdata;
73 1 : BOOST_CHECK(m_wallet.FillPSBT(psbtx, complete, SIGHASH_ALL, true, true) != TransactionError::OK);
74 1 : }
75 :
76 148 : BOOST_AUTO_TEST_CASE(parse_hd_keypath)
77 : {
78 1 : std::vector<uint32_t> keypath;
79 :
80 1 : 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));
81 1 : BOOST_CHECK(!ParseHDKeypath("///////////////////////////", keypath));
82 :
83 1 : 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));
84 1 : BOOST_CHECK(!ParseHDKeypath("//////////////////////////'/", keypath));
85 :
86 1 : 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));
87 1 : BOOST_CHECK(!ParseHDKeypath("1///////////////////////////", keypath));
88 :
89 1 : 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));
90 1 : BOOST_CHECK(!ParseHDKeypath("1/'//////////////////////////", keypath));
91 :
92 1 : BOOST_CHECK(ParseHDKeypath("", keypath));
93 1 : BOOST_CHECK(!ParseHDKeypath(" ", keypath));
94 :
95 1 : BOOST_CHECK(ParseHDKeypath("0", keypath));
96 1 : BOOST_CHECK(!ParseHDKeypath("O", keypath));
97 :
98 1 : BOOST_CHECK(ParseHDKeypath("0000'/0000'/0000'", keypath));
99 1 : BOOST_CHECK(!ParseHDKeypath("0000,/0000,/0000,", keypath));
100 :
101 1 : BOOST_CHECK(ParseHDKeypath("01234", keypath));
102 1 : BOOST_CHECK(!ParseHDKeypath("0x1234", keypath));
103 :
104 1 : BOOST_CHECK(ParseHDKeypath("1", keypath));
105 1 : BOOST_CHECK(!ParseHDKeypath(" 1", keypath));
106 :
107 1 : BOOST_CHECK(ParseHDKeypath("42", keypath));
108 1 : BOOST_CHECK(!ParseHDKeypath("m42", keypath));
109 :
110 1 : BOOST_CHECK(ParseHDKeypath("4294967295", keypath)); // 4294967295 == 0xFFFFFFFF (uint32_t max)
111 1 : BOOST_CHECK(!ParseHDKeypath("4294967296", keypath)); // 4294967296 == 0xFFFFFFFF (uint32_t max) + 1
112 :
113 1 : BOOST_CHECK(ParseHDKeypath("m", keypath));
114 1 : BOOST_CHECK(!ParseHDKeypath("n", keypath));
115 :
116 1 : BOOST_CHECK(ParseHDKeypath("m/", keypath));
117 1 : BOOST_CHECK(!ParseHDKeypath("n/", keypath));
118 :
119 1 : BOOST_CHECK(ParseHDKeypath("m/0", keypath));
120 1 : BOOST_CHECK(!ParseHDKeypath("n/0", keypath));
121 :
122 1 : BOOST_CHECK(ParseHDKeypath("m/0'", keypath));
123 1 : BOOST_CHECK(!ParseHDKeypath("m/0''", keypath));
124 :
125 1 : BOOST_CHECK(ParseHDKeypath("m/0'/0'", keypath));
126 1 : BOOST_CHECK(!ParseHDKeypath("m/'0/0'", keypath));
127 :
128 1 : BOOST_CHECK(ParseHDKeypath("m/0/0", keypath));
129 1 : BOOST_CHECK(!ParseHDKeypath("n/0/0", keypath));
130 :
131 1 : BOOST_CHECK(ParseHDKeypath("m/0/0/00", keypath));
132 1 : BOOST_CHECK(!ParseHDKeypath("m/0/0/f00", keypath));
133 :
134 1 : BOOST_CHECK(ParseHDKeypath("m/0/0/000000000000000000000000000000000000000000000000000000000000000000000000000000000000", keypath));
135 1 : BOOST_CHECK(!ParseHDKeypath("m/1/1/111111111111111111111111111111111111111111111111111111111111111111111111111111111111", keypath));
136 :
137 1 : BOOST_CHECK(ParseHDKeypath("m/0/00/0", keypath));
138 1 : BOOST_CHECK(!ParseHDKeypath("m/0'/00/'0", keypath));
139 :
140 1 : BOOST_CHECK(ParseHDKeypath("m/1/", keypath));
141 1 : BOOST_CHECK(!ParseHDKeypath("m/1//", keypath));
142 :
143 1 : BOOST_CHECK(ParseHDKeypath("m/0/4294967295", keypath)); // 4294967295 == 0xFFFFFFFF (uint32_t max)
144 1 : BOOST_CHECK(!ParseHDKeypath("m/0/4294967296", keypath)); // 4294967296 == 0xFFFFFFFF (uint32_t max) + 1
145 :
146 1 : BOOST_CHECK(ParseHDKeypath("m/4294967295", keypath)); // 4294967295 == 0xFFFFFFFF (uint32_t max)
147 1 : BOOST_CHECK(!ParseHDKeypath("m/4294967296", keypath)); // 4294967296 == 0xFFFFFFFF (uint32_t max) + 1
148 1 : }
149 :
150 146 : BOOST_AUTO_TEST_SUITE_END()
151 : } // namespace wallet
|
