Changeset View
Changeset View
Standalone View
Standalone View
test/functional/p2p_invalid_block.py
#!/usr/bin/env python3 | #!/usr/bin/env python3 | ||||
# Copyright (c) 2015-2016 The Bitcoin Core developers | # Copyright (c) 2015-2016 The Bitcoin Core developers | ||||
# Distributed under the MIT software license, see the accompanying | # Distributed under the MIT software license, see the accompanying | ||||
# file COPYING or http://www.opensource.org/licenses/mit-license.php. | # file COPYING or http://www.opensource.org/licenses/mit-license.php. | ||||
"""Test node responses to invalid blocks. | """Test node responses to invalid blocks. | ||||
In this test we connect to one node over p2p, and test block requests: | In this test we connect to one node over p2p, and test block requests: | ||||
1) Valid blocks should be requested and become chain tip. | 1) Valid blocks should be requested and become chain tip. | ||||
2) Invalid block with duplicated transaction should be re-requested. | 2) Invalid block with duplicated transaction should be re-requested. | ||||
3) Invalid block with bad coinbase value should be rejected and not | 3) Invalid block with bad coinbase value should be rejected and not | ||||
re-requested. | re-requested. | ||||
""" | """ | ||||
import copy | import copy | ||||
import time | |||||
from test_framework.blocktools import ( | from test_framework.blocktools import ( | ||||
create_block, | create_block, | ||||
create_coinbase, | create_coinbase, | ||||
create_transaction, | create_transaction, | ||||
) | ) | ||||
from test_framework.comptool import RejectResult, TestInstance, TestManager | |||||
from test_framework.messages import COIN | from test_framework.messages import COIN | ||||
from test_framework.mininode import network_thread_start | from test_framework.mininode import network_thread_start, P2PDataStore | ||||
from test_framework.test_framework import ComparisonTestFramework | from test_framework.test_framework import BitcoinTestFramework | ||||
from test_framework.util import assert_equal | from test_framework.util import assert_equal | ||||
# Use the ComparisonTestFramework with 1 node: only use --testbinary. | class InvalidBlockRequestTest(BitcoinTestFramework): | ||||
class InvalidBlockRequestTest(ComparisonTestFramework): | |||||
''' Can either run this test as 1 node with expected answers, or two and compare them. | |||||
Change the "outcome" variable from each TestInstance object to only do the comparison. ''' | |||||
def set_test_params(self): | def set_test_params(self): | ||||
self.num_nodes = 1 | self.num_nodes = 1 | ||||
self.setup_clean_chain = True | self.setup_clean_chain = True | ||||
self.extra_args = [["-whitelist=127.0.0.1"]] | |||||
def run_test(self): | def run_test(self): | ||||
test = TestManager(self, self.options.tmpdir) | # Add p2p connection to node0 | ||||
test.add_all_connections(self.nodes) | node = self.nodes[0] # convenience reference to the node | ||||
self.tip = None | node.add_p2p_connection(P2PDataStore()) | ||||
self.block_time = None | |||||
self.extra_args = [["-whitelist=127.0.0.1"]] | |||||
network_thread_start() | network_thread_start() | ||||
test.run() | node.p2p.wait_for_verack() | ||||
best_block = node.getblock(node.getbestblockhash()) | |||||
tip = int(node.getbestblockhash(), 16) | |||||
height = best_block["height"] + 1 | |||||
block_time = best_block["time"] + 1 | |||||
self.log.info("Create a new block with an anyone-can-spend coinbase") | |||||
def get_tests(self): | |||||
if self.tip is None: | |||||
self.tip = int("0x" + self.nodes[0].getbestblockhash(), 0) | |||||
self.block_time = int(time.time()) + 1 | |||||
''' | |||||
Create a new block with an anyone-can-spend coinbase | |||||
''' | |||||
height = 1 | height = 1 | ||||
block = create_block( | block = create_block(tip, create_coinbase(height), block_time) | ||||
self.tip, create_coinbase(height), self.block_time) | |||||
self.block_time += 1 | |||||
block.solve() | block.solve() | ||||
self.tip = block.sha256 | # Save the coinbase for later | ||||
height += 1 | |||||
yield TestInstance([[block, True]]) | |||||
block1 = block | block1 = block | ||||
tip = block.sha256 | |||||
node.p2p.send_blocks_and_test([block1], node, True) | |||||
''' | self.log.info("Mature the block.") | ||||
Now we need that block to mature so we can spend the coinbase. | node.generate(100) | ||||
''' | |||||
test = TestInstance(sync_every_block=False) | best_block = node.getblock(node.getbestblockhash()) | ||||
for i in range(100): | tip = int(node.getbestblockhash(), 16) | ||||
block = create_block( | height = best_block["height"] + 1 | ||||
self.tip, create_coinbase(height), self.block_time) | block_time = best_block["time"] + 1 | ||||
block.solve() | |||||
self.tip = block.sha256 | # Use merkle-root malleability to generate an invalid block with | ||||
self.block_time += 1 | # same blockheader. | ||||
test.blocks_and_transactions.append([block, True]) | # Manufacture a block with 3 transactions (coinbase, spend of prior | ||||
height += 1 | # coinbase, spend of that spend). Duplicate the 3rd transaction to | ||||
yield test | # leave merkle root and blockheader unchanged but invalidate the block. | ||||
self.log.info("Test merkle root malleability.") | |||||
assert(block.sha256 == int(self.nodes[0].getbestblockhash(), 16)) | |||||
block2 = create_block(tip, create_coinbase(height), block_time) | |||||
''' | block_time += 1 | ||||
Now we use merkle-root malleability to generate an invalid block with | |||||
same blockheader. | |||||
Manufacture a block with 3 transactions (coinbase, spend of prior | |||||
coinbase, spend of that spend). Duplicate the 3rd transaction to | |||||
leave merkle root and blockheader unchanged but invalidate the block. | |||||
''' | |||||
block2 = create_block( | |||||
self.tip, create_coinbase(height), self.block_time) | |||||
self.block_time += 1 | |||||
# b'0x51' is OP_TRUE | # b'0x51' is OP_TRUE | ||||
tx1 = create_transaction(block1.vtx[0], 0, b'', 50 * COIN) | tx1 = create_transaction(block1.vtx[0], 0, b'', 50 * COIN) | ||||
tx2 = create_transaction(tx1, 0, b'\x51', 50 * COIN) | tx2 = create_transaction(tx1, 0, b'\x51', 50 * COIN) | ||||
block2.vtx.extend([tx1, tx2]) | block2.vtx.extend([tx1, tx2]) | ||||
block2.vtx = [block2.vtx[0]] + \ | block2.vtx = [block2.vtx[0]] + \ | ||||
sorted(block2.vtx[1:], key=lambda tx: tx.get_id()) | sorted(block2.vtx[1:], key=lambda tx: tx.get_id()) | ||||
block2.hashMerkleRoot = block2.calc_merkle_root() | block2.hashMerkleRoot = block2.calc_merkle_root() | ||||
block2.rehash() | block2.rehash() | ||||
block2.solve() | block2.solve() | ||||
orig_hash = block2.sha256 | orig_hash = block2.sha256 | ||||
block2_orig = copy.deepcopy(block2) | block2_orig = copy.deepcopy(block2) | ||||
# Mutate block 2 | # Mutate block 2 | ||||
block2.vtx.append(block2.vtx[2]) | block2.vtx.append(block2.vtx[2]) | ||||
assert_equal(block2.hashMerkleRoot, block2.calc_merkle_root()) | assert_equal(block2.hashMerkleRoot, block2.calc_merkle_root()) | ||||
assert_equal(orig_hash, block2.rehash()) | assert_equal(orig_hash, block2.rehash()) | ||||
assert(block2_orig.vtx != block2.vtx) | assert(block2_orig.vtx != block2.vtx) | ||||
self.tip = block2.sha256 | node.p2p.send_blocks_and_test( | ||||
[block2], node, False, False, 16, b'bad-txns-duplicate') | |||||
yield TestInstance([[block2, RejectResult(16, b'bad-txns-duplicate')]]) | |||||
yield TestInstance([[block2_orig, True]]) | |||||
height += 1 | |||||
# Check transactions for duplicate inputs | # Check transactions for duplicate inputs | ||||
self.log.info("Test duplicate input block.") | self.log.info("Test duplicate input block.") | ||||
block2_orig.vtx[2].vin.append(block2_orig.vtx[2].vin[0]) | block2_orig.vtx[2].vin.append(block2_orig.vtx[2].vin[0]) | ||||
block2.vtx = [block2.vtx[0]] + \ | block2.vtx = [block2.vtx[0]] + \ | ||||
sorted(block2.vtx[1:], key=lambda tx: tx.get_id()) | sorted(block2.vtx[1:], key=lambda tx: tx.get_id()) | ||||
block2_orig.vtx[2].rehash() | block2_orig.vtx[2].rehash() | ||||
block2_orig.hashMerkleRoot = block2_orig.calc_merkle_root() | block2_orig.hashMerkleRoot = block2_orig.calc_merkle_root() | ||||
block2_orig.rehash() | block2_orig.rehash() | ||||
block2_orig.solve() | block2_orig.solve() | ||||
yield TestInstance([[block2_orig, RejectResult(16, b'bad-txns-inputs-duplicate')]]) | node.p2p.send_blocks_and_test( | ||||
[block2_orig], node, False, False, 16, b'bad-txns-inputs-duplicate') | |||||
self.log.info("Test very broken block.") | |||||
''' | block3 = create_block(tip, create_coinbase(height), block_time) | ||||
Make sure that a totally screwed up block is not valid. | block_time += 1 | ||||
''' | block3.vtx[0].vout[0].nValue = 100 * COIN # Too high! | ||||
block3 = create_block( | |||||
self.tip, create_coinbase(height), self.block_time) | |||||
self.block_time += 1 | |||||
block3.vtx[0].vout[0].nValue = 51 * COIN # Too high! | |||||
block3.vtx[0].sha256 = None | block3.vtx[0].sha256 = None | ||||
block3.vtx[0].calc_sha256() | block3.vtx[0].calc_sha256() | ||||
block3.hashMerkleRoot = block3.calc_merkle_root() | block3.hashMerkleRoot = block3.calc_merkle_root() | ||||
block3.rehash() | block3.rehash() | ||||
block3.solve() | block3.solve() | ||||
yield TestInstance([[block3, RejectResult(16, b'bad-cb-amount')]]) | |||||
node.p2p.send_blocks_and_test( | |||||
[block3], node, False, False, 16, b'bad-cb-amount') | |||||
if __name__ == '__main__': | if __name__ == '__main__': | ||||
InvalidBlockRequestTest().main() | InvalidBlockRequestTest().main() |