Branch data Line data Source code
1 : : // Copyright (c) 2022-present The Bitcoin Core developers
2 : : // Distributed under the MIT software license, see the accompanying
3 : : // file COPYING or https://www.opensource.org/licenses/mit-license.php.
4 : :
5 : : #include <wallet/test/util.h>
6 : : #include <wallet/wallet.h>
7 : : #include <test/util/logging.h>
8 : : #include <test/util/setup_common.h>
9 : :
10 : : #include <boost/test/unit_test.hpp>
11 : :
12 : : namespace wallet {
13 : :
14 : : BOOST_AUTO_TEST_SUITE(walletload_tests)
15 : :
16 : : class DummyDescriptor final : public Descriptor {
17 : : private:
18 : : std::string desc;
19 : : public:
20 [ - + ]: 4 : explicit DummyDescriptor(const std::string& descriptor) : desc(descriptor) {};
21 : 2 : ~DummyDescriptor() = default;
22 : :
23 [ - + ]: 4 : std::string ToString(bool compat_format) const override { return desc; }
24 : 0 : std::optional<OutputType> GetOutputType() const override { return OutputType::UNKNOWN; }
25 : :
26 : 0 : bool IsRange() const override { return false; }
27 : 0 : bool IsSolvable() const override { return false; }
28 : 0 : bool IsSingleType() const override { return true; }
29 : 0 : bool ToPrivateString(const SigningProvider& provider, std::string& out) const override { return false; }
30 : 0 : bool ToNormalizedString(const SigningProvider& provider, std::string& out, const DescriptorCache* cache = nullptr) const override { return false; }
31 : 0 : bool Expand(int pos, const SigningProvider& provider, std::vector<CScript>& output_scripts, FlatSigningProvider& out, DescriptorCache* write_cache = nullptr) const override { return false; };
32 : 0 : bool ExpandFromCache(int pos, const DescriptorCache& read_cache, std::vector<CScript>& output_scripts, FlatSigningProvider& out) const override { return false; }
33 : 0 : void ExpandPrivate(int pos, const SigningProvider& provider, FlatSigningProvider& out) const override {}
34 : 0 : std::optional<int64_t> ScriptSize() const override { return {}; }
35 : 0 : std::optional<int64_t> MaxSatisfactionWeight(bool) const override { return {}; }
36 : 0 : std::optional<int64_t> MaxSatisfactionElems() const override { return {}; }
37 : 0 : void GetPubKeys(std::set<CPubKey>& pubkeys, std::set<CExtPubKey>& ext_pubs) const override {}
38 : 0 : std::vector<std::string> Warnings() const override { return {}; }
39 : : };
40 : :
41 [ + - + - : 7 : BOOST_FIXTURE_TEST_CASE(wallet_load_descriptors, TestingSetup)
+ - + - -
+ + - + -
+ - + - +
- + - - +
+ - + - +
- + - + -
+ - - + +
- + - + -
+ - + - +
- + - - +
+ - + - +
- + - + -
+ - - + +
- ]
42 : : {
43 [ + - ]: 1 : std::unique_ptr<WalletDatabase> database = CreateMockableWalletDatabase();
44 : 1 : {
45 : : // Write unknown active descriptor
46 [ + - ]: 1 : WalletBatch batch(*database);
47 [ + - ]: 1 : std::string unknown_desc = "trx(tpubD6NzVbkrYhZ4Y4S7m6Y5s9GD8FqEMBy56AGphZXuagajudVZEnYyBahZMgHNCTJc2at82YX6s8JiL1Lohu5A3v1Ur76qguNH4QVQ7qYrBQx/86'/1'/0'/0/*)#8pn8tzdt";
48 [ + - + - : 2 : WalletDescriptor wallet_descriptor(std::make_shared<DummyDescriptor>(unknown_desc), 0, 0, 0, 0);
- + ]
49 [ + - + - : 2 : BOOST_CHECK(batch.WriteDescriptor(uint256(), wallet_descriptor));
+ - + - ]
50 [ + - + - : 2 : BOOST_CHECK(batch.WriteActiveScriptPubKeyMan(static_cast<uint8_t>(OutputType::UNKNOWN), uint256(), false));
+ - ]
51 : 1 : }
52 : :
53 : 1 : {
54 : : // Now try to load the wallet and verify the error.
55 [ + - + - : 2 : const std::shared_ptr<CWallet> wallet(new CWallet(m_node.chain.get(), "", std::move(database)));
+ - + - -
- ]
56 [ + - + - : 1 : BOOST_CHECK_EQUAL(wallet->LoadWallet(), DBErrors::UNKNOWN_DESCRIPTOR);
+ - + - ]
57 : 0 : }
58 : :
59 : : // Test 2
60 : : // Now write a valid descriptor with an invalid ID.
61 : : // As the software produces another ID for the descriptor, the loading process must be aborted.
62 [ + - ]: 2 : database = CreateMockableWalletDatabase();
63 : :
64 : : // Verify the error
65 : 1 : bool found = false;
66 [ + - ]: 4 : DebugLogHelper logHelper("The descriptor ID calculated by the wallet differs from the one in DB", [&](const std::string* s) {
67 : 2 : found = true;
68 : 2 : return false;
69 [ + - ]: 1 : });
70 : :
71 : 1 : {
72 : : // Write valid descriptor with invalid ID
73 [ + - ]: 1 : WalletBatch batch(*database);
74 [ + - ]: 1 : std::string desc = "wpkh([d34db33f/84h/0h/0h]xpub6DJ2dNUysrn5Vt36jH2KLBT2i1auw1tTSSomg8PhqNiUtx8QX2SvC9nrHu81fT41fvDUnhMjEzQgXnQjKEu3oaqMSzhSrHMxyyoEAmUHQbY/0/*)#cjjspncu";
75 [ + - + - : 2 : WalletDescriptor wallet_descriptor(std::make_shared<DummyDescriptor>(desc), 0, 0, 0, 0);
- + ]
76 [ + - + - : 2 : BOOST_CHECK(batch.WriteDescriptor(uint256::ONE, wallet_descriptor));
+ - ]
77 : 1 : }
78 : :
79 : 1 : {
80 : : // Now try to load the wallet and verify the error.
81 [ + - + - : 2 : const std::shared_ptr<CWallet> wallet(new CWallet(m_node.chain.get(), "", std::move(database)));
+ - + - -
- ]
82 [ + - + - : 1 : BOOST_CHECK_EQUAL(wallet->LoadWallet(), DBErrors::CORRUPT);
+ - ]
83 [ + - + - : 2 : BOOST_CHECK(found); // The error must be logged
+ - + - ]
84 : 0 : }
85 : 1 : }
86 : :
87 : : BOOST_AUTO_TEST_SUITE_END()
88 : : } // namespace wallet
|
