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

#2689 Implemented fixes to _check_connection following PR

Browse Source
This commit is contained in:
Archer Bowen
2024-08-19 13:08:31 +01:00
parent 87332873d2
commit f595f44ce9
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -463,26 +463,26 @@ class AbstractC2(Application, identifier="AbstractC2"):
:return: A tuple containing a boolean True/False and a corresponding Request Response :return: A tuple containing a boolean True/False and a corresponding Request Response
:rtype: tuple[bool, RequestResponse] :rtype: tuple[bool, RequestResponse]
""" """
if self._can_perform_network_action == False: if not self._can_perform_network_action:
self.sys_log.warning(f"{self.name}: Unable to make leverage networking resources. Rejecting Command.") self.sys_log.warning(f"{self.name}: Unable to make leverage networking resources. Rejecting Command.")
return [ return (
False, False,
RequestResponse( RequestResponse(
status="failure", data={"Reason": "Unable to access networking resources. Unable to send command."} status="failure", data={"Reason": "Unable to access networking resources. Unable to send command."}
), ),
] )
if self.c2_remote_connection is False: if self.c2_remote_connection is False:
self.sys_log.warning(f"{self.name}: C2 Application has yet to establish connection. Rejecting command.") self.sys_log.warning(f"{self.name}: C2 Application has yet to establish connection. Rejecting command.")
return [ return (
False, False,
RequestResponse( RequestResponse(
status="failure", status="failure",
data={"Reason": "C2 Application has yet to establish connection. Unable to send command."}, data={"Reason": "C2 Application has yet to establish connection. Unable to send command."},
), ),
] )
else: else:
return [ return (
True, True,
RequestResponse(status="success", data={"Reason": "C2 Application is able to send connections."}), RequestResponse(status="success", data={"Reason": "C2 Application is able to send connections."}),
] )