varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-12 16:10:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor creation of identify response to make testing easier

Browse Source
This commit is contained in:
Alex Stokes
2019-11-06 15:18:40 -08:00
parent 071eccc995
commit 58f360167d
1 changed files with 17 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
libp2p/identity/identify/protocol.py
View File

@ -19,16 +19,12 @@ def _multiaddr_to_bytes(maddr: Multiaddr) -> bytes:
return maddr.to_bytes() return maddr.to_bytes()
def identify_handler_for(host: IHost) -> StreamHandlerFn: def _mk_identify_protobuf(host: IHost) -> Identify:
async def handle_identify(stream: INetStream) -> None:
peer_id = stream.muxed_conn.peer_id
logger.debug("received a request for %s from %s", ID, peer_id)
public_key = host.get_public_key() public_key = host.get_public_key()
laddrs = host.get_addrs() laddrs = host.get_addrs()
protocols = host.get_mux().get_protocols() protocols = host.get_mux().get_protocols()
protobuf = Identify( return Identify(
protocol_version=PROTOCOL_VERSION, protocol_version=PROTOCOL_VERSION,
agent_version=AGENT_VERSION, agent_version=AGENT_VERSION,
public_key=public_key.serialize(), public_key=public_key.serialize(),
@ -37,6 +33,14 @@ def identify_handler_for(host: IHost) -> StreamHandlerFn:
observed_addr=b"", observed_addr=b"",
protocols=protocols, protocols=protocols,
) )
def identify_handler_for(host: IHost) -> StreamHandlerFn:
async def handle_identify(stream: INetStream) -> None:
peer_id = stream.muxed_conn.peer_id
logger.debug("received a request for %s from %s", ID, peer_id)
protobuf = _mk_identify_protobuf(host)
response = protobuf.SerializeToString() response = protobuf.SerializeToString()
await stream.write(response) await stream.write(response)