GlcoinCore/squeaknode
1
0
Fork 0
mirror of https://github.com/yzernik/squeaknode.git synced 2026-08-16 13:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Parse the correct ip addr when saving sent offer (#597)

Browse source
This commit is contained in:
Jonathan Zernik 2021-01-11 07:58:15 -08:00 • committed by GitHub
parent b176b7fdfc
commit 534b94e53d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
squeaknode/server/squeak_server_servicer.py
View file

@ -7,6 +7,7 @@ from squeak.core import CSqueak
from proto import squeak_server_pb2
from proto import squeak_server_pb2_grpc
from squeaknode.core.util import get_hash
from squeaknode.server.util import parse_ip_address
logger = logging.getLogger(__name__)
@ -72,8 +73,9 @@ class SqueakServerServicer(squeak_server_pb2_grpc.SqueakServerServicer):
squeak_hash = bytes.fromhex(squeak_hash_str)
# TODO: check if hash is valid
client_addr = context.peer()
ip_addr = parse_ip_address(client_addr)
buy_response = self.handler.handle_get_offer(squeak_hash, client_addr)
buy_response = self.handler.handle_get_offer(squeak_hash, ip_addr)
if buy_response is None:
context.set_code(grpc.StatusCode.NOT_FOUND)

4
squeaknode/server/util.py Normal file
View file

@ -0,0 +1,4 @@
def parse_ip_address(grpc_addr_str: str) -> str:
return grpc_addr_str.split(':')[1]