301 lines
14 KiB
Python
Executable file
301 lines
14 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# Copyright (c) 2017-2018 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 that the wallet can send and receive using all combinations of address types.
|
|
|
|
There are 5 nodes-under-test:
|
|
- node0 uses legacy addresses
|
|
- node1 uses p2sh/segwit addresses
|
|
- node2 uses p2sh/segwit addresses and bech32 addresses for change
|
|
- node3 uses bech32 addresses
|
|
- node4 uses a p2sh/segwit addresses for change
|
|
|
|
node5 exists to generate new blocks.
|
|
|
|
## Multisig address test
|
|
|
|
Test that adding a multisig address with:
|
|
- an uncompressed pubkey always gives a legacy address
|
|
- only compressed pubkeys gives the an `-addresstype` address
|
|
|
|
## Sending to address types test
|
|
|
|
A series of tests, iterating over node0-node4. In each iteration of the test, one node sends:
|
|
- 10/101th of its balance to itself (using getrawchangeaddress for single key addresses)
|
|
- 20/101th to the next node
|
|
- 30/101th to the node after that
|
|
- 40/101th to the remaining node
|
|
- 1/101th remains as fee+change
|
|
|
|
Iterate over each node for single key addresses, and then over each node for
|
|
multisig addresses.
|
|
|
|
Repeat test, but with explicit address_type parameters passed to getnewaddress
|
|
and getrawchangeaddress:
|
|
- node0 and node3 send to p2sh.
|
|
- node1 sends to bech32.
|
|
- node2 sends to legacy.
|
|
|
|
As every node sends coins after receiving, this also
|
|
verifies that spending coins sent to all these address types works.
|
|
|
|
## Change type test
|
|
|
|
Test that the nodes generate the correct change address type:
|
|
- node0 always uses a legacy change address.
|
|
- node1 uses a bech32 addresses for change if any destination address is bech32.
|
|
- node2 always uses a bech32 address for change
|
|
- node3 always uses a bech32 address for change
|
|
- node4 always uses p2sh/segwit output for change.
|
|
"""
|
|
|
|
from decimal import Decimal
|
|
import itertools
|
|
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
from test_framework.util import (
|
|
assert_equal,
|
|
assert_greater_than,
|
|
assert_raises_rpc_error,
|
|
connect_nodes_bi,
|
|
sync_blocks,
|
|
sync_mempools,
|
|
)
|
|
|
|
|
|
class AddressTypeTest(BitcoinTestFramework):
|
|
def set_test_params(self):
|
|
self.num_nodes = 6
|
|
self.extra_args = [
|
|
["-addresstype=legacy"],
|
|
["-addresstype=p2sh-segwit"],
|
|
["-addresstype=p2sh-segwit", "-changetype=bech32"],
|
|
["-addresstype=bech32"],
|
|
["-changetype=p2sh-segwit"],
|
|
[],
|
|
]
|
|
|
|
def skip_test_if_missing_module(self):
|
|
self.skip_if_no_wallet()
|
|
|
|
def setup_network(self):
|
|
self.setup_nodes()
|
|
|
|
# Fully mesh-connect nodes for faster mempool sync
|
|
for i, j in itertools.product(range(self.num_nodes), repeat=2):
|
|
if i > j:
|
|
connect_nodes_bi(self.nodes, i, j)
|
|
self.sync_all()
|
|
|
|
def get_balances(self, confirmed=True):
|
|
"""Return a list of confirmed or unconfirmed balances."""
|
|
if confirmed:
|
|
return [self.nodes[i].getbalance() for i in range(4)]
|
|
else:
|
|
return [self.nodes[i].getunconfirmedbalance() for i in range(4)]
|
|
|
|
def test_address(self, node, address, multisig, typ):
|
|
"""Run sanity checks on an address."""
|
|
info = self.nodes[node].getaddressinfo(address)
|
|
assert(self.nodes[node].validateaddress(address)['isvalid'])
|
|
if not multisig and typ == 'legacy':
|
|
# P2PKH
|
|
assert(not info['isscript'])
|
|
assert(not info['iswitness'])
|
|
assert('pubkey' in info)
|
|
elif not multisig and typ == 'p2sh-segwit':
|
|
# P2SH-P2WPKH
|
|
assert(info['isscript'])
|
|
assert(not info['iswitness'])
|
|
assert_equal(info['script'], 'witness_v0_keyhash')
|
|
assert('pubkey' in info)
|
|
elif not multisig and typ == 'bech32':
|
|
# P2WPKH
|
|
assert(not info['isscript'])
|
|
assert(info['iswitness'])
|
|
assert_equal(info['witness_version'], 0)
|
|
assert_equal(len(info['witness_program']), 40)
|
|
assert('pubkey' in info)
|
|
elif typ == 'legacy':
|
|
# P2SH-multisig
|
|
assert(info['isscript'])
|
|
assert_equal(info['script'], 'multisig')
|
|
assert(not info['iswitness'])
|
|
assert('pubkeys' in info)
|
|
elif typ == 'p2sh-segwit':
|
|
# P2SH-P2WSH-multisig
|
|
assert(info['isscript'])
|
|
assert_equal(info['script'], 'witness_v0_scripthash')
|
|
assert(not info['iswitness'])
|
|
assert(info['embedded']['isscript'])
|
|
assert_equal(info['embedded']['script'], 'multisig')
|
|
assert(info['embedded']['iswitness'])
|
|
assert_equal(info['embedded']['witness_version'], 0)
|
|
assert_equal(len(info['embedded']['witness_program']), 64)
|
|
assert('pubkeys' in info['embedded'])
|
|
elif typ == 'bech32':
|
|
# P2WSH-multisig
|
|
assert(info['isscript'])
|
|
assert_equal(info['script'], 'multisig')
|
|
assert(info['iswitness'])
|
|
assert_equal(info['witness_version'], 0)
|
|
assert_equal(len(info['witness_program']), 64)
|
|
assert('pubkeys' in info)
|
|
else:
|
|
# Unknown type
|
|
assert(False)
|
|
|
|
def test_change_output_type(self, node_sender, destinations, expected_type):
|
|
txid = self.nodes[node_sender].sendmany(fromaccount="", amounts=dict.fromkeys(destinations, 0.001))
|
|
raw_tx = self.nodes[node_sender].getrawtransaction(txid)
|
|
tx = self.nodes[node_sender].decoderawtransaction(raw_tx)
|
|
|
|
# Make sure the transaction has change:
|
|
assert_equal(len(tx["vout"]), len(destinations) + 1)
|
|
|
|
# Make sure the destinations are included, and remove them:
|
|
output_addresses = [vout['scriptPubKey']['addresses'][0] for vout in tx["vout"]]
|
|
change_addresses = [d for d in output_addresses if d not in destinations]
|
|
assert_equal(len(change_addresses), 1)
|
|
|
|
self.log.debug("Check if change address " + change_addresses[0] + " is " + expected_type)
|
|
self.test_address(node_sender, change_addresses[0], multisig=False, typ=expected_type)
|
|
|
|
def run_test(self):
|
|
# Mine 101 blocks on node5 to bring nodes out of IBD and make sure that
|
|
# no coinbases are maturing for the nodes-under-test during the test
|
|
self.nodes[5].generate(101)
|
|
sync_blocks(self.nodes)
|
|
|
|
uncompressed_1 = "0496b538e853519c726a2c91e61ec11600ae1390813a627c66fb8be7947be63c52da7589379515d4e0a604f8141781e62294721166bf621e73a82cbf2342c858ee"
|
|
uncompressed_2 = "047211a824f55b505228e4c3d5194c1fcfaa15a456abdf37f9b9d97a4040afc073dee6c89064984f03385237d92167c13e236446b417ab79a0fcae412ae3316b77"
|
|
compressed_1 = "0296b538e853519c726a2c91e61ec11600ae1390813a627c66fb8be7947be63c52"
|
|
compressed_2 = "037211a824f55b505228e4c3d5194c1fcfaa15a456abdf37f9b9d97a4040afc073"
|
|
|
|
# addmultisigaddress with at least 1 uncompressed key should return a legacy address.
|
|
for node in range(4):
|
|
self.test_address(node, self.nodes[node].addmultisigaddress(2, [uncompressed_1, uncompressed_2])['address'], True, 'legacy')
|
|
self.test_address(node, self.nodes[node].addmultisigaddress(2, [compressed_1, uncompressed_2])['address'], True, 'legacy')
|
|
self.test_address(node, self.nodes[node].addmultisigaddress(2, [uncompressed_1, compressed_2])['address'], True, 'legacy')
|
|
# addmultisigaddress with all compressed keys should return the appropriate address type (even when the keys are not ours).
|
|
self.test_address(0, self.nodes[0].addmultisigaddress(2, [compressed_1, compressed_2])['address'], True, 'legacy')
|
|
self.test_address(1, self.nodes[1].addmultisigaddress(2, [compressed_1, compressed_2])['address'], True, 'p2sh-segwit')
|
|
self.test_address(2, self.nodes[2].addmultisigaddress(2, [compressed_1, compressed_2])['address'], True, 'p2sh-segwit')
|
|
self.test_address(3, self.nodes[3].addmultisigaddress(2, [compressed_1, compressed_2])['address'], True, 'bech32')
|
|
|
|
for explicit_type, multisig, from_node in itertools.product([False, True], [False, True], range(4)):
|
|
address_type = None
|
|
if explicit_type and not multisig:
|
|
if from_node == 1:
|
|
address_type = 'bech32'
|
|
elif from_node == 0 or from_node == 3:
|
|
address_type = 'p2sh-segwit'
|
|
else:
|
|
address_type = 'legacy'
|
|
self.log.info("Sending from node {} ({}) with{} multisig using {}".format(from_node, self.extra_args[from_node], "" if multisig else "out", "default" if address_type is None else address_type))
|
|
old_balances = self.get_balances()
|
|
self.log.debug("Old balances are {}".format(old_balances))
|
|
to_send = (old_balances[from_node] / 101).quantize(Decimal("0.00000001"))
|
|
sends = {}
|
|
|
|
self.log.debug("Prepare sends")
|
|
for n, to_node in enumerate(range(from_node, from_node + 4)):
|
|
to_node %= 4
|
|
change = False
|
|
if not multisig:
|
|
if from_node == to_node:
|
|
# When sending non-multisig to self, use getrawchangeaddress
|
|
address = self.nodes[to_node].getrawchangeaddress(address_type=address_type)
|
|
change = True
|
|
else:
|
|
address = self.nodes[to_node].getnewaddress(address_type=address_type)
|
|
else:
|
|
addr1 = self.nodes[to_node].getnewaddress()
|
|
addr2 = self.nodes[to_node].getnewaddress()
|
|
address = self.nodes[to_node].addmultisigaddress(2, [addr1, addr2])['address']
|
|
|
|
# Do some sanity checking on the created address
|
|
if address_type is not None:
|
|
typ = address_type
|
|
elif to_node == 0:
|
|
typ = 'legacy'
|
|
elif to_node == 1 or (to_node == 2 and not change):
|
|
typ = 'p2sh-segwit'
|
|
else:
|
|
typ = 'bech32'
|
|
self.test_address(to_node, address, multisig, typ)
|
|
|
|
# Output entry
|
|
sends[address] = to_send * 10 * (1 + n)
|
|
|
|
self.log.debug("Sending: {}".format(sends))
|
|
self.nodes[from_node].sendmany("", sends)
|
|
sync_mempools(self.nodes)
|
|
|
|
unconf_balances = self.get_balances(False)
|
|
self.log.debug("Check unconfirmed balances: {}".format(unconf_balances))
|
|
assert_equal(unconf_balances[from_node], 0)
|
|
for n, to_node in enumerate(range(from_node + 1, from_node + 4)):
|
|
to_node %= 4
|
|
assert_equal(unconf_balances[to_node], to_send * 10 * (2 + n))
|
|
|
|
# node5 collects fee and block subsidy to keep accounting simple
|
|
self.nodes[5].generate(1)
|
|
sync_blocks(self.nodes)
|
|
|
|
new_balances = self.get_balances()
|
|
self.log.debug("Check new balances: {}".format(new_balances))
|
|
# We don't know what fee was set, so we can only check bounds on the balance of the sending node
|
|
assert_greater_than(new_balances[from_node], to_send * 10)
|
|
assert_greater_than(to_send * 11, new_balances[from_node])
|
|
for n, to_node in enumerate(range(from_node + 1, from_node + 4)):
|
|
to_node %= 4
|
|
assert_equal(new_balances[to_node], old_balances[to_node] + to_send * 10 * (2 + n))
|
|
|
|
# Get one p2sh/segwit address from node2 and two bech32 addresses from node3:
|
|
to_address_p2sh = self.nodes[2].getnewaddress()
|
|
to_address_bech32_1 = self.nodes[3].getnewaddress()
|
|
to_address_bech32_2 = self.nodes[3].getnewaddress()
|
|
|
|
# Fund node 4:
|
|
self.nodes[5].sendtoaddress(self.nodes[4].getnewaddress(), Decimal("1"))
|
|
self.nodes[5].generate(1)
|
|
sync_blocks(self.nodes)
|
|
assert_equal(self.nodes[4].getbalance(), 1)
|
|
|
|
self.log.info("Nodes with addresstype=legacy never use a P2WPKH change output")
|
|
self.test_change_output_type(0, [to_address_bech32_1], 'legacy')
|
|
|
|
self.log.info("Nodes with addresstype=p2sh-segwit only use a P2WPKH change output if any destination address is bech32:")
|
|
self.test_change_output_type(1, [to_address_p2sh], 'p2sh-segwit')
|
|
self.test_change_output_type(1, [to_address_bech32_1], 'bech32')
|
|
self.test_change_output_type(1, [to_address_p2sh, to_address_bech32_1], 'bech32')
|
|
self.test_change_output_type(1, [to_address_bech32_1, to_address_bech32_2], 'bech32')
|
|
|
|
self.log.info("Nodes with change_type=bech32 always use a P2WPKH change output:")
|
|
self.test_change_output_type(2, [to_address_bech32_1], 'bech32')
|
|
self.test_change_output_type(2, [to_address_p2sh], 'bech32')
|
|
|
|
self.log.info("Nodes with addresstype=bech32 always use a P2WPKH change output (unless changetype is set otherwise):")
|
|
self.test_change_output_type(3, [to_address_bech32_1], 'bech32')
|
|
self.test_change_output_type(3, [to_address_p2sh], 'bech32')
|
|
|
|
self.log.info('getrawchangeaddress defaults to addresstype if -changetype is not set and argument is absent')
|
|
self.test_address(3, self.nodes[3].getrawchangeaddress(), multisig=False, typ='bech32')
|
|
|
|
self.log.info('test invalid address type arguments')
|
|
assert_raises_rpc_error(-5, "Unknown address type ''", self.nodes[3].addmultisigaddress, 2, [compressed_1, compressed_2], None, '')
|
|
assert_raises_rpc_error(-5, "Unknown address type ''", self.nodes[3].getnewaddress, None, '')
|
|
assert_raises_rpc_error(-5, "Unknown address type ''", self.nodes[3].getrawchangeaddress, '')
|
|
assert_raises_rpc_error(-5, "Unknown address type 'bech23'", self.nodes[3].getrawchangeaddress, 'bech23')
|
|
|
|
self.log.info("Nodes with changetype=p2sh-segwit never use a P2WPKH change output")
|
|
self.test_change_output_type(4, [to_address_bech32_1], 'p2sh-segwit')
|
|
self.test_address(4, self.nodes[4].getrawchangeaddress(), multisig=False, typ='p2sh-segwit')
|
|
self.log.info("Except for getrawchangeaddress if specified:")
|
|
self.test_address(4, self.nodes[4].getrawchangeaddress(), multisig=False, typ='p2sh-segwit')
|
|
self.test_address(4, self.nodes[4].getrawchangeaddress('bech32'), multisig=False, typ='bech32')
|
|
|
|
if __name__ == '__main__':
|
|
AddressTypeTest().main()
|