fix: Removed cyclical imports.

GoSecure · May 21, 2020 · 255bb33 · Res260 · May 21, 2020 · alxbl
1 parent 8cb9ad9
commit 255bb33
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/pyrdp/layer/tcp.py b/pyrdp/layer/tcp.py
@@ -16,7 +16,6 @@
 from pyrdp.logging import LOGGER_NAMES, getSSLLogger
 from pyrdp.parser.tcp import TCPParser
 from pyrdp.pdu import PDU
-from pyrdp.mitm import MITMConfig
 
 
 TLS_RECORD = 0x16
@@ -46,7 +45,7 @@ class TwistedTCPLayer(IntermediateLayer, Protocol):
     TCP observers are notified when a connection is made.
     """
 
-    def __init__(self, config: MITMConfig):
+    def __init__(self, config):
         self.log = logging.getLogger(LOGGER_NAMES.PYRDP)
         super().__init__(TCPParser())
         self.connectedEvent = asyncio.Event()

diff --git a/pyrdp/mitm/layerset.py b/pyrdp/mitm/layerset.py
@@ -5,17 +5,17 @@
 #
 
 from pyrdp.enum import SegmentationPDUType
-from pyrdp.mitm import MITMConfig
 from pyrdp.layer import FastPathLayer, LayerChainItem, MCSLayer, SecurityLayer, SegmentationLayer, SlowPathLayer, \
     TPKTLayer, TwistedTCPLayer, X224Layer
+# from pyrdp.mitm.config import MITMConfig
 
 
 class RDPLayerSet:
     """
     Class that handles initialization of regular (non-virtual channel) RDP layers.
     """
 
-    def __init__(self, config: MITMConfig):
+    def __init__(self, config):
         self.tcp = TwistedTCPLayer(config)
         self.segmentation = SegmentationLayer()
         self.tpkt = TPKTLayer()