#2888: Fixed C2Beacon test failures.

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

View File

@@ -56,12 +56,6 @@ class C2Beacon(AbstractC2, identifier="C2Beacon"):
def __init__(self, **kwargs):
kwargs["name"] = "C2Beacon"
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
def _host_terminal(self) -> Optional[Terminal]:

View File

@@ -495,6 +495,13 @@ def test_c2_suite_yaml():
computer_b: Computer = yaml_network.get_node_by_hostname("node_b")
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