1 #!/usr/bin/env python3
2 # Copyright (c) 2022-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 Test encrypted v2 p2p proposed in BIP 324
7 """
8 from test_framework.blocktools import (
9 create_block,
10 )
11 from test_framework.p2p import (
12 P2PDataStore,
13 P2PInterface,
14 )
15 from test_framework.test_framework import BitcoinTestFramework
16 from test_framework.util import (
17 assert_equal,
18 assert_greater_than,
19 check_node_connections,
20 )
21 from test_framework.crypto.chacha20 import REKEY_INTERVAL
22 23 24 class P2PEncrypted(BitcoinTestFramework):
25 def set_test_params(self):
26 self.num_nodes = 2
27 self.extra_args = [["-v2transport=1"], ["-v2transport=1"]]
28 29 def setup_network(self):
30 self.setup_nodes()
31 32 def generate_blocks(self, node, number):
33 test_blocks = []
34 last_block = node.getbestblockhash()
35 tip = int(last_block, 16)
36 tipheight = node.getblockcount()
37 last_block_time = node.getblock(last_block)['time']
38 for _ in range(number):
39 # Create some blocks
40 block = create_block(tip, height=tipheight + 1, ntime=last_block_time + 1)
41 block.solve()
42 test_blocks.append(block)
43 tip = block.hash_int
44 tipheight += 1
45 last_block_time += 1
46 return test_blocks
47 48 def create_test_block(self, txs):
49 block = create_block(self.tip, height=self.tipheight + 1, ntime=self.last_block_time + 600, txlist=txs)
50 block.solve()
51 return block
52 53 def run_test(self):
54 node0, node1 = self.nodes[0], self.nodes[1]
55 self.log.info("Check inbound connection to v2 TestNode from v2 P2PConnection is v2")
56 peer1 = node0.add_p2p_connection(P2PInterface(), wait_for_verack=True, supports_v2_p2p=True)
57 assert peer1.supports_v2_p2p
58 assert_equal(node0.getpeerinfo()[-1]["transport_protocol_type"], "v2")
59 60 self.log.info("Check inbound connection to v2 TestNode from v1 P2PConnection is v1")
61 peer2 = node0.add_p2p_connection(P2PInterface(), wait_for_verack=True, supports_v2_p2p=False)
62 assert not peer2.supports_v2_p2p
63 assert_equal(node0.getpeerinfo()[-1]["transport_protocol_type"], "v1")
64 65 self.log.info("Check outbound connection from v2 TestNode to v1 P2PConnection advertised as v1 is v1")
66 peer3 = node0.add_outbound_p2p_connection(P2PInterface(), p2p_idx=0, supports_v2_p2p=False, advertise_v2_p2p=False)
67 assert not peer3.supports_v2_p2p
68 assert_equal(node0.getpeerinfo()[-1]["transport_protocol_type"], "v1")
69 70 # v2 TestNode performs downgrading here
71 self.log.info("Check outbound connection from v2 TestNode to v1 P2PConnection advertised as v2 is v1")
72 peer4 = node0.add_outbound_p2p_connection(P2PInterface(), p2p_idx=1, supports_v2_p2p=False, advertise_v2_p2p=True)
73 assert not peer4.supports_v2_p2p
74 assert_equal(node0.getpeerinfo()[-1]["transport_protocol_type"], "v1")
75 76 self.log.info("Check outbound connection from v2 TestNode to v2 P2PConnection advertised as v2 is v2")
77 peer5 = node0.add_outbound_p2p_connection(P2PInterface(), p2p_idx=2, supports_v2_p2p=True, advertise_v2_p2p=True)
78 assert peer5.supports_v2_p2p
79 assert_equal(node0.getpeerinfo()[-1]["transport_protocol_type"], "v2")
80 81 self.log.info("Check if version is sent and verack is received in inbound/outbound connections")
82 assert_equal(len(node0.getpeerinfo()), 5) # check if above 5 connections are present in node0's getpeerinfo()
83 for peer in node0.getpeerinfo():
84 assert_greater_than(peer['bytessent_per_msg']['version'], 0)
85 assert_greater_than(peer['bytesrecv_per_msg']['verack'], 0)
86 87 self.log.info("Testing whether blocks propagate - check if tips sync when number of blocks >= REKEY_INTERVAL")
88 # tests whether rekeying (which happens every REKEY_INTERVAL packets) works correctly
89 test_blocks = self.generate_blocks(node0, REKEY_INTERVAL+1)
90 91 for i in range(2):
92 peer6 = node0.add_p2p_connection(P2PDataStore(), supports_v2_p2p=True)
93 assert peer6.supports_v2_p2p
94 assert_equal(node0.getpeerinfo()[-1]["transport_protocol_type"], "v2")
95 96 # Consider: node0 <-- peer6. node0 and node1 aren't connected here.
97 # Construct the following topology: node1 <--> node0 <-- peer6
98 # and test that blocks produced by peer6 will be received by node1 if sent normally
99 # and won't be received by node1 if sent as decoy messages
100 101 # First, check whether blocks produced be peer6 are received by node0 if sent normally
102 # and not received by node0 if sent as decoy messages.
103 if i:
104 # check that node0 receives blocks produced by peer6
105 self.log.info("Check if blocks produced by node0's p2p connection is received by node0")
106 peer6.send_blocks_and_test(test_blocks, node0, success=True) # node0's tip advances
107 else:
108 # check that node0 doesn't receive blocks produced by peer6 since they are sent as decoy messages
109 self.log.info("Check if blocks produced by node0's p2p connection sent as decoys aren't received by node0")
110 peer6.send_blocks_and_test(test_blocks, node0, success=False, is_decoy=True) # node0's tip doesn't advance
111 112 # Then, connect node0 and node1 using v2 and check whether the blocks are received by node1
113 self.connect_nodes(0, 1, peer_advertises_v2=True)
114 self.log.info("Wait for node1 to receive all the blocks from node0")
115 self.sync_all()
116 self.log.info("Make sure node0 and node1 have same block tips")
117 assert_equal(node0.getbestblockhash(), node1.getbestblockhash())
118 119 self.disconnect_nodes(0, 1)
120 121 self.log.info("Check the connections opened as expected")
122 check_node_connections(node=node0, num_in=4, num_out=3)
123 124 self.log.info("Check inbound connection to v1 TestNode from v2 P2PConnection is v1")
125 self.restart_node(0, ["-v2transport=0"])
126 peer1 = node0.add_p2p_connection(P2PInterface(), wait_for_verack=True, supports_v2_p2p=True)
127 assert not peer1.supports_v2_p2p
128 assert_equal(node0.getpeerinfo()[-1]["transport_protocol_type"], "v1")
129 check_node_connections(node=node0, num_in=1, num_out=0)
130 131 132 if __name__ == '__main__':
133 P2PEncrypted(__file__).main()
134