GlcoinCore/squeaknode
1
0
Fork 0
mirror of https://github.com/yzernik/squeaknode.git synced 2026-08-15 12:50:47 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Simplify handle_getdata in peer message handler (#1161)

Browse source
This commit is contained in:
Jonathan Zernik 2021-09-03 05:49:07 -07:00 • committed by GitHub
parent 86c0b4ab86
commit 21efa2d4d9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 25 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

49
squeaknode/network/peer_message_handler.py
View file

@ -132,30 +132,11 @@ class PeerMessageHandler:
invs = msg.inv
not_found = []
for inv in invs:
if inv.type == 1:
squeak = self.squeak_controller.get_squeak(inv.hash)
if squeak is None:
not_found.append(inv)
else:
squeak.ClearDecryptionKey()
squeak_msg = msg_squeak(squeak=squeak)
self.peer.send_msg(squeak_msg)
if inv.type == 2:
offer = self.squeak_controller.get_buy_offer(
squeak_hash=inv.hash,
peer_address=self.peer.remote_address,
)
if offer is None:
not_found.append(inv)
else:
offer_msg = msg_offer(
hashSqk=inv.hash,
nonce=offer.nonce,
strPaymentInfo=offer.payment_request.encode('utf-8'),
host=offer.host.encode('utf-8'),
port=offer.port,
)
self.peer.send_msg(offer_msg)
reply_msg = self._get_inv_reply(inv)
if reply_msg is not None:
self.peer.send_msg(reply_msg)
else:
not_found.append(inv)
if not_found:
notfound_msg = msg_notfound(inv=not_found)
self.peer.send_msg(notfound_msg)
@ -211,3 +192,23 @@ class PeerMessageHandler:
if invs:
inv_msg = msg_inv(inv=invs)
self.peer.send_msg(inv_msg)
def _get_inv_reply(self, inv):
if inv.type == 1:
squeak = self.squeak_controller.get_squeak(inv.hash)
if squeak is not None:
squeak.ClearDecryptionKey()
return msg_squeak(squeak=squeak)
if inv.type == 2:
offer = self.squeak_controller.get_buy_offer(
squeak_hash=inv.hash,
peer_address=self.peer.remote_address,
)
if offer is not None:
return msg_offer(
hashSqk=inv.hash,
nonce=offer.nonce,
strPaymentInfo=offer.payment_request.encode('utf-8'),
host=offer.host.encode('utf-8'),
port=offer.port,
)