Changeset View
Changeset View
Standalone View
Standalone View
qa/rpc-tests/maxblocksinflight.py
| #!/usr/bin/env python3 | #!/usr/bin/env python3 | ||||
| # Copyright (c) 2015-2016 The Bitcoin Core developers | # Copyright (c) 2015-2016 The Bitcoin Core developers | ||||
| # Distributed under the MIT software license, see the accompanying | # Distributed under the MIT software license, see the accompanying | ||||
| # file COPYING or http://www.opensource.org/licenses/mit-license.php. | # file COPYING or http://www.opensource.org/licenses/mit-license.php. | ||||
| from test_framework.mininode import * | from test_framework.mininode import * | ||||
| from test_framework.test_framework import BitcoinTestFramework | from test_framework.test_framework import BitcoinTestFramework | ||||
| from test_framework.util import * | from test_framework.util import * | ||||
| import logging | |||||
| ''' | ''' | ||||
| In this test we connect to one node over p2p, send it numerous inv's, and | In this test we connect to one node over p2p, send it numerous inv's, and | ||||
| compare the resulting number of getdata requests to a max allowed value. We | compare the resulting number of getdata requests to a max allowed value. We | ||||
| test for exceeding 128 blocks in flight, which was the limit an 0.9 client will | test for exceeding 128 blocks in flight, which was the limit an 0.9 client will | ||||
| reach. [0.10 clients shouldn't request more than 16 from a single peer.] | reach. [0.10 clients shouldn't request more than 16 from a single peer.] | ||||
| ''' | ''' | ||||
| MAX_REQUESTS = 128 | MAX_REQUESTS = 128 | ||||
| Show All 11 Lines | def on_getdata(self, conn, message): | ||||
| self.blockReqCounts[inv.hash] += 1 | self.blockReqCounts[inv.hash] += 1 | ||||
| def on_close(self, conn): | def on_close(self, conn): | ||||
| if not self.disconnectOkay: | if not self.disconnectOkay: | ||||
| raise EarlyDisconnectError(0) | raise EarlyDisconnectError(0) | ||||
| def __init__(self): | def __init__(self): | ||||
| NodeConnCB.__init__(self) | NodeConnCB.__init__(self) | ||||
| self.log = logging.getLogger("BlockRelayTest") | |||||
| def add_new_connection(self, connection): | def add_new_connection(self, connection): | ||||
| self.connection = connection | self.connection = connection | ||||
| self.blockReqCounts = {} | self.blockReqCounts = {} | ||||
| self.disconnectOkay = False | self.disconnectOkay = False | ||||
| def run(self): | def run(self): | ||||
| self.connection.rpc.generate(1) # Leave IBD | self.connection.rpc.generate(1) # Leave IBD | ||||
| Show All 17 Lines | def run(self): | ||||
| for key in self.blockReqCounts: | for key in self.blockReqCounts: | ||||
| total_requests += self.blockReqCounts[key] | total_requests += self.blockReqCounts[key] | ||||
| if self.blockReqCounts[key] > 1: | if self.blockReqCounts[key] > 1: | ||||
| raise AssertionError( | raise AssertionError( | ||||
| "Error, test failed: block %064x requested more than once" % key) | "Error, test failed: block %064x requested more than once" % key) | ||||
| if total_requests > MAX_REQUESTS: | if total_requests > MAX_REQUESTS: | ||||
| raise AssertionError( | raise AssertionError( | ||||
| "Error, too many blocks (%d) requested" % total_requests) | "Error, too many blocks (%d) requested" % total_requests) | ||||
| print("Round %d: success (total requests: %d)" % | self.log.info("Round %d: success (total requests: %d)" % | ||||
| (count, total_requests)) | (count, total_requests)) | ||||
| self.disconnectOkay = True | self.disconnectOkay = True | ||||
| self.connection.disconnect_node() | self.connection.disconnect_node() | ||||
| class MaxBlocksInFlightTest(BitcoinTestFramework): | class MaxBlocksInFlightTest(BitcoinTestFramework): | ||||
| def add_options(self, parser): | def add_options(self, parser): | ||||
| parser.add_option("--testbinary", dest="testbinary", | parser.add_option("--testbinary", dest="testbinary", | ||||
| default=os.getenv("BITCOIND", "bitcoind"), | default=os.getenv("BITCOIND", "bitcoind"), | ||||
| help="Binary to test max block requests behavior") | help="Binary to test max block requests behavior") | ||||
| def __init__(self): | def __init__(self): | ||||
| super().__init__() | super().__init__() | ||||
| self.setup_clean_chain = True | self.setup_clean_chain = True | ||||
| self.num_nodes = 1 | self.num_nodes = 1 | ||||
| def setup_network(self): | def setup_network(self): | ||||
| self.nodes = start_nodes(self.num_nodes, self.options.tmpdir, | self.nodes = start_nodes(self.num_nodes, self.options.tmpdir, | ||||
| extra_args=[ | extra_args=[['-whitelist=127.0.0.1']], | ||||
| ['-debug', '-whitelist=127.0.0.1']], | |||||
| binary=[self.options.testbinary]) | binary=[self.options.testbinary]) | ||||
| def run_test(self): | def run_test(self): | ||||
| test = TestManager() | test = TestManager() | ||||
| # pass log handler through to the test manager object | |||||
| test.log = self.log | |||||
| test.add_new_connection( | test.add_new_connection( | ||||
| NodeConn('127.0.0.1', p2p_port(0), self.nodes[0], test)) | NodeConn('127.0.0.1', p2p_port(0), self.nodes[0], test)) | ||||
| NetworkThread().start() # Start up network handling in another thread | NetworkThread().start() # Start up network handling in another thread | ||||
| test.run() | test.run() | ||||
| if __name__ == '__main__': | if __name__ == '__main__': | ||||
| MaxBlocksInFlightTest().main() | MaxBlocksInFlightTest().main() | ||||