Fixed BC-SECURITY#764 delay jitter python agent but

janit0rjoe · Dec 17, 2024 · 6db5293 · 6db5293
1 parent 5aeb633
commit 6db5293
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/empire/server/core/agent_task_service.py b/empire/server/core/agent_task_service.py
@@ -270,7 +270,7 @@ def create_task_update_sleep(
                 db,
                 agent,
                 "TASK_CMD_WAIT",
-                f"global delay; global jitter; delay={delay}; jitter={jitter}; print('delay/jitter set to {delay}/{jitter}')",
+                f"global agent; agent.delay={delay}; agent.jitter={jitter}; print('delay/jitter set to {delay}/{jitter}')",
                 user_id=user_id,
             )
         if agent.language == "csharp":

diff --git a/empire/server/data/agent/agent.py b/empire/server/data/agent/agent.py
@@ -621,6 +621,7 @@ def dynamic_code_execute_wait_nosave(self, data, result_id):
         Task 100
         """
         try:
+            globals().update({'agent':self})
             buffer = StringIO()
             sys.stdout = buffer
             code_obj = compile(data, "<string>", "exec")