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

#2689 Improving comments in abstract c2

Browse Source
This commit is contained in:
Archer.Bowen
2024-07-30 17:18:28 +01:00
parent 4c03a20154
commit e4358b02bc
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -161,8 +161,10 @@ class AbstractC2(Application):
Returns False if a keep alive was unable to be sent. Returns False if a keep alive was unable to be sent.
Returns True if a keep alive was successfully sent or already has been sent this timestep. Returns True if a keep alive was successfully sent or already has been sent this timestep.
""" """
self.sys_log.info(f"{self.name}: Keep Alive Received from {self.c2_remote_connection}")
# Using this guard clause to prevent packet storms and recognise that we've achieved a connection. # Using this guard clause to prevent packet storms and recognise that we've achieved a connection.
if self.keep_alive_sent: if self.keep_alive_sent:
self.sys_log.info(f"{self.name}: Connection successfully established with {self.c2_remote_connection}")
self.c2_connection_active = True # Sets the connection to active self.c2_connection_active = True # Sets the connection to active
self.keep_alive_inactivity = 0 # Sets the keep alive inactivity to zero self.keep_alive_inactivity = 0 # Sets the keep alive inactivity to zero
@@ -174,14 +176,11 @@ class AbstractC2(Application):
# If this method returns true then we have sent successfully sent a keep alive. # If this method returns true then we have sent successfully sent a keep alive.
if self._send_keep_alive(self): if self._send_keep_alive(self):
# debugging/info logging that we successfully sent a keep alive self.keep_alive_sent = True # Setting the guard clause to true (prevents packet storms.)
# Now when the returning keep_alive comes back we won't send another keep alive
self.keep_alive_sent = True
return True return True
# Return false if we're unable to send handle the keep alive correctly.
else: else:
# debugging/info logging that we unsuccessfully sent a keep alive.
return False return False
def receive(self, payload: MasqueradePacket, session_id: Optional[str] = None) -> bool: def receive(self, payload: MasqueradePacket, session_id: Optional[str] = None) -> bool: