Update server.py

BC-SECURITY · Nov 12, 2024 · 0308dc9 · 0308dc9
1 parent d2e7f90
commit 0308dc9
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/empire/server/server.py b/empire/server/server.py
@@ -2,6 +2,7 @@
 import logging
 import os
 import pathlib
+import platform
 import pwd
 import shutil
 import signal
@@ -70,7 +71,15 @@ def setup_logging(args):
 INVOKE_OBFS_SRC_DIR_BASE = os.path.join(
     os.path.dirname(__file__), "data/Invoke-Obfuscation"
 )
-INVOKE_OBFS_DST_DIR_BASE = "/usr/local/share/powershell/Modules/Invoke-Obfuscation"
+
+user_home = Path.home()
+if platform.system() == "Windows":
+    # Windows-specific user module path
+    INVOKE_OBFS_DST_DIR_BASE = user_home / "Documents" / "PowerShell" / "Modules" / "Invoke-Obfuscation"
+else:
+    # Linux/macOS-specific user module path
+    INVOKE_OBFS_DST_DIR_BASE = user_home / ".local" / "share" / "powershell" / "Modules" / "Invoke-Obfuscation"
+
 
 
 def reset():