b93974c3f3
comptool.py creates a tool for running a test suite on top of the mininode p2p framework. It supports two types of tests: those for which we expect certain behavior (acceptance or rejection of a block or transaction) and those for which we are just comparing that the behavior of 2 or more nodes is the same. blockstore.py defines BlockStore and TxStore, which provide db-backed maps between block/tx hashes and the corresponding block or tx. blocktools.py defines utility functions for creating and manipulating blocks and transactions. invalidblockrequest.py is an example test in the comptool framework, which tests the behavior of a single node when sent two different types of invalid blocks (a block with a duplicated transaction and a block with a bad coinbase value).
65 lines
2 KiB
Python
65 lines
2 KiB
Python
# blocktools.py - utilities for manipulating blocks and transactions
|
|
#
|
|
# Distributed under the MIT/X11 software license, see the accompanying
|
|
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
#
|
|
|
|
from mininode import *
|
|
from script import CScript, CScriptOp
|
|
|
|
# Create a block (with regtest difficulty)
|
|
def create_block(hashprev, coinbase, nTime=None):
|
|
block = CBlock()
|
|
if nTime is None:
|
|
import time
|
|
block.nTime = int(time.time()+600)
|
|
else:
|
|
block.nTime = nTime
|
|
block.hashPrevBlock = hashprev
|
|
block.nBits = 0x207fffff # Will break after a difficulty adjustment...
|
|
block.vtx.append(coinbase)
|
|
block.hashMerkleRoot = block.calc_merkle_root()
|
|
block.calc_sha256()
|
|
return block
|
|
|
|
def serialize_script_num(value):
|
|
r = bytearray(0)
|
|
if value == 0:
|
|
return r
|
|
neg = value < 0
|
|
absvalue = -value if neg else value
|
|
while (absvalue):
|
|
r.append(chr(absvalue & 0xff))
|
|
absvalue >>= 8
|
|
if r[-1] & 0x80:
|
|
r.append(0x80 if neg else 0)
|
|
elif neg:
|
|
r[-1] |= 0x80
|
|
return r
|
|
|
|
counter=1
|
|
# Create an anyone-can-spend coinbase transaction, assuming no miner fees
|
|
def create_coinbase(heightAdjust = 0):
|
|
global counter
|
|
coinbase = CTransaction()
|
|
coinbase.vin.append(CTxIn(COutPoint(0, 0xffffffff),
|
|
ser_string(serialize_script_num(counter+heightAdjust)), 0xffffffff))
|
|
counter += 1
|
|
coinbaseoutput = CTxOut()
|
|
coinbaseoutput.nValue = 50*100000000
|
|
halvings = int((counter+heightAdjust)/150) # regtest
|
|
coinbaseoutput.nValue >>= halvings
|
|
coinbaseoutput.scriptPubKey = ""
|
|
coinbase.vout = [ coinbaseoutput ]
|
|
coinbase.calc_sha256()
|
|
return coinbase
|
|
|
|
# Create a transaction with an anyone-can-spend output, that spends the
|
|
# nth output of prevtx.
|
|
def create_transaction(prevtx, n, sig, value):
|
|
tx = CTransaction()
|
|
assert(n < len(prevtx.vout))
|
|
tx.vin.append(CTxIn(COutPoint(prevtx.sha256, n), sig, 0xffffffff))
|
|
tx.vout.append(CTxOut(value, ""))
|
|
tx.calc_sha256()
|
|
return tx
|