mirror of https://github.com/bitcoin/bitcoin
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
567 lines
29 KiB
567 lines
29 KiB
#!/usr/bin/env python3
|
|
# Copyright (c) 2018-2022 The Bitcoin Core developers
|
|
# Distributed under the MIT software license, see the accompanying
|
|
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
"""Test bitcoin-wallet."""
|
|
|
|
import os
|
|
import platform
|
|
import stat
|
|
import subprocess
|
|
import textwrap
|
|
|
|
from collections import OrderedDict
|
|
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
from test_framework.util import (
|
|
assert_equal,
|
|
assert_greater_than,
|
|
sha256sum_file,
|
|
)
|
|
|
|
|
|
class ToolWalletTest(BitcoinTestFramework):
|
|
def add_options(self, parser):
|
|
self.add_wallet_options(parser)
|
|
parser.add_argument("--bdbro", action="store_true", help="Use the BerkeleyRO internal parser when dumping a Berkeley DB wallet file")
|
|
parser.add_argument("--swap-bdb-endian", action="store_true",help="When making Legacy BDB wallets, always make then byte swapped internally")
|
|
|
|
def set_test_params(self):
|
|
self.num_nodes = 1
|
|
self.setup_clean_chain = True
|
|
self.rpc_timeout = 120
|
|
if self.options.swap_bdb_endian:
|
|
self.extra_args = [["-swapbdbendian"]]
|
|
|
|
def skip_test_if_missing_module(self):
|
|
self.skip_if_no_wallet()
|
|
self.skip_if_no_wallet_tool()
|
|
|
|
def bitcoin_wallet_process(self, *args):
|
|
default_args = ['-datadir={}'.format(self.nodes[0].datadir_path), '-chain=%s' % self.chain]
|
|
if not self.options.descriptors and 'create' in args:
|
|
default_args.append('-legacy')
|
|
if "dump" in args and self.options.bdbro:
|
|
default_args.append("-withinternalbdb")
|
|
|
|
return subprocess.Popen([self.options.bitcoinwallet] + default_args + list(args), stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
|
|
|
|
def assert_raises_tool_error(self, error, *args):
|
|
p = self.bitcoin_wallet_process(*args)
|
|
stdout, stderr = p.communicate()
|
|
assert_equal(stdout, '')
|
|
if isinstance(error, tuple):
|
|
assert_equal(p.poll(), error[0])
|
|
assert error[1] in stderr.strip()
|
|
else:
|
|
assert_equal(p.poll(), 1)
|
|
assert error in stderr.strip()
|
|
|
|
def assert_tool_output(self, output, *args):
|
|
p = self.bitcoin_wallet_process(*args)
|
|
stdout, stderr = p.communicate()
|
|
assert_equal(stderr, '')
|
|
assert_equal(stdout, output)
|
|
assert_equal(p.poll(), 0)
|
|
|
|
def wallet_shasum(self):
|
|
return sha256sum_file(self.wallet_path).hex()
|
|
|
|
def wallet_timestamp(self):
|
|
return os.path.getmtime(self.wallet_path)
|
|
|
|
def wallet_permissions(self):
|
|
return oct(os.lstat(self.wallet_path).st_mode)[-3:]
|
|
|
|
def log_wallet_timestamp_comparison(self, old, new):
|
|
result = 'unchanged' if new == old else 'increased!'
|
|
self.log.debug('Wallet file timestamp {}'.format(result))
|
|
|
|
def get_expected_info_output(self, name="", transactions=0, keypool=2, address=0, imported_privs=0):
|
|
wallet_name = self.default_wallet_name if name == "" else name
|
|
if self.options.descriptors:
|
|
output_types = 4 # p2pkh, p2sh, segwit, bech32m
|
|
return textwrap.dedent('''\
|
|
Wallet info
|
|
===========
|
|
Name: %s
|
|
Format: sqlite
|
|
Descriptors: yes
|
|
Encrypted: no
|
|
HD (hd seed available): yes
|
|
Keypool Size: %d
|
|
Transactions: %d
|
|
Address Book: %d
|
|
''' % (wallet_name, keypool * output_types, transactions, imported_privs * 3 + address))
|
|
else:
|
|
output_types = 3 # p2pkh, p2sh, segwit. Legacy wallets do not support bech32m.
|
|
return textwrap.dedent('''\
|
|
Wallet info
|
|
===========
|
|
Name: %s
|
|
Format: bdb
|
|
Descriptors: no
|
|
Encrypted: no
|
|
HD (hd seed available): yes
|
|
Keypool Size: %d
|
|
Transactions: %d
|
|
Address Book: %d
|
|
''' % (wallet_name, keypool, transactions, (address + imported_privs) * output_types))
|
|
|
|
def read_dump(self, filename):
|
|
dump = OrderedDict()
|
|
with open(filename, "r", encoding="utf8") as f:
|
|
for row in f:
|
|
row = row.strip()
|
|
key, value = row.split(',')
|
|
dump[key] = value
|
|
return dump
|
|
|
|
def assert_is_sqlite(self, filename):
|
|
with open(filename, 'rb') as f:
|
|
file_magic = f.read(16)
|
|
assert file_magic == b'SQLite format 3\x00'
|
|
|
|
def assert_is_bdb(self, filename):
|
|
with open(filename, 'rb') as f:
|
|
f.seek(12, 0)
|
|
file_magic = f.read(4)
|
|
assert file_magic == b'\x00\x05\x31\x62' or file_magic == b'\x62\x31\x05\x00'
|
|
|
|
def write_dump(self, dump, filename, magic=None, skip_checksum=False):
|
|
if magic is None:
|
|
magic = "BITCOIN_CORE_WALLET_DUMP"
|
|
with open(filename, "w", encoding="utf8") as f:
|
|
row = ",".join([magic, dump[magic]]) + "\n"
|
|
f.write(row)
|
|
for k, v in dump.items():
|
|
if k == magic or k == "checksum":
|
|
continue
|
|
row = ",".join([k, v]) + "\n"
|
|
f.write(row)
|
|
if not skip_checksum:
|
|
row = ",".join(["checksum", dump["checksum"]]) + "\n"
|
|
f.write(row)
|
|
|
|
def assert_dump(self, expected, received):
|
|
e = expected.copy()
|
|
r = received.copy()
|
|
|
|
# BDB will add a "version" record that is not present in sqlite
|
|
# In that case, we should ignore this record in both
|
|
# But because this also effects the checksum, we also need to drop that.
|
|
v_key = "0776657273696f6e" # Version key
|
|
if v_key in e and v_key not in r:
|
|
del e[v_key]
|
|
del e["checksum"]
|
|
del r["checksum"]
|
|
if v_key not in e and v_key in r:
|
|
del r[v_key]
|
|
del e["checksum"]
|
|
del r["checksum"]
|
|
|
|
assert_equal(len(e), len(r))
|
|
for k, v in e.items():
|
|
assert_equal(v, r[k])
|
|
|
|
def do_tool_createfromdump(self, wallet_name, dumpfile, file_format=None):
|
|
dumppath = self.nodes[0].datadir_path / dumpfile
|
|
rt_dumppath = self.nodes[0].datadir_path / "rt-{}.dump".format(wallet_name)
|
|
|
|
dump_data = self.read_dump(dumppath)
|
|
|
|
args = ["-wallet={}".format(wallet_name),
|
|
"-dumpfile={}".format(dumppath)]
|
|
if file_format is not None:
|
|
args.append("-format={}".format(file_format))
|
|
args.append("createfromdump")
|
|
|
|
load_output = ""
|
|
if file_format is not None and file_format != dump_data["format"]:
|
|
load_output += "Warning: Dumpfile wallet format \"{}\" does not match command line specified format \"{}\".\n".format(dump_data["format"], file_format)
|
|
self.assert_tool_output(load_output, *args)
|
|
assert (self.nodes[0].wallets_path / wallet_name).is_dir()
|
|
|
|
self.assert_tool_output("The dumpfile may contain private keys. To ensure the safety of your Bitcoin, do not share the dumpfile.\n", '-wallet={}'.format(wallet_name), '-dumpfile={}'.format(rt_dumppath), 'dump')
|
|
|
|
rt_dump_data = self.read_dump(rt_dumppath)
|
|
wallet_dat = self.nodes[0].wallets_path / wallet_name / "wallet.dat"
|
|
if rt_dump_data["format"] == "bdb":
|
|
self.assert_is_bdb(wallet_dat)
|
|
else:
|
|
self.assert_is_sqlite(wallet_dat)
|
|
|
|
def test_invalid_tool_commands_and_args(self):
|
|
self.log.info('Testing that various invalid commands raise with specific error messages')
|
|
self.assert_raises_tool_error("Error parsing command line arguments: Invalid command 'foo'", 'foo')
|
|
# `bitcoin-wallet help` raises an error. Use `bitcoin-wallet -help`.
|
|
self.assert_raises_tool_error("Error parsing command line arguments: Invalid command 'help'", 'help')
|
|
self.assert_raises_tool_error('Error: Additional arguments provided (create). Methods do not take arguments. Please refer to `-help`.', 'info', 'create')
|
|
self.assert_raises_tool_error('Error parsing command line arguments: Invalid parameter -foo', '-foo')
|
|
self.assert_raises_tool_error('No method provided. Run `bitcoin-wallet -help` for valid methods.')
|
|
self.assert_raises_tool_error('Wallet name must be provided when creating a new wallet.', 'create')
|
|
locked_dir = self.nodes[0].wallets_path
|
|
error = 'Error initializing wallet database environment "{}"!'.format(locked_dir)
|
|
if self.options.descriptors:
|
|
error = f"SQLiteDatabase: Unable to obtain an exclusive lock on the database, is it being used by another instance of {self.config['environment']['PACKAGE_NAME']}?"
|
|
self.assert_raises_tool_error(
|
|
error,
|
|
'-wallet=' + self.default_wallet_name,
|
|
'info',
|
|
)
|
|
path = self.nodes[0].wallets_path / "nonexistent.dat"
|
|
self.assert_raises_tool_error("Failed to load database path '{}'. Path does not exist.".format(path), '-wallet=nonexistent.dat', 'info')
|
|
|
|
def test_tool_wallet_info(self):
|
|
# Stop the node to close the wallet to call the info command.
|
|
self.stop_node(0)
|
|
self.log.info('Calling wallet tool info, testing output')
|
|
#
|
|
# TODO: Wallet tool info should work with wallet file permissions set to
|
|
# read-only without raising:
|
|
# "Error loading wallet.dat. Is wallet being used by another process?"
|
|
# The following lines should be uncommented and the tests still succeed:
|
|
#
|
|
# self.log.debug('Setting wallet file permissions to 400 (read-only)')
|
|
# os.chmod(self.wallet_path, stat.S_IRUSR)
|
|
# assert self.wallet_permissions() in ['400', '666'] # Sanity check. 666 because Appveyor.
|
|
# shasum_before = self.wallet_shasum()
|
|
timestamp_before = self.wallet_timestamp()
|
|
self.log.debug('Wallet file timestamp before calling info: {}'.format(timestamp_before))
|
|
out = self.get_expected_info_output(imported_privs=1)
|
|
self.assert_tool_output(out, '-wallet=' + self.default_wallet_name, 'info')
|
|
timestamp_after = self.wallet_timestamp()
|
|
self.log.debug('Wallet file timestamp after calling info: {}'.format(timestamp_after))
|
|
self.log_wallet_timestamp_comparison(timestamp_before, timestamp_after)
|
|
self.log.debug('Setting wallet file permissions back to 600 (read/write)')
|
|
os.chmod(self.wallet_path, stat.S_IRUSR | stat.S_IWUSR)
|
|
assert self.wallet_permissions() in ['600', '666'] # Sanity check. 666 because Appveyor.
|
|
#
|
|
# TODO: Wallet tool info should not write to the wallet file.
|
|
# The following lines should be uncommented and the tests still succeed:
|
|
#
|
|
# assert_equal(timestamp_before, timestamp_after)
|
|
# shasum_after = self.wallet_shasum()
|
|
# assert_equal(shasum_before, shasum_after)
|
|
# self.log.debug('Wallet file shasum unchanged\n')
|
|
|
|
def test_tool_wallet_info_after_transaction(self):
|
|
"""
|
|
Mutate the wallet with a transaction to verify that the info command
|
|
output changes accordingly.
|
|
"""
|
|
self.start_node(0)
|
|
self.log.info('Generating transaction to mutate wallet')
|
|
self.generate(self.nodes[0], 1)
|
|
self.stop_node(0)
|
|
|
|
self.log.info('Calling wallet tool info after generating a transaction, testing output')
|
|
shasum_before = self.wallet_shasum()
|
|
timestamp_before = self.wallet_timestamp()
|
|
self.log.debug('Wallet file timestamp before calling info: {}'.format(timestamp_before))
|
|
out = self.get_expected_info_output(transactions=1, imported_privs=1)
|
|
self.assert_tool_output(out, '-wallet=' + self.default_wallet_name, 'info')
|
|
shasum_after = self.wallet_shasum()
|
|
timestamp_after = self.wallet_timestamp()
|
|
self.log.debug('Wallet file timestamp after calling info: {}'.format(timestamp_after))
|
|
self.log_wallet_timestamp_comparison(timestamp_before, timestamp_after)
|
|
#
|
|
# TODO: Wallet tool info should not write to the wallet file.
|
|
# This assertion should be uncommented and succeed:
|
|
# assert_equal(timestamp_before, timestamp_after)
|
|
assert_equal(shasum_before, shasum_after)
|
|
self.log.debug('Wallet file shasum unchanged\n')
|
|
|
|
def test_tool_wallet_create_on_existing_wallet(self):
|
|
self.log.info('Calling wallet tool create on an existing wallet, testing output')
|
|
shasum_before = self.wallet_shasum()
|
|
timestamp_before = self.wallet_timestamp()
|
|
self.log.debug('Wallet file timestamp before calling create: {}'.format(timestamp_before))
|
|
out = "Topping up keypool...\n" + self.get_expected_info_output(name="foo", keypool=2000)
|
|
self.assert_tool_output(out, '-wallet=foo', 'create')
|
|
shasum_after = self.wallet_shasum()
|
|
timestamp_after = self.wallet_timestamp()
|
|
self.log.debug('Wallet file timestamp after calling create: {}'.format(timestamp_after))
|
|
self.log_wallet_timestamp_comparison(timestamp_before, timestamp_after)
|
|
assert_equal(timestamp_before, timestamp_after)
|
|
assert_equal(shasum_before, shasum_after)
|
|
self.log.debug('Wallet file shasum unchanged\n')
|
|
|
|
def test_getwalletinfo_on_different_wallet(self):
|
|
self.log.info('Starting node with arg -wallet=foo')
|
|
self.start_node(0, ['-nowallet', '-wallet=foo'])
|
|
|
|
self.log.info('Calling getwalletinfo on a different wallet ("foo"), testing output')
|
|
shasum_before = self.wallet_shasum()
|
|
timestamp_before = self.wallet_timestamp()
|
|
self.log.debug('Wallet file timestamp before calling getwalletinfo: {}'.format(timestamp_before))
|
|
out = self.nodes[0].getwalletinfo()
|
|
self.stop_node(0)
|
|
|
|
shasum_after = self.wallet_shasum()
|
|
timestamp_after = self.wallet_timestamp()
|
|
self.log.debug('Wallet file timestamp after calling getwalletinfo: {}'.format(timestamp_after))
|
|
|
|
assert_equal(0, out['txcount'])
|
|
if not self.options.descriptors:
|
|
assert_equal(1000, out['keypoolsize'])
|
|
assert_equal(1000, out['keypoolsize_hd_internal'])
|
|
assert_equal(True, 'hdseedid' in out)
|
|
else:
|
|
assert_equal(4000, out['keypoolsize'])
|
|
assert_equal(4000, out['keypoolsize_hd_internal'])
|
|
|
|
self.log_wallet_timestamp_comparison(timestamp_before, timestamp_after)
|
|
assert_equal(timestamp_before, timestamp_after)
|
|
assert_equal(shasum_after, shasum_before)
|
|
self.log.debug('Wallet file shasum unchanged\n')
|
|
|
|
def test_salvage(self):
|
|
# TODO: Check salvage actually salvages and doesn't break things. https://github.com/bitcoin/bitcoin/issues/7463
|
|
self.log.info('Check salvage')
|
|
self.start_node(0)
|
|
self.nodes[0].createwallet("salvage")
|
|
self.stop_node(0)
|
|
|
|
self.assert_tool_output('', '-wallet=salvage', 'salvage')
|
|
|
|
def test_dump_createfromdump(self):
|
|
self.start_node(0)
|
|
self.nodes[0].createwallet("todump")
|
|
file_format = self.nodes[0].get_wallet_rpc("todump").getwalletinfo()["format"]
|
|
self.nodes[0].createwallet("todump2")
|
|
self.stop_node(0)
|
|
|
|
self.log.info('Checking dump arguments')
|
|
self.assert_raises_tool_error('No dump file provided. To use dump, -dumpfile=<filename> must be provided.', '-wallet=todump', 'dump')
|
|
|
|
self.log.info('Checking basic dump')
|
|
wallet_dump = self.nodes[0].datadir_path / "wallet.dump"
|
|
self.assert_tool_output('The dumpfile may contain private keys. To ensure the safety of your Bitcoin, do not share the dumpfile.\n', '-wallet=todump', '-dumpfile={}'.format(wallet_dump), 'dump')
|
|
|
|
dump_data = self.read_dump(wallet_dump)
|
|
orig_dump = dump_data.copy()
|
|
# Check the dump magic
|
|
assert_equal(dump_data['BITCOIN_CORE_WALLET_DUMP'], '1')
|
|
# Check the file format
|
|
assert_equal(dump_data["format"], file_format)
|
|
|
|
self.log.info('Checking that a dumpfile cannot be overwritten')
|
|
self.assert_raises_tool_error('File {} already exists. If you are sure this is what you want, move it out of the way first.'.format(wallet_dump), '-wallet=todump2', '-dumpfile={}'.format(wallet_dump), 'dump')
|
|
|
|
self.log.info('Checking createfromdump arguments')
|
|
self.assert_raises_tool_error('No dump file provided. To use createfromdump, -dumpfile=<filename> must be provided.', '-wallet=todump', 'createfromdump')
|
|
non_exist_dump = self.nodes[0].datadir_path / "wallet.nodump"
|
|
self.assert_raises_tool_error('Unknown wallet file format "notaformat" provided. Please provide one of "bdb" or "sqlite".', '-wallet=todump', '-format=notaformat', '-dumpfile={}'.format(wallet_dump), 'createfromdump')
|
|
self.assert_raises_tool_error('Dump file {} does not exist.'.format(non_exist_dump), '-wallet=todump', '-dumpfile={}'.format(non_exist_dump), 'createfromdump')
|
|
wallet_path = self.nodes[0].wallets_path / "todump2"
|
|
self.assert_raises_tool_error('Failed to create database path \'{}\'. Database already exists.'.format(wallet_path), '-wallet=todump2', '-dumpfile={}'.format(wallet_dump), 'createfromdump')
|
|
self.assert_raises_tool_error("The -descriptors option can only be used with the 'create' command.", '-descriptors', '-wallet=todump2', '-dumpfile={}'.format(wallet_dump), 'createfromdump')
|
|
|
|
self.log.info('Checking createfromdump')
|
|
self.do_tool_createfromdump("load", "wallet.dump")
|
|
if self.is_bdb_compiled():
|
|
self.do_tool_createfromdump("load-bdb", "wallet.dump", "bdb")
|
|
if self.is_sqlite_compiled():
|
|
self.do_tool_createfromdump("load-sqlite", "wallet.dump", "sqlite")
|
|
|
|
self.log.info('Checking createfromdump handling of magic and versions')
|
|
bad_ver_wallet_dump = self.nodes[0].datadir_path / "wallet-bad_ver1.dump"
|
|
dump_data["BITCOIN_CORE_WALLET_DUMP"] = "0"
|
|
self.write_dump(dump_data, bad_ver_wallet_dump)
|
|
self.assert_raises_tool_error('Error: Dumpfile version is not supported. This version of bitcoin-wallet only supports version 1 dumpfiles. Got dumpfile with version 0', '-wallet=badload', '-dumpfile={}'.format(bad_ver_wallet_dump), 'createfromdump')
|
|
assert not (self.nodes[0].wallets_path / "badload").is_dir()
|
|
bad_ver_wallet_dump = self.nodes[0].datadir_path / "wallet-bad_ver2.dump"
|
|
dump_data["BITCOIN_CORE_WALLET_DUMP"] = "2"
|
|
self.write_dump(dump_data, bad_ver_wallet_dump)
|
|
self.assert_raises_tool_error('Error: Dumpfile version is not supported. This version of bitcoin-wallet only supports version 1 dumpfiles. Got dumpfile with version 2', '-wallet=badload', '-dumpfile={}'.format(bad_ver_wallet_dump), 'createfromdump')
|
|
assert not (self.nodes[0].wallets_path / "badload").is_dir()
|
|
bad_magic_wallet_dump = self.nodes[0].datadir_path / "wallet-bad_magic.dump"
|
|
del dump_data["BITCOIN_CORE_WALLET_DUMP"]
|
|
dump_data["not_the_right_magic"] = "1"
|
|
self.write_dump(dump_data, bad_magic_wallet_dump, "not_the_right_magic")
|
|
self.assert_raises_tool_error('Error: Dumpfile identifier record is incorrect. Got "not_the_right_magic", expected "BITCOIN_CORE_WALLET_DUMP".', '-wallet=badload', '-dumpfile={}'.format(bad_magic_wallet_dump), 'createfromdump')
|
|
assert not (self.nodes[0].wallets_path / "badload").is_dir()
|
|
|
|
self.log.info('Checking createfromdump handling of checksums')
|
|
bad_sum_wallet_dump = self.nodes[0].datadir_path / "wallet-bad_sum1.dump"
|
|
dump_data = orig_dump.copy()
|
|
checksum = dump_data["checksum"]
|
|
dump_data["checksum"] = "1" * 64
|
|
self.write_dump(dump_data, bad_sum_wallet_dump)
|
|
self.assert_raises_tool_error('Error: Dumpfile checksum does not match. Computed {}, expected {}'.format(checksum, "1" * 64), '-wallet=bad', '-dumpfile={}'.format(bad_sum_wallet_dump), 'createfromdump')
|
|
assert not (self.nodes[0].wallets_path / "badload").is_dir()
|
|
bad_sum_wallet_dump = self.nodes[0].datadir_path / "wallet-bad_sum2.dump"
|
|
del dump_data["checksum"]
|
|
self.write_dump(dump_data, bad_sum_wallet_dump, skip_checksum=True)
|
|
self.assert_raises_tool_error('Error: Missing checksum', '-wallet=badload', '-dumpfile={}'.format(bad_sum_wallet_dump), 'createfromdump')
|
|
assert not (self.nodes[0].wallets_path / "badload").is_dir()
|
|
bad_sum_wallet_dump = self.nodes[0].datadir_path / "wallet-bad_sum3.dump"
|
|
dump_data["checksum"] = "2" * 10
|
|
self.write_dump(dump_data, bad_sum_wallet_dump)
|
|
self.assert_raises_tool_error('Error: Checksum is not the correct size', '-wallet=badload', '-dumpfile={}'.format(bad_sum_wallet_dump), 'createfromdump')
|
|
assert not (self.nodes[0].wallets_path / "badload").is_dir()
|
|
dump_data["checksum"] = "3" * 66
|
|
self.write_dump(dump_data, bad_sum_wallet_dump)
|
|
self.assert_raises_tool_error('Error: Checksum is not the correct size', '-wallet=badload', '-dumpfile={}'.format(bad_sum_wallet_dump), 'createfromdump')
|
|
assert not (self.nodes[0].wallets_path / "badload").is_dir()
|
|
|
|
def test_chainless_conflicts(self):
|
|
self.log.info("Test wallet tool when wallet contains conflicting transactions")
|
|
self.restart_node(0)
|
|
self.generate(self.nodes[0], 101)
|
|
|
|
def_wallet = self.nodes[0].get_wallet_rpc(self.default_wallet_name)
|
|
|
|
self.nodes[0].createwallet("conflicts")
|
|
wallet = self.nodes[0].get_wallet_rpc("conflicts")
|
|
def_wallet.sendtoaddress(wallet.getnewaddress(), 10)
|
|
self.generate(self.nodes[0], 1)
|
|
|
|
# parent tx
|
|
parent_txid = wallet.sendtoaddress(wallet.getnewaddress(), 9)
|
|
parent_txid_bytes = bytes.fromhex(parent_txid)[::-1]
|
|
conflict_utxo = wallet.gettransaction(txid=parent_txid, verbose=True)["decoded"]["vin"][0]
|
|
|
|
# The specific assertion in MarkConflicted being tested requires that the parent tx is already loaded
|
|
# by the time the child tx is loaded. Since transactions end up being loaded in txid order due to how both
|
|
# and sqlite store things, we can just grind the child tx until it has a txid that is greater than the parent's.
|
|
locktime = 500000000 # Use locktime as nonce, starting at unix timestamp minimum
|
|
addr = wallet.getnewaddress()
|
|
while True:
|
|
child_send_res = wallet.send(outputs=[{addr: 8}], add_to_wallet=False, locktime=locktime)
|
|
child_txid = child_send_res["txid"]
|
|
child_txid_bytes = bytes.fromhex(child_txid)[::-1]
|
|
if (child_txid_bytes > parent_txid_bytes):
|
|
wallet.sendrawtransaction(child_send_res["hex"])
|
|
break
|
|
locktime += 1
|
|
|
|
# conflict with parent
|
|
conflict_unsigned = self.nodes[0].createrawtransaction(inputs=[conflict_utxo], outputs=[{wallet.getnewaddress(): 9.9999}])
|
|
conflict_signed = wallet.signrawtransactionwithwallet(conflict_unsigned)["hex"]
|
|
conflict_txid = self.nodes[0].sendrawtransaction(conflict_signed)
|
|
self.generate(self.nodes[0], 1)
|
|
assert_equal(wallet.gettransaction(txid=parent_txid)["confirmations"], -1)
|
|
assert_equal(wallet.gettransaction(txid=child_txid)["confirmations"], -1)
|
|
assert_equal(wallet.gettransaction(txid=conflict_txid)["confirmations"], 1)
|
|
|
|
self.stop_node(0)
|
|
|
|
# Wallet tool should successfully give info for this wallet
|
|
expected_output = textwrap.dedent(f'''\
|
|
Wallet info
|
|
===========
|
|
Name: conflicts
|
|
Format: {"sqlite" if self.options.descriptors else "bdb"}
|
|
Descriptors: {"yes" if self.options.descriptors else "no"}
|
|
Encrypted: no
|
|
HD (hd seed available): yes
|
|
Keypool Size: {"8" if self.options.descriptors else "1"}
|
|
Transactions: 4
|
|
Address Book: 4
|
|
''')
|
|
self.assert_tool_output(expected_output, "-wallet=conflicts", "info")
|
|
|
|
def test_dump_endianness(self):
|
|
self.log.info("Testing dumps of the same contents with different BDB endianness")
|
|
|
|
self.start_node(0)
|
|
self.nodes[0].createwallet("endian")
|
|
self.stop_node(0)
|
|
|
|
wallet_dump = self.nodes[0].datadir_path / "endian.dump"
|
|
self.assert_tool_output("The dumpfile may contain private keys. To ensure the safety of your Bitcoin, do not share the dumpfile.\n", "-wallet=endian", f"-dumpfile={wallet_dump}", "dump")
|
|
expected_dump = self.read_dump(wallet_dump)
|
|
|
|
self.do_tool_createfromdump("native_endian", "endian.dump", "bdb")
|
|
native_dump = self.read_dump(self.nodes[0].datadir_path / "rt-native_endian.dump")
|
|
self.assert_dump(expected_dump, native_dump)
|
|
|
|
self.do_tool_createfromdump("other_endian", "endian.dump", "bdb_swap")
|
|
other_dump = self.read_dump(self.nodes[0].datadir_path / "rt-other_endian.dump")
|
|
self.assert_dump(expected_dump, other_dump)
|
|
|
|
def test_dump_very_large_records(self):
|
|
self.log.info("Test that wallets with large records are successfully dumped")
|
|
|
|
self.start_node(0)
|
|
self.nodes[0].createwallet("bigrecords")
|
|
wallet = self.nodes[0].get_wallet_rpc("bigrecords")
|
|
|
|
# Both BDB and sqlite have maximum page sizes of 65536 bytes, with defaults of 4096
|
|
# When a record exceeds some size threshold, both BDB and SQLite will store the data
|
|
# in one or more overflow pages. We want to make sure that our tooling can dump such
|
|
# records, even when they span multiple pages. To make a large record, we just need
|
|
# to make a very big transaction.
|
|
self.generate(self.nodes[0], 101)
|
|
def_wallet = self.nodes[0].get_wallet_rpc(self.default_wallet_name)
|
|
outputs = {}
|
|
for i in range(500):
|
|
outputs[wallet.getnewaddress(address_type="p2sh-segwit")] = 0.01
|
|
def_wallet.sendmany(amounts=outputs)
|
|
self.generate(self.nodes[0], 1)
|
|
send_res = wallet.sendall([def_wallet.getnewaddress()])
|
|
self.generate(self.nodes[0], 1)
|
|
assert_equal(send_res["complete"], True)
|
|
tx = wallet.gettransaction(txid=send_res["txid"], verbose=True)
|
|
assert_greater_than(tx["decoded"]["size"], 70000)
|
|
|
|
self.stop_node(0)
|
|
|
|
wallet_dump = self.nodes[0].datadir_path / "bigrecords.dump"
|
|
self.assert_tool_output("The dumpfile may contain private keys. To ensure the safety of your Bitcoin, do not share the dumpfile.\n", "-wallet=bigrecords", f"-dumpfile={wallet_dump}", "dump")
|
|
dump = self.read_dump(wallet_dump)
|
|
for k,v in dump.items():
|
|
if tx["hex"] in v:
|
|
break
|
|
else:
|
|
assert False, "Big transaction was not found in wallet dump"
|
|
|
|
def test_dump_unclean_lsns(self):
|
|
if not self.options.bdbro:
|
|
return
|
|
self.log.info("Test that a legacy wallet that has not been compacted is not dumped by bdbro")
|
|
|
|
self.start_node(0, extra_args=["-flushwallet=0"])
|
|
self.nodes[0].createwallet("unclean_lsn")
|
|
wallet = self.nodes[0].get_wallet_rpc("unclean_lsn")
|
|
# First unload and load normally to make sure everything is written
|
|
wallet.unloadwallet()
|
|
self.nodes[0].loadwallet("unclean_lsn")
|
|
# Next cause a bunch of writes by filling the keypool
|
|
wallet.keypoolrefill(wallet.getwalletinfo()["keypoolsize"] + 100)
|
|
# Lastly kill bitcoind so that the LSNs don't get reset
|
|
self.nodes[0].process.kill()
|
|
self.nodes[0].wait_until_stopped(expected_ret_code=1 if platform.system() == "Windows" else -9)
|
|
assert self.nodes[0].is_node_stopped()
|
|
|
|
wallet_dump = self.nodes[0].datadir_path / "unclean_lsn.dump"
|
|
self.assert_raises_tool_error("LSNs are not reset, this database is not completely flushed. Please reopen then close the database with a version that has BDB support", "-wallet=unclean_lsn", f"-dumpfile={wallet_dump}", "dump")
|
|
|
|
# File can be dumped after reload it normally
|
|
self.start_node(0)
|
|
self.nodes[0].loadwallet("unclean_lsn")
|
|
self.stop_node(0)
|
|
self.assert_tool_output("The dumpfile may contain private keys. To ensure the safety of your Bitcoin, do not share the dumpfile.\n", "-wallet=unclean_lsn", f"-dumpfile={wallet_dump}", "dump")
|
|
|
|
def run_test(self):
|
|
self.wallet_path = self.nodes[0].wallets_path / self.default_wallet_name / self.wallet_data_filename
|
|
self.test_invalid_tool_commands_and_args()
|
|
# Warning: The following tests are order-dependent.
|
|
self.test_tool_wallet_info()
|
|
self.test_tool_wallet_info_after_transaction()
|
|
self.test_tool_wallet_create_on_existing_wallet()
|
|
self.test_getwalletinfo_on_different_wallet()
|
|
if not self.options.descriptors:
|
|
# Salvage is a legacy wallet only thing
|
|
self.test_salvage()
|
|
self.test_dump_endianness()
|
|
self.test_dump_unclean_lsns()
|
|
self.test_dump_createfromdump()
|
|
self.test_chainless_conflicts()
|
|
self.test_dump_very_large_records()
|
|
|
|
if __name__ == '__main__':
|
|
ToolWalletTest().main()
|