mirror of
https://github.com/bitcoin/bitcoin.git
synced 2026-05-05 03:18:50 +02:00
Merge #8872: Remove block-request logic from INV message processing
037159cRemove block-request logic from INV message processing (Matt Corallo)3451203[qa] Respond to getheaders and do not assume a getdata on inv (Matt Corallo)d768f15[qa] Make comptool push blocks instead of relying on inv-fetch (mrbandrews)
This commit is contained in:
@@ -27,6 +27,7 @@ class TestNode(SingleNodeConnCB):
|
||||
self.last_cmpctblock = None
|
||||
self.block_announced = False
|
||||
self.last_getdata = None
|
||||
self.last_getheaders = None
|
||||
self.last_getblocktxn = None
|
||||
self.last_block = None
|
||||
self.last_blocktxn = None
|
||||
@@ -64,6 +65,9 @@ class TestNode(SingleNodeConnCB):
|
||||
def on_getdata(self, conn, message):
|
||||
self.last_getdata = message
|
||||
|
||||
def on_getheaders(self, conn, message):
|
||||
self.last_getheaders = message
|
||||
|
||||
def on_getblocktxn(self, conn, message):
|
||||
self.last_getblocktxn = message
|
||||
|
||||
@@ -396,6 +400,9 @@ class CompactBlocksTest(BitcoinTestFramework):
|
||||
|
||||
if announce == "inv":
|
||||
test_node.send_message(msg_inv([CInv(2, block.sha256)]))
|
||||
success = wait_until(lambda: test_node.last_getheaders is not None, timeout=30)
|
||||
assert(success)
|
||||
test_node.send_header_for_blocks([block])
|
||||
else:
|
||||
test_node.send_header_for_blocks([block])
|
||||
success = wait_until(lambda: test_node.last_getdata is not None, timeout=30)
|
||||
|
||||
@@ -64,6 +64,9 @@ class TestNode(NodeConnCB):
|
||||
self.getdataset.add(inv.hash)
|
||||
self.last_getdata = message
|
||||
|
||||
def on_getheaders(self, conn, message):
|
||||
self.last_getheaders = message
|
||||
|
||||
def on_pong(self, conn, message):
|
||||
self.last_pong = message
|
||||
|
||||
@@ -97,6 +100,10 @@ class TestNode(NodeConnCB):
|
||||
test_function = lambda: self.last_getdata != None
|
||||
self.sync(test_function, timeout)
|
||||
|
||||
def wait_for_getheaders(self, timeout=60):
|
||||
test_function = lambda: self.last_getheaders != None
|
||||
self.sync(test_function, timeout)
|
||||
|
||||
def wait_for_inv(self, expected_inv, timeout=60):
|
||||
test_function = lambda: self.last_inv != expected_inv
|
||||
self.sync(test_function, timeout)
|
||||
@@ -111,12 +118,15 @@ class TestNode(NodeConnCB):
|
||||
def announce_block_and_wait_for_getdata(self, block, use_header, timeout=60):
|
||||
with mininode_lock:
|
||||
self.last_getdata = None
|
||||
self.last_getheaders = None
|
||||
msg = msg_headers()
|
||||
msg.headers = [ CBlockHeader(block) ]
|
||||
if use_header:
|
||||
msg = msg_headers()
|
||||
msg.headers = [ CBlockHeader(block) ]
|
||||
self.send_message(msg)
|
||||
else:
|
||||
self.send_message(msg_inv(inv=[CInv(2, block.sha256)]))
|
||||
self.wait_for_getheaders()
|
||||
self.send_message(msg)
|
||||
self.wait_for_getdata()
|
||||
return
|
||||
|
||||
|
||||
@@ -348,14 +348,13 @@ class SendHeadersTest(BitcoinTestFramework):
|
||||
if j == 0:
|
||||
# Announce via inv
|
||||
test_node.send_block_inv(tip)
|
||||
test_node.wait_for_getdata([tip], timeout=5)
|
||||
test_node.wait_for_getheaders(timeout=5)
|
||||
# Should have received a getheaders now
|
||||
test_node.send_header_for_blocks(blocks)
|
||||
# Test that duplicate inv's won't result in duplicate
|
||||
# getdata requests, or duplicate headers announcements
|
||||
inv_node.send_block_inv(tip)
|
||||
# Should have received a getheaders as well!
|
||||
test_node.send_header_for_blocks(blocks)
|
||||
test_node.wait_for_getdata([x.sha256 for x in blocks[0:-1]], timeout=5)
|
||||
[ inv_node.send_block_inv(x.sha256) for x in blocks[0:-1] ]
|
||||
[ inv_node.send_block_inv(x.sha256) for x in blocks ]
|
||||
test_node.wait_for_getdata([x.sha256 for x in blocks], timeout=5)
|
||||
inv_node.sync_with_ping()
|
||||
else:
|
||||
# Announce via headers
|
||||
|
||||
@@ -111,6 +111,11 @@ class TestNode(NodeConnCB):
|
||||
m.locator = self.block_store.get_locator(self.bestblockhash)
|
||||
self.conn.send_message(m)
|
||||
|
||||
def send_header(self, header):
|
||||
m = msg_headers()
|
||||
m.headers.append(header)
|
||||
self.conn.send_message(m)
|
||||
|
||||
# This assumes BIP31
|
||||
def send_ping(self, nonce):
|
||||
self.pingMap[nonce] = True
|
||||
@@ -345,8 +350,16 @@ class TestManager(object):
|
||||
# Either send inv's to each node and sync, or add
|
||||
# to invqueue for later inv'ing.
|
||||
if (test_instance.sync_every_block):
|
||||
[ c.cb.send_inv(block) for c in self.connections ]
|
||||
self.sync_blocks(block.sha256, 1)
|
||||
# if we expect success, send inv and sync every block
|
||||
# if we expect failure, just push the block and see what happens.
|
||||
if outcome == True:
|
||||
[ c.cb.send_inv(block) for c in self.connections ]
|
||||
self.sync_blocks(block.sha256, 1)
|
||||
else:
|
||||
[ c.send_message(msg_block(block)) for c in self.connections ]
|
||||
[ c.cb.send_ping(self.ping_counter) for c in self.connections ]
|
||||
self.wait_for_pings(self.ping_counter)
|
||||
self.ping_counter += 1
|
||||
if (not self.check_results(tip, outcome)):
|
||||
raise AssertionError("Test failed at test %d" % test_number)
|
||||
else:
|
||||
@@ -354,6 +367,8 @@ class TestManager(object):
|
||||
elif isinstance(b_or_t, CBlockHeader):
|
||||
block_header = b_or_t
|
||||
self.block_store.add_header(block_header)
|
||||
[ c.cb.send_header(block_header) for c in self.connections ]
|
||||
|
||||
else: # Tx test runner
|
||||
assert(isinstance(b_or_t, CTransaction))
|
||||
tx = b_or_t
|
||||
|
||||
Reference in New Issue
Block a user