Changeset View
Changeset View
Standalone View
Standalone View
test/functional/abc_p2p_compactproofs.py
| #!/usr/bin/env python3 | #!/usr/bin/env python3 | ||||
| # Copyright (c) 2022 The Bitcoin developers | # Copyright (c) 2022 The Bitcoin 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 proof inventory relaying | Test proof inventory relaying | ||||
| """ | """ | ||||
| import random | |||||
| from test_framework.avatools import ( | from test_framework.avatools import ( | ||||
| AvaP2PInterface, | AvaP2PInterface, | ||||
| gen_proof, | gen_proof, | ||||
| get_ava_p2p_interface, | |||||
| get_proof_ids, | get_proof_ids, | ||||
| wait_for_proof, | wait_for_proof, | ||||
| ) | ) | ||||
| from test_framework.messages import ( | from test_framework.messages import ( | ||||
| MSG_AVA_PROOF, | MSG_AVA_PROOF, | ||||
| MSG_TYPE_MASK, | MSG_TYPE_MASK, | ||||
| AvalanchePrefilledProof, | |||||
| msg_avaproofs, | |||||
| msg_getavaproofs, | msg_getavaproofs, | ||||
| ) | ) | ||||
| from test_framework.p2p import P2PInterface, p2p_lock | from test_framework.p2p import P2PInterface, p2p_lock | ||||
| from test_framework.siphash import siphash256 | from test_framework.siphash import siphash256 | ||||
| from test_framework.test_framework import BitcoinTestFramework | from test_framework.test_framework import BitcoinTestFramework | ||||
| from test_framework.util import assert_equal | from test_framework.util import assert_equal | ||||
| ▲ Show 20 Lines • Show All 58 Lines • ▼ Show 20 Lines | def test_respond_getavaproofs(self): | ||||
| avaproofs.key0, | avaproofs.key0, | ||||
| avaproofs.key1, | avaproofs.key1, | ||||
| proofid) & 0x0000ffffffffffff for proofid in sorted(proofids)] | proofid) & 0x0000ffffffffffff for proofid in sorted(proofids)] | ||||
| assert_equal(expected_shortids, avaproofs.shortids) | assert_equal(expected_shortids, avaproofs.shortids) | ||||
| # Don't expect any prefilled proof for now | # Don't expect any prefilled proof for now | ||||
| assert_equal(len(avaproofs.prefilled_proofs), 0) | assert_equal(len(avaproofs.prefilled_proofs), 0) | ||||
| def test_request_missing_proofs(self): | |||||
| self.log.info( | |||||
| "Check the node requests the missing proofs after receiving an avaproofs message") | |||||
| node = self.nodes[0] | |||||
| key0 = random.randint(0, 2**64 - 1) | |||||
| key1 = random.randint(0, 2**64 - 1) | |||||
| proofs = [gen_proof(node)[1] for _ in range(10)] | |||||
| # Build a map from proofid to shortid. Use sorted proofids so we don't | |||||
| # have the same indices than the `proofs` list. | |||||
| proofids = [p.proofid for p in proofs] | |||||
| shortid_map = {} | |||||
| for proofid in sorted(proofids): | |||||
| shortid_map[proofid] = siphash256( | |||||
| key0, key1, proofid) & 0x0000ffffffffffff | |||||
| self.log.info("The node ignores unsollicited avaproofs") | |||||
| spam_peer = get_ava_p2p_interface(node) | |||||
| msg = msg_avaproofs() | |||||
| msg.key0 = key0 | |||||
| msg.key1 = key1 | |||||
| msg.shortids = list(shortid_map.values()) | |||||
| msg.prefilled_proofs = [] | |||||
| with node.assert_debug_log(["Ignoring unsollicited avaproofs"]): | |||||
| spam_peer.send_message(msg) | |||||
| self.restart_node( | |||||
| 0, | |||||
| extra_args=self.extra_args[0] + | |||||
| ['-acceptunsollicitedavaproofs=1']) | |||||
| def received_avaproofsreq(peer): | |||||
| with p2p_lock: | |||||
| return peer.last_message.get("avaproofsreq") | |||||
| def expect_indices(shortids, expected_indices, prefilled_proofs=None): | |||||
| if prefilled_proofs is None: | |||||
| prefilled_proofs = [] | |||||
| msg = msg_avaproofs() | |||||
| msg.key0 = key0 | |||||
| msg.key1 = key1 | |||||
| msg.shortids = shortids | |||||
| msg.prefilled_proofs = prefilled_proofs | |||||
| peer = get_ava_p2p_interface(node) | |||||
| peer.send_message(msg) | |||||
| self.wait_until(lambda: received_avaproofsreq(peer), timeout=1) | |||||
| avaproofsreq = received_avaproofsreq(peer) | |||||
| assert_equal(avaproofsreq.indices, expected_indices) | |||||
| self.log.info("Check no proof is requested if there is no shortid") | |||||
| expect_indices([], []) | |||||
| self.log.info( | |||||
| "Check the node requests all the proofs if it known none") | |||||
| expect_indices( | |||||
| list(shortid_map.values()), | |||||
| [i for i in range(len(shortid_map))] | |||||
| ) | |||||
| self.log.info( | |||||
| "Check the node requests only the missing proofs") | |||||
| known_proofids = [] | |||||
| for proof in proofs[:5]: | |||||
| node.sendavalancheproof(proof.serialize().hex()) | |||||
| known_proofids.append(proof.proofid) | |||||
| expected_indices = [i for i, proofid in enumerate( | |||||
| shortid_map) if proofid not in known_proofids] | |||||
| expect_indices(list(shortid_map.values()), expected_indices) | |||||
| self.log.info( | |||||
| "Check the node don't request prefilled proofs") | |||||
| # Get the indices for a couple of proofs | |||||
| indice_proof5 = list(shortid_map.keys()).index(proofids[5]) | |||||
| indice_proof6 = list(shortid_map.keys()).index(proofids[6]) | |||||
| prefilled_proofs = [ | |||||
| AvalanchePrefilledProof(indice_proof5, proofs[5]), | |||||
| AvalanchePrefilledProof(indice_proof6, proofs[6]), | |||||
| ] | |||||
| prefilled_proofs = sorted( | |||||
| prefilled_proofs, | |||||
| key=lambda prefilled_proof: prefilled_proof.index) | |||||
| remaining_shortids = [shortid for proofid, shortid in shortid_map.items( | |||||
| ) if proofid not in proofids[5:7]] | |||||
| known_proofids.extend(proofids[5:7]) | |||||
| expected_indices = [i for i, proofid in enumerate( | |||||
| shortid_map) if proofid not in known_proofids] | |||||
| expect_indices( | |||||
| remaining_shortids, | |||||
| expected_indices, | |||||
| prefilled_proofs=prefilled_proofs) | |||||
| self.log.info( | |||||
| "Check the node requests no proof if it knows all of them") | |||||
| for proof in proofs[5:]: | |||||
| node.sendavalancheproof(proof.serialize().hex()) | |||||
| known_proofids.append(proof.proofid) | |||||
| expect_indices(list(shortid_map.values()), []) | |||||
| self.log.info("Check out of bounds index") | |||||
| bad_peer = get_ava_p2p_interface(node) | |||||
| msg = msg_avaproofs() | |||||
| msg.key0 = key0 | |||||
| msg.key1 = key1 | |||||
| msg.shortids = list(shortid_map.values()) | |||||
| msg.prefilled_proofs = [ | |||||
| AvalanchePrefilledProof( | |||||
| len(shortid_map) + 1, | |||||
| gen_proof(node)[1])] | |||||
| with node.assert_debug_log(["Misbehaving", "avaproofs-index-out-of-bound"]): | |||||
| bad_peer.send_message(msg) | |||||
| self.log.info("An invalid prefilled proof will trigger a ban") | |||||
| _, no_stake = gen_proof(node) | |||||
| no_stake.stakes = [] | |||||
| msg = msg_avaproofs() | |||||
| msg.key0 = key0 | |||||
| msg.key1 = key1 | |||||
| msg.shortids = list(shortid_map.values()) | |||||
| msg.prefilled_proofs = [ | |||||
| AvalanchePrefilledProof(len(shortid_map), no_stake), | |||||
| ] | |||||
| with node.assert_debug_log(["Misbehaving", "invalid-proof"]): | |||||
| bad_peer.send_message(msg) | |||||
| def run_test(self): | def run_test(self): | ||||
| self.test_respond_getavaproofs() | self.test_respond_getavaproofs() | ||||
| self.test_request_missing_proofs() | |||||
| if __name__ == '__main__': | if __name__ == '__main__': | ||||
| CompactProofsTest().main() | CompactProofsTest().main() | ||||