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

Make mplex.read_message handle RawConnError

Browse Source
This commit is contained in:
mhchia
2019-09-21 18:17:00 +08:00
parent e44c2145cc
commit 539047be2d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libp2p/stream_muxer/mplex/mplex.py
View File

@ -4,6 +4,7 @@ from typing import Awaitable, Dict, List, Optional, Tuple
from libp2p.exceptions import ParseError from libp2p.exceptions import ParseError
from libp2p.io.exceptions import IncompleteReadError from libp2p.io.exceptions import IncompleteReadError
from libp2p.network.connection.exceptions import RawConnError
from libp2p.peer.id import ID from libp2p.peer.id import ID
from libp2p.security.secure_conn_interface import ISecureConn from libp2p.security.secure_conn_interface import ISecureConn
from libp2p.stream_muxer.abc import IMuxedConn, IMuxedStream from libp2p.stream_muxer.abc import IMuxedConn, IMuxedStream
@ -203,8 +204,7 @@ class Mplex(IMuxedConn):
message = await asyncio.wait_for( message = await asyncio.wait_for(
read_varint_prefixed_bytes(self.secured_conn), timeout=5 read_varint_prefixed_bytes(self.secured_conn), timeout=5
) )
# TODO: Catch RawConnError? except (ParseError, RawConnError, IncompleteReadError) as error:
except (ParseError, IncompleteReadError) as error:
raise MplexUnavailable( raise MplexUnavailable(
"failed to read messages correctly from the underlying connection" "failed to read messages correctly from the underlying connection"
) from error ) from error