Apply suggestions from code review

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

View File

@@ -461,7 +461,7 @@ class PrimaiteGame:
opt = application_cfg["options"]
new_application.configure(
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")
else 5,
masquerade_protocol=IPProtocol[(opt.get("masquerade_protocol"))]

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.
@abstractmethod
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
# Abstract method

View File

@@ -14,7 +14,7 @@ class C2Server(AbstractC2, identifier="C2Server"):
"""
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.
The C2 Server must be installed and be in a running state before it's able to receive