Skip to content

Commit faec09b

Browse files
committed
Merge #8894: [Testing] Include fRelay in mininode version messages
e5d682f Fix mininode version message format (jnewbery)
2 parents fb15610 + e5d682f commit faec09b

File tree

1 file changed

+20
-6
lines changed

1 file changed

+20
-6
lines changed

qa/rpc-tests/test_framework/mininode.py

Lines changed: 20 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,7 @@
4141
BIP0031_VERSION = 60000
4242
MY_VERSION = 70014 # past bip-31 for ping/pong
4343
MY_SUBVERSION = b"/python-mininode-tester:0.0.3/"
44+
MY_RELAY = 1 # from version 70001 onwards, fRelay should be appended to version messages (BIP37)
4445

4546
MAX_INV_SZ = 50000
4647
MAX_BLOCK_SIZE = 1000000
@@ -951,6 +952,7 @@ def __init__(self):
951952
self.nNonce = random.getrandbits(64)
952953
self.strSubVer = MY_SUBVERSION
953954
self.nStartingHeight = -1
955+
self.nRelay = MY_RELAY
954956

955957
def deserialize(self, f):
956958
self.nVersion = struct.unpack("<i", f.read(4))[0]
@@ -960,21 +962,32 @@ def deserialize(self, f):
960962
self.nTime = struct.unpack("<q", f.read(8))[0]
961963
self.addrTo = CAddress()
962964
self.addrTo.deserialize(f)
965+
963966
if self.nVersion >= 106:
964967
self.addrFrom = CAddress()
965968
self.addrFrom.deserialize(f)
966969
self.nNonce = struct.unpack("<Q", f.read(8))[0]
967970
self.strSubVer = deser_string(f)
968-
if self.nVersion >= 209:
969-
self.nStartingHeight = struct.unpack("<i", f.read(4))[0]
970-
else:
971-
self.nStartingHeight = None
972971
else:
973972
self.addrFrom = None
974973
self.nNonce = None
975974
self.strSubVer = None
976975
self.nStartingHeight = None
977976

977+
if self.nVersion >= 209:
978+
self.nStartingHeight = struct.unpack("<i", f.read(4))[0]
979+
else:
980+
self.nStartingHeight = None
981+
982+
if self.nVersion >= 70001:
983+
# Relay field is optional for version 70001 onwards
984+
try:
985+
self.nRelay = struct.unpack("<b", f.read(1))[0]
986+
except:
987+
self.nRelay = 0
988+
else:
989+
self.nRelay = 0
990+
978991
def serialize(self):
979992
r = b""
980993
r += struct.pack("<i", self.nVersion)
@@ -985,13 +998,14 @@ def serialize(self):
985998
r += struct.pack("<Q", self.nNonce)
986999
r += ser_string(self.strSubVer)
9871000
r += struct.pack("<i", self.nStartingHeight)
1001+
r += struct.pack("<b", self.nRelay)
9881002
return r
9891003

9901004
def __repr__(self):
991-
return 'msg_version(nVersion=%i nServices=%i nTime=%s addrTo=%s addrFrom=%s nNonce=0x%016X strSubVer=%s nStartingHeight=%i)' \
1005+
return 'msg_version(nVersion=%i nServices=%i nTime=%s addrTo=%s addrFrom=%s nNonce=0x%016X strSubVer=%s nStartingHeight=%i nRelay=%i)' \
9921006
% (self.nVersion, self.nServices, time.ctime(self.nTime),
9931007
repr(self.addrTo), repr(self.addrFrom), self.nNonce,
994-
self.strSubVer, self.nStartingHeight)
1008+
self.strSubVer, self.nStartingHeight, self.nRelay)
9951009

9961010

9971011
class msg_verack(object):

0 commit comments

Comments
 (0)