#2681 Updated to include yaml file tests + include listening on multiports.

tests/assets/configs/basic_c2_setup.yaml

@@ -0,0 +1,76 @@
+#  Basic Switched network
+#
+#  --------------      --------------      --------------
+#  |  node_a  |------|  switch_1  |------|  node_b  |
+#  --------------      --------------      --------------
+#
+io_settings:
+  save_step_metadata: false
+  save_pcap_logs: true
+  save_sys_logs: true
+  sys_log_level: WARNING
+  agent_log_level: INFO
+  save_agent_logs: true
+  write_agent_log_to_terminal: True
+
+
+game:
+  max_episode_length: 256
+  ports:
+  - ARP
+  - DNS
+  - HTTP
+  - POSTGRES_SERVER
+  protocols:
+  - ICMP
+  - TCP
+  - UDP
+
+simulation:
+  network:
+    nodes:
+
+    - type: switch
+      hostname: switch_1
+      num_ports: 8
+
+    - hostname: node_a
+      type: computer
+      ip_address: 192.168.10.21
+      subnet_mask: 255.255.255.0
+      default_gateway: 192.168.10.1
+      applications:
+      - type: C2Server
+        options:
+          listen_on_ports:
+            - 80
+            - 53
+            - 21
+    - hostname: node_b
+      type: computer
+      ip_address: 192.168.10.22
+      subnet_mask: 255.255.255.0
+      default_gateway: 192.168.10.1
+      applications:
+      - type: C2Beacon
+        options:
+          c2_server_ip_address: 192.168.10.21
+          keep_alive_frequency: 5
+          masquerade_protocol: TCP
+          masquerade_port: HTTP
+          listen_on_ports:
+            - 80
+            - 53
+            - 21
+
+    links:
+    - endpoint_a_hostname: switch_1
+      endpoint_a_port: 1
+      endpoint_b_hostname: node_a
+      endpoint_b_port: 1
+      bandwidth: 200
+    - endpoint_a_hostname: switch_1
+      endpoint_a_port: 2
+      endpoint_b_hostname: node_b
+      endpoint_b_port: 1
+      bandwidth: 200

tests/integration_tests/system/red_applications/test_c2_suite_integration.py

@@ -3,6 +3,7 @@ from ipaddress import IPv4Address
 from typing import Tuple
 
 import pytest
+import yaml
 
 from primaite.game.agent.interface import ProxyAgent
 from primaite.game.game import PrimaiteGame
@@ -22,6 +23,7 @@ from primaite.simulator.system.applications.red_applications.ransomware_script i
 from primaite.simulator.system.services.database.database_service import DatabaseService
 from primaite.simulator.system.services.dns.dns_server import DNSServer
 from primaite.simulator.system.services.web_server.web_server import WebServer
+from tests import TEST_ASSETS_ROOT
 
 
 @pytest.fixture(scope="function")
@@ -463,3 +465,35 @@ def test_c2_suite_acl_bypass(basic_network):
     assert c2_packets_blocked == blocking_acl.match_count
     assert c2_server.c2_connection_active is True
     assert c2_beacon.c2_connection_active is True
+
+
+def test_c2_suite_yaml():
+    """Tests that the C2 Suite is can be configured correctly via the Yaml."""
+    with open(TEST_ASSETS_ROOT / "configs" / "basic_c2_setup.yaml") as f:
+        cfg = yaml.safe_load(f)
+    game = PrimaiteGame.from_config(cfg)
+
+    yaml_network = game.simulation.network
+    computer_a: Computer = yaml_network.get_node_by_hostname("node_a")
+    c2_server: C2Server = computer_a.software_manager.software.get("C2Server")
+
+    computer_b: Computer = yaml_network.get_node_by_hostname("node_b")
+    c2_beacon: C2Beacon = computer_b.software_manager.software.get("C2Beacon")
+
+    assert c2_server.operating_state == ApplicationOperatingState.RUNNING
+
+    assert c2_beacon.c2_remote_connection == IPv4Address("192.168.10.21")
+
+    c2_beacon.establish()
+
+    # Asserting that the c2 beacon has established a c2 connection
+    assert c2_beacon.c2_connection_active is True
+    # Asserting that the c2 server has established a c2 connection.
+    assert c2_server.c2_connection_active is True
+    assert c2_server.c2_remote_connection == IPv4Address("192.168.10.22")
+
+    for i in range(50):
+        yaml_network.apply_timestep(i)
+
+    assert c2_beacon.c2_connection_active is True
+    assert c2_server.c2_connection_active is True