mirror of
https://github.com/dashpay/dash.git
synced 2024-12-27 13:03:17 +01:00
588b8e5caf
66f861aAdd a test for P2P inactivity timeouts (Matt Corallo)b436f92qa: Expose on-connection to mininode listeners (Matt Corallo)8aaba7aqa: mininode learns when a socket connects, not its first action (Matt Corallo)2cbd119Disconnect peers which we do not receive VERACKs from within 60 sec (Matt Corallo)
104 lines
3.4 KiB
Python
Executable File
104 lines
3.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright (c) 2016 The Bitcoin Core developers
|
|
# Distributed under the MIT software license, see the accompanying
|
|
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
""" TimeoutsTest -- test various net timeouts (only in extended tests)
|
|
|
|
- Create three bitcoind nodes:
|
|
|
|
no_verack_node - we never send a verack in response to their version
|
|
no_version_node - we never send a version (only a ping)
|
|
no_send_node - we never send any P2P message.
|
|
|
|
- Start all three nodes
|
|
- Wait 1 second
|
|
- Assert that we're connected
|
|
- Send a ping to no_verack_node and no_version_node
|
|
- Wait 30 seconds
|
|
- Assert that we're still connected
|
|
- Send a ping to no_verack_node and no_version_node
|
|
- Wait 31 seconds
|
|
- Assert that we're no longer connected (timeout to receive version/verack is 60 seconds)
|
|
"""
|
|
|
|
from time import sleep
|
|
|
|
from test_framework.mininode import *
|
|
from test_framework.test_framework import BitcoinTestFramework
|
|
from test_framework.util import *
|
|
|
|
class TestNode(SingleNodeConnCB):
|
|
def __init__(self):
|
|
SingleNodeConnCB.__init__(self)
|
|
self.connected = False
|
|
self.received_version = False
|
|
|
|
def on_open(self, conn):
|
|
self.connected = True
|
|
|
|
def on_close(self, conn):
|
|
self.connected = False
|
|
|
|
def on_version(self, conn, message):
|
|
# Don't send a verack in response
|
|
self.received_version = True
|
|
|
|
class TimeoutsTest(BitcoinTestFramework):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.setup_clean_chain = True
|
|
self.num_nodes = 1
|
|
|
|
def setup_network(self):
|
|
self.nodes = []
|
|
|
|
# Start up node0 to be a version 1, pre-segwit node.
|
|
self.nodes = start_nodes(self.num_nodes, self.options.tmpdir,
|
|
[["-debug", "-logtimemicros=1"]])
|
|
|
|
def run_test(self):
|
|
# Setup the p2p connections and start up the network thread.
|
|
self.no_verack_node = TestNode() # never send verack
|
|
self.no_version_node = TestNode() # never send version (just ping)
|
|
self.no_send_node = TestNode() # never send anything
|
|
|
|
connections = []
|
|
connections.append(NodeConn('127.0.0.1', p2p_port(0), self.nodes[0], self.no_verack_node))
|
|
connections.append(NodeConn('127.0.0.1', p2p_port(0), self.nodes[0], self.no_version_node, send_version=False))
|
|
connections.append(NodeConn('127.0.0.1', p2p_port(0), self.nodes[0], self.no_send_node, send_version=False))
|
|
self.no_verack_node.add_connection(connections[0])
|
|
self.no_version_node.add_connection(connections[1])
|
|
self.no_send_node.add_connection(connections[2])
|
|
|
|
NetworkThread().start() # Start up network handling in another thread
|
|
|
|
sleep(1)
|
|
|
|
assert(self.no_verack_node.connected)
|
|
assert(self.no_version_node.connected)
|
|
assert(self.no_send_node.connected)
|
|
|
|
ping_msg = msg_ping()
|
|
connections[0].send_message(ping_msg)
|
|
connections[1].send_message(ping_msg)
|
|
|
|
sleep(30)
|
|
|
|
assert(self.no_verack_node.received_version)
|
|
|
|
assert(self.no_verack_node.connected)
|
|
assert(self.no_version_node.connected)
|
|
assert(self.no_send_node.connected)
|
|
|
|
connections[0].send_message(ping_msg)
|
|
connections[1].send_message(ping_msg)
|
|
|
|
sleep(31)
|
|
|
|
assert(not self.no_verack_node.connected)
|
|
assert(not self.no_version_node.connected)
|
|
assert(not self.no_send_node.connected)
|
|
|
|
if __name__ == '__main__':
|
|
TimeoutsTest().main()
|