varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-08 14:10:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Restore initiator flag to BaseSession type

Browse Source
This commit is contained in:
Alex Stokes
2019-09-08 15:37:41 -04:00
parent f38899e26e
commit 2025a5c7f1
3 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -27,9 +27,10 @@ class InsecureSession(BaseSession):
local_peer: ID,
local_private_key: PrivateKey,
conn: ReadWriteCloser,
initiator: bool,
peer_id: Optional[ID] = None,
) -> None:
super().__init__(local_peer, local_private_key, peer_id)
super().__init__(local_peer, local_private_key, initiator, peer_id)
self.conn = conn
async def write(self, data: bytes) -> int:
@ -99,7 +100,7 @@ class InsecureTransport(BaseSecureTransport):
for an inbound connection (i.e. we are not the initiator)
:return: secure connection object (that implements secure_conn_interface)
"""
session = InsecureSession(self.local_peer, self.local_private_key, conn)
session = InsecureSession(self.local_peer, self.local_private_key, conn, False)
await session.run_handshake()
return session
@ -110,7 +111,7 @@ class InsecureTransport(BaseSecureTransport):
:return: secure connection object (that implements secure_conn_interface)
"""
session = InsecureSession(
self.local_peer, self.local_private_key, conn, peer_id
self.local_peer, self.local_private_key, conn, True, peer_id
)
await session.run_handshake()
return session