42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
|
import logging
|
||
|
|
|
||
|
|
from libp2p.network.stream.net_stream_interface import INetStream
|
||
|
|
from libp2p.peer.id import ID
|
||
|
|
|
||
|
|
ID = "/ipfs/ping/1.0.0"
|
||
|
|
PING_LENGTH = 32
|
||
|
|
RESP_TIMEOUT = 60
|
||
|
|
|
||
|
|
logger = logging.getLogger("libp2p.host.ping")
|
||
|
|
|
||
|
|
|
||
|
|
async def _handle_ping(stream: INetStream, peer_id: ID) -> None:
|
||
|
|
try:
|
||
|
|
payload = await asyncio.wait_for(stream.read(PING_LENGTH), RESP_TIMEOUT)
|
||
|
|
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 Exception as 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())
|
||
|
|
|
||
|
|
await stream.write(payload)
|
||
|
|
|
||
|
|
|
||
|
|
async def handle_ping(self, stream: INetStream) -> None:
|
||
|
|
"""
|
||
|
|
``handle_ping`` responds to incoming ping requests until one side
|
||
|
|
errors or closes the ``stream``.
|
||
|
|
"""
|
||
|
|
peer_id = stream.muxed_conn.peer_id
|
||
|
|
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
await _handle_ping(stream, peer_id)
|
||
|
|
except Exception:
|
||
|
|
await stream.reset()
|
||
|
|
return
|