45 lines
1.2 KiB
Python
45 lines
1.2 KiB
Python
from .muxed_stream import MuxedStream
|
|
from .muxed_connection import MuxedConn
|
|
|
|
class Multiplex(object):
|
|
"""
|
|
reference: https://github.com/whyrusleeping/go-smux-multiplex/blob/master/multiplex.go
|
|
"""
|
|
def __init__(self, conn, initiator):
|
|
"""
|
|
:param conn: an instance of raw connection
|
|
:param initiator: boolean to prevent multiplex with self
|
|
"""
|
|
self.muxed_conn = MuxedConn(conn, initiator)
|
|
|
|
def close(self):
|
|
"""
|
|
close the stream muxer and underlying raw connection
|
|
"""
|
|
return self.muxed_conn.close()
|
|
|
|
def is_closed(self):
|
|
"""
|
|
check connection is fully closed
|
|
:return: true if successful
|
|
"""
|
|
return self.muxed_conn.is_closed()
|
|
|
|
def open_stream(self, protocol_id, stream_name):
|
|
"""
|
|
creates a new muxed_stream
|
|
:return: a new stream
|
|
"""
|
|
return self.muxed_conn.open_stream(protocol_id, stream_name)
|
|
|
|
def accept_stream(self, _muxed_stream):
|
|
"""
|
|
accepts a muxed stream opened by the other end
|
|
:param _muxed_stream: stream to be accepted
|
|
:return: the accepted stream
|
|
"""
|
|
pass
|
|
|
|
# def new_conn(raw_conn, is_server):
|
|
# pass
|