From c9d62d512c174ef4edca44832ef7b0c60bd2e8d3 Mon Sep 17 00:00:00 2001 From: Archer Bowen Date: Tue, 20 Aug 2024 11:15:04 +0100 Subject: [PATCH] #2689 fixed mismerge --- src/primaite/game/agent/actions.py | 13 ++----------- 1 file changed, 2 insertions(+), 11 deletions(-) diff --git a/src/primaite/game/agent/actions.py b/src/primaite/game/agent/actions.py index 42ba25b4..2e6189c0 100644 --- a/src/primaite/game/agent/actions.py +++ b/src/primaite/game/agent/actions.py @@ -1166,19 +1166,10 @@ class NodeSessionsRemoteLogoutAction(AbstractAction): def __init__(self, manager: "ActionManager", **kwargs) -> None: 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.""" node_name = self.manager.get_node_name_by_idx(node_id) - return [ - "network", - "node", - node_name, - "service", - "Terminal", - "send_remote_command", - remote_ip, - {"command": command}, - ] + return ["network", "node", node_name, "service", "Terminal", "remote_logoff", remote_ip] class RansomwareConfigureC2ServerAction(AbstractAction):