mirror of
https://github.com/dashpay/dash.git
synced 2024-12-28 05:23:01 +01:00
b6544791b4
fa5b58ea01fac1adb6336b8b6b5217193295c695 test: Avoid intermittent block download timeout in p2p_ibd_stalling (MarcoFalke) Pull request description: Fixes #30704 The goal of the test is to check the stalling timeout, not the block download timeout. On extremely slow hardware (for example qemu virtual hardware), downloading the 1023 blocks may take longer than the block download timeout. Fix it by pinning the time using mocktime, and only advance it when testing the stalling timeout. ACKs for top commit: tdb3: CR ACK fa5b58ea01fac1adb6336b8b6b5217193295c695 brunoerg: utACK fa5b58ea01fac1adb6336b8b6b5217193295c695 Tree-SHA512: 9a9221f264bea52be5e9fe81fd319f5a6970cd315cc5e9f5e2e049c5d84619b19b9f6f075cda8d34565c2d6c17a88fb57e195c66c271e40f73119a77caecb6d7
170 lines
6.5 KiB
Python
Executable File
170 lines
6.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright (c) 2022- The Bitcoin Core developers
|
|
# Distributed under the MIT software license, see the accompanying
|
|
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
"""
|
|
Test stalling logic during IBD
|
|
"""
|
|
|
|
import time
|
|
|
|
from test_framework.blocktools import (
|
|
create_block,
|
|
create_coinbase
|
|
)
|
|
from test_framework.messages import (
|
|
MSG_BLOCK,
|
|
MSG_TYPE_MASK,
|
|
NODE_NETWORK,
|
|
NODE_BLOOM,
|
|
)
|
|
from test_framework.p2p import (
|
|
CBlockHeader,
|
|
msg_block,
|
|
msg_headers,
|
|
P2PDataStore,
|
|
)
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
from test_framework.util import (
|
|
assert_equal,
|
|
)
|
|
|
|
|
|
class P2PStaller(P2PDataStore):
|
|
def __init__(self, stall_block):
|
|
self.stall_block = stall_block
|
|
super().__init__()
|
|
|
|
def on_getdata(self, message):
|
|
for inv in message.inv:
|
|
self.getdata_requests.append(inv.hash)
|
|
if (inv.type & MSG_TYPE_MASK) == MSG_BLOCK:
|
|
if (inv.hash != self.stall_block):
|
|
self.send_message(msg_block(self.block_store[inv.hash]))
|
|
|
|
def on_getheaders(self, message):
|
|
pass
|
|
|
|
class P2PIBDStallingTest(BitcoinTestFramework):
|
|
def set_test_params(self):
|
|
self.disable_mocktime = True
|
|
self.extra_args = [["-dip3params=2000:2000", "-testactivationheight=v20@2000"]]
|
|
self.setup_clean_chain = True
|
|
self.num_nodes = 1
|
|
|
|
def run_test(self):
|
|
NUM_BLOCKS = 1025
|
|
NUM_PEERS = 4
|
|
node = self.nodes[0]
|
|
tip = int(node.getbestblockhash(), 16)
|
|
blocks = []
|
|
height = 1
|
|
block_time = node.getblock(node.getbestblockhash())['time'] + 1
|
|
self.log.info("Prepare blocks without sending them to the node")
|
|
block_dict = {}
|
|
for _ in range(NUM_BLOCKS):
|
|
blocks.append(create_block(tip, create_coinbase(height), block_time))
|
|
blocks[-1].solve()
|
|
tip = blocks[-1].sha256
|
|
block_time += 1
|
|
height += 1
|
|
block_dict[blocks[-1].sha256] = blocks[-1]
|
|
stall_block = blocks[0].sha256
|
|
|
|
headers_message = msg_headers()
|
|
headers_message.headers = [CBlockHeader(b) for b in blocks[:NUM_BLOCKS-1]]
|
|
peers = []
|
|
|
|
self.log.info("Check that a staller does not get disconnected if the 1024 block lookahead buffer is filled")
|
|
self.mocktime = int(time.time()) + 1
|
|
for id in range(NUM_PEERS):
|
|
peers.append(node.add_outbound_p2p_connection(P2PStaller(stall_block), services = NODE_NETWORK | NODE_BLOOM, p2p_idx=id, connection_type="outbound-full-relay"))
|
|
peers[-1].block_store = block_dict
|
|
peers[-1].send_message(headers_message)
|
|
|
|
# Need to wait until 1023 blocks are received - the magic total bytes number is a workaround in lack of an rpc
|
|
# returning the number of downloaded (but not connected) blocks.
|
|
bytes_recv = 172761 if not self.options.v2transport else 169692
|
|
self.wait_until(lambda: self.total_bytes_recv_for_blocks() == bytes_recv)
|
|
|
|
self.all_sync_send_with_ping(peers)
|
|
# If there was a peer marked for stalling, it would get disconnected
|
|
self.mocktime += 3
|
|
node.setmocktime(self.mocktime)
|
|
self.all_sync_send_with_ping(peers)
|
|
assert_equal(node.num_test_p2p_connections(), NUM_PEERS)
|
|
|
|
self.log.info("Check that increasing the window beyond 1024 blocks triggers stalling logic")
|
|
headers_message.headers = [CBlockHeader(b) for b in blocks]
|
|
with node.assert_debug_log(expected_msgs=['Stall started']):
|
|
for p in peers:
|
|
p.send_message(headers_message)
|
|
self.all_sync_send_with_ping(peers)
|
|
|
|
self.log.info("Check that the stalling peer is disconnected after 2 seconds")
|
|
self.mocktime += 3
|
|
node.setmocktime(self.mocktime)
|
|
peers[0].wait_for_disconnect()
|
|
assert_equal(node.num_test_p2p_connections(), NUM_PEERS - 1)
|
|
self.wait_until(lambda: self.is_block_requested(peers, stall_block))
|
|
# Make sure that SendMessages() is invoked, which assigns the missing block
|
|
# to another peer and starts the stalling logic for them
|
|
self.all_sync_send_with_ping(peers)
|
|
|
|
self.log.info("Check that the stalling timeout gets doubled to 4 seconds for the next staller")
|
|
# No disconnect after just 3 seconds
|
|
self.mocktime += 3
|
|
node.setmocktime(self.mocktime)
|
|
self.all_sync_send_with_ping(peers)
|
|
assert_equal(node.num_test_p2p_connections(), NUM_PEERS - 1)
|
|
|
|
self.mocktime += 2
|
|
node.setmocktime(self.mocktime)
|
|
self.wait_until(lambda: sum(x.is_connected for x in node.p2ps) == NUM_PEERS - 2)
|
|
self.wait_until(lambda: self.is_block_requested(peers, stall_block))
|
|
self.all_sync_send_with_ping(peers)
|
|
|
|
self.log.info("Check that the stalling timeout gets doubled to 8 seconds for the next staller")
|
|
# No disconnect after just 7 seconds
|
|
self.mocktime += 7
|
|
node.setmocktime(self.mocktime)
|
|
self.all_sync_send_with_ping(peers)
|
|
assert_equal(node.num_test_p2p_connections(), NUM_PEERS - 2)
|
|
|
|
self.mocktime += 2
|
|
node.setmocktime(self.mocktime)
|
|
self.wait_until(lambda: sum(x.is_connected for x in node.p2ps) == NUM_PEERS - 3)
|
|
self.wait_until(lambda: self.is_block_requested(peers, stall_block))
|
|
self.all_sync_send_with_ping(peers)
|
|
|
|
self.log.info("Provide the withheld block and check that stalling timeout gets reduced back to 2 seconds")
|
|
with node.assert_debug_log(expected_msgs=['Decreased stalling timeout to 2 seconds']):
|
|
for p in peers:
|
|
if p.is_connected and (stall_block in p.getdata_requests):
|
|
p.send_message(msg_block(block_dict[stall_block]))
|
|
|
|
self.log.info("Check that all outstanding blocks get connected")
|
|
self.wait_until(lambda: node.getblockcount() == NUM_BLOCKS)
|
|
|
|
def total_bytes_recv_for_blocks(self):
|
|
total = 0
|
|
for info in self.nodes[0].getpeerinfo():
|
|
if ("block" in info["bytesrecv_per_msg"].keys()):
|
|
total += info["bytesrecv_per_msg"]["block"]
|
|
return total
|
|
|
|
def all_sync_send_with_ping(self, peers):
|
|
for p in peers:
|
|
if p.is_connected:
|
|
p.sync_send_with_ping()
|
|
|
|
def is_block_requested(self, peers, hash):
|
|
for p in peers:
|
|
if p.is_connected and (hash in p.getdata_requests):
|
|
return True
|
|
return False
|
|
|
|
|
|
if __name__ == '__main__':
|
|
P2PIBDStallingTest().main()
|