2 # Copyright (c) 2014-2016 The Bitcoin Core developers
3 # Distributed under the MIT software license, see the accompanying
4 # file COPYING or http://www.opensource.org/licenses/mit-license.php.
5 """Test the wallet accounts properly when there are cloned transactions with malleated scriptsigs."""
7 from test_framework
.test_framework
import BitcoinTestFramework
8 from test_framework
.util
import *
10 class TxnMallTest(BitcoinTestFramework
):
11 def add_options(self
, parser
):
12 parser
.add_option("--mineblock", dest
="mine_block", default
=False, action
="store_true",
13 help="Test double-spend of 1-confirmed transaction")
15 def setup_network(self
):
16 # Start with split network:
17 super(TxnMallTest
, self
).setup_network()
18 disconnect_nodes(self
.nodes
[1], 2)
19 disconnect_nodes(self
.nodes
[2], 1)
22 # All nodes should start with 1,250 BTC:
23 starting_balance
= 1250
25 assert_equal(self
.nodes
[i
].getbalance(), starting_balance
)
26 self
.nodes
[i
].getnewaddress("") # bug workaround, coins generated assigned to first getnewaddress!
28 # Assign coins to foo and bar accounts:
29 self
.nodes
[0].settxfee(.001)
31 node0_address_foo
= self
.nodes
[0].getnewaddress("foo")
32 fund_foo_txid
= self
.nodes
[0].sendfrom("", node0_address_foo
, 1219)
33 fund_foo_tx
= self
.nodes
[0].gettransaction(fund_foo_txid
)
35 node0_address_bar
= self
.nodes
[0].getnewaddress("bar")
36 fund_bar_txid
= self
.nodes
[0].sendfrom("", node0_address_bar
, 29)
37 fund_bar_tx
= self
.nodes
[0].gettransaction(fund_bar_txid
)
39 assert_equal(self
.nodes
[0].getbalance(""),
40 starting_balance
- 1219 - 29 + fund_foo_tx
["fee"] + fund_bar_tx
["fee"])
42 # Coins are sent to node1_address
43 node1_address
= self
.nodes
[1].getnewaddress("from0")
45 # Send tx1, and another transaction tx2 that won't be cloned
46 txid1
= self
.nodes
[0].sendfrom("foo", node1_address
, 40, 0)
47 txid2
= self
.nodes
[0].sendfrom("bar", node1_address
, 20, 0)
49 # Construct a clone of tx1, to be malleated
50 rawtx1
= self
.nodes
[0].getrawtransaction(txid1
,1)
51 clone_inputs
= [{"txid":rawtx1
["vin"][0]["txid"],"vout":rawtx1
["vin"][0]["vout"]}]
52 clone_outputs
= {rawtx1
["vout"][0]["scriptPubKey"]["addresses"][0]:rawtx1
["vout"][0]["value"],
53 rawtx1
["vout"][1]["scriptPubKey"]["addresses"][0]:rawtx1
["vout"][1]["value"]}
54 clone_locktime
= rawtx1
["locktime"]
55 clone_raw
= self
.nodes
[0].createrawtransaction(clone_inputs
, clone_outputs
, clone_locktime
)
57 # createrawtransaction randomizes the order of its outputs, so swap them if necessary.
58 # output 0 is at version+#inputs+input+sigstub+sequence+#outputs
59 # 40 BTC serialized is 00286bee00000000
60 pos0
= 2*(4+1+36+1+4+1)
61 hex40
= "00286bee00000000"
62 output_len
= 16 + 2 + 2 * int("0x" + clone_raw
[pos0
+ 16 : pos0
+ 16 + 2], 0)
63 if (rawtx1
["vout"][0]["value"] == 40 and clone_raw
[pos0
: pos0
+ 16] != hex40
or
64 rawtx1
["vout"][0]["value"] != 40 and clone_raw
[pos0
: pos0
+ 16] == hex40
):
65 output0
= clone_raw
[pos0
: pos0
+ output_len
]
66 output1
= clone_raw
[pos0
+ output_len
: pos0
+ 2 * output_len
]
67 clone_raw
= clone_raw
[:pos0
] + output1
+ output0
+ clone_raw
[pos0
+ 2 * output_len
:]
69 # Use a different signature hash type to sign. This creates an equivalent but malleated clone.
70 # Don't send the clone anywhere yet
71 tx1_clone
= self
.nodes
[0].signrawtransaction(clone_raw
, None, None, "ALL|ANYONECANPAY")
72 assert_equal(tx1_clone
["complete"], True)
74 # Have node0 mine a block, if requested:
75 if (self
.options
.mine_block
):
76 self
.nodes
[0].generate(1)
77 sync_blocks(self
.nodes
[0:2])
79 tx1
= self
.nodes
[0].gettransaction(txid1
)
80 tx2
= self
.nodes
[0].gettransaction(txid2
)
82 # Node0's balance should be starting balance, plus 50BTC for another
83 # matured block, minus tx1 and tx2 amounts, and minus transaction fees:
84 expected
= starting_balance
+ fund_foo_tx
["fee"] + fund_bar_tx
["fee"]
85 if self
.options
.mine_block
: expected
+= 50
86 expected
+= tx1
["amount"] + tx1
["fee"]
87 expected
+= tx2
["amount"] + tx2
["fee"]
88 assert_equal(self
.nodes
[0].getbalance(), expected
)
90 # foo and bar accounts should be debited:
91 assert_equal(self
.nodes
[0].getbalance("foo", 0), 1219 + tx1
["amount"] + tx1
["fee"])
92 assert_equal(self
.nodes
[0].getbalance("bar", 0), 29 + tx2
["amount"] + tx2
["fee"])
94 if self
.options
.mine_block
:
95 assert_equal(tx1
["confirmations"], 1)
96 assert_equal(tx2
["confirmations"], 1)
97 # Node1's "from0" balance should be both transaction amounts:
98 assert_equal(self
.nodes
[1].getbalance("from0"), -(tx1
["amount"] + tx2
["amount"]))
100 assert_equal(tx1
["confirmations"], 0)
101 assert_equal(tx2
["confirmations"], 0)
103 # Send clone and its parent to miner
104 self
.nodes
[2].sendrawtransaction(fund_foo_tx
["hex"])
105 txid1_clone
= self
.nodes
[2].sendrawtransaction(tx1_clone
["hex"])
106 # ... mine a block...
107 self
.nodes
[2].generate(1)
109 # Reconnect the split network, and sync chain:
110 connect_nodes(self
.nodes
[1], 2)
111 self
.nodes
[2].sendrawtransaction(fund_bar_tx
["hex"])
112 self
.nodes
[2].sendrawtransaction(tx2
["hex"])
113 self
.nodes
[2].generate(1) # Mine another block to make sure we sync
114 sync_blocks(self
.nodes
)
116 # Re-fetch transaction info:
117 tx1
= self
.nodes
[0].gettransaction(txid1
)
118 tx1_clone
= self
.nodes
[0].gettransaction(txid1_clone
)
119 tx2
= self
.nodes
[0].gettransaction(txid2
)
121 # Verify expected confirmations
122 assert_equal(tx1
["confirmations"], -2)
123 assert_equal(tx1_clone
["confirmations"], 2)
124 assert_equal(tx2
["confirmations"], 1)
126 # Check node0's total balance; should be same as before the clone, + 100 BTC for 2 matured,
127 # less possible orphaned matured subsidy
129 if (self
.options
.mine_block
):
131 assert_equal(self
.nodes
[0].getbalance(), expected
)
132 assert_equal(self
.nodes
[0].getbalance("*", 0), expected
)
134 # Check node0's individual account balances.
135 # "foo" should have been debited by the equivalent clone of tx1
136 assert_equal(self
.nodes
[0].getbalance("foo"), 1219 + tx1
["amount"] + tx1
["fee"])
137 # "bar" should have been debited by (possibly unconfirmed) tx2
138 assert_equal(self
.nodes
[0].getbalance("bar", 0), 29 + tx2
["amount"] + tx2
["fee"])
139 # "" should have starting balance, less funding txes, plus subsidies
140 assert_equal(self
.nodes
[0].getbalance("", 0), starting_balance
147 # Node1's "from0" account balance
148 assert_equal(self
.nodes
[1].getbalance("from0", 0), -(tx1
["amount"] + tx2
["amount"]))
150 if __name__
== '__main__':