phantom/bitcoin
1
0
Fork 0
mirror of https://github.com/bitcoin/bitcoin.git synced 2025-01-10 03:47:29 -03:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

test: support WTX INVs from P2PDataStore and fix a comment

Browse source
This commit is contained in:
Vasil Dimov 2023-05-24 08:34:57 +02:00
parent ebe42c00aa
commit 22cd0e888c
No known key found for this signature in database
GPG key ID: 54DF06F64B55CBBF
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
test/functional/test_framework/p2p.py
View file

@ -808,12 +808,13 @@ class P2PDataStore(P2PInterface):
self.getdata_requests = [] self.getdata_requests = []
def on_getdata(self, message): def on_getdata(self, message):
"""Check for the tx/block in our stores and if found, reply with an inv message.""" """Check for the tx/block in our stores and if found, reply with MSG_TX or MSG_BLOCK."""
for inv in message.inv: for inv in message.inv:
self.getdata_requests.append(inv.hash) self.getdata_requests.append(inv.hash)
if (inv.type & MSG_TYPE_MASK) == MSG_TX and inv.hash in self.tx_store.keys(): invtype = inv.type & MSG_TYPE_MASK
if (invtype == MSG_TX or invtype == MSG_WTX) and inv.hash in self.tx_store.keys():
self.send_message(msg_tx(self.tx_store[inv.hash])) self.send_message(msg_tx(self.tx_store[inv.hash]))
elif (inv.type & MSG_TYPE_MASK) == MSG_BLOCK and inv.hash in self.block_store.keys(): elif invtype == MSG_BLOCK and inv.hash in self.block_store.keys():
self.send_message(msg_block(self.block_store[inv.hash])) self.send_message(msg_block(self.block_store[inv.hash]))
else: else:
logger.debug('getdata message type {} received.'.format(hex(inv.type))) logger.debug('getdata message type {} received.'.format(hex(inv.type)))