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 ( | ||||
MAX_HEADERS_RESULTS, | MAX_HEADERS_RESULTS, | ||||
MAX_INV_SIZE, | MAX_INV_SIZE, | ||||
MAX_PROTOCOL_MESSAGE_LENGTH, | MAX_PROTOCOL_MESSAGE_LENGTH, | ||||
MSG_TX, | MSG_TX, | ||||
CBlockHeader, | CBlockHeader, | ||||
CInv, | CInv, | ||||
msg_avahello, | |||||
msg_avapoll, | |||||
msg_avaresponse, | |||||
msg_getdata, | msg_getdata, | ||||
msg_headers, | msg_headers, | ||||
msg_inv, | msg_inv, | ||||
msg_ping, | msg_ping, | ||||
ser_string, | ser_string, | ||||
) | ) | ||||
from test_framework.p2p import P2PDataStore, P2PInterface | from test_framework.p2p import P2PDataStore, P2PInterface | ||||
from test_framework.test_framework import BitcoinTestFramework | from test_framework.test_framework import BitcoinTestFramework | ||||
Show All 37 Lines | def run_test(self): | ||||
self.test_msgtype() | self.test_msgtype() | ||||
self.test_addrv2_empty() | self.test_addrv2_empty() | ||||
self.test_addrv2_no_addresses() | self.test_addrv2_no_addresses() | ||||
self.test_addrv2_too_long_address() | self.test_addrv2_too_long_address() | ||||
self.test_addrv2_unrecognized_network() | self.test_addrv2_unrecognized_network() | ||||
self.test_oversized_inv_msg() | self.test_oversized_inv_msg() | ||||
self.test_oversized_getdata_msg() | self.test_oversized_getdata_msg() | ||||
self.test_oversized_headers_msg() | self.test_oversized_headers_msg() | ||||
self.test_unsolicited_ava_messages() | |||||
self.test_resource_exhaustion() | self.test_resource_exhaustion() | ||||
def test_buffer(self): | def test_buffer(self): | ||||
self.log.info( | self.log.info( | ||||
"Test message with header split across two buffers is received") | "Test message with header split across two buffers is received") | ||||
conn = self.nodes[0].add_p2p_connection(P2PDataStore()) | conn = self.nodes[0].add_p2p_connection(P2PDataStore()) | ||||
# Create valid message | # Create valid message | ||||
msg = conn.build_message(msg_ping(nonce=12345)) | msg = conn.build_message(msg_ping(nonce=12345)) | ||||
▲ Show 20 Lines • Show All 183 Lines • ▼ Show 20 Lines | class InvalidMessagesTest(BitcoinTestFramework): | ||||
def test_oversized_getdata_msg(self): | def test_oversized_getdata_msg(self): | ||||
size = MAX_INV_SIZE + 1 | size = MAX_INV_SIZE + 1 | ||||
self.test_oversized_msg(msg_getdata([CInv(MSG_TX, 1)] * size), size) | self.test_oversized_msg(msg_getdata([CInv(MSG_TX, 1)] * size), size) | ||||
def test_oversized_headers_msg(self): | def test_oversized_headers_msg(self): | ||||
size = MAX_HEADERS_RESULTS + 1 | size = MAX_HEADERS_RESULTS + 1 | ||||
self.test_oversized_msg(msg_headers([CBlockHeader()] * size), size) | self.test_oversized_msg(msg_headers([CBlockHeader()] * size), size) | ||||
def test_unsolicited_ava_messages(self): | |||||
"""Node 0 has avalanche disabled by default. If a node does not | |||||
advertise the avalanche service flag, it does not expect to receive | |||||
any avalanche related message and should consider it as spam. | |||||
""" | |||||
conn = self.nodes[0].add_p2p_connection(P2PInterface()) | |||||
with self.nodes[0].assert_debug_log( | |||||
['Misbehaving', '(0 -> 20): unsolicited-avahello']): | |||||
msg = msg_avahello() | |||||
conn.send_and_ping(msg) | |||||
with self.nodes[0].assert_debug_log( | |||||
['Misbehaving', '(20 -> 40): unsolicited-avapoll']): | |||||
msg = msg_avapoll() | |||||
conn.send_and_ping(msg) | |||||
with self.nodes[0].assert_debug_log( | |||||
['Misbehaving', '(40 -> 60): unsolicited-avaresponse']): | |||||
msg = msg_avaresponse() | |||||
conn.send_and_ping(msg) | |||||
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 | ||||
self.log.info( | self.log.info( | ||||
Show All 23 Lines |