Updated library import for new Empire 4.0 server directory structure (#4

) * Updated library import for new Empire 4.0 server directory structure * Added check for new Empire directory structure to load appropriate modules for new Empire version with backward compat for older versions
BC-SECURITY · Apr 2, 2021 · 526ccbe · 526ccbe
1 parent 0e6ea6f
commit 526ccbe
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/plugins/socksproxyserver.py b/plugins/socksproxyserver.py
@@ -1,7 +1,12 @@
 from __future__ import print_function
+import importlib
 
-from lib.common.plugins import Plugin
-import lib.common.helpers as helpers
+if importlib.util.find_spec("empire") is not None:
+    from empire.server.common.plugins import Plugin
+    import empire.server.common.helpers as helpers    
+else:
+    from lib.common.plugins import Plugin
+    import lib.common.helpers as helpers    
 
 import socket
 import _thread