Differential D183 Diff 424 qa/rpc-tests/abc-p2p-fullblocktest.py

Changeset View

Standalone View

qa/rpc-tests/abc-p2p-fullblocktest.py

This file was moved from qa/rpc-tests/abc-p2p-fullblocktest-simple.py.

Show All 12 Lines

from test_framework.test_framework import ComparisonTestFramework		from test_framework.test_framework import ComparisonTestFramework
from test_framework.util import *		from test_framework.util import *
from test_framework.comptool import TestManager, TestInstance, RejectResult		from test_framework.comptool import TestManager, TestInstance, RejectResult
from test_framework.blocktools import *		from test_framework.blocktools import *
import time		import time
from test_framework.key import CECKey		from test_framework.key import CECKey
from test_framework.script import *		from test_framework.script import *
		from test_framework.cdefs import MAX_BLOCK_SIGOPS_PER_MB, MAX_TX_SIGOPS_COUNT


class PreviousSpendableOutput(object):		class PreviousSpendableOutput(object):
def __init__(self, tx = CTransaction(), n = -1):		def __init__(self, tx = CTransaction(), n = -1):
self.tx = tx		self.tx = tx
self.n = n # the output we're spending		self.n = n # the output we're spending


▲ Show 20 Lines • Show All 54 Lines • ▼ Show 20 Lines	def sign_tx(self, tx, spend_tx, n):
tx.vin[0].scriptSig = CScript([self.coinbase_key.sign(sighash) + bytes(bytearray([SIGHASH_ALL]))])		tx.vin[0].scriptSig = CScript([self.coinbase_key.sign(sighash) + bytes(bytearray([SIGHASH_ALL]))])

def create_and_sign_transaction(self, spend_tx, n, value, script=CScript([OP_TRUE])):		def create_and_sign_transaction(self, spend_tx, n, value, script=CScript([OP_TRUE])):
tx = self.create_tx(spend_tx, n, value, script)		tx = self.create_tx(spend_tx, n, value, script)
self.sign_tx(tx, spend_tx, n)		self.sign_tx(tx, spend_tx, n)
tx.rehash()		tx.rehash()
return tx		return tx

def next_block(self, number, spend=None, additional_coinbase_value=0, script=None, block_size=0, solve=True):		def next_block(self, number, spend=None, additional_coinbase_value=0, script=None, extra_sigops=0, block_size=0, solve=True):
"""		"""
Create a block on top of self.tip, and advance self.tip to point to the new block		Create a block on top of self.tip, and advance self.tip to point to the new block
if spend is specified, then 1 satoshi will be spent from that to an anyone-can-spend		if spend is specified, then 1 satoshi will be spent from that to an anyone-can-spend
output, and rest will go to fees.		output, and rest will go to fees.
"""		"""
if self.tip == None:		if self.tip == None:
base_block_hash = self.genesis_hash		base_block_hash = self.genesis_hash
block_time = int(time.time())+1		block_time = int(time.time())+1
Show All 36 Lines	def next_block(self, number, spend=None, additional_coinbase_value=0, script=None, extra_sigops=0, block_size=0, solve=True):
self.add_transactions_to_block(block, [tx])		self.add_transactions_to_block(block, [tx])
block.hashMerkleRoot = block.calc_merkle_root()		block.hashMerkleRoot = block.calc_merkle_root()
if spendable_output != None and block_size > 0:		if spendable_output != None and block_size > 0:
while len(block.serialize()) < block_size:		while len(block.serialize()) < block_size:
tx = CTransaction()		tx = CTransaction()
script_length = block_size - len(block.serialize()) - 79		script_length = block_size - len(block.serialize()) - 79
if script_length > 510000:		if script_length > 510000:
script_length = 500000		script_length = 500000
script_output = CScript([b'\x00' * script_length])		tx_sigops = min(extra_sigops, script_length, MAX_TX_SIGOPS_COUNT)
		extra_sigops -= tx_sigops
		script_pad_len = script_length - tx_sigops
		script_output = CScript([b'\x00' * script_pad_len] + [OP_CHECKSIG] * tx_sigops)
tx.vout.append(CTxOut(0, CScript([OP_TRUE])))		tx.vout.append(CTxOut(0, CScript([OP_TRUE])))
tx.vout.append(CTxOut(0, script_output))		tx.vout.append(CTxOut(0, script_output))
tx.vin.append(CTxIn(COutPoint(spendable_output.tx.sha256, spendable_output.n)))		tx.vin.append(CTxIn(COutPoint(spendable_output.tx.sha256, spendable_output.n)))
spendable_output = PreviousSpendableOutput(tx, 0)		spendable_output = PreviousSpendableOutput(tx, 0)
self.add_transactions_to_block(block, [tx])		self.add_transactions_to_block(block, [tx])
block.hashMerkleRoot = block.calc_merkle_root()		block.hashMerkleRoot = block.calc_merkle_root()
# Make sure the math above worked out to produce the correct block size		# Make sure the math above worked out to produce the correct block size
# (the math will fail if there are too many transactions in the block)		# (the math will fail if there are too many transactions in the block)
assert_equal(len(block.serialize()), block_size)		assert_equal(len(block.serialize()), block_size)
		# Make sure all the requested sigops have been included
		assert_equal(extra_sigops, 0)
