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

#2689 Adding slight changes to c2_Beacon & terminal that appeared when merging from dev.

Browse Source
This commit is contained in:
Archer Bowen
2024-08-09 12:14:57 +01:00
parent 507abe46c2
commit 4241118d26
3 changed files with 9 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/primaite/notebooks/Command-&-Control-E2E-Demonstration.ipynb
View File

@@ -114,8 +114,8 @@
" node_id: 1\n", " node_id: 1\n",
" ip_address:\n", " ip_address:\n",
" account:\n", " account:\n",
" username: test123\n", " username: admin\n",
" password: pass123\n", " password: admin\n",
" commands:\n", " commands:\n",
" - \n", " - \n",
" - software_manager\n", " - software_manager\n",

14
src/primaite/simulator/system/applications/red_applications/c2/c2_beacon.py
View File

@@ -49,10 +49,10 @@ class C2Beacon(AbstractC2, identifier="C2Beacon"):
"The frequency at which ``Keep Alive`` packets are sent to the C2 Server from the C2 Beacon." "The frequency at which ``Keep Alive`` packets are sent to the C2 Server from the C2 Beacon."
local_terminal_session: LocalTerminalConnection = None local_terminal_session: LocalTerminalConnection = None
"""#TODO""" "The currently in use local terminal session."
remote_terminal_session: RemoteTerminalConnection = None remote_terminal_session: RemoteTerminalConnection = None
"""#TODO""" "The currently in use remote terminal session"
@property @property
def _host_terminal(self) -> Optional[Terminal]: def _host_terminal(self) -> Optional[Terminal]:
@@ -199,7 +199,7 @@ class C2Beacon(AbstractC2, identifier="C2Beacon"):
---------------------|------------------------ ---------------------|------------------------
RANSOMWARE_CONFIGURE | self._command_ransomware_config() RANSOMWARE_CONFIGURE | self._command_ransomware_config()
RANSOMWARE_LAUNCH | self._command_ransomware_launch() RANSOMWARE_LAUNCH | self._command_ransomware_launch()
Terminal | self._command_terminal() TERMINAL | self._command_terminal()
Please see each method individually for further information regarding Please see each method individually for further information regarding
the implementation of these commands. the implementation of these commands.
@@ -340,14 +340,13 @@ class C2Beacon(AbstractC2, identifier="C2Beacon"):
data={"Reason": "Host does not seem to have terminal installed. Unable to resolve command."}, data={"Reason": "Host does not seem to have terminal installed. Unable to resolve command."},
) )
# TODO: Placeholder until further details on handling user sessions.
given_commands = payload.payload.get("commands") given_commands = payload.payload.get("commands")
given_username = payload.payload.get("username") given_username = payload.payload.get("username")
given_password = payload.payload.get("password") given_password = payload.payload.get("password")
remote_ip = payload.payload.get("ip_address") remote_ip = payload.payload.get("ip_address")
# Creating a remote terminal session if given an IP Address, otherwise using a local terminal session. # Creating a remote terminal session if given an IP Address, otherwise using a local terminal session.
if payload.payload.get("ip_address") is None: if remote_ip is None:
terminal_session = self.get_terminal_session(username=given_username, password=given_password) terminal_session = self.get_terminal_session(username=given_username, password=given_password)
else: else:
terminal_session = self.get_remote_terminal_session( terminal_session = self.get_remote_terminal_session(
@@ -355,9 +354,8 @@ class C2Beacon(AbstractC2, identifier="C2Beacon"):
) )
if terminal_session is None: if terminal_session is None:
RequestResponse( return RequestResponse(
status="failure", status="failure", data={"reason": "Terminal Login failed. Cannot create a terminal session."}
data={"Reason": "Host cannot is unable to connect to terminal. Unable to resolve command."},
) )
for index, given_command in enumerate(given_commands): for index, given_command in enumerate(given_commands):

2
src/primaite/simulator/system/services/terminal/terminal.py
View File

@@ -92,7 +92,7 @@ class LocalTerminalConnection(TerminalClientConnection):
if not self.is_active: if not self.is_active:
self.parent_terminal.sys_log.warning("Connection inactive, cannot execute") self.parent_terminal.sys_log.warning("Connection inactive, cannot execute")
return None return None
return self.parent_terminal.execute(command, connection_id=self.connection_uuid) return self.parent_terminal.execute(command)
class RemoteTerminalConnection(TerminalClientConnection): class RemoteTerminalConnection(TerminalClientConnection):