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

Apply suggestions from code review

Browse Source
This commit is contained in:
Archer Bowen
2024-08-13 15:43:21 +00:00
parent ead302c95d
commit d6e2994d6b
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/primaite/game/game.py
View File

@@ -461,7 +461,7 @@ class PrimaiteGame:
opt = application_cfg["options"] opt = application_cfg["options"]
new_application.configure( new_application.configure(
c2_server_ip_address=IPv4Address(opt.get("c2_server_ip_address")), c2_server_ip_address=IPv4Address(opt.get("c2_server_ip_address")),
keep_alive_frequency=(opt.get("keep_alive_frequency")) keep_alive_frequency=(opt.get("keep_alive_frequency", 5))
if opt.get("keep_alive_frequency") if opt.get("keep_alive_frequency")
else 5, else 5,
masquerade_protocol=IPProtocol[(opt.get("masquerade_protocol"))] masquerade_protocol=IPProtocol[(opt.get("masquerade_protocol"))]

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

@@ -187,7 +187,7 @@ class AbstractC2(Application, identifier="AbstractC2"):
# Used in C2 server to parse and receive the output of commands sent to the c2 beacon. # Used in C2 server to parse and receive the output of commands sent to the c2 beacon.
@abstractmethod @abstractmethod
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
# Abstract method # Abstract method

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

@@ -14,7 +14,7 @@ class C2Server(AbstractC2, identifier="C2Server"):
""" """
C2 Server Application. C2 Server Application.
Represents a vendor generic C2 Server is used in conjunction with the C2 beacon Represents a vendor generic C2 Server used in conjunction with the C2 beacon
to simulate malicious communications and infrastructure within primAITE. to simulate malicious communications and infrastructure within primAITE.
The C2 Server must be installed and be in a running state before it's able to receive The C2 Server must be installed and be in a running state before it's able to receive