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

Raise read_delim exception with different msgs

Browse Source 
Separate `len(msg_bytes) == 0` and `msg_bytes[-1:] != b"\n"`, to raise
`ParseError` with different messages.
This commit is contained in:
mhchia 2019-09-24 13:22:25 +08:00
parent 37bee9fb16
commit 7405f078e6
No known key found for this signature in database
GPG Key ID: 389EFBEA1362589A
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libp2p/utils.py
View File

@ -73,8 +73,12 @@ def encode_delim(msg: bytes) -> bytes:
async def read_delim(reader: Reader) -> bytes: async def read_delim(reader: Reader) -> bytes:
msg_bytes = await read_varint_prefixed_bytes(reader) msg_bytes = await read_varint_prefixed_bytes(reader)
if len(msg_bytes) == 0 or msg_bytes[-1:] != b"\n": if len(msg_bytes) == 0:
raise ParseError(f'msg_bytes is not delimited by b"\\n": msg_bytes={msg_bytes}') raise ParseError(f"`len(msg_bytes)` should not be 0")
if msg_bytes[-1:] != b"\n":
raise ParseError(
f'`msg_bytes` is not delimited by b"\\n": `msg_bytes`={msg_bytes}'
)
return msg_bytes[:-1] return msg_bytes[:-1]