From 7b1584ccb7a33fc7854d223805a81c786b2935a5 Mon Sep 17 00:00:00 2001 From: Archer Bowen Date: Mon, 19 Aug 2024 15:24:24 +0100 Subject: [PATCH] #2689 Updated following PR --- .../applications/red_applications/c2/abstract_c2.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/src/primaite/simulator/system/applications/red_applications/c2/abstract_c2.py b/src/primaite/simulator/system/applications/red_applications/c2/abstract_c2.py index 354976b7..7316dd63 100644 --- a/src/primaite/simulator/system/applications/red_applications/c2/abstract_c2.py +++ b/src/primaite/simulator/system/applications/red_applications/c2/abstract_c2.py @@ -472,7 +472,7 @@ class AbstractC2(Application, identifier="AbstractC2"): ), ) - if self.c2_remote_connection is False: + if self.c2_remote_connection is None: self.sys_log.warning(f"{self.name}: C2 Application has yet to establish connection. Rejecting command.") return ( False, @@ -481,8 +481,7 @@ class AbstractC2(Application, identifier="AbstractC2"): data={"Reason": "C2 Application has yet to establish connection. Unable to send command."}, ), ) - else: - return ( - True, - RequestResponse(status="success", data={"Reason": "C2 Application is able to send connections."}), - ) + return ( + True, + RequestResponse(status="success", data={"Reason": "C2 Application is able to send connections."}), + )