varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix msg encoding

Browse Source 
- Change varint-prefix encode to fixedint-prefix(4 bytes) encode.
This commit is contained in:
mhchia
2019-08-17 21:41:17 +08:00
parent 22b1a5395d
commit bb7d37fd4f
4 changed files with 38 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
libp2p/security/insecure/transport.py
View File

@ -5,7 +5,7 @@ from libp2p.security.base_session import BaseSession
from libp2p.security.base_transport import BaseSecureTransport
from libp2p.security.secure_conn_interface import ISecureConn
from libp2p.typing import TProtocol
from libp2p.utils import encode_varint_prefixed, read_varint_prefixed_bytes
from libp2p.utils import encode_fixedint_prefixed, read_fixedint_prefixed
from .exceptions import UpgradeFailure
from .pb import plaintext_pb2
@ -20,12 +20,15 @@ class InsecureSession(BaseSession):
# FIXME: Update the read/write to `BaseSession`
async def run_handshake(self):
msg = make_exchange_message(self.local_private_key.get_public_key())
self.writer.write(encode_varint_prefixed(msg.SerializeToString()))
msg_bytes = msg.SerializeToString()
encoded_msg_bytes = encode_fixedint_prefixed(msg_bytes)
self.writer.write(encoded_msg_bytes)
await self.writer.drain()
msg_bytes_other_side = await read_varint_prefixed_bytes(self.reader)
msg_bytes_other_side = await read_fixedint_prefixed(self.reader)
msg_other_side = plaintext_pb2.Exchange()
msg_other_side.ParseFromString(msg_bytes_other_side)
# TODO: Verify public key with peer id
# TODO: Store public key
self.remote_peer_id = ID(msg_other_side.id)