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

#2888: Fixed C2Beacon test failures.

Browse Source
This commit is contained in:
Nick Todd
2025-01-07 14:21:07 +00:00
parent d0c357355c
commit 0203a8699a
2 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -56,12 +56,6 @@ class C2Beacon(AbstractC2, identifier="C2Beacon"):
def __init__(self, **kwargs): def __init__(self, **kwargs):
kwargs["name"] = "C2Beacon" kwargs["name"] = "C2Beacon"
super().__init__(**kwargs) super().__init__(**kwargs)
self.configure(
c2_server_ip_address=self.config.c2_server_ip_address,
keep_alive_frequency=self.config.keep_alive_frequency,
masquerade_port=self.config.masquerade_port,
masquerade_protocol=self.config.masquerade_protocol,
)
@property @property
def _host_terminal(self) -> Optional[Terminal]: def _host_terminal(self) -> Optional[Terminal]:

7
tests/integration_tests/system/red_applications/test_c2_suite_integration.py
View File

@@ -495,6 +495,13 @@ def test_c2_suite_yaml():
computer_b: Computer = yaml_network.get_node_by_hostname("node_b") computer_b: Computer = yaml_network.get_node_by_hostname("node_b")
c2_beacon: C2Beacon = computer_b.software_manager.software.get("C2Beacon") c2_beacon: C2Beacon = computer_b.software_manager.software.get("C2Beacon")
c2_beacon.configure(
c2_server_ip_address=c2_beacon.config.c2_server_ip_address,
keep_alive_frequency=c2_beacon.config.keep_alive_frequency,
masquerade_port=c2_beacon.config.masquerade_port,
masquerade_protocol=c2_beacon.config.masquerade_protocol,
)
assert c2_server.operating_state == ApplicationOperatingState.RUNNING assert c2_server.operating_state == ApplicationOperatingState.RUNNING