Changeset View
Changeset View
Standalone View
Standalone View
test/functional/p2p_blocksonly.py
Show All 23 Lines | def run_test(self): | ||||
# confirmed coins | # confirmed coins | ||||
self.miniwallet.rescan_utxos() | self.miniwallet.rescan_utxos() | ||||
self.blocksonly_mode_tests() | self.blocksonly_mode_tests() | ||||
self.blocks_relay_conn_tests() | self.blocks_relay_conn_tests() | ||||
def blocksonly_mode_tests(self): | def blocksonly_mode_tests(self): | ||||
self.log.info("Tests with node running in -blocksonly mode") | self.log.info("Tests with node running in -blocksonly mode") | ||||
assert_equal(self.nodes[0].getnetworkinfo()['localrelay'], False) | assert_equal(self.nodes[0].getnetworkinfo()["localrelay"], False) | ||||
self.nodes[0].add_p2p_connection(P2PInterface()) | self.nodes[0].add_p2p_connection(P2PInterface()) | ||||
tx, txid, tx_hex = self.check_p2p_tx_violation() | tx, txid, tx_hex = self.check_p2p_tx_violation() | ||||
self.log.info('Check that tx invs also violate the protocol') | self.log.info("Check that tx invs also violate the protocol") | ||||
self.nodes[0].add_p2p_connection(P2PInterface()) | self.nodes[0].add_p2p_connection(P2PInterface()) | ||||
with self.nodes[0].assert_debug_log(['transaction (0000000000000000000000000000000000000000000000000000000000001234) inv sent in violation of protocol, disconnecting peer']): | with self.nodes[0].assert_debug_log( | ||||
self.nodes[0].p2ps[0].send_message( | [ | ||||
msg_inv([CInv(t=MSG_TX, h=0x1234)])) | "transaction" | ||||
" (0000000000000000000000000000000000000000000000000000000000001234)" | |||||
" inv sent in violation of protocol, disconnecting peer" | |||||
] | |||||
): | |||||
self.nodes[0].p2ps[0].send_message(msg_inv([CInv(t=MSG_TX, h=0x1234)])) | |||||
self.nodes[0].p2ps[0].wait_for_disconnect() | self.nodes[0].p2ps[0].wait_for_disconnect() | ||||
del self.nodes[0].p2ps[0] | del self.nodes[0].p2ps[0] | ||||
self.log.info( | self.log.info( | ||||
'Check that txs from rpc are not rejected and relayed to other peers') | "Check that txs from rpc are not rejected and relayed to other peers" | ||||
) | |||||
tx_relay_peer = self.nodes[0].add_p2p_connection(P2PInterface()) | tx_relay_peer = self.nodes[0].add_p2p_connection(P2PInterface()) | ||||
assert_equal(self.nodes[0].getpeerinfo()[0]['relaytxes'], True) | assert_equal(self.nodes[0].getpeerinfo()[0]["relaytxes"], True) | ||||
assert_equal( | assert_equal(self.nodes[0].testmempoolaccept([tx_hex])[0]["allowed"], True) | ||||
self.nodes[0].testmempoolaccept( | with self.nodes[0].assert_debug_log([f"received getdata for: tx {txid} peer"]): | ||||
[tx_hex])[0]['allowed'], True) | |||||
with self.nodes[0].assert_debug_log([f'received getdata for: tx {txid} peer']): | |||||
self.nodes[0].sendrawtransaction(tx_hex) | self.nodes[0].sendrawtransaction(tx_hex) | ||||
tx_relay_peer.wait_for_tx(txid) | tx_relay_peer.wait_for_tx(txid) | ||||
assert_equal(self.nodes[0].getmempoolinfo()['size'], 1) | assert_equal(self.nodes[0].getmempoolinfo()["size"], 1) | ||||
self.log.info("Restarting node 0 with relay permission and blocksonly") | self.log.info("Restarting node 0 with relay permission and blocksonly") | ||||
self.restart_node(0, | self.restart_node( | ||||
["-persistmempool=0", | 0, | ||||
[ | |||||
"-persistmempool=0", | |||||
"-whitelist=relay@127.0.0.1", | "-whitelist=relay@127.0.0.1", | ||||
"-blocksonly", | "-blocksonly", | ||||
'-deprecatedrpc=whitelisted']) | "-deprecatedrpc=whitelisted", | ||||
], | |||||
) | |||||
assert_equal(self.nodes[0].getrawmempool(), []) | assert_equal(self.nodes[0].getrawmempool(), []) | ||||
first_peer = self.nodes[0].add_p2p_connection(P2PInterface()) | first_peer = self.nodes[0].add_p2p_connection(P2PInterface()) | ||||
second_peer = self.nodes[0].add_p2p_connection(P2PInterface()) | second_peer = self.nodes[0].add_p2p_connection(P2PInterface()) | ||||
peer_1_info = self.nodes[0].getpeerinfo()[0] | peer_1_info = self.nodes[0].getpeerinfo()[0] | ||||
assert_equal(peer_1_info['permissions'], ['relay']) | assert_equal(peer_1_info["permissions"], ["relay"]) | ||||
peer_2_info = self.nodes[0].getpeerinfo()[1] | peer_2_info = self.nodes[0].getpeerinfo()[1] | ||||
assert_equal(peer_2_info['permissions'], ['relay']) | assert_equal(peer_2_info["permissions"], ["relay"]) | ||||
assert_equal( | assert_equal(self.nodes[0].testmempoolaccept([tx_hex])[0]["allowed"], True) | ||||
self.nodes[0].testmempoolaccept( | |||||
[tx_hex])[0]['allowed'], True) | |||||
self.log.info( | self.log.info( | ||||
'Check that the tx from first_peer with relay-permission is ' | "Check that the tx from first_peer with relay-permission is " | ||||
'relayed to others (ie.second_peer)') | "relayed to others (ie.second_peer)" | ||||
) | |||||
with self.nodes[0].assert_debug_log(["received getdata"]): | with self.nodes[0].assert_debug_log(["received getdata"]): | ||||
# Note that normally, first_peer would never send us transactions | # Note that normally, first_peer would never send us transactions | ||||
# since we're a blocksonly node. By activating blocksonly, we | # since we're a blocksonly node. By activating blocksonly, we | ||||
# explicitly tell our peers that they should not send us | # explicitly tell our peers that they should not send us | ||||
# transactions, and Bitcoin ABC respects that choice and will not | # transactions, and Bitcoin ABC respects that choice and will not | ||||
# send transactions. | # send transactions. | ||||
# But if, for some reason, first_peer decides to relay transactions | # But if, for some reason, first_peer decides to relay transactions | ||||
# to us anyway, we should relay them to second_peer since we gave | # to us anyway, we should relay them to second_peer since we gave | ||||
# relay permission to first_peer. | # relay permission to first_peer. | ||||
# See https://github.com/bitcoin/bitcoin/issues/19943 for details. | # See https://github.com/bitcoin/bitcoin/issues/19943 for details. | ||||
first_peer.send_message(msg_tx(tx)) | first_peer.send_message(msg_tx(tx)) | ||||
self.log.info( | self.log.info( | ||||
'Check that the peer with relay-permission is still connected' | "Check that the peer with relay-permission is still connected" | ||||
' after sending the transaction') | " after sending the transaction" | ||||
) | |||||
assert_equal(first_peer.is_connected, True) | assert_equal(first_peer.is_connected, True) | ||||
second_peer.wait_for_tx(txid) | second_peer.wait_for_tx(txid) | ||||
assert_equal(self.nodes[0].getmempoolinfo()['size'], 1) | assert_equal(self.nodes[0].getmempoolinfo()["size"], 1) | ||||
self.log.info( | self.log.info("Relay-permission peer's transaction is accepted and relayed") | ||||
"Relay-permission peer's transaction is accepted and relayed") | |||||
self.nodes[0].disconnect_p2ps() | self.nodes[0].disconnect_p2ps() | ||||
self.generate(self.nodes[0], 1) | self.generate(self.nodes[0], 1) | ||||
def blocks_relay_conn_tests(self): | def blocks_relay_conn_tests(self): | ||||
self.log.info( | self.log.info( | ||||
'Tests with node in normal mode with block-relay-only connections') | "Tests with node in normal mode with block-relay-only connections" | ||||
) | |||||
# disables blocks only mode | # disables blocks only mode | ||||
self.restart_node(0, ["-noblocksonly"]) | self.restart_node(0, ["-noblocksonly"]) | ||||
assert_equal(self.nodes[0].getnetworkinfo()['localrelay'], True) | assert_equal(self.nodes[0].getnetworkinfo()["localrelay"], True) | ||||
# Ensure we disconnect if a block-relay-only connection sends us a | # Ensure we disconnect if a block-relay-only connection sends us a | ||||
# transaction | # transaction | ||||
self.nodes[0].add_outbound_p2p_connection( | self.nodes[0].add_outbound_p2p_connection( | ||||
P2PInterface(), p2p_idx=0, connection_type="block-relay-only") | P2PInterface(), p2p_idx=0, connection_type="block-relay-only" | ||||
assert_equal(self.nodes[0].getpeerinfo()[0]['relaytxes'], False) | ) | ||||
assert_equal(self.nodes[0].getpeerinfo()[0]["relaytxes"], False) | |||||
_, txid, tx_hex = self.check_p2p_tx_violation() | _, txid, tx_hex = self.check_p2p_tx_violation() | ||||
self.log.info( | self.log.info("Check that txs from RPC are not sent to blockrelay connection") | ||||
"Check that txs from RPC are not sent to blockrelay connection") | |||||
conn = self.nodes[0].add_outbound_p2p_connection( | conn = self.nodes[0].add_outbound_p2p_connection( | ||||
P2PTxInvStore(), p2p_idx=1, connection_type="block-relay-only") | P2PTxInvStore(), p2p_idx=1, connection_type="block-relay-only" | ||||
) | |||||
self.nodes[0].sendrawtransaction(tx_hex) | self.nodes[0].sendrawtransaction(tx_hex) | ||||
# Bump time forward to ensure nNextInvSend timer pops | # Bump time forward to ensure nNextInvSend timer pops | ||||
self.nodes[0].setmocktime(int(time.time()) + 60) | self.nodes[0].setmocktime(int(time.time()) + 60) | ||||
conn.sync_send_with_ping() | conn.sync_send_with_ping() | ||||
assert int(txid, 16) not in conn.get_invs() | assert int(txid, 16) not in conn.get_invs() | ||||
def check_p2p_tx_violation(self): | def check_p2p_tx_violation(self): | ||||
self.log.info( | self.log.info("Check that txs from P2P are rejected and result in disconnect") | ||||
'Check that txs from P2P are rejected and result in disconnect') | |||||
spendtx = self.miniwallet.create_self_transfer(from_node=self.nodes[0]) | spendtx = self.miniwallet.create_self_transfer(from_node=self.nodes[0]) | ||||
with self.nodes[0].assert_debug_log(['transaction sent in violation of protocol peer=0']): | with self.nodes[0].assert_debug_log( | ||||
self.nodes[0].p2ps[0].send_message(msg_tx(spendtx['tx'])) | ["transaction sent in violation of protocol peer=0"] | ||||
): | |||||
self.nodes[0].p2ps[0].send_message(msg_tx(spendtx["tx"])) | |||||
self.nodes[0].p2ps[0].wait_for_disconnect() | self.nodes[0].p2ps[0].wait_for_disconnect() | ||||
assert_equal(self.nodes[0].getmempoolinfo()['size'], 0) | assert_equal(self.nodes[0].getmempoolinfo()["size"], 0) | ||||
# Remove the disconnected peer | # Remove the disconnected peer | ||||
del self.nodes[0].p2ps[0] | del self.nodes[0].p2ps[0] | ||||
return spendtx['tx'], spendtx['txid'], spendtx['hex'] | return spendtx["tx"], spendtx["txid"], spendtx["hex"] | ||||
if __name__ == '__main__': | if __name__ == "__main__": | ||||
P2PBlocksOnly().main() | P2PBlocksOnly().main() |