Changeset View
Changeset View
Standalone View
Standalone View
test/functional/p2p_getdata.py
| #!/usr/bin/env python3 | #!/usr/bin/env python3 | ||||
| # Copyright (c) 2020 The Bitcoin Core developers | # Copyright (c) 2020 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. | ||||
| """Test GETDATA processing behavior""" | """Test GETDATA processing behavior""" | ||||
| from collections import defaultdict | from collections import defaultdict | ||||
| from test_framework.messages import ( | from test_framework.messages import ( | ||||
| CInv, | CInv, | ||||
| msg_getdata, | msg_getdata, | ||||
| ) | ) | ||||
| from test_framework.mininode import ( | from test_framework.mininode import P2PInterface | ||||
| mininode_lock, | |||||
| P2PInterface, | |||||
| ) | |||||
| from test_framework.test_framework import BitcoinTestFramework | from test_framework.test_framework import BitcoinTestFramework | ||||
| from test_framework.util import wait_until | |||||
| class P2PStoreBlock(P2PInterface): | class P2PStoreBlock(P2PInterface): | ||||
| def __init__(self): | def __init__(self): | ||||
| super().__init__() | super().__init__() | ||||
| self.blocks = defaultdict(int) | self.blocks = defaultdict(int) | ||||
| def on_block(self, message): | def on_block(self, message): | ||||
| message.block.calc_sha256() | message.block.calc_sha256() | ||||
| self.blocks[message.block.sha256] += 1 | self.blocks[message.block.sha256] += 1 | ||||
| class GetdataTest(BitcoinTestFramework): | class GetdataTest(BitcoinTestFramework): | ||||
| def set_test_params(self): | def set_test_params(self): | ||||
| self.num_nodes = 1 | self.num_nodes = 1 | ||||
| def run_test(self): | def run_test(self): | ||||
| self.nodes[0].add_p2p_connection(P2PStoreBlock()) | p2p_block_store = self.nodes[0].add_p2p_connection(P2PStoreBlock()) | ||||
| self.log.info( | self.log.info( | ||||
| "test that an invalid GETDATA doesn't prevent processing of future messages") | "test that an invalid GETDATA doesn't prevent processing of future messages") | ||||
| # Send invalid message and verify that node responds to later ping | # Send invalid message and verify that node responds to later ping | ||||
| invalid_getdata = msg_getdata() | invalid_getdata = msg_getdata() | ||||
| invalid_getdata.inv.append(CInv(t=0, h=0)) # INV type 0 is invalid. | invalid_getdata.inv.append(CInv(t=0, h=0)) # INV type 0 is invalid. | ||||
| self.nodes[0].p2ps[0].send_and_ping(invalid_getdata) | p2p_block_store.send_and_ping(invalid_getdata) | ||||
| # Check getdata still works by fetching tip block | # Check getdata still works by fetching tip block | ||||
| best_block = int(self.nodes[0].getbestblockhash(), 16) | best_block = int(self.nodes[0].getbestblockhash(), 16) | ||||
| good_getdata = msg_getdata() | good_getdata = msg_getdata() | ||||
| good_getdata.inv.append(CInv(t=2, h=best_block)) | good_getdata.inv.append(CInv(t=2, h=best_block)) | ||||
| self.nodes[0].p2ps[0].send_and_ping(good_getdata) | p2p_block_store.send_and_ping(good_getdata) | ||||
| wait_until( | p2p_block_store.wait_until( | ||||
| lambda: self.nodes[0].p2ps[0].blocks[best_block] == 1, | lambda: self.nodes[0].p2ps[0].blocks[best_block] == 1) | ||||
| timeout=30, | |||||
| lock=mininode_lock) | |||||
| if __name__ == '__main__': | if __name__ == '__main__': | ||||
| GetdataTest().main() | GetdataTest().main() | ||||