mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-01-15 22:32:37 -03:00
e1e7a90d5f
Adds a new dump command to bitcoin-wallet which prints out all of the wallet's records in hex.
96 lines
2.9 KiB
C++
96 lines
2.9 KiB
C++
// Copyright (c) 2020 The Bitcoin Core developers
|
|
// Distributed under the MIT software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
#include <wallet/dump.h>
|
|
|
|
#include <util/translation.h>
|
|
#include <wallet/wallet.h>
|
|
|
|
static const std::string DUMP_MAGIC = "BITCOIN_CORE_WALLET_DUMP";
|
|
uint32_t DUMP_VERSION = 1;
|
|
|
|
bool DumpWallet(CWallet& wallet, bilingual_str& error)
|
|
{
|
|
// Get the dumpfile
|
|
std::string dump_filename = gArgs.GetArg("-dumpfile", "");
|
|
if (dump_filename.empty()) {
|
|
error = _("No dump file provided. To use dump, -dumpfile=<filename> must be provided.");
|
|
return false;
|
|
}
|
|
|
|
fs::path path = dump_filename;
|
|
path = fs::absolute(path);
|
|
if (fs::exists(path)) {
|
|
error = strprintf(_("File %s already exists. If you are sure this is what you want, move it out of the way first."), path.string());
|
|
return false;
|
|
}
|
|
fsbridge::ofstream dump_file;
|
|
dump_file.open(path);
|
|
if (dump_file.fail()) {
|
|
error = strprintf(_("Unable to open %s for writing"), path.string());
|
|
return false;
|
|
}
|
|
|
|
CHashWriter hasher(0, 0);
|
|
|
|
WalletDatabase& db = wallet.GetDatabase();
|
|
std::unique_ptr<DatabaseBatch> batch = db.MakeBatch();
|
|
|
|
bool ret = true;
|
|
if (!batch->StartCursor()) {
|
|
error = _("Error: Couldn't create cursor into database");
|
|
ret = false;
|
|
}
|
|
|
|
// Write out a magic string with version
|
|
std::string line = strprintf("%s,%u\n", DUMP_MAGIC, DUMP_VERSION);
|
|
dump_file.write(line.data(), line.size());
|
|
hasher.write(line.data(), line.size());
|
|
|
|
// Write out the file format
|
|
line = strprintf("%s,%s\n", "format", db.Format());
|
|
dump_file.write(line.data(), line.size());
|
|
hasher.write(line.data(), line.size());
|
|
|
|
if (ret) {
|
|
|
|
// Read the records
|
|
while (true) {
|
|
CDataStream ss_key(SER_DISK, CLIENT_VERSION);
|
|
CDataStream ss_value(SER_DISK, CLIENT_VERSION);
|
|
bool complete;
|
|
ret = batch->ReadAtCursor(ss_key, ss_value, complete);
|
|
if (complete) {
|
|
ret = true;
|
|
break;
|
|
} else if (!ret) {
|
|
error = _("Error reading next record from wallet database");
|
|
break;
|
|
}
|
|
std::string key_str = HexStr(ss_key);
|
|
std::string value_str = HexStr(ss_value);
|
|
line = strprintf("%s,%s\n", key_str, value_str);
|
|
dump_file.write(line.data(), line.size());
|
|
hasher.write(line.data(), line.size());
|
|
}
|
|
}
|
|
|
|
batch->CloseCursor();
|
|
batch.reset();
|
|
|
|
// Close the wallet after we're done with it. The caller won't be doing this
|
|
wallet.Close();
|
|
|
|
if (ret) {
|
|
// Write the hash
|
|
tfm::format(dump_file, "checksum,%s\n", HexStr(hasher.GetHash()));
|
|
dump_file.close();
|
|
} else {
|
|
// Remove the dumpfile on failure
|
|
dump_file.close();
|
|
fs::remove(path);
|
|
}
|
|
|
|
return ret;
|
|
}
|