37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
from libp2p.stream_muxer.mplex.mplex import Mplex
|
|
from libp2p.security.security_multistream import SecurityMultistream
|
|
|
|
|
|
class TransportUpgrader:
|
|
def __init__(self, secOpt, muxerOpt):
|
|
# Store security option
|
|
self.security_multistream = SecurityMultistream()
|
|
for key in secOpt:
|
|
self.security_multistream.add_transport(key, secOpt[key])
|
|
|
|
# Store muxer option
|
|
self.muxer = muxerOpt
|
|
|
|
def upgrade_listener(self, transport, listeners):
|
|
"""
|
|
Upgrade multiaddr listeners to libp2p-transport listeners
|
|
"""
|
|
|
|
async def upgrade_security(self, raw_conn, peer_id, initiator):
|
|
"""
|
|
Upgrade conn to be a secured connection
|
|
"""
|
|
if initiator:
|
|
return await self.security_multistream.secure_outbound(raw_conn, peer_id)
|
|
|
|
return await self.security_multistream.secure_inbound(raw_conn)
|
|
|
|
def upgrade_connection(self, conn, generic_protocol_handler, peer_id):
|
|
"""
|
|
Upgrade raw connection to muxed connection
|
|
"""
|
|
|
|
# For PoC, no security, default to mplex
|
|
# TODO do exchange to determine multiplexer
|
|
return Mplex(conn, generic_protocol_handler, peer_id)
|