You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
3.4 KiB
103 lines
3.4 KiB
#!/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. |
|
"""Test various net timeouts. |
|
|
|
- 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()
|
|
|