feature_nulldummy.py raw
1 #!/usr/bin/env python3
2 # Copyright (c) 2016-present 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 NULLDUMMY softfork.
6
7 Connect to a single node.
8 Generate 2 blocks (save the coinbases for later).
9 Generate COINBASE_MATURITY (CB) more blocks to ensure the coinbases are mature.
10 [Policy/Consensus] Check that NULLDUMMY compliant transactions are accepted in block CB + 3.
11 [Policy] Check that non-NULLDUMMY transactions are rejected before activation.
12 [Consensus] Check that the new NULLDUMMY rules are not enforced on block CB + 4.
13 [Policy/Consensus] Check that the new NULLDUMMY rules are enforced on block CB + 5.
14 """
15 import time
16
17 from test_framework.address import address_to_scriptpubkey
18 from test_framework.blocktools import (
19 COINBASE_MATURITY,
20 NORMAL_GBT_REQUEST_PARAMS,
21 add_witness_commitment,
22 create_block,
23 )
24 from test_framework.messages import (
25 CTransaction,
26 tx_from_hex,
27 )
28 from test_framework.script import (
29 OP_0,
30 OP_TRUE,
31 )
32 from test_framework.test_framework import BitcoinTestFramework
33 from test_framework.util import (
34 assert_equal,
35 assert_raises_rpc_error,
36 )
37 from test_framework.wallet import getnewdestination
38 from test_framework.wallet_util import generate_keypair
39
40 NULLDUMMY_TX_ERROR = "mempool-script-verify-flag-failed (Dummy CHECKMULTISIG argument must be zero)"
41 NULLDUMMY_BLK_ERROR = "block-script-verify-flag-failed (Dummy CHECKMULTISIG argument must be zero)"
42
43 def invalidate_nulldummy_tx(tx):
44 """Transform a NULLDUMMY compliant tx (i.e. scriptSig starts with OP_0)
45 to be non-NULLDUMMY compliant by replacing the dummy with OP_TRUE"""
46 assert_equal(tx.vin[0].scriptSig[0], OP_0)
47 tx.vin[0].scriptSig = bytes([OP_TRUE]) + tx.vin[0].scriptSig[1:]
48
49
50 class NULLDUMMYTest(BitcoinTestFramework):
51 def set_test_params(self):
52 self.num_nodes = 1
53 self.setup_clean_chain = True
54 # This script tests NULLDUMMY activation, which is part of the 'segwit' deployment, so we go through
55 # normal segwit activation here (and don't use the default always-on behaviour).
56 self.extra_args = [[
57 f'-testactivationheight=segwit@{COINBASE_MATURITY + 5}',
58 '-addresstype=legacy',
59 ]]
60
61 def create_transaction(self, *, txid, input_details=None, addr, amount, privkey):
62 input = {"txid": txid, "vout": 0}
63 output = {addr: amount}
64 rawtx = self.nodes[0].createrawtransaction([input], output)
65 # Details only needed for scripthash or witness spends
66 input = None if not input_details else [{**input, **input_details}]
67 signedtx = self.nodes[0].signrawtransactionwithkey(rawtx, [privkey], input)
68 return tx_from_hex(signedtx["hex"])
69
70 def run_test(self):
71 self.privkey, self.pubkey = generate_keypair(wif=True)
72 cms = self.nodes[0].createmultisig(1, [self.pubkey.hex()])
73 wms = self.nodes[0].createmultisig(1, [self.pubkey.hex()], 'p2sh-segwit')
74 self.ms_address = cms["address"]
75 ms_unlock_details = {"scriptPubKey": address_to_scriptpubkey(self.ms_address).hex(),
76 "redeemScript": cms["redeemScript"]}
77 self.wit_ms_address = wms['address']
78
79 self.coinbase_blocks = self.generate(self.nodes[0], 2) # block height = 2
80 coinbase_txid = []
81 for i in self.coinbase_blocks:
82 coinbase_txid.append(self.nodes[0].getblock(i)['tx'][0])
83 self.generate(self.nodes[0], COINBASE_MATURITY) # block height = COINBASE_MATURITY + 2
84 self.lastblockhash = self.nodes[0].getbestblockhash()
85 self.lastblockheight = COINBASE_MATURITY + 2
86 self.lastblocktime = int(time.time()) + self.lastblockheight
87
88 self.log.info(f"Test 1: NULLDUMMY compliant base transactions should be accepted to mempool and mined before activation [{COINBASE_MATURITY + 3}]")
89 test1txs = [self.create_transaction(txid=coinbase_txid[0], addr=self.ms_address, amount=49,
90 privkey=self.nodes[0].get_deterministic_priv_key().key)]
91 txid1 = self.nodes[0].sendrawtransaction(test1txs[0].serialize_with_witness().hex(), 0)
92 test1txs.append(self.create_transaction(txid=txid1, input_details=ms_unlock_details,
93 addr=self.ms_address, amount=48,
94 privkey=self.privkey))
95 txid2 = self.nodes[0].sendrawtransaction(test1txs[1].serialize_with_witness().hex(), 0)
96 test1txs.append(self.create_transaction(txid=coinbase_txid[1],
97 addr=self.wit_ms_address, amount=49,
98 privkey=self.nodes[0].get_deterministic_priv_key().key))
99 txid3 = self.nodes[0].sendrawtransaction(test1txs[2].serialize_with_witness().hex(), 0)
100 self.block_submit(self.nodes[0], test1txs, accept=True)
101
102 self.log.info("Test 2: Non-NULLDUMMY base multisig transaction should not be accepted to mempool before activation")
103 test2tx = self.create_transaction(txid=txid2, input_details=ms_unlock_details,
104 addr=self.ms_address, amount=47,
105 privkey=self.privkey)
106 invalidate_nulldummy_tx(test2tx)
107 assert_raises_rpc_error(-26, NULLDUMMY_TX_ERROR, self.nodes[0].sendrawtransaction, test2tx.serialize_with_witness().hex(), 0)
108
109 self.log.info(f"Test 3: Non-NULLDUMMY base transactions should be accepted in a block before activation [{COINBASE_MATURITY + 4}]")
110 self.block_submit(self.nodes[0], [test2tx], accept=True)
111
112 self.log.info("Test 4: Non-NULLDUMMY base multisig transaction is invalid after activation")
113 test4tx = self.create_transaction(txid=test2tx.txid_hex, input_details=ms_unlock_details,
114 addr=getnewdestination()[2], amount=46,
115 privkey=self.privkey)
116 test6txs = [CTransaction(test4tx)]
117 invalidate_nulldummy_tx(test4tx)
118 assert_raises_rpc_error(-26, NULLDUMMY_TX_ERROR, self.nodes[0].sendrawtransaction, test4tx.serialize_with_witness().hex(), 0)
119 self.block_submit(self.nodes[0], [test4tx], accept=False)
120
121 self.log.info("Test 5: Non-NULLDUMMY P2WSH multisig transaction invalid after activation")
122 test5tx = self.create_transaction(txid=txid3, input_details={"scriptPubKey": test1txs[2].vout[0].scriptPubKey.hex(),
123 "amount": 49, "witnessScript": wms["redeemScript"]},
124 addr=getnewdestination(address_type='p2sh-segwit')[2], amount=48,
125 privkey=self.privkey)
126 test6txs.append(CTransaction(test5tx))
127 test5tx.wit.vtxinwit[0].scriptWitness.stack[0] = b'\x01'
128 assert_raises_rpc_error(-26, NULLDUMMY_TX_ERROR, self.nodes[0].sendrawtransaction, test5tx.serialize_with_witness().hex(), 0)
129 self.block_submit(self.nodes[0], [test5tx], with_witness=True, accept=False)
130
131 self.log.info(f"Test 6: NULLDUMMY compliant base/witness transactions should be accepted to mempool and in block after activation [{COINBASE_MATURITY + 5}]")
132 for i in test6txs:
133 self.nodes[0].sendrawtransaction(i.serialize_with_witness().hex(), 0)
134 self.block_submit(self.nodes[0], test6txs, with_witness=True, accept=True)
135
136 def block_submit(self, node, txs, *, with_witness=False, accept):
137 tmpl = node.getblocktemplate(NORMAL_GBT_REQUEST_PARAMS)
138 assert_equal(tmpl['previousblockhash'], self.lastblockhash)
139 assert_equal(tmpl['height'], self.lastblockheight + 1)
140 block = create_block(tmpl=tmpl, ntime=self.lastblocktime + 1, txlist=txs)
141 if with_witness:
142 add_witness_commitment(block)
143 block.solve()
144 assert_equal(None if accept else NULLDUMMY_BLK_ERROR, node.submitblock(block.serialize().hex()))
145 if accept:
146 assert_equal(node.getbestblockhash(), block.hash_hex)
147 self.lastblockhash = block.hash_hex
148 self.lastblocktime += 1
149 self.lastblockheight += 1
150 else:
151 assert_equal(node.getbestblockhash(), self.lastblockhash)
152
153
154 if __name__ == '__main__':
155 NULLDUMMYTest(__file__).main()
156