Browse Source

upgrade connection

pull/47/head
Alex Haynes 6 years ago
parent
commit
1bfa1f1ba5
  1. 2
      network/swarm.py
  2. 9
      transport/upgrader.py

2
network/swarm.py

@ -43,7 +43,7 @@ class Swarm(INetwork):
raw_conn = self.transport.dial(first_addr)
# Use upgrader to upgrade raw conn to muxed conn
muxed_conn = self.upgrader.upgrade_connection(raw_conn)
muxed_conn = self.upgrader.upgrade_connection(raw_conn, True)
# Store muxed connection in connections
self.connections[peer_id] = muxed_conn

9
transport/upgrader.py

@ -1,3 +1,5 @@
from muxer.mplex.muxed_connection import MuxedConn
class TransportUpgrader(object):
def __init__(self, secOpt, muxerOpt):
@ -14,11 +16,14 @@ class TransportUpgrader(object):
def upgrade_security(self):
pass
def upgrade_connection(self, conn):
def upgrade_connection(self, conn, initiator):
"""
upgrade raw connection to muxed connection
"""
# For PoC, no security
# Default to mplex
pass
# TODO do exchange to determine multiplexer
return MuxedConn(conn, initiator)
Loading…
Cancel
Save