b0dec4a04a
-debug and -logtimemicros are now set by default. Individual test cases no longer need to set these parameters manually.
188 lines
6.7 KiB
Python
Executable file
188 lines
6.7 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# Copyright (c) 2015-2016 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 BIP66 (DER SIG).
|
|
|
|
Connect to a single node.
|
|
Mine 2 (version 2) blocks (save the coinbases for later).
|
|
Generate 98 more version 2 blocks, verify the node accepts.
|
|
Mine 749 version 3 blocks, verify the node accepts.
|
|
Check that the new DERSIG rules are not enforced on the 750th version 3 block.
|
|
Check that the new DERSIG rules are enforced on the 751st version 3 block.
|
|
Mine 199 new version blocks.
|
|
Mine 1 old-version block.
|
|
Mine 1 new version block.
|
|
Mine 1 old version block, see that the node rejects.
|
|
"""
|
|
|
|
from test_framework.test_framework import ComparisonTestFramework
|
|
from test_framework.util import *
|
|
from test_framework.mininode import CTransaction, NetworkThread
|
|
from test_framework.blocktools import create_coinbase, create_block
|
|
from test_framework.comptool import TestInstance, TestManager
|
|
from test_framework.script import CScript
|
|
from io import BytesIO
|
|
import time
|
|
|
|
# A canonical signature consists of:
|
|
# <30> <total len> <02> <len R> <R> <02> <len S> <S> <hashtype>
|
|
def unDERify(tx):
|
|
"""
|
|
Make the signature in vin 0 of a tx non-DER-compliant,
|
|
by adding padding after the S-value.
|
|
"""
|
|
scriptSig = CScript(tx.vin[0].scriptSig)
|
|
newscript = []
|
|
for i in scriptSig:
|
|
if (len(newscript) == 0):
|
|
newscript.append(i[0:-1] + b'\0' + i[-1:])
|
|
else:
|
|
newscript.append(i)
|
|
tx.vin[0].scriptSig = CScript(newscript)
|
|
|
|
class BIP66Test(ComparisonTestFramework):
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.num_nodes = 1
|
|
|
|
def setup_network(self):
|
|
# Must set the blockversion for this test
|
|
self.nodes = start_nodes(self.num_nodes, self.options.tmpdir,
|
|
extra_args=[['-whitelist=127.0.0.1', '-blockversion=2']],
|
|
binary=[self.options.testbinary])
|
|
|
|
def run_test(self):
|
|
test = TestManager(self, self.options.tmpdir)
|
|
test.add_all_connections(self.nodes)
|
|
NetworkThread().start() # Start up network handling in another thread
|
|
test.run()
|
|
|
|
def create_transaction(self, node, coinbase, to_address, amount):
|
|
from_txid = node.getblock(coinbase)['tx'][0]
|
|
inputs = [{ "txid" : from_txid, "vout" : 0}]
|
|
outputs = { to_address : amount }
|
|
rawtx = node.createrawtransaction(inputs, outputs)
|
|
signresult = node.signrawtransaction(rawtx)
|
|
tx = CTransaction()
|
|
f = BytesIO(hex_str_to_bytes(signresult['hex']))
|
|
tx.deserialize(f)
|
|
return tx
|
|
|
|
def get_tests(self):
|
|
|
|
self.coinbase_blocks = self.nodes[0].generate(2)
|
|
height = 3 # height of the next block to build
|
|
self.tip = int("0x" + self.nodes[0].getbestblockhash(), 0)
|
|
self.nodeaddress = self.nodes[0].getnewaddress()
|
|
self.last_block_time = int(time.time())
|
|
|
|
''' 298 more version 2 blocks '''
|
|
test_blocks = []
|
|
for i in range(298):
|
|
block = create_block(self.tip, create_coinbase(height), self.last_block_time + 1)
|
|
block.nVersion = 2
|
|
block.rehash()
|
|
block.solve()
|
|
test_blocks.append([block, True])
|
|
self.last_block_time += 1
|
|
self.tip = block.sha256
|
|
height += 1
|
|
yield TestInstance(test_blocks, sync_every_block=False)
|
|
|
|
''' Mine 749 version 3 blocks '''
|
|
test_blocks = []
|
|
for i in range(749):
|
|
block = create_block(self.tip, create_coinbase(height), self.last_block_time + 1)
|
|
block.nVersion = 3
|
|
block.rehash()
|
|
block.solve()
|
|
test_blocks.append([block, True])
|
|
self.last_block_time += 1
|
|
self.tip = block.sha256
|
|
height += 1
|
|
yield TestInstance(test_blocks, sync_every_block=False)
|
|
|
|
'''
|
|
Check that the new DERSIG rules are not enforced in the 750th
|
|
version 3 block.
|
|
'''
|
|
spendtx = self.create_transaction(self.nodes[0],
|
|
self.coinbase_blocks[0], self.nodeaddress, 1.0)
|
|
unDERify(spendtx)
|
|
spendtx.rehash()
|
|
|
|
block = create_block(self.tip, create_coinbase(height), self.last_block_time + 1)
|
|
block.nVersion = 3
|
|
block.vtx.append(spendtx)
|
|
block.hashMerkleRoot = block.calc_merkle_root()
|
|
block.rehash()
|
|
block.solve()
|
|
|
|
self.last_block_time += 1
|
|
self.tip = block.sha256
|
|
height += 1
|
|
yield TestInstance([[block, True]])
|
|
|
|
''' Mine 199 new version blocks on last valid tip '''
|
|
test_blocks = []
|
|
for i in range(199):
|
|
block = create_block(self.tip, create_coinbase(height), self.last_block_time + 1)
|
|
block.nVersion = 3
|
|
block.rehash()
|
|
block.solve()
|
|
test_blocks.append([block, True])
|
|
self.last_block_time += 1
|
|
self.tip = block.sha256
|
|
height += 1
|
|
yield TestInstance(test_blocks, sync_every_block=False)
|
|
|
|
''' Mine 1 old version block '''
|
|
block = create_block(self.tip, create_coinbase(height), self.last_block_time + 1)
|
|
block.nVersion = 2
|
|
block.rehash()
|
|
block.solve()
|
|
self.last_block_time += 1
|
|
self.tip = block.sha256
|
|
height += 1
|
|
yield TestInstance([[block, True]])
|
|
|
|
''' Mine 1 new version block '''
|
|
block = create_block(self.tip, create_coinbase(height), self.last_block_time + 1)
|
|
block.nVersion = 3
|
|
block.rehash()
|
|
block.solve()
|
|
self.last_block_time += 1
|
|
self.tip = block.sha256
|
|
height += 1
|
|
yield TestInstance([[block, True]])
|
|
|
|
'''
|
|
Check that the new DERSIG rules are enforced in the 951st version 3
|
|
block.
|
|
'''
|
|
spendtx = self.create_transaction(self.nodes[0],
|
|
self.coinbase_blocks[1], self.nodeaddress, 1.0)
|
|
unDERify(spendtx)
|
|
spendtx.rehash()
|
|
|
|
block = create_block(self.tip, create_coinbase(height), self.last_block_time + 1)
|
|
block.nVersion = 3
|
|
block.vtx.append(spendtx)
|
|
block.hashMerkleRoot = block.calc_merkle_root()
|
|
block.rehash()
|
|
block.solve()
|
|
self.last_block_time += 1
|
|
yield TestInstance([[block, False]])
|
|
|
|
''' Mine 1 old version block, should be invalid '''
|
|
block = create_block(self.tip, create_coinbase(height), self.last_block_time + 1)
|
|
block.nVersion = 2
|
|
block.rehash()
|
|
block.solve()
|
|
self.last_block_time += 1
|
|
yield TestInstance([[block, False]])
|
|
|
|
if __name__ == '__main__':
|
|
BIP66Test().main()
|