StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests 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
No known key found for this signature in database
GPG Key ID: 389EFBEA1362589A
1 changed files with 2 additions and 2 deletions
Show Stats 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