Initial commit of v1.0.0. Updated the .gitignore for the standard Python gitignore. Added Azure DevOps release pipeline for proper artifact release from the start.

PRIMAITE/common/__init__.py

@@ -0,0 +1,2 @@
+
+# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence.

PRIMAITE/common/config_values_main.py

@@ -0,0 +1,59 @@
+# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence.
+"""
+The config class
+"""
+
+class config_values_main(object):
+    """
+    Class to hold main config values
+    """
+
+    def __init__(self):
+        """
+        Init
+        """
+
+        # Generic
+        self.agent_identifier = ""                  # the agent in use
+        self.num_episodes = 0                       # number of episodes to train over
+        self.num_steps = 0                          # number of steps in an episode
+        self.time_delay = 0                         # delay between steps (ms) - applies to generic agents only
+        self.config_filename_use_case = ""          # the filename for the Use Case config file
+
+        # Environment
+        self.observation_space_high_value = 0       # The high value for the observation space
+
+        # Reward values
+        # Generic
+        self.all_ok = 0 
+        # Node Operating State
+        self.off_should_be_on = 0
+        self.off_should_be_resetting = 0
+        self.on_should_be_off = 0
+        self.on_should_be_resetting = 0
+        self.resetting_should_be_on = 0
+        self.resetting_should_be_off = 0 
+        # Node O/S or Service State
+        self.good_should_be_patching = 0
+        self.good_should_be_compromised = 0
+        self.good_should_be_overwhelmed = 0
+        self.patching_should_be_good = 0
+        self.patching_should_be_compromised = 0
+        self.patching_should_be_overwhelmed = 0
+        self.compromised_should_be_good = 0
+        self.compromised_should_be_patching = 0
+        self.compromised_should_be_overwhelmed = 0
+        self.compromised = 0     
+        self.overwhelmed_should_be_good = 0
+        self.overwhelmed_should_be_patching = 0
+        self.overwhelmed_should_be_compromised = 0
+        self.overwhelmed = 0
+        # IER status
+        self.red_ier_running = 0
+        self.green_ier_blocked = 0
+
+        # Patching / Reset
+        self.os_patching_duration = 0                         # The time taken to patch the OS
+        self.node_reset_duration = 0                          # The time taken to reset a node (hardware)
+        self.service_patching_duration = 0                    # The time taken to patch a service
+

PRIMAITE/common/enums.py

@@ -0,0 +1,84 @@
+# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence.
+"""
+Enumerations for APE
+"""
+
+from enum import Enum
+
+class TYPE(Enum):
+    """
+    Node type enumeration
+    """
+
+    CCTV = 1
+    SWITCH = 2
+    COMPUTER = 3
+    LINK = 4
+    MONITOR = 5
+    PRINTER = 6
+    LOP = 7
+    RTU = 8
+    ACTUATOR = 9
+    SERVER = 10
+
+class PRIORITY(Enum):
+    """
+    Node priority enumeration
+    """
+
+    P1 = 1
+    P2 = 2
+    P3 = 3
+    P4 = 4
+    P5 = 5
+
+class HARDWARE_STATE(Enum):
+    """
+    Node hardware state enumeration
+    """
+
+    ON = 1
+    OFF = 2
+    RESETTING = 3
+
+class SOFTWARE_STATE(Enum):
+    """
+    O/S or Service state enumeration
+    """
+
+    GOOD = 1
+    PATCHING = 2
+    COMPROMISED = 3
+    OVERWHELMED = 4
+
+class NODE_POL_TYPE(Enum):
+    """
+    Node Pattern of Life type enumeration
+    """
+
+    OPERATING = 1
+    OS = 2
+    SERVICE = 3
+
+class PROTOCOL(Enum):
+    """
+    Service protocol enumeration
+    """
+
+    LDAP = 0
+    FTP = 1
+    HTTPS = 2
+    SMTP = 3
+    RTP = 4
+    IPP = 5
+    TCP = 6
+    NONE = 7
+
+class ACTION_TYPE(Enum):
+    """
+    Action type enumeration
+    """
+
+    NODE = 0
+    ACL = 1
+

PRIMAITE/common/protocol.py

@@ -0,0 +1,59 @@
+# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence.
+"""
+The protocol class
+"""
+
+class Protocol(object):
+    """
+    Protocol class
+    """
+
+    def __init__(self, _name):
+        """
+        Init
+
+        Args:
+            _name: The protocol name
+        """
+
+        self.name = _name
+        self.load = 0       # bps
+
+    def get_name(self):
+        """
+        Gets the protocol name
+
+        Returns:
+             The protocol name
+        """
+
+        return self.name
+
+    def get_load(self):
+        """
+        Gets the protocol load
+
+        Returns:
+             The protocol load (bps)
+        """
+
+        return self.load
+
+    def add_load(self, _load):
+        """
+        Adds load to the protocol
+
+        Args:
+            _load: The load to add
+        """
+
+        self.load += _load
+
+    def clear_load(self):
+        """
+        Clears the load on this protocol
+        """
+
+        self.load = 0
+
+

PRIMAITE/common/service.py

@@ -0,0 +1,100 @@
+# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence.
+"""
+The Service class
+"""
+
+from common.enums import SOFTWARE_STATE
+
+class Service(object):
+    """
+    Service class
+    """
+
+    def __init__(self, _name, _port, _state):
+        """
+        Init
+
+        Args:
+            _name: The service name
+            _port: The service port
+            _state: The service state
+        """
+
+        self.name = _name
+        self.port = _port
+        self.state = _state
+        self.patching_count = 0
+
+    def set_name(self, _name):
+        """
+        Sets the service name
+
+        Args:
+            _name: The service name
+        """
+
+        self.name = _name
+
+    def get_name(self):
+        """
+        Gets the service name
+
+        Returns:
+             The service name
+        """
+
+        return self.name
+
+    def set_port(self, _port):
+        """
+        Sets the service port
+
+        Args:
+            _port: The service port
+        """
+
+        self.port = _port
+
+    def get_port(self):
+        """
+        Gets the service port
+
+        Returns:
+             The service port
+        """
+
+        return self.port
+
+    def set_state(self, _state):
+        """
+        Sets the service state
+
+        Args:
+            _state: The service state
+        """
+
+        self.state = _state
+
+    def get_state(self):
+        """
+        Gets the service state
+
+        Returns:
+             The service state
+        """
+
+        return self.state
+
+    def reduce_patching_count(self):
+        """
+        Reduces the patching count for the service
+        """
+
+        self.patching_count -= 1
+        if self.patching_count <= 0:
+            self.patching_count = 0
+            self.state = SOFTWARE_STATE.GOOD
+
+
+
+