Changeset View
Changeset View
Standalone View
Standalone View
test/functional/p2p_feefilter.py
#!/usr/bin/env python3 | #!/usr/bin/env python3 | ||||
# Copyright (c) 2016-2019 The Bitcoin Core developers | # Copyright (c) 2016-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 processing of feefilter messages.""" | """Test processing of feefilter messages.""" | ||||
from decimal import Decimal | from decimal import Decimal | ||||
import time | import time | ||||
from test_framework.messages import MSG_TX, msg_feefilter | from test_framework.messages import MSG_TX, msg_feefilter | ||||
from test_framework.mininode import ( | from test_framework.mininode import ( | ||||
mininode_lock, | mininode_lock, | ||||
P2PInterface, | P2PInterface, | ||||
) | ) | ||||
from test_framework.test_framework import BitcoinTestFramework | from test_framework.test_framework import BitcoinTestFramework | ||||
from test_framework.util import assert_equal | |||||
def hashToHex(hash): | def hashToHex(hash): | ||||
return format(hash, '064x') | return format(hash, '064x') | ||||
# Wait up to 60 secs to see if the testnode has received all the expected invs | # Wait up to 60 secs to see if the testnode has received all the expected invs | ||||
def allInvsMatch(invsExpected, testnode): | def allInvsMatch(invsExpected, testnode): | ||||
for x in range(60): | for x in range(60): | ||||
with mininode_lock: | with mininode_lock: | ||||
if (sorted(invsExpected) == sorted(testnode.txinvs)): | if (sorted(invsExpected) == sorted(testnode.txinvs)): | ||||
return True | return True | ||||
time.sleep(1) | time.sleep(1) | ||||
return False | return False | ||||
class FeefilterConn(P2PInterface): | |||||
feefilter_received = False | |||||
def on_feefilter(self, message): | |||||
self.feefilter_received = True | |||||
def assert_feefilter_received(self, recv: bool): | |||||
with mininode_lock: | |||||
assert_equal(self.feefilter_received, recv) | |||||
class TestP2PConn(P2PInterface): | class TestP2PConn(P2PInterface): | ||||
def __init__(self): | def __init__(self): | ||||
super().__init__() | super().__init__() | ||||
self.txinvs = [] | self.txinvs = [] | ||||
def on_inv(self, message): | def on_inv(self, message): | ||||
for i in message.inv: | for i in message.inv: | ||||
if (i.type == MSG_TX): | if (i.type == MSG_TX): | ||||
Show All 14 Lines | def set_test_params(self): | ||||
# See issue #16499 | # See issue #16499 | ||||
self.extra_args = [["-minrelaytxfee=0.00000100", | self.extra_args = [["-minrelaytxfee=0.00000100", | ||||
"-mintxfee=0.00000100"]] * self.num_nodes | "-mintxfee=0.00000100"]] * self.num_nodes | ||||
def skip_test_if_missing_module(self): | def skip_test_if_missing_module(self): | ||||
self.skip_if_no_wallet() | self.skip_if_no_wallet() | ||||
def run_test(self): | def run_test(self): | ||||
self.test_feefilter_forcerelay() | |||||
self.test_feefilter() | |||||
def test_feefilter_forcerelay(self): | |||||
self.log.info( | |||||
'Check that peers without forcerelay permission (default) get a feefilter message') | |||||
self.nodes[0].add_p2p_connection( | |||||
FeefilterConn()).assert_feefilter_received(True) | |||||
self.log.info( | |||||
'Check that peers with forcerelay permission do not get a feefilter message') | |||||
self.restart_node(0, extra_args=['-whitelist=forcerelay@127.0.0.1']) | |||||
self.nodes[0].add_p2p_connection( | |||||
FeefilterConn()).assert_feefilter_received(False) | |||||
# Restart to disconnect peers and load default extra_args | |||||
self.restart_node(0) | |||||
self.connect_nodes(1, 0) | |||||
def test_feefilter(self): | |||||
node1 = self.nodes[1] | node1 = self.nodes[1] | ||||
node0 = self.nodes[0] | node0 = self.nodes[0] | ||||
# Get out of IBD | |||||
node1.generate(1) | |||||
self.sync_blocks() | |||||
self.nodes[0].add_p2p_connection(TestP2PConn()) | conn = self.nodes[0].add_p2p_connection(TestP2PConn()) | ||||
# Test that invs are received by test connection for all txs at | # Test that invs are received by test connection for all txs at | ||||
# feerate of .2 sat/byte | # feerate of .2 sat/byte | ||||
node1.settxfee(Decimal("0.00000200")) | node1.settxfee(Decimal("0.00000200")) | ||||
txids = [node1.sendtoaddress(node1.getnewaddress(), 1) | txids = [node1.sendtoaddress(node1.getnewaddress(), 1) | ||||
for x in range(3)] | for x in range(3)] | ||||
assert allInvsMatch(txids, self.nodes[0].p2p) | assert allInvsMatch(txids, conn) | ||||
self.nodes[0].p2p.clear_invs() | conn.clear_invs() | ||||
# Set a filter of .15 sat/byte on test connection | # Set a filter of .15 sat/byte on test connection | ||||
self.nodes[0].p2p.send_and_ping(msg_feefilter(150)) | conn.send_and_ping(msg_feefilter(150)) | ||||
# Test that txs are still being received by test connection | # Test that txs are still being received by test connection | ||||
# (paying .15 sat/byte) | # (paying .15 sat/byte) | ||||
node1.settxfee(Decimal("0.00000150")) | node1.settxfee(Decimal("0.00000150")) | ||||
txids = [node1.sendtoaddress(node1.getnewaddress(), 1) | txids = [node1.sendtoaddress(node1.getnewaddress(), 1) | ||||
for x in range(3)] | for x in range(3)] | ||||
assert allInvsMatch(txids, self.nodes[0].p2p) | assert allInvsMatch(txids, conn) | ||||
self.nodes[0].p2p.clear_invs() | conn.clear_invs() | ||||
# Change tx fee rate to .1 sat/byte and test they are no longer received | # Change tx fee rate to .1 sat/byte and test they are no longer received | ||||
# by the test connection | # by the test connection | ||||
node1.settxfee(Decimal("0.00000100")) | node1.settxfee(Decimal("0.00000100")) | ||||
[node1.sendtoaddress(node1.getnewaddress(), 1) for x in range(3)] | [node1.sendtoaddress(node1.getnewaddress(), 1) for x in range(3)] | ||||
self.sync_mempools() # must be sure node 0 has received all txs | self.sync_mempools() # must be sure node 0 has received all txs | ||||
# Send one transaction from node0 that should be received, so that we | # Send one transaction from node0 that should be received, so that we | ||||
# we can sync the test on receipt (if node1's txs were relayed, they'd | # we can sync the test on receipt (if node1's txs were relayed, they'd | ||||
# be received by the time this node0 tx is received). This is | # be received by the time this node0 tx is received). This is | ||||
# unfortunately reliant on the current relay behavior where we batch up | # unfortunately reliant on the current relay behavior where we batch up | ||||
# to 35 entries in an inv, which means that when this next transaction | # to 35 entries in an inv, which means that when this next transaction | ||||
# is eligible for relay, the prior transactions from node1 are eligible | # is eligible for relay, the prior transactions from node1 are eligible | ||||
# as well. | # as well. | ||||
node0.settxfee(Decimal("0.00020000")) | node0.settxfee(Decimal("0.00020000")) | ||||
txids = [node0.sendtoaddress(node0.getnewaddress(), 1)] | txids = [node0.sendtoaddress(node0.getnewaddress(), 1)] | ||||
assert allInvsMatch(txids, self.nodes[0].p2p) | assert allInvsMatch(txids, conn) | ||||
self.nodes[0].p2p.clear_invs() | conn.clear_invs() | ||||
# Remove fee filter and check that txs are received again | # Remove fee filter and check that txs are received again | ||||
self.nodes[0].p2p.send_and_ping(msg_feefilter(0)) | conn.send_and_ping(msg_feefilter(0)) | ||||
txids = [node1.sendtoaddress(node1.getnewaddress(), 1) | txids = [node1.sendtoaddress(node1.getnewaddress(), 1) | ||||
for x in range(3)] | for x in range(3)] | ||||
assert allInvsMatch(txids, self.nodes[0].p2p) | assert allInvsMatch(txids, conn) | ||||
self.nodes[0].p2p.clear_invs() | conn.clear_invs() | ||||
if __name__ == '__main__': | if __name__ == '__main__': | ||||
FeeFilterTest().main() | FeeFilterTest().main() |