Tests: Edit bloated varint test and add option for 'barely expensive' tests

This commit is contained in:
mrbandrews 2016-06-08 15:09:01 -04:00
parent 12c5a16c4e
commit ff2dcf274d

View file

@ -65,6 +65,10 @@ class FullBlockTest(ComparisonTestFramework):
self.tip = None self.tip = None
self.blocks = {} self.blocks = {}
def add_options(self, parser):
super().add_options(parser)
parser.add_option("--runbarelyexpensive", dest="runbarelyexpensive", default=True)
def run_test(self): def run_test(self):
self.test = TestManager(self, self.options.tmpdir) self.test = TestManager(self, self.options.tmpdir)
self.test.add_all_connections(self.nodes) self.test.add_all_connections(self.nodes)
@ -875,10 +879,13 @@ class FullBlockTest(ComparisonTestFramework):
yield rejected(RejectResult(16, b'bad-txns-nonfinal')) yield rejected(RejectResult(16, b'bad-txns-nonfinal'))
# This checks that a block with a bloated VARINT between the block_header and the array of tx is rejected # This checks that a block with a bloated VARINT between the block_header and the array of tx such that
# (previous behavior was that it was accepted.) It also checks that if you subsequently send that block # the block is > MAX_BLOCK_SIZE with the bloated varint, but <= MAX_BLOCK_SIZE without the bloated varint,
# with correct encoding, it should be accepted (i.e., the receiving node should not reject it on the # does not cause a subsequent, identical block with canonical encoding to be rejected. The test does not
# basis that it's the same as an already-rejected block, which would be a DoS vulnerability.) # care whether the bloated block is accepted or rejected; it only cares that the second block is accepted.
#
# What matters is that the receiving node should not reject the bloated block, and then reject the canonical
# block on the basis that it's the same as an already-rejected block (which would be a consensus failure.)
# #
# -> b39 (11) -> b42 (12) -> b43 (13) -> b53 (14) -> b55 (15) -> b57 (16) -> b60 (17) -> b64 (18) # -> b39 (11) -> b42 (12) -> b43 (13) -> b53 (14) -> b55 (15) -> b57 (16) -> b60 (17) -> b64 (18)
# \ # \
@ -903,7 +910,7 @@ class FullBlockTest(ComparisonTestFramework):
tx.vin.append(CTxIn(COutPoint(b64a.vtx[1].sha256, 0))) tx.vin.append(CTxIn(COutPoint(b64a.vtx[1].sha256, 0)))
b64a = update_block("64a", [tx]) b64a = update_block("64a", [tx])
assert_equal(len(b64a.serialize()), MAX_BLOCK_SIZE + 8) assert_equal(len(b64a.serialize()), MAX_BLOCK_SIZE + 8)
yield rejected() yield TestInstance([[self.tip, None]])
# comptool workaround: to make sure b64 is delivered, manually erase b64a from blockstore # comptool workaround: to make sure b64 is delivered, manually erase b64a from blockstore
self.test.block_store.erase(b64a.sha256) self.test.block_store.erase(b64a.sha256)
@ -941,7 +948,6 @@ class FullBlockTest(ComparisonTestFramework):
update_block(66, [tx2, tx1]) update_block(66, [tx2, tx1])
yield rejected(RejectResult(16, b'bad-txns-inputs-missingorspent')) yield rejected(RejectResult(16, b'bad-txns-inputs-missingorspent'))
# Attempt to double-spend a transaction created in a block # Attempt to double-spend a transaction created in a block
# #
# -> b43 (13) -> b53 (14) -> b55 (15) -> b57 (16) -> b60 (17) -> b64 (18) -> b65 (19) # -> b43 (13) -> b53 (14) -> b55 (15) -> b57 (16) -> b60 (17) -> b64 (18) -> b65 (19)
@ -1239,46 +1245,48 @@ class FullBlockTest(ComparisonTestFramework):
# Test re-org of a week's worth of blocks (1088 blocks) # Test re-org of a week's worth of blocks (1088 blocks)
# This test takes a minute or two and can be accomplished in memory # This test takes a minute or two and can be accomplished in memory
# #
tip(88) if self.options.runbarelyexpensive:
LARGE_REORG_SIZE = 1088 tip(88)
test1 = TestInstance(sync_every_block=False) LARGE_REORG_SIZE = 1088
spend=out[32] test1 = TestInstance(sync_every_block=False)
for i in range(89, LARGE_REORG_SIZE + 89): spend=out[32]
b = block(i, spend) for i in range(89, LARGE_REORG_SIZE + 89):
tx = CTransaction() b = block(i, spend)
script_length = MAX_BLOCK_SIZE - len(b.serialize()) - 69 tx = CTransaction()
script_output = CScript([b'\x00' * script_length]) script_length = MAX_BLOCK_SIZE - len(b.serialize()) - 69
tx.vout.append(CTxOut(0, script_output)) script_output = CScript([b'\x00' * script_length])
tx.vin.append(CTxIn(COutPoint(b.vtx[1].sha256, 0))) tx.vout.append(CTxOut(0, script_output))
b = update_block(i, [tx]) tx.vin.append(CTxIn(COutPoint(b.vtx[1].sha256, 0)))
assert_equal(len(b.serialize()), MAX_BLOCK_SIZE) b = update_block(i, [tx])
test1.blocks_and_transactions.append([self.tip, True]) assert_equal(len(b.serialize()), MAX_BLOCK_SIZE)
save_spendable_output() test1.blocks_and_transactions.append([self.tip, True])
spend = get_spendable_output() save_spendable_output()
spend = get_spendable_output()
yield test1 yield test1
chain1_tip = i chain1_tip = i
# now create alt chain of same length # now create alt chain of same length
tip(88) tip(88)
test2 = TestInstance(sync_every_block=False) test2 = TestInstance(sync_every_block=False)
for i in range(89, LARGE_REORG_SIZE + 89): for i in range(89, LARGE_REORG_SIZE + 89):
block("alt"+str(i)) block("alt"+str(i))
test2.blocks_and_transactions.append([self.tip, False]) test2.blocks_and_transactions.append([self.tip, False])
yield test2 yield test2
# extend alt chain to trigger re-org # extend alt chain to trigger re-org
block("alt" + str(chain1_tip + 1)) block("alt" + str(chain1_tip + 1))
yield accepted() yield accepted()
# ... and re-org back to the first chain # ... and re-org back to the first chain
tip(chain1_tip) tip(chain1_tip)
block(chain1_tip + 1) block(chain1_tip + 1)
yield rejected() yield rejected()
block(chain1_tip + 2) block(chain1_tip + 2)
yield accepted() yield accepted()
chain1_tip += 2
chain1_tip += 2
if __name__ == '__main__': if __name__ == '__main__':