Changeset View
Changeset View
Standalone View
Standalone View
test/functional/p2p_eviction.py
| Show First 20 Lines • Show All 47 Lines • ▼ Show 20 Lines | def on_ping(self, message): | ||||
| time.sleep(0.1) | time.sleep(0.1) | ||||
| self.send_message(msg_pong(message.nonce)) | self.send_message(msg_pong(message.nonce)) | ||||
| class P2PEvict(BitcoinTestFramework): | class P2PEvict(BitcoinTestFramework): | ||||
| def set_test_params(self): | def set_test_params(self): | ||||
| self.setup_clean_chain = True | self.setup_clean_chain = True | ||||
| self.num_nodes = 1 | self.num_nodes = 1 | ||||
| # The choice of maxconnections=164 results in a maximum of 153 inbound | # The choice of maxconnections=168 results in a maximum of 157 inbound | ||||
| # connections (164 - 10 outbound - 1 feeler). 152 inbound peers are | # connections (164 - 10 outbound - 1 feeler). 156 inbound peers are | ||||
| # protected from eviction: | # protected from eviction: | ||||
| # 4 by netgroup, 4 that sent us blocks, 4 that sent us proofs, 4 that | # 4 by netgroup, 4 that sent us blocks, 4 that sent us proofs, 4 that | ||||
| # sent us transactions, 8 via lowest ping time, 128 with the best | # sent us transactions, 8 via lowest ping time, 128 with the best | ||||
| # avalanche availability score | # avalanche availability score, 4 with the best availability score from | ||||
| self.extra_args = [['-maxconnections=164', "-enableavalanche=1"]] | # each of 4 avalanche peers | ||||
| self.extra_args = [['-maxconnections=168', "-enableavalanche=1"]] | |||||
| def run_test(self): | def run_test(self): | ||||
| # peers that we expect to be protected from eviction | # peers that we expect to be protected from eviction | ||||
| protected_peers = set() | protected_peers = set() | ||||
| current_peer = -1 | current_peer = -1 | ||||
| node = self.nodes[0] | node = self.nodes[0] | ||||
| blocks = node.generatetoaddress( | blocks = node.generatetoaddress( | ||||
| 101, node.get_deterministic_priv_key().address) | 101, node.get_deterministic_priv_key().address) | ||||
| ▲ Show 20 Lines • Show All 72 Lines • ▼ Show 20 Lines | def run_test(self): | ||||
| self.log.info( | self.log.info( | ||||
| "Create 8 peers and protect them from eviction by having faster pings") | "Create 8 peers and protect them from eviction by having faster pings") | ||||
| for _ in range(8): | for _ in range(8): | ||||
| fastpeer = node.add_p2p_connection(P2PInterface()) | fastpeer = node.add_p2p_connection(P2PInterface()) | ||||
| current_peer += 1 | current_peer += 1 | ||||
| wait_until(lambda: "ping" in fastpeer.last_message, timeout=10) | wait_until(lambda: "ping" in fastpeer.last_message, timeout=10) | ||||
| self.log.info( | def build_proof_and_delegation(stakes): | ||||
| "Create 128 peers and protect them from eviction by sending an avahello message") | |||||
| proof = node.buildavalancheproof( | proof = node.buildavalancheproof( | ||||
| 42, 2000000000, pubkey.get_bytes().hex(), [stakes[0]]) | 42, 2000000000, pubkey.get_bytes().hex(), stakes) | ||||
| proof_obj = FromHex(AvalancheProof(), proof) | proof_obj = FromHex(AvalancheProof(), proof) | ||||
| delegation = node.delegateavalancheproof( | delegation = node.delegateavalancheproof( | ||||
| f"{proof_obj.limited_proofid:064x}", | f"{proof_obj.limited_proofid:064x}", | ||||
| bytes_to_wif(privkey.get_bytes()), | bytes_to_wif(privkey.get_bytes()), | ||||
| pubkey.get_bytes().hex(), | pubkey.get_bytes().hex(), | ||||
| ) | ) | ||||
| return proof, delegation | |||||
| for _ in range(128): | def add_avalanche_connection(): | ||||
| avapeer = node.add_p2p_connection(SlowAvaP2PInterface()) | avapeer = node.add_p2p_connection(SlowAvaP2PInterface()) | ||||
| current_peer += 1 | |||||
| avapeer.sync_with_ping() | avapeer.sync_with_ping() | ||||
| avapeer.send_avahello(delegation, privkey) | avapeer.send_avahello(delegation, privkey) | ||||
| self.log.info( | |||||
| "Create 128 peers and protect them from eviction by sending an avahello message") | |||||
| proof, delegation = build_proof_and_delegation([stakes[0]]) | |||||
| for _ in range(128): | |||||
| add_avalanche_connection() | |||||
| current_peer += 1 | |||||
| self.log.info( | |||||
| "Create 4 more peers, each with a different proof known by our node") | |||||
| for i in range(1, 5): | |||||
| proof, delegation = build_proof_and_delegation([stakes[i]]) | |||||
| add_avalanche_connection() | |||||
| current_peer += 1 | |||||
| # Make sure by asking the node what the actual min pings are | # Make sure by asking the node what the actual min pings are | ||||
| peerinfo = node.getpeerinfo() | peerinfo = node.getpeerinfo() | ||||
| pings = {} | pings = {} | ||||
| for i in range(len(peerinfo)): | for i in range(len(peerinfo)): | ||||
| pings[i] = peerinfo[i]['minping'] if 'minping' in peerinfo[i] else 1000000 | pings[i] = peerinfo[i]['minping'] if 'minping' in peerinfo[i] else 1000000 | ||||
| sorted_pings = sorted(pings.items(), key=lambda x: x[1]) | sorted_pings = sorted(pings.items(), key=lambda x: x[1]) | ||||
| # Usually the 8 fast peers are protected. In rare case of unreliable pings, | # Usually the 8 fast peers are protected. In rare case of unreliable pings, | ||||
| Show All 33 Lines | |||||