mirror of
https://github.com/varun-r-mallya/py-libp2p.git
synced 2025-12-31 20:36:24 +00:00
28f6de37ee
* ignore kad * fix swarm, and minor * fix init and swarm * ignore pb * enable mypy * fix basic host * fix tcp * fix mplex * add typing for pb * skip format pyi * [mypy] no need to ignore pb now * add typing to chat
59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
from typing import Mapping, Sequence
|
|
|
|
from libp2p.network.connection.raw_connection_interface import IRawConnection
|
|
from libp2p.network.typing import GenericProtocolHandlerFn
|
|
from libp2p.peer.id import ID
|
|
from libp2p.security.secure_conn_interface import ISecureConn
|
|
from libp2p.security.secure_transport_interface import ISecureTransport
|
|
from libp2p.security.security_multistream import SecurityMultistream
|
|
from libp2p.stream_muxer.mplex.mplex import Mplex
|
|
from libp2p.typing import TProtocol
|
|
|
|
from .listener_interface import IListener
|
|
from .transport_interface import ITransport
|
|
|
|
|
|
class TransportUpgrader:
|
|
security_multistream: SecurityMultistream
|
|
muxer: Sequence[str]
|
|
|
|
def __init__(
|
|
self, secOpt: Mapping[TProtocol, ISecureTransport], muxerOpt: Sequence[str]
|
|
) -> None:
|
|
# 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: ITransport, listeners: IListener) -> None:
|
|
"""
|
|
Upgrade multiaddr listeners to libp2p-transport listeners
|
|
"""
|
|
pass
|
|
|
|
async def upgrade_security(
|
|
self, raw_conn: IRawConnection, peer_id: ID, initiator: bool
|
|
) -> ISecureConn:
|
|
"""
|
|
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)
|
|
|
|
@staticmethod
|
|
def upgrade_connection(
|
|
conn: ISecureConn, generic_protocol_handler: GenericProtocolHandlerFn, peer_id: ID
|
|
) -> Mplex:
|
|
"""
|
|
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)
|