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 <chainparams.h>
8 : : #include <chainparamsbase.h>
9 : : #include <clientversion.h>
10 : : #include <common/args.h>
11 : : #include <common/system.h>
12 : : #include <compat/compat.h>
13 : : #include <interfaces/init.h>
14 : : #include <key.h>
15 : : #include <logging.h>
16 : : #include <pubkey.h>
17 : : #include <tinyformat.h>
18 : : #include <util/exception.h>
19 : : #include <util/translation.h>
20 : : #include <wallet/wallettool.h>
21 : :
22 : : #include <exception>
23 : : #include <functional>
24 : : #include <string>
25 : : #include <tuple>
26 : :
27 : : using util::Join;
28 : :
29 : : const std::function<std::string(const char*)> G_TRANSLATION_FUN = nullptr;
30 : :
31 : 32 : static void SetupWalletToolArgs(ArgsManager& argsman)
32 : : {
33 : 32 : SetupHelpOptions(argsman);
34 : 32 : SetupChainParamsBaseOptions(argsman);
35 : :
36 [ + - + - ]: 64 : argsman.AddArg("-version", "Print version and exit", ArgsManager::ALLOW_ANY, OptionsCategory::OPTIONS);
37 [ + - + - ]: 64 : argsman.AddArg("-datadir=<dir>", "Specify data directory", ArgsManager::ALLOW_ANY, OptionsCategory::OPTIONS);
38 [ + - + - ]: 64 : argsman.AddArg("-wallet=<wallet-name>", "Specify wallet name", ArgsManager::ALLOW_ANY | ArgsManager::NETWORK_ONLY, OptionsCategory::OPTIONS);
39 [ + - + - ]: 64 : argsman.AddArg("-dumpfile=<file name>", "When used with 'dump', writes out the records to this file. When used with 'createfromdump', loads the records into a new wallet.", ArgsManager::ALLOW_ANY | ArgsManager::DISALLOW_NEGATION, OptionsCategory::OPTIONS);
40 [ + - + - ]: 64 : argsman.AddArg("-debug=<category>", "Output debugging information (default: 0).", ArgsManager::ALLOW_ANY, OptionsCategory::DEBUG_TEST);
41 [ + - + - ]: 64 : argsman.AddArg("-descriptors", "Create descriptors wallet. Only for 'create'", ArgsManager::ALLOW_ANY, OptionsCategory::OPTIONS);
42 [ + - + - ]: 64 : argsman.AddArg("-legacy", "Create legacy wallet. Only for 'create'", ArgsManager::ALLOW_ANY, OptionsCategory::OPTIONS);
43 [ + - + - ]: 64 : argsman.AddArg("-format=<format>", "The format of the wallet file to create. Either \"bdb\" or \"sqlite\". Only used with 'createfromdump'", ArgsManager::ALLOW_ANY, OptionsCategory::OPTIONS);
44 [ + - + - ]: 64 : argsman.AddArg("-printtoconsole", "Send trace/debug info to console (default: 1 when no -debug is true, 0 otherwise).", ArgsManager::ALLOW_ANY, OptionsCategory::DEBUG_TEST);
45 [ + - + - ]: 64 : argsman.AddArg("-withinternalbdb", "Use the internal Berkeley DB parser when dumping a Berkeley DB wallet file (default: false)", ArgsManager::ALLOW_ANY, OptionsCategory::DEBUG_TEST);
46 : :
47 [ + - + - ]: 64 : argsman.AddCommand("info", "Get wallet info");
48 [ + - + - ]: 64 : argsman.AddCommand("create", "Create new wallet file");
49 [ + - + - ]: 64 : argsman.AddCommand("salvage", "Attempt to recover private keys from a corrupt wallet. Warning: 'salvage' is experimental.");
50 [ + - + - ]: 64 : argsman.AddCommand("dump", "Print out all of the wallet key-value records");
51 [ + - + - ]: 64 : argsman.AddCommand("createfromdump", "Create new wallet file from dumped records");
52 : 32 : }
53 : :
54 : 32 : static std::optional<int> WalletAppInit(ArgsManager& args, int argc, char* argv[])
55 : : {
56 : 32 : SetupWalletToolArgs(args);
57 [ + - ]: 32 : std::string error_message;
58 [ + - + + ]: 32 : if (!args.ParseParameters(argc, argv, error_message)) {
59 [ + - ]: 3 : tfm::format(std::cerr, "Error parsing command line arguments: %s\n", error_message);
60 : 3 : return EXIT_FAILURE;
61 : : }
62 : 29 : const bool missing_args{argc < 2};
63 [ + - + - : 58 : if (missing_args || HelpRequested(args) || args.IsArgSet("-version")) {
+ - + - +
- + - -
+ ]
64 [ # # # # : 0 : std::string strUsage = strprintf("%s bitcoin-wallet version", CLIENT_NAME) + " " + FormatFullVersion() + "\n";
# # ]
65 : :
66 [ # # # # : 0 : if (args.IsArgSet("-version")) {
# # ]
67 [ # # # # ]: 0 : strUsage += FormatParagraph(LicenseInfo());
68 : : } else {
69 [ # # ]: 0 : strUsage += "\n"
70 : : "bitcoin-wallet is an offline tool for creating and interacting with " CLIENT_NAME " wallet files.\n"
71 : : "By default bitcoin-wallet will act on wallets in the default mainnet wallet directory in the datadir.\n"
72 : : "To change the target wallet, use the -datadir, -wallet and -regtest/-signet/-testnet/-testnet4 arguments.\n\n"
73 : : "Usage:\n"
74 : 0 : " bitcoin-wallet [options] <command>\n";
75 [ # # # # ]: 0 : strUsage += "\n" + args.GetHelpMessage();
76 : : }
77 [ # # ]: 0 : tfm::format(std::cout, "%s", strUsage);
78 [ # # ]: 0 : if (missing_args) {
79 [ # # ]: 0 : tfm::format(std::cerr, "Error: too few parameters\n");
80 : 0 : return EXIT_FAILURE;
81 : : }
82 : 0 : return EXIT_SUCCESS;
83 : 0 : }
84 : :
85 : : // check for printtoconsole, allow -debug
86 [ + - + - : 29 : LogInstance().m_print_to_console = args.GetBoolArg("-printtoconsole", args.GetBoolArg("-debug", false));
+ - + - +
- ]
87 : :
88 [ + - - + ]: 29 : if (!CheckDataDirOption(args)) {
89 [ # # # # : 0 : tfm::format(std::cerr, "Error: Specified data directory \"%s\" does not exist.\n", args.GetArg("-datadir", ""));
# # # # ]
90 : 0 : return EXIT_FAILURE;
91 : : }
92 : : // Check for chain settings (Params() calls are only valid after this clause)
93 [ + - + - ]: 29 : SelectParams(args.GetChainType());
94 : :
95 : 29 : return std::nullopt;
96 : 32 : }
97 : :
98 : 32 : MAIN_FUNCTION
99 : : {
100 : 32 : ArgsManager& args = gArgs;
101 : : #ifdef WIN32
102 : : common::WinCmdLineArgs winArgs;
103 : : std::tie(argc, argv) = winArgs.get();
104 : : #endif
105 : :
106 : 32 : int exit_status;
107 : 32 : std::unique_ptr<interfaces::Init> init = interfaces::MakeWalletInit(argc, argv, exit_status);
108 [ - + ]: 32 : if (!init) {
109 : 0 : return exit_status;
110 : : }
111 : :
112 [ + - ]: 32 : SetupEnvironment();
113 [ + - ]: 32 : RandomInit();
114 : 32 : try {
115 [ + - + + ]: 32 : if (const auto maybe_exit{WalletAppInit(args, argc, argv)}) return *maybe_exit;
116 [ - - ]: 0 : } catch (const std::exception& e) {
117 [ - - ]: 0 : PrintExceptionContinue(&e, "WalletAppInit()");
118 : 0 : return EXIT_FAILURE;
119 : 0 : } catch (...) {
120 [ - - ]: 0 : PrintExceptionContinue(nullptr, "WalletAppInit()");
121 : 0 : return EXIT_FAILURE;
122 [ - - ]: 0 : }
123 : :
124 [ + - ]: 29 : const auto command = args.GetCommand();
125 [ + + ]: 29 : if (!command) {
126 [ + - ]: 1 : tfm::format(std::cerr, "No method provided. Run `bitcoin-wallet -help` for valid methods.\n");
127 : : return EXIT_FAILURE;
128 : : }
129 [ + + ]: 28 : if (command->args.size() != 0) {
130 [ + - + - ]: 1 : tfm::format(std::cerr, "Error: Additional arguments provided (%s). Methods do not take arguments. Please refer to `-help`.\n", Join(command->args, ", "));
131 : 1 : return EXIT_FAILURE;
132 : : }
133 : :
134 [ + - ]: 27 : ECC_Context ecc_context{};
135 [ + - + + ]: 27 : if (!wallet::WalletTool::ExecuteWalletToolFunc(args, command->command)) {
136 : 17 : return EXIT_FAILURE;
137 : : }
138 : : return EXIT_SUCCESS;
139 : 59 : }
|
