Changeset View
Changeset View
Standalone View
Standalone View
test/functional/p2p_invalid_messages.py
| #!/usr/bin/env python3 | #!/usr/bin/env python3 | ||||
| # Copyright (c) 2015-2019 The Bitcoin Core developers | # Copyright (c) 2015-2019 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 node responses to invalid network messages.""" | """Test node responses to invalid network messages.""" | ||||
| import struct | import struct | ||||
| import time | import time | ||||
| from test_framework.messages import ( | from test_framework.messages import ( | ||||
| CBlockHeader, | CBlockHeader, | ||||
| CInv, | CInv, | ||||
| MAX_HEADERS_RESULTS, | MAX_HEADERS_RESULTS, | ||||
| MAX_INV_SIZE, | MAX_INV_SIZE, | ||||
| MAX_PROTOCOL_MESSAGE_LENGTH, | MAX_PROTOCOL_MESSAGE_LENGTH, | ||||
| msg_avahello, | msg_avahello, | ||||
| msg_avapoll, | msg_avapoll, | ||||
| msg_avaproof, | |||||
| msg_avaresponse, | msg_avaresponse, | ||||
| msg_getdata, | msg_getdata, | ||||
| msg_headers, | msg_headers, | ||||
| msg_inv, | msg_inv, | ||||
| msg_ping, | msg_ping, | ||||
| MSG_TX, | MSG_TX, | ||||
| ser_string, | ser_string, | ||||
| ) | ) | ||||
| ▲ Show 20 Lines • Show All 259 Lines • ▼ Show 20 Lines | def test_unsolicited_ava_messages(self): | ||||
| with self.nodes[0].assert_debug_log( | with self.nodes[0].assert_debug_log( | ||||
| ['Misbehaving', '(20 -> 40): unsolicited-avapoll']): | ['Misbehaving', '(20 -> 40): unsolicited-avapoll']): | ||||
| msg = msg_avapoll() | msg = msg_avapoll() | ||||
| conn.send_and_ping(msg) | conn.send_and_ping(msg) | ||||
| with self.nodes[0].assert_debug_log( | with self.nodes[0].assert_debug_log( | ||||
| ['Misbehaving', '(40 -> 60): unsolicited-avaresponse']): | ['Misbehaving', '(40 -> 60): unsolicited-avaresponse']): | ||||
| msg = msg_avaresponse() | msg = msg_avaresponse() | ||||
| conn.send_and_ping(msg) | conn.send_and_ping(msg) | ||||
| with self.nodes[0].assert_debug_log( | |||||
| ['Misbehaving', '(60 -> 80): unsolicited-avaproof']): | |||||
| msg = msg_avaproof() | |||||
| conn.send_and_ping(msg) | |||||
| self.nodes[0].disconnect_p2ps() | self.nodes[0].disconnect_p2ps() | ||||
| def test_resource_exhaustion(self): | def test_resource_exhaustion(self): | ||||
| self.log.info("Test node stays up despite many large junk messages") | self.log.info("Test node stays up despite many large junk messages") | ||||
| conn = self.nodes[0].add_p2p_connection(P2PDataStore()) | conn = self.nodes[0].add_p2p_connection(P2PDataStore()) | ||||
| conn2 = self.nodes[0].add_p2p_connection(P2PDataStore()) | conn2 = self.nodes[0].add_p2p_connection(P2PDataStore()) | ||||
| msg_at_size = msg_unrecognized(str_data="b" * VALID_DATA_LIMIT) | msg_at_size = msg_unrecognized(str_data="b" * VALID_DATA_LIMIT) | ||||
| assert len(msg_at_size.serialize()) == MAX_PROTOCOL_MESSAGE_LENGTH | assert len(msg_at_size.serialize()) == MAX_PROTOCOL_MESSAGE_LENGTH | ||||
| Show All 25 Lines | |||||