#2258: testing individual application install

src/primaite/game/game.py

@@ -16,6 +16,7 @@ from primaite.simulator.network.hardware.nodes.computer import Computer
 from primaite.simulator.network.hardware.nodes.router import Router
 from primaite.simulator.network.hardware.nodes.server import Server
 from primaite.simulator.network.hardware.nodes.switch import Switch
+from primaite.simulator.network.transmission.transport_layer import Port
 from primaite.simulator.sim_container import Simulation
 from primaite.simulator.system.applications.database_client import DatabaseClient
 from primaite.simulator.system.applications.red_applications.data_manipulation_bot import DataManipulationBot
@@ -310,7 +311,15 @@ class PrimaiteGame:
                     elif application_type == "DoSBot":
                         if "options" in application_cfg:
                             opt = application_cfg["options"]
-                            new_application.target_ip_address = opt.get("target_ip_address")
+                            new_application.configure(
+                                target_ip_address=IPv4Address(opt.get("target_ip_address")),
+                                target_port=Port(opt.get("target_port", Port.POSTGRES_SERVER.value)),
+                                payload=opt.get("payload"),
+                                repeat=bool(opt.get("repeat")),
+                                port_scan_p_of_success=float(opt.get("port_scan_p_of_success", "0.1")),
+                                dos_intensity=float(opt.get("dos_intensity", "1.0")),
+                                max_sessions=int(opt.get("max_sessions", "1000")),
+                            )
             if "nics" in node_cfg:
                 for nic_num, nic_cfg in node_cfg["nics"].items():
                     new_node.connect_nic(NIC(ip_address=nic_cfg["ip_address"], subnet_mask=nic_cfg["subnet_mask"]))

tests/assets/configs/basic_switched_network.yaml

@@ -87,11 +87,13 @@ simulation:
           port_scan_p_of_success: 0.8
           data_manipulation_p_of_success: 0.8
           payload: "DELETE"
-          server_ip: 192.168.1.14
+          server_ip: 192.168.1.21
       - ref: dos_bot
         type: DoSBot
         options:
           target_ip_address: 192.168.10.21
+          payload: SPOOF DATA
+          port_scan_p_of_success: 0.8
       services:
       - ref: client_1_dns_server
         type: DNSServer

tests/integration_tests/game_configuration.py

@@ -1,3 +1,4 @@
+from ipaddress import IPv4Address
 from pathlib import Path
 from typing import Union
 
@@ -9,6 +10,8 @@ from primaite.game.agent.interface import ProxyAgent, RandomAgent
 from primaite.game.game import APPLICATION_TYPES_MAPPING, PrimaiteGame, SERVICE_TYPES_MAPPING
 from primaite.simulator.network.container import Network
 from primaite.simulator.network.hardware.nodes.computer import Computer
+from primaite.simulator.system.applications.red_applications.data_manipulation_bot import DataManipulationBot
+from primaite.simulator.system.applications.red_applications.dos_bot import DoSBot
 from primaite.simulator.system.applications.web_browser import WebBrowser
 from primaite.simulator.system.services.dns.dns_client import DNSClient
 from primaite.simulator.system.services.ftp.ftp_client import FTPClient
@@ -76,3 +79,41 @@ def test_node_software_install():
     # check that services have been installed on client 1
     for service in SERVICE_TYPES_MAPPING:
         assert client_1.software_manager.software.get(service) is not None
+
+
+def test_web_browser_install():
+    """Test that the web browser can be configured via config."""
+    game = load_config(BASIC_CONFIG)
+    client_1: Computer = game.simulation.network.get_node_by_hostname("client_1")
+
+    web_browser: WebBrowser = client_1.software_manager.software.get("WebBrowser")
+
+    assert web_browser.target_url == "http://arcd.com/users/"
+
+
+def test_data_manipulation_bot_install():
+    """Test that the data manipulation bot can be configured via config."""
+    game = load_config(BASIC_CONFIG)
+    client_1: Computer = game.simulation.network.get_node_by_hostname("client_1")
+
+    data_manipulation_bot: DataManipulationBot = client_1.software_manager.software.get("DataManipulationBot")
+
+    assert data_manipulation_bot.server_ip_address == IPv4Address("192.168.1.21")
+    assert data_manipulation_bot.payload == "DELETE"
+    assert data_manipulation_bot.data_manipulation_p_of_success == 0.8
+    assert data_manipulation_bot.port_scan_p_of_success == 0.8
+
+
+def test_dos_bot_install():
+    """Test that the denial of service bot can be configured via config."""
+    game = load_config(BASIC_CONFIG)
+    client_1: Computer = game.simulation.network.get_node_by_hostname("client_1")
+
+    dos_bot: DoSBot = client_1.software_manager.software.get("DoSBot")
+
+    assert dos_bot.target_ip_address == IPv4Address("192.168.10.21")
+    assert dos_bot.payload == "SPOOF DATA"
+    assert dos_bot.port_scan_p_of_success == 0.8
+    assert dos_bot.dos_intensity == 1.0  # default
+    assert dos_bot.max_sessions == 1000  # default
+    assert dos_bot.repeat is False  # default