From fbb4eba6b74cbd766fc4c9ecb1bb19cc866c896e Mon Sep 17 00:00:00 2001
From: Marek Wolan <marek.wolan@methods.co.uk>
Date: Wed, 27 Mar 2024 03:27:44 +0000
Subject: [PATCH] Draft new observation space config

---
 .../_package_data/data_manipulation.yaml      | 49 +++++++++++++++++--
 1 file changed, 45 insertions(+), 4 deletions(-)

diff --git a/src/primaite/config/_package_data/data_manipulation.yaml b/src/primaite/config/_package_data/data_manipulation.yaml
index 12f60b63..06028ee1 100644
--- a/src/primaite/config/_package_data/data_manipulation.yaml
+++ b/src/primaite/config/_package_data/data_manipulation.yaml
@@ -176,13 +176,54 @@ agents:
     team: BLUE
     type: ProxyAgent
 
+    observation_space:
+      - type: NODES
+        label: NODES # What is the dictionary key called
+        options:
+          hosts:
+            - hostname: domain_controller
+            - hostname: web_server
+            - hostname: database_server
+            - hostname: backup_server
+            - hostname: security_suite
+            - hostname: client_1
+            - hostname: client_2
+          routers:
+            - hostname: router_1
+          firewalls: {}
+
+          num_host_services: 1
+          num_host_applications: 0
+          num_host_folders: 1
+          num_host_files: 1
+          num_host_network_interfaces: 2
+          num_router_ports: 4
+          num_acl_rules: 10
+          num_firewall_ports: 4
+          firewalls_internal_inbound_acl: true
+          firewalls_internal_outbound_acl: true
+          firewalls_dmz_inbound_acl: true
+          firewalls_dmz_outbound_acl: true
+          firewalls_external_inbound_acl: true
+          firewalls_external_outbound_acl: true
+      - type: LINKS
+        label: "LINKS"
+        options:
+          links:
+            - link_ref: router_1___switch_1
+            - link_ref: router_1___switch_2
+            - link_ref: switch_1___domain_controller
+            - link_ref: switch_1___web_server
+            - link_ref: switch_1___database_server
+            - link_ref: switch_1___backup_server
+            - link_ref: switch_1___security_suite
+            - link_ref: switch_2___client_1
+            - link_ref: switch_2___client_2
+            - link_ref: switch_2___security_suite
+
     observation_space:
       type: UC2BlueObservation
       options:
-        num_services_per_node: 1
-        num_folders_per_node: 1
-        num_files_per_folder: 1
-        num_nics_per_node: 2
         nodes:
         - node_hostname: domain_controller
           services: