p2p_net_deadlock.py
1 #!/usr/bin/env python3 2 # Copyright (c) 2023-present The Bitcoin Core developers 3 # Distributed under the MIT software license, see the accompanying 4 # file COPYING or http://www.opensource.org/licenses/mit-license.php. 5 6 import threading 7 from test_framework.test_framework import BitcoinTestFramework 8 from random import randbytes 9 10 11 class NetDeadlockTest(BitcoinTestFramework): 12 def set_test_params(self): 13 self.setup_clean_chain = True 14 self.num_nodes = 2 15 16 def run_test(self): 17 node0 = self.nodes[0] 18 node1 = self.nodes[1] 19 20 self.log.info("Simultaneously send a large message on both sides") 21 rand_msg = randbytes(4000000).hex() 22 23 thread0 = threading.Thread(target=node0.sendmsgtopeer, args=(0, "unknown", rand_msg)) 24 thread1 = threading.Thread(target=node1.sendmsgtopeer, args=(0, "unknown", rand_msg)) 25 26 thread0.start() 27 thread1.start() 28 thread0.join() 29 thread1.join() 30 31 self.log.info("Check whether a deadlock happened") 32 self.generate(node0, 1) 33 self.sync_blocks() 34 35 36 if __name__ == '__main__': 37 NetDeadlockTest(__file__).main()