Browse Source
0.1466f861a
Add a test for P2P inactivity timeouts (Matt Corallo)b436f92
qa: Expose on-connection to mininode listeners (Matt Corallo)8aaba7a
qa: mininode learns when a socket connects, not its first action (Matt Corallo)2cbd119
Disconnect peers which we do not receive VERACKs from within 60 sec (Matt Corallo)
Wladimir J. van der Laan
8 years ago
4 changed files with 135 additions and 12 deletions
@ -0,0 +1,103 @@
@@ -0,0 +1,103 @@
|
||||
#!/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() |
Loading…
Reference in new issue