Merge pull request #277 from BC-SECURITY/sponsors-dev

4.3.2

VERSION

@@ -1 +1 @@
-4.3.2
+4.3.3

changelog

@@ -1,3 +1,8 @@
+1/24/2022
+------------
+- Version 4.3.3 Master Release
+        - Added a hook for when an agent is fully checked in (stage2) (@Vinnybod)
+
 1/14/2022
 ------------
 - Version 4.3.2 Master Release

empire/server/common/agents.py

@@ -1225,6 +1225,8 @@ def handle_agent_staging(self, sessionID, language, meta, additional, encData, s
             if self.mainMenu.socketio:
                 self.mainMenu.socketio.emit('agents/stage2', agent, broadcast=True)
 
+            hooks.run_hooks(hooks.AFTER_AGENT_STAGE2_HOOK, self.get_agent_from_name_or_session_id(sessionID))
+
             # save the initial sysinfo information in the agent log
             output = messages.display_agent(agent, returnAsString=True)
             output += "[+] Agent %s now active:\n" % (sessionID)

empire/server/common/empire.py

@@ -44,7 +44,7 @@
 from empire.server.database import models
 from sqlalchemy import or_, func, and_
 
-VERSION = "4.3.2 BC Security Fork"
+VERSION = "4.3.3 BC Security Fork"
 
 
 class MainMenu(cmd.Cmd):

empire/server/common/hooks.py

@@ -28,6 +28,11 @@ class Hooks(object):
     # It has one argument (agent: models.Agent)
     AFTER_AGENT_CHECKIN_HOOK = 'after_agent_checkin_hook'
 
+    # This event is triggered after the agent has completed the stage2 of the checkin process,
+    # and the sysinfo has been written to the database.
+    # It has one argument (agent: models.Agent)
+    AFTER_AGENT_STAGE2_HOOK = 'after_agent_stage2_hook'
+
     def __init__(self):
         self.hooks: Dict[str, Dict[str, Callable]] = {}
         self.filters: Dict[str, Dict[str, Callable]] = {}

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "empire-bc-security-fork"
-version = "4.3.2"
+version = "4.3.3"
 description = ""
 authors = ["BC Security <[email protected]>"]
 readme = "README.md"