mirror of
https://github.com/ElementsProject/elements.git
synced 2026-08-16 13:01:19 +02:00
de81c0d97e
I also moved some RPC methods that were not at the end of the file with the other ELEMENTS RPC methods. This makes the diff on my machine look crazy. Let's hope GitHub has more intelligent diffs..
149 lines
6.5 KiB
Python
Executable file
149 lines
6.5 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# Copyright (c) 2014-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 the wallet accounts properly when there are cloned transactions with malleated scriptsigs."""
|
|
|
|
import io
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
from test_framework.messages import CTransaction, COIN
|
|
from test_framework.util import (
|
|
assert_equal,
|
|
bytes_to_hex_str as b2x,
|
|
connect_nodes,
|
|
disconnect_nodes,
|
|
sync_blocks,
|
|
)
|
|
|
|
class TxnMallTest(BitcoinTestFramework):
|
|
def set_test_params(self):
|
|
self.num_nodes = 4
|
|
|
|
def skip_test_if_missing_module(self):
|
|
self.skip_if_no_wallet()
|
|
|
|
def add_options(self, parser):
|
|
parser.add_argument("--mineblock", dest="mine_block", default=False, action="store_true",
|
|
help="Test double-spend of 1-confirmed transaction")
|
|
parser.add_argument("--segwit", dest="segwit", default=False, action="store_true",
|
|
help="Test behaviour with SegWit txn (which should fail")
|
|
|
|
def setup_network(self):
|
|
# Start with split network:
|
|
super(TxnMallTest, self).setup_network()
|
|
disconnect_nodes(self.nodes[1], 2)
|
|
disconnect_nodes(self.nodes[2], 1)
|
|
|
|
def run_test(self):
|
|
if self.options.segwit:
|
|
output_type = "p2sh-segwit"
|
|
else:
|
|
output_type = "legacy"
|
|
|
|
# All nodes should start with 1,250 BTC:
|
|
starting_balance = 1250
|
|
for i in range(4):
|
|
assert_equal(self.nodes[i].getbalance()['bitcoin'], starting_balance)
|
|
self.nodes[i].getnewaddress() # bug workaround, coins generated assigned to first getnewaddress!
|
|
|
|
self.nodes[0].settxfee(.001)
|
|
|
|
node0_address1 = self.nodes[0].getnewaddress(address_type=output_type)
|
|
node0_txid1 = self.nodes[0].sendtoaddress(node0_address1, 1219)
|
|
node0_tx1 = self.nodes[0].gettransaction(node0_txid1)
|
|
|
|
node0_address2 = self.nodes[0].getnewaddress(address_type=output_type)
|
|
node0_txid2 = self.nodes[0].sendtoaddress(node0_address2, 29)
|
|
node0_tx2 = self.nodes[0].gettransaction(node0_txid2)
|
|
|
|
assert_equal(self.nodes[0].getbalance()['bitcoin'],
|
|
starting_balance + node0_tx1["fee"]['bitcoin'] + node0_tx2["fee"]['bitcoin'])
|
|
|
|
# Coins are sent to node1_address
|
|
node1_address = self.nodes[1].getnewaddress()
|
|
|
|
# Send tx1, and another transaction tx2 that won't be cloned
|
|
txid1 = self.nodes[0].sendtoaddress(node1_address, 40)
|
|
txid2 = self.nodes[0].sendtoaddress(node1_address, 20)
|
|
|
|
# Construct a clone of tx1, to be malleated
|
|
rawtx1 = self.nodes[0].getrawtransaction(txid1, 1)
|
|
clone_inputs = [{"txid": rawtx1["vin"][0]["txid"], "vout": rawtx1["vin"][0]["vout"], "sequence": rawtx1["vin"][0]["sequence"]}]
|
|
clone_outputs = {rawtx1["vout"][0]["scriptPubKey"]["addresses"][0]: rawtx1["vout"][0]["value"],
|
|
rawtx1["vout"][1]["scriptPubKey"]["addresses"][0]: rawtx1["vout"][1]["value"]}
|
|
assert_equal(rawtx1["vout"][2]["scriptPubKey"]["type"], "fee")
|
|
clone_outputs["fee"] = rawtx1["vout"][2]["value"]
|
|
clone_locktime = rawtx1["locktime"]
|
|
clone_raw = self.nodes[0].createrawtransaction(clone_inputs, clone_outputs, clone_locktime)
|
|
|
|
# createrawtransaction randomizes the order of its outputs, so swap them if necessary.
|
|
clone_tx = CTransaction()
|
|
clone_tx.deserialize(io.BytesIO(bytes.fromhex(clone_raw)))
|
|
if (rawtx1["vout"][0]["value"] == 40 and clone_tx.vout[0].nValue.getAmount() != 40*COIN or rawtx1["vout"][0]["value"] != 40 and clone_tx.vout[0].nValue.getAmount() == 40*COIN):
|
|
(clone_tx.vout[0], clone_tx.vout[1]) = (clone_tx.vout[1], clone_tx.vout[0])
|
|
|
|
# Use a different signature hash type to sign. This creates an equivalent but malleated clone.
|
|
# Don't send the clone anywhere yet
|
|
tx1_clone = self.nodes[0].signrawtransactionwithwallet(b2x(clone_tx.serialize()), None, "ALL|ANYONECANPAY")
|
|
assert_equal(tx1_clone["complete"], True)
|
|
|
|
# Have node0 mine a block, if requested:
|
|
if (self.options.mine_block):
|
|
self.nodes[0].generate(1)
|
|
sync_blocks(self.nodes[0:2])
|
|
|
|
tx1 = self.nodes[0].gettransaction(txid1)
|
|
tx2 = self.nodes[0].gettransaction(txid2)
|
|
|
|
# Node0's balance should be starting balance, plus 50BTC for another
|
|
# matured block, minus tx1 and tx2 amounts, and minus transaction fees:
|
|
expected = starting_balance + node0_tx1["fee"]['bitcoin'] + node0_tx2["fee"]['bitcoin']
|
|
if self.options.mine_block:
|
|
expected += 50
|
|
expected += tx1["amount"]['bitcoin'] + tx1["fee"]['bitcoin']
|
|
expected += tx2["amount"]['bitcoin'] + tx2["fee"]['bitcoin']
|
|
assert_equal(self.nodes[0].getbalance()['bitcoin'], expected)
|
|
|
|
if self.options.mine_block:
|
|
assert_equal(tx1["confirmations"], 1)
|
|
assert_equal(tx2["confirmations"], 1)
|
|
else:
|
|
assert_equal(tx1["confirmations"], 0)
|
|
assert_equal(tx2["confirmations"], 0)
|
|
|
|
# Send clone and its parent to miner
|
|
self.nodes[2].sendrawtransaction(node0_tx1["hex"])
|
|
txid1_clone = self.nodes[2].sendrawtransaction(tx1_clone["hex"])
|
|
if self.options.segwit:
|
|
assert_equal(txid1, txid1_clone)
|
|
return
|
|
|
|
# ... mine a block...
|
|
self.nodes[2].generate(1)
|
|
|
|
# Reconnect the split network, and sync chain:
|
|
connect_nodes(self.nodes[1], 2)
|
|
self.nodes[2].sendrawtransaction(node0_tx2["hex"])
|
|
self.nodes[2].sendrawtransaction(tx2["hex"])
|
|
self.nodes[2].generate(1) # Mine another block to make sure we sync
|
|
sync_blocks(self.nodes)
|
|
|
|
# Re-fetch transaction info:
|
|
tx1 = self.nodes[0].gettransaction(txid1)
|
|
tx1_clone = self.nodes[0].gettransaction(txid1_clone)
|
|
tx2 = self.nodes[0].gettransaction(txid2)
|
|
|
|
# Verify expected confirmations
|
|
assert_equal(tx1["confirmations"], -2)
|
|
assert_equal(tx1_clone["confirmations"], 2)
|
|
assert_equal(tx2["confirmations"], 1)
|
|
|
|
# Check node0's total balance; should be same as before the clone, + 100 BTC for 2 matured,
|
|
# less possible orphaned matured subsidy
|
|
expected += 100
|
|
if (self.options.mine_block):
|
|
expected -= 50
|
|
assert_equal(self.nodes[0].getbalance()['bitcoin'], expected)
|
|
|
|
if __name__ == '__main__':
|
|
TxnMallTest().main()
|