if solve:		if solve:
block.solve()		block.solve()
self.tip = block		self.tip = block
self.block_heights[block.sha256] = height		self.block_heights[block.sha256] = height
assert number not in self.blocks		assert number not in self.blocks
self.blocks[number] = block		self.blocks[number] = block
return block		return block

▲ Show 20 Lines • Show All 69 Lines • ▼ Show 20 Lines	def get_tests(self):

# block of maximal size		# block of maximal size
block(17, spend=out[16], block_size=self.excessive_block_size)		block(17, spend=out[16], block_size=self.excessive_block_size)
yield accepted()		yield accepted()

# Reject oversized blocks with bad-blk-length error		# Reject oversized blocks with bad-blk-length error
block(18, spend=out[17], block_size=self.excessive_block_size + 1)		block(18, spend=out[17], block_size=self.excessive_block_size + 1)
yield rejected(RejectResult(16, b'bad-blk-length'))		yield rejected(RejectResult(16, b'bad-blk-length'))

		# Rewind bad block.
		tip(17)

		# Accept many sigops
		lots_of_checksigs = CScript([OP_CHECKSIG] * (MAX_BLOCK_SIGOPS_PER_MB - 1))
		block(19, spend=out[17], script=lots_of_checksigs, block_size=ONE_MEGABYTE)
		yield accepted()

		too_many_blk_checksigs = CScript([OP_CHECKSIG] * MAX_BLOCK_SIGOPS_PER_MB)
		block(20, spend=out[18], script=too_many_blk_checksigs, block_size=ONE_MEGABYTE)
		yield rejected(RejectResult(16, b'bad-blk-sigops'))

		# Rewind bad block
		tip(19)

		# Accept 40k sigops per block > 1MB and <= 2MB
		block(21, spend=out[18], script=lots_of_checksigs, extra_sigops=MAX_BLOCK_SIGOPS_PER_MB, block_size=ONE_MEGABYTE + 1)
		yield accepted()

		# Accept 40k sigops per block > 1MB and <= 2MB
		block(22, spend=out[19], script=lots_of_checksigs, extra_sigops=MAX_BLOCK_SIGOPS_PER_MB, block_size=2*ONE_MEGABYTE)
		yield accepted()

		# Reject more than 40k sigops per block > 1MB and <= 2MB.
		block(23, spend=out[20], script=lots_of_checksigs, extra_sigops=MAX_BLOCK_SIGOPS_PER_MB + 1, block_size=ONE_MEGABYTE + 1)
		yield rejected(RejectResult(16, b'bad-blk-sigops'))

		# Rewind bad block
		tip(22)

		# Reject more than 40k sigops per block > 1MB and <= 2MB.
		block(24, spend=out[20], script=lots_of_checksigs, extra_sigops=MAX_BLOCK_SIGOPS_PER_MB + 1, block_size=2*ONE_MEGABYTE)
		yield rejected(RejectResult(16, b'bad-blk-sigops'))

		# Rewind bad block
		tip(22)

		# Accept 60k sigops per block > 2MB and <= 3MB
		block(25, spend=out[20], script=lots_of_checksigs, extra_sigops=2MAX_BLOCK_SIGOPS_PER_MB, block_size=2ONE_MEGABYTE + 1)
		yield accepted()

		# Accept 60k sigops per block > 2MB and <= 3MB
		block(26, spend=out[21], script=lots_of_checksigs, extra_sigops=2MAX_BLOCK_SIGOPS_PER_MB, block_size=3ONE_MEGABYTE)
		yield accepted()

		# Reject more than 40k sigops per block > 1MB and <= 2MB.
		block(27, spend=out[22], script=lots_of_checksigs, extra_sigops=2MAX_BLOCK_SIGOPS_PER_MB + 1, block_size=2ONE_MEGABYTE + 1)
		yield rejected(RejectResult(16, b'bad-blk-sigops'))

		# Rewind bad block
		tip(26)

		# Reject more than 40k sigops per block > 1MB and <= 2MB.
		block(28, spend=out[22], script=lots_of_checksigs, extra_sigops=2MAX_BLOCK_SIGOPS_PER_MB + 1, block_size=3ONE_MEGABYTE)
		yield rejected(RejectResult(16, b'bad-blk-sigops'))

		# Rewind bad block
		tip(26)

		# Too many sigops in one txn
		too_many_tx_checksigs = CScript([OP_CHECKSIG] * (MAX_BLOCK_SIGOPS_PER_MB + 1))
		block(29, spend=out[22], script=too_many_tx_checksigs, block_size=ONE_MEGABYTE + 1)
		yield rejected(RejectResult(16, b'bad-txn-sigops'))


if __name__ == '__main__':		if __name__ == '__main__':
FullBlockTest().main()		FullBlockTest().main()