mirror of
https://github.com/dashpay/dash.git
synced 2024-12-25 20:12:57 +01:00
merge bitcoin#22211: relay I2P addresses even if not reachable (by us)
This commit is contained in:
parent
7e08db55fe
commit
fe66202c05
@ -232,7 +232,7 @@ public:
|
|||||||
*/
|
*/
|
||||||
bool IsRelayable() const
|
bool IsRelayable() const
|
||||||
{
|
{
|
||||||
return IsIPv4() || IsIPv6() || IsTor();
|
return IsIPv4() || IsIPv6() || IsTor() || IsI2P();
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -15,6 +15,10 @@ from test_framework.p2p import P2PInterface
|
|||||||
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
|
||||||
|
|
||||||
|
I2P_ADDR = "c4gfnttsuwqomiygupdqqqyy5y5emnk5c73hrfvatri67prd7vyq.b32.i2p"
|
||||||
|
|
||||||
|
ADDRS = []
|
||||||
|
|
||||||
|
|
||||||
class AddrReceiver(P2PInterface):
|
class AddrReceiver(P2PInterface):
|
||||||
addrv2_received_and_checked = False
|
addrv2_received_and_checked = False
|
||||||
@ -23,11 +27,8 @@ class AddrReceiver(P2PInterface):
|
|||||||
super().__init__(support_addrv2 = True)
|
super().__init__(support_addrv2 = True)
|
||||||
|
|
||||||
def on_addrv2(self, message):
|
def on_addrv2(self, message):
|
||||||
for addr in message.addrs:
|
if ADDRS == message.addrs:
|
||||||
assert_equal(addr.nServices, 1)
|
self.addrv2_received_and_checked = True
|
||||||
assert addr.ip.startswith('123.123.123.')
|
|
||||||
assert 8333 <= addr.port < 8343
|
|
||||||
self.addrv2_received_and_checked = True
|
|
||||||
|
|
||||||
def wait_for_addrv2(self):
|
def wait_for_addrv2(self):
|
||||||
self.wait_until(lambda: "addrv2" in self.last_message)
|
self.wait_until(lambda: "addrv2" in self.last_message)
|
||||||
@ -39,18 +40,21 @@ class AddrTest(BitcoinTestFramework):
|
|||||||
self.num_nodes = 1
|
self.num_nodes = 1
|
||||||
|
|
||||||
def run_test(self):
|
def run_test(self):
|
||||||
ADDRS = []
|
|
||||||
for i in range(10):
|
for i in range(10):
|
||||||
addr = CAddress()
|
addr = CAddress()
|
||||||
addr.time = int(self.mocktime) + i
|
addr.time = int(self.mocktime) + i
|
||||||
addr.nServices = NODE_NETWORK
|
addr.nServices = NODE_NETWORK
|
||||||
addr.ip = "123.123.123.{}".format(i % 256)
|
# Add one I2P address at an arbitrary position.
|
||||||
|
if i == 5:
|
||||||
|
addr.net = addr.NET_I2P
|
||||||
|
addr.ip = I2P_ADDR
|
||||||
|
else:
|
||||||
|
addr.ip = f"123.123.123.{i % 256}"
|
||||||
addr.port = 8333 + i
|
addr.port = 8333 + i
|
||||||
ADDRS.append(addr)
|
ADDRS.append(addr)
|
||||||
|
|
||||||
self.log.info('Create connection that sends addrv2 messages')
|
self.log.info('Create connection that sends addrv2 messages')
|
||||||
addr_source = self.nodes[0].add_p2p_connection(P2PInterface())
|
addr_source = self.nodes[0].add_p2p_connection(P2PInterface())
|
||||||
|
|
||||||
msg = msg_addrv2()
|
msg = msg_addrv2()
|
||||||
|
|
||||||
self.log.info('Send too-large addrv2 message')
|
self.log.info('Send too-large addrv2 message')
|
||||||
@ -63,22 +67,24 @@ class AddrTest(BitcoinTestFramework):
|
|||||||
self.log.info('Check that addrv2 message content is relayed and added to addrman')
|
self.log.info('Check that addrv2 message content is relayed and added to addrman')
|
||||||
addr_source = self.nodes[0].add_p2p_connection(P2PInterface())
|
addr_source = self.nodes[0].add_p2p_connection(P2PInterface())
|
||||||
addr_receiver = self.nodes[0].add_p2p_connection(AddrReceiver())
|
addr_receiver = self.nodes[0].add_p2p_connection(AddrReceiver())
|
||||||
|
|
||||||
msg.addrs = ADDRS
|
msg.addrs = ADDRS
|
||||||
with self.nodes[0].assert_debug_log([
|
with self.nodes[0].assert_debug_log([
|
||||||
'Added 10 addresses from 127.0.0.1: 0 tried',
|
# The I2P address is not added to node's own addrman because it has no
|
||||||
'received: addrv2 (131 bytes) peer=1',
|
# I2P reachability (thus 10 - 1 = 9).
|
||||||
|
'Added 9 addresses from 127.0.0.1: 0 tried',
|
||||||
|
'received: addrv2 (159 bytes) peer=1',
|
||||||
]):
|
]):
|
||||||
addr_source.send_and_ping(msg)
|
addr_source.send_and_ping(msg)
|
||||||
|
|
||||||
# Wait until "Added ..." before bumping mocktime to make sure addv2 is (almost) fully processed
|
# Wait until "Added ..." before bumping mocktime to make sure addv2 is (almost) fully processed
|
||||||
with self.nodes[0].assert_debug_log([
|
with self.nodes[0].assert_debug_log([
|
||||||
'sending addrv2 (131 bytes) peer=2',
|
'sending addrv2 (159 bytes) peer=2',
|
||||||
]):
|
]):
|
||||||
self.bump_mocktime(30 * 60)
|
self.bump_mocktime(30 * 60)
|
||||||
addr_receiver.wait_for_addrv2()
|
addr_receiver.wait_for_addrv2()
|
||||||
|
|
||||||
assert addr_receiver.addrv2_received_and_checked
|
assert addr_receiver.addrv2_received_and_checked
|
||||||
|
assert_equal(len(self.nodes[0].getnodeaddresses(count=0, network="i2p")), 0)
|
||||||
|
|
||||||
self.nodes[0].disconnect_p2ps()
|
self.nodes[0].disconnect_p2ps()
|
||||||
|
|
||||||
|
@ -16,7 +16,7 @@ ser_*, deser_*: functions that handle serialization/deserialization.
|
|||||||
Classes use __slots__ to ensure extraneous attributes aren't accidentally added
|
Classes use __slots__ to ensure extraneous attributes aren't accidentally added
|
||||||
by tests, compromising their intended effect.
|
by tests, compromising their intended effect.
|
||||||
"""
|
"""
|
||||||
|
from base64 import b32decode, b32encode
|
||||||
import copy
|
import copy
|
||||||
from collections import namedtuple
|
from collections import namedtuple
|
||||||
import hashlib
|
import hashlib
|
||||||
@ -251,15 +251,20 @@ class CAddress:
|
|||||||
|
|
||||||
# see https://github.com/bitcoin/bips/blob/master/bip-0155.mediawiki
|
# see https://github.com/bitcoin/bips/blob/master/bip-0155.mediawiki
|
||||||
NET_IPV4 = 1
|
NET_IPV4 = 1
|
||||||
|
NET_I2P = 5
|
||||||
|
|
||||||
ADDRV2_NET_NAME = {
|
ADDRV2_NET_NAME = {
|
||||||
NET_IPV4: "IPv4"
|
NET_IPV4: "IPv4",
|
||||||
|
NET_I2P: "I2P"
|
||||||
}
|
}
|
||||||
|
|
||||||
ADDRV2_ADDRESS_LENGTH = {
|
ADDRV2_ADDRESS_LENGTH = {
|
||||||
NET_IPV4: 4
|
NET_IPV4: 4,
|
||||||
|
NET_I2P: 32
|
||||||
}
|
}
|
||||||
|
|
||||||
|
I2P_PAD = "===="
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.time = 0
|
self.time = 0
|
||||||
self.nServices = 1
|
self.nServices = 1
|
||||||
@ -267,6 +272,9 @@ class CAddress:
|
|||||||
self.ip = "0.0.0.0"
|
self.ip = "0.0.0.0"
|
||||||
self.port = 0
|
self.port = 0
|
||||||
|
|
||||||
|
def __eq__(self, other):
|
||||||
|
return self.net == other.net and self.ip == other.ip and self.nServices == other.nServices and self.port == other.port and self.time == other.time
|
||||||
|
|
||||||
def deserialize(self, f, *, with_time=True):
|
def deserialize(self, f, *, with_time=True):
|
||||||
"""Deserialize from addrv1 format (pre-BIP155)"""
|
"""Deserialize from addrv1 format (pre-BIP155)"""
|
||||||
if with_time:
|
if with_time:
|
||||||
@ -299,24 +307,33 @@ class CAddress:
|
|||||||
self.nServices = deser_compact_size(f)
|
self.nServices = deser_compact_size(f)
|
||||||
|
|
||||||
self.net = struct.unpack("B", f.read(1))[0]
|
self.net = struct.unpack("B", f.read(1))[0]
|
||||||
assert self.net == self.NET_IPV4
|
assert self.net in (self.NET_IPV4, self.NET_I2P)
|
||||||
|
|
||||||
address_length = deser_compact_size(f)
|
address_length = deser_compact_size(f)
|
||||||
assert address_length == self.ADDRV2_ADDRESS_LENGTH[self.net]
|
assert address_length == self.ADDRV2_ADDRESS_LENGTH[self.net]
|
||||||
|
|
||||||
self.ip = socket.inet_ntoa(f.read(4))
|
addr_bytes = f.read(address_length)
|
||||||
|
if self.net == self.NET_IPV4:
|
||||||
|
self.ip = socket.inet_ntoa(addr_bytes)
|
||||||
|
else:
|
||||||
|
self.ip = b32encode(addr_bytes)[0:-len(self.I2P_PAD)].decode("ascii").lower() + ".b32.i2p"
|
||||||
|
|
||||||
self.port = struct.unpack(">H", f.read(2))[0]
|
self.port = struct.unpack(">H", f.read(2))[0]
|
||||||
|
|
||||||
def serialize_v2(self):
|
def serialize_v2(self):
|
||||||
"""Serialize in addrv2 format (BIP155)"""
|
"""Serialize in addrv2 format (BIP155)"""
|
||||||
assert self.net == self.NET_IPV4
|
assert self.net in (self.NET_IPV4, self.NET_I2P)
|
||||||
r = b""
|
r = b""
|
||||||
r += struct.pack("<I", self.time)
|
r += struct.pack("<I", self.time)
|
||||||
r += ser_compact_size(self.nServices)
|
r += ser_compact_size(self.nServices)
|
||||||
r += struct.pack("B", self.net)
|
r += struct.pack("B", self.net)
|
||||||
r += ser_compact_size(self.ADDRV2_ADDRESS_LENGTH[self.net])
|
r += ser_compact_size(self.ADDRV2_ADDRESS_LENGTH[self.net])
|
||||||
r += socket.inet_aton(self.ip)
|
if self.net == self.NET_IPV4:
|
||||||
|
r += socket.inet_aton(self.ip)
|
||||||
|
else:
|
||||||
|
sfx = ".b32.i2p"
|
||||||
|
assert self.ip.endswith(sfx)
|
||||||
|
r += b32decode(self.ip[0:-len(sfx)] + self.I2P_PAD, True)
|
||||||
r += struct.pack(">H", self.port)
|
r += struct.pack(">H", self.port)
|
||||||
return r
|
return r
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user