StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Handle StreamClosed in ping protocol handler

Browse Source
This commit is contained in:
NIC619 2019-11-12 18:33:44 +08:00
parent 9be9b4bbfc
commit 86e0fa4563
No known key found for this signature in database
GPG Key ID: 570C35F5C2D51B17
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
libp2p/host/ping.py
View File

@ -1,7 +1,7 @@
import asyncio
import logging
from libp2p.network.stream.exceptions import StreamEOF, StreamReset
from libp2p.network.stream.exceptions import StreamClosed, StreamEOF, StreamReset
from libp2p.network.stream.net_stream_interface import INetStream
from libp2p.peer.id import ID as PeerID
from libp2p.typing import TProtocol
@ -25,17 +25,24 @@ async def _handle_ping(stream: INetStream, peer_id: PeerID) -> bool:
logger.debug("Other side closed while waiting for ping from %s", peer_id)
return False
except StreamReset as error:
print("peer", peer_id, "ping reset")
logger.debug(
"Other side reset while waiting for ping from %s: %s", peer_id, error
)
raise
except Exception as error:
print("peer", peer_id, "ping", type(error))
logger.debug("Error while waiting to read ping for %s: %s", peer_id, error)
raise
logger.debug("Received ping from %s with data: 0x%s", peer_id, payload.hex())
print("receive ping from", peer_id)
await stream.write(payload)
try:
await stream.write(payload)
except StreamClosed:
logger.debug("Fail to respond to ping from %s: stream closed", peer_id)
raise
return True
@ -44,11 +51,13 @@ async def handle_ping(stream: INetStream) -> None:
or closes the ``stream``."""
peer_id = stream.muxed_conn.peer_id
print("handling ping from", peer_id)
while True:
try:
should_continue = await _handle_ping(stream, peer_id)
if not should_continue:
return
except Exception:
print("error finish ping")
await stream.reset()
return