fake server: only import FakeServer if it is actually needed

GoSecure · Dec 14, 2022 · ecd5d15 · ecd5d15
1 parent 41e9b31
commit ecd5d15
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyrdp/mitm/state.py b/pyrdp/mitm/state.py
@@ -14,7 +14,6 @@
 from pyrdp.pdu import ClientChannelDefinition
 from pyrdp.security import RC4CrypterProxy, SecuritySettings
 from pyrdp.mitm import MITMConfig
-from pyrdp.mitm.FakeServer import FakeServer
 
 
 class RDPMITMState:
@@ -135,6 +134,7 @@ def useRedirectionHost(self):
         self.effectiveTargetPort = self.config.redirectionPort
 
     def useFakeServer(self):
+        from pyrdp.mitm.FakeServer import FakeServer
         self.fakeServer = FakeServer()
         self.effectiveTargetHost = "127.0.0.1"
         self.effectiveTargetPort = self.fakeServer.port