Branch data Line data Source code
1 : : // Copyright (c) 2016-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 <bitcoin-build-config.h> // IWYU pragma: keep
6 : :
7 : : #include <wallet/wallettool.h>
8 : :
9 : : #include <common/args.h>
10 : : #include <util/fs.h>
11 : : #include <util/translation.h>
12 : : #include <wallet/dump.h>
13 : : #include <wallet/salvage.h>
14 : : #include <wallet/wallet.h>
15 : : #include <wallet/walletutil.h>
16 : :
17 : : namespace wallet {
18 : : namespace WalletTool {
19 : :
20 : : // The standard wallet deleter function blocks on the validation interface
21 : : // queue, which doesn't exist for the bitcoin-wallet. Define our own
22 : : // deleter here.
23 : 4 : static void WalletToolReleaseWallet(CWallet* wallet)
24 : : {
25 : 4 : wallet->WalletLogPrintf("Releasing wallet\n");
26 : 4 : wallet->Close();
27 [ + - ]: 4 : delete wallet;
28 : 4 : }
29 : :
30 : 1 : static void WalletCreate(CWallet* wallet_instance, uint64_t wallet_creation_flags)
31 : : {
32 : 1 : LOCK(wallet_instance->cs_wallet);
33 : :
34 [ + - ]: 1 : wallet_instance->SetMinVersion(FEATURE_LATEST);
35 [ + - ]: 1 : wallet_instance->InitWalletFlags(wallet_creation_flags);
36 : :
37 [ + - - + ]: 1 : if (!wallet_instance->IsWalletFlagSet(WALLET_FLAG_DESCRIPTORS)) {
38 [ # # ]: 0 : auto spk_man = wallet_instance->GetOrCreateLegacyScriptPubKeyMan();
39 [ # # ]: 0 : spk_man->SetupGeneration(false);
40 : : } else {
41 [ + - ]: 1 : wallet_instance->SetupDescriptorScriptPubKeyMans();
42 : : }
43 : :
44 [ + - ]: 1 : tfm::format(std::cout, "Topping up keypool...\n");
45 [ + - ]: 1 : wallet_instance->TopUpKeyPool();
46 : 1 : }
47 : :
48 : 6 : static std::shared_ptr<CWallet> MakeWallet(const std::string& name, const fs::path& path, DatabaseOptions options)
49 : : {
50 : 6 : DatabaseStatus status;
51 [ + - ]: 6 : bilingual_str error;
52 [ + - ]: 6 : std::unique_ptr<WalletDatabase> database = MakeDatabase(path, options, status, error);
53 [ + + ]: 6 : if (!database) {
54 [ + - ]: 2 : tfm::format(std::cerr, "%s\n", error.original);
55 : 2 : return nullptr;
56 : : }
57 : :
58 : : // dummy chain interface
59 [ + - + - : 8 : std::shared_ptr<CWallet> wallet_instance{new CWallet(/*chain=*/nullptr, name, std::move(database)), WalletToolReleaseWallet};
+ - - - ]
60 : 4 : DBErrors load_wallet_ret;
61 : 4 : try {
62 [ + - ]: 4 : load_wallet_ret = wallet_instance->LoadWallet();
63 [ - - ]: 0 : } catch (const std::runtime_error&) {
64 [ - - ]: 0 : tfm::format(std::cerr, "Error loading %s. Is wallet being used by another process?\n", name);
65 : 0 : return nullptr;
66 : 0 : }
67 : :
68 [ - - - - : 4 : if (load_wallet_ret != DBErrors::LOAD_OK) {
- - + ]
69 : : if (load_wallet_ret == DBErrors::CORRUPT) {
70 [ # # ]: 0 : tfm::format(std::cerr, "Error loading %s: Wallet corrupted", name);
71 : 0 : return nullptr;
72 : : } else if (load_wallet_ret == DBErrors::NONCRITICAL_ERROR) {
73 [ # # ]: 0 : tfm::format(std::cerr, "Error reading %s! All keys read correctly, but transaction data"
74 : : " or address book entries might be missing or incorrect.",
75 : : name);
76 : : } else if (load_wallet_ret == DBErrors::TOO_NEW) {
77 [ # # ]: 0 : tfm::format(std::cerr, "Error loading %s: Wallet requires newer version of %s",
78 : : name, CLIENT_NAME);
79 : 0 : return nullptr;
80 : : } else if (load_wallet_ret == DBErrors::NEED_REWRITE) {
81 [ # # ]: 0 : tfm::format(std::cerr, "Wallet needed to be rewritten: restart %s to complete", CLIENT_NAME);
82 : 0 : return nullptr;
83 : : } else if (load_wallet_ret == DBErrors::NEED_RESCAN) {
84 [ # # ]: 0 : tfm::format(std::cerr, "Error reading %s! Some transaction data might be missing or"
85 : : " incorrect. Wallet requires a rescan.",
86 : : name);
87 : : } else {
88 [ # # ]: 0 : tfm::format(std::cerr, "Error loading %s", name);
89 : 0 : return nullptr;
90 : : }
91 : : }
92 : :
93 [ + + + - ]: 4 : if (options.require_create) WalletCreate(wallet_instance.get(), options.create_flags);
94 : :
95 : 4 : return wallet_instance;
96 : 12 : }
97 : :
98 : 4 : static void WalletShowInfo(CWallet* wallet_instance)
99 : : {
100 : 4 : LOCK(wallet_instance->cs_wallet);
101 : :
102 [ + - ]: 4 : tfm::format(std::cout, "Wallet info\n===========\n");
103 [ + - ]: 4 : tfm::format(std::cout, "Name: %s\n", wallet_instance->GetName());
104 [ + - + - ]: 4 : tfm::format(std::cout, "Format: %s\n", wallet_instance->GetDatabase().Format());
105 [ + - - + : 4 : tfm::format(std::cout, "Descriptors: %s\n", wallet_instance->IsWalletFlagSet(WALLET_FLAG_DESCRIPTORS) ? "yes" : "no");
+ - ]
106 [ + - + - : 8 : tfm::format(std::cout, "Encrypted: %s\n", wallet_instance->IsCrypted() ? "yes" : "no");
+ - ]
107 [ + - - + : 4 : tfm::format(std::cout, "HD (hd seed available): %s\n", wallet_instance->IsHDEnabled() ? "yes" : "no");
+ - ]
108 [ + - + - ]: 4 : tfm::format(std::cout, "Keypool Size: %u\n", wallet_instance->GetKeyPoolSize());
109 [ + - ]: 4 : tfm::format(std::cout, "Transactions: %zu\n", wallet_instance->mapWallet.size());
110 [ + - ]: 4 : tfm::format(std::cout, "Address Book: %zu\n", wallet_instance->m_address_book.size());
111 : 4 : }
112 : :
113 : 27 : bool ExecuteWalletToolFunc(const ArgsManager& args, const std::string& command)
114 : : {
115 [ + - + + : 54 : if (args.IsArgSet("-format") && command != "createfromdump") {
+ - - + ]
116 : 0 : tfm::format(std::cerr, "The -format option can only be used with the \"createfromdump\" command.\n");
117 : 0 : return false;
118 : : }
119 [ + - + + : 54 : if (args.IsArgSet("-dumpfile") && command != "dump" && command != "createfromdump") {
+ + + - -
+ ]
120 : 0 : tfm::format(std::cerr, "The -dumpfile option can only be used with the \"dump\" and \"createfromdump\" commands.\n");
121 : 0 : return false;
122 : : }
123 [ + - + + : 54 : if (args.IsArgSet("-descriptors") && command != "create") {
- + + + ]
124 : 1 : tfm::format(std::cerr, "The -descriptors option can only be used with the 'create' command.\n");
125 : 1 : return false;
126 : : }
127 [ + - - + : 52 : if (args.IsArgSet("-legacy") && command != "create") {
- - - + ]
128 : 0 : tfm::format(std::cerr, "The -legacy option can only be used with the 'create' command.\n");
129 : 0 : return false;
130 : : }
131 [ + + + - : 28 : if (command == "create" && !args.IsArgSet("-wallet")) {
+ + + + ]
132 : 1 : tfm::format(std::cerr, "Wallet name must be provided when creating a new wallet.\n");
133 : 1 : return false;
134 : : }
135 [ + - + - ]: 50 : const std::string name = args.GetArg("-wallet", "");
136 [ + - + - : 50 : const fs::path path = fsbridge::AbsPathJoin(GetWalletDir(), fs::PathFromString(name));
+ - ]
137 : :
138 [ + + ]: 25 : if (command == "create") {
139 [ + - ]: 1 : DatabaseOptions options;
140 [ + - ]: 1 : ReadDatabaseArgs(args, options);
141 : 1 : options.require_create = true;
142 : : // If -legacy is set, use it. Otherwise default to false.
143 [ + - + - ]: 1 : bool make_legacy = args.GetBoolArg("-legacy", false);
144 : : // If neither -legacy nor -descriptors is set, default to true. If -descriptors is set, use its value.
145 [ + - + - : 2 : bool make_descriptors = (!args.IsArgSet("-descriptors") && !args.IsArgSet("-legacy")) || (args.IsArgSet("-descriptors") && args.GetBoolArg("-descriptors", true));
+ - + - +
- - + - -
- - - - -
- - - - -
- + + - -
- - - -
- ]
146 [ - + ]: 1 : if (make_legacy && make_descriptors) {
147 [ # # ]: 0 : tfm::format(std::cerr, "Only one of -legacy or -descriptors can be set to true, not both\n");
148 : : return false;
149 : : }
150 [ - + ]: 1 : if (!make_legacy && !make_descriptors) {
151 [ # # ]: 0 : tfm::format(std::cerr, "One of -legacy or -descriptors must be set to true (or omitted)\n");
152 : : return false;
153 : : }
154 [ + - ]: 1 : if (make_descriptors) {
155 : 1 : options.create_flags |= WALLET_FLAG_DESCRIPTORS;
156 : 1 : options.require_format = DatabaseFormat::SQLITE;
157 : : }
158 : :
159 [ + - + - ]: 1 : const std::shared_ptr<CWallet> wallet_instance = MakeWallet(name, path, options);
160 [ + - ]: 1 : if (wallet_instance) {
161 [ + - ]: 1 : WalletShowInfo(wallet_instance.get());
162 [ + - ]: 1 : wallet_instance->Close();
163 : : }
164 [ + + ]: 25 : } else if (command == "info") {
165 [ + - ]: 5 : DatabaseOptions options;
166 [ + - ]: 5 : ReadDatabaseArgs(args, options);
167 : 5 : options.require_existing = true;
168 [ + - + - ]: 5 : const std::shared_ptr<CWallet> wallet_instance = MakeWallet(name, path, options);
169 [ + + - + ]: 5 : if (!wallet_instance) return false;
170 [ + - ]: 3 : WalletShowInfo(wallet_instance.get());
171 [ + - ]: 3 : wallet_instance->Close();
172 [ - + ]: 24 : } else if (command == "salvage") {
173 : : #ifdef USE_BDB
174 : : bilingual_str error;
175 : : std::vector<bilingual_str> warnings;
176 : : bool ret = RecoverDatabaseFile(args, path, error, warnings);
177 : : if (!ret) {
178 : : for (const auto& warning : warnings) {
179 : : tfm::format(std::cerr, "%s\n", warning.original);
180 : : }
181 : : if (!error.empty()) {
182 : : tfm::format(std::cerr, "%s\n", error.original);
183 : : }
184 : : }
185 : : return ret;
186 : : #else
187 [ # # ]: 0 : tfm::format(std::cerr, "Salvage command is not available as BDB support is not compiled");
188 : : return false;
189 : : #endif
190 [ + + ]: 19 : } else if (command == "dump") {
191 [ + - + - ]: 12 : DatabaseOptions options;
192 [ + - ]: 6 : ReadDatabaseArgs(args, options);
193 : 6 : options.require_existing = true;
194 : 6 : DatabaseStatus status;
195 : :
196 [ + - + - : 12 : if (args.GetBoolArg("-withinternalbdb", false) && IsBDBFile(BDBDataFile(path))) {
- + - - -
- - - - +
- - ]
197 : 0 : options.require_format = DatabaseFormat::BERKELEY_RO;
198 : : }
199 : :
200 [ + - ]: 6 : bilingual_str error;
201 [ + - ]: 6 : std::unique_ptr<WalletDatabase> database = MakeDatabase(path, options, status, error);
202 [ - + ]: 6 : if (!database) {
203 [ # # ]: 0 : tfm::format(std::cerr, "%s\n", error.original);
204 : : return false;
205 : : }
206 : :
207 [ + - ]: 6 : bool ret = DumpWallet(args, *database, error);
208 [ + + - + ]: 6 : if (!ret && !error.empty()) {
209 [ + - ]: 2 : tfm::format(std::cerr, "%s\n", error.original);
210 : : return ret;
211 : : }
212 [ + - ]: 4 : tfm::format(std::cout, "The dumpfile may contain private keys. To ensure the safety of your Bitcoin, do not share the dumpfile.\n");
213 : : return ret;
214 [ + - ]: 25 : } else if (command == "createfromdump") {
215 [ + - ]: 13 : bilingual_str error;
216 : 13 : std::vector<bilingual_str> warnings;
217 [ + - ]: 13 : bool ret = CreateFromDump(args, name, path, error, warnings);
218 [ - + ]: 13 : for (const auto& warning : warnings) {
219 [ # # ]: 0 : tfm::format(std::cout, "%s\n", warning.original);
220 : : }
221 [ + + + - ]: 13 : if (!ret && !error.empty()) {
222 [ + - ]: 11 : tfm::format(std::cerr, "%s\n", error.original);
223 : : }
224 : 13 : return ret;
225 : 26 : } else {
226 [ # # ]: 0 : tfm::format(std::cerr, "Invalid command: %s\n", command);
227 : : return false;
228 : : }
229 : :
230 : : return true;
231 : 25 : }
232 : : } // namespace WalletTool
233 : : } // namespace wallet
|
