Skip to content

Commit

Permalink
Fix for renaming agents in client (#543)
Browse files Browse the repository at this point in the history
* fixed rename description for agents

* fixed client issue when sessionid and name dont match

* fixed issue with renamed agents not being killed
  • Loading branch information
Cx01N authored Jan 31, 2023
1 parent 2cbc69d commit 31c5eb6
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 5 deletions.
5 changes: 3 additions & 2 deletions empire/client/src/menus/AgentMenu.py
Original file line number Diff line number Diff line change
Expand Up @@ -135,7 +135,7 @@ def hide(self) -> None:
@command
def rename(self, agent_name: str, new_agent_name: str) -> None:
"""
Rename selected listener
Rename selected agent
Usage: rename <agent_name> <new_agent_name>
"""
Expand All @@ -150,7 +150,8 @@ def rename(self, agent_name: str, new_agent_name: str) -> None:

@staticmethod
def kill_agent(agent_name: str) -> None:
response = state.kill_agent(agent_name)
session_id = state.agents[agent_name]["session_id"]
response = state.kill_agent(session_id)
if response.status_code == 201:
log.info("Kill command sent to agent " + agent_name)
elif "detail" in response:
Expand Down
7 changes: 4 additions & 3 deletions empire/client/src/menus/InteractMenu.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,7 +140,7 @@ def on_enter(self, **kwargs) -> bool:
return True

def get_prompt(self) -> str:
joined = "/".join([self.display_name, self.selected]).strip("/")
joined = "/".join([self.display_name, self.name]).strip("/")
return f"(Empire: <ansired>{joined}</ansired>) > "

def display_cached_results(self) -> None:
Expand All @@ -162,8 +162,9 @@ def use(self, agent_name: str) -> None:
"""
state.get_agents()
if agent_name in state.agents.keys():
self.selected = agent_name
self.session_id = state.agents[self.selected]["session_id"]
self.name = agent_name
self.selected = state.agents[agent_name]["session_id"]
self.session_id = state.agents[agent_name]["session_id"]
self.agent_options = state.agents[agent_name] # todo rename agent_options
self.agent_language = self.agent_options["language"]

Expand Down

0 comments on commit 31c5eb6

Please sign in to comment.