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

Add "msgio" functions

Browse Source
This commit is contained in:
Alex Stokes 2019-08-22 17:54:49 +02:00
parent 26165b0729
commit b59c5d6ca1
No known key found for this signature in database
GPG Key ID: 51CE1721B245C086
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
libp2p/io/__init__.py Normal file
View File

16
libp2p/io/msgio.py Normal file
View File

@ -0,0 +1,16 @@
import asyncio
SIZE_LEN_BYTES = 4
# TODO unify w/ https://github.com/libp2p/py-libp2p/blob/1aed52856f56a4b791696bbcbac31b5f9c2e88c9/libp2p/utils.py#L85-L99
def encode(msg_bytes: bytes) -> bytes:
len_prefix = len(msg_bytes).to_bytes(SIZE_LEN_BYTES, "big")
return len_prefix + msg_bytes
async def read_next_message(reader: asyncio.StreamReader) -> bytes:
len_bytes = await reader.readexactly(SIZE_LEN_BYTES)
len_int = int.from_bytes(len_bytes, "big")
return await reader.readexactly(len_int)