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

#2689 Added TODOs for future reference.

Browse Source
This commit is contained in:
Archer.Bowen
2024-07-30 13:10:23 +01:00
parent 8320ec524b
commit 8a00a2a29d
3 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -11,6 +11,13 @@ from primaite.simulator.network.transmission.network_layer import IPProtocol
from primaite.simulator.network.transmission.transport_layer import Port from primaite.simulator.network.transmission.transport_layer import Port
from primaite.simulator.system.applications.application import Application from primaite.simulator.system.applications.application import Application
# TODO:
# Complete C2 Server and C2 Beacon TODOs
# Create test that leverage all the functionality needed for the different TAPs
# Create a .RST doc
# Potentially? A notebook which demonstrates a custom red agent using the c2 server for various means.
class C2Command(Enum): class C2Command(Enum):
""" """
Enumerations representing the different commands the C2 suite currently supports. Enumerations representing the different commands the C2 suite currently supports.

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

@@ -29,8 +29,11 @@ class C2Beacon(AbstractC2):
keep_alive_frequency: int = 5 keep_alive_frequency: int = 5
"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."
# TODO:
# Uncomment the Import and this Property after terminal PR # Implement the placeholder command methods
# Implement the keep alive frequency.
# Implement a command output method that sends the RequestResponse to the C2 server.
# Uncomment the terminal Import and the terminal property after terminal PR
#@property #@property
#def _host_db_client(self) -> Terminal: #def _host_db_client(self) -> Terminal:

5
src/primaite/simulator/system/applications/red_applications/c2/c2_server.py
View File

@@ -4,7 +4,10 @@ from primaite.simulator.network.protocols.masquerade import C2Payload, Masquerad
class C2Server(AbstractC2): class C2Server(AbstractC2):
# TODO:
# Implement the request manager and agent actions.
# Implement the output handling methods. (These need to interface with the actions)
def _handle_command_output(payload): def _handle_command_output(payload):
"""Abstract Method: Used in C2 server to prase and receive the output of commands sent to the c2 beacon.""" """Abstract Method: Used in C2 server to parse and receive the output of commands sent to the c2 beacon."""
pass pass