oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#2689 fixed mismerge

Browse Source
This commit is contained in:
Archer Bowen
2024-08-20 11:15:04 +01:00
parent b8767da61e
commit c9d62d512c
1 changed files with 2 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/primaite/game/agent/actions.py
View File

@@ -1166,19 +1166,10 @@ class NodeSessionsRemoteLogoutAction(AbstractAction):
def __init__(self, manager: "ActionManager", **kwargs) -> None: def __init__(self, manager: "ActionManager", **kwargs) -> None:
super().__init__(manager=manager) super().__init__(manager=manager)
def form_request(self, node_id: int, remote_ip: str, command: RequestFormat) -> RequestFormat: def form_request(self, node_id: str, remote_ip: str) -> RequestFormat:
"""Return the action formatted as a request which can be ingested by the PrimAITE simulation.""" """Return the action formatted as a request which can be ingested by the PrimAITE simulation."""
node_name = self.manager.get_node_name_by_idx(node_id) node_name = self.manager.get_node_name_by_idx(node_id)
return [ return ["network", "node", node_name, "service", "Terminal", "remote_logoff", remote_ip]
"network",
"node",
node_name,
"service",
"Terminal",
"send_remote_command",
remote_ip,
{"command": command},
]
class RansomwareConfigureC2ServerAction(AbstractAction): class RansomwareConfigureC2ServerAction(AbstractAction):