From f24b488f79b0597dfde3eedd982bb5ec854d4231 Mon Sep 17 00:00:00 2001 From: Alex Stokes Date: Tue, 24 Sep 2019 19:04:18 -0700 Subject: [PATCH] handle other side closing their end of the connection during `ping` --- libp2p/host/ping.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/libp2p/host/ping.py b/libp2p/host/ping.py index a01609d..9f43d16 100644 --- a/libp2p/host/ping.py +++ b/libp2p/host/ping.py @@ -1,5 +1,6 @@ import logging +from libp2p.network.stream.exceptions import StreamEOF, StreamReset from libp2p.network.stream.net_stream_interface import INetStream from libp2p.peer.id import ID @@ -16,7 +17,11 @@ async def _handle_ping(stream: INetStream, peer_id: ID) -> None: except asyncio.TimeoutError as error: logger.debug("Timed out waiting for ping from %s: %s", peer_id, error) raise - # TODO: handle the other end closing the stream + except (StreamEOF, StreamReset) as error: + logger.debug( + "Other side closed while waiting for ping from %s: %s", peer_id, error + ) + raise except Exception as error: logger.debug("Error while waiting to read ping for %s: %s", peer_id, error) raise