Committed the v1.1.0 code provided by James Short. Had to add setuptools==66 to setup.py as older versions of Gym are uninstallable with setuptools>=67

PRIMAITE/common/config_values_main.py

@@ -19,6 +19,7 @@ class config_values_main(object):
         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
+        self.session_type = ""                      # the session type to run (TRAINING or EVALUATION)
 
         # Environment
         self.observation_space_high_value = 0       # The high value for the observation space
@@ -33,6 +34,7 @@ class config_values_main(object):
         self.on_should_be_resetting = 0
         self.resetting_should_be_on = 0
         self.resetting_should_be_off = 0 
+        self.resetting = 0
         # Node O/S or Service State
         self.good_should_be_patching = 0
         self.good_should_be_compromised = 0
@@ -40,6 +42,7 @@ class config_values_main(object):
         self.patching_should_be_good = 0
         self.patching_should_be_compromised = 0
         self.patching_should_be_overwhelmed = 0
+        self.patching = 0
         self.compromised_should_be_good = 0
         self.compromised_should_be_patching = 0
         self.compromised_should_be_overwhelmed = 0
@@ -48,6 +51,32 @@ class config_values_main(object):
         self.overwhelmed_should_be_patching = 0
         self.overwhelmed_should_be_compromised = 0
         self.overwhelmed = 0
+        # Node File System State
+        self.good_should_be_repairing = 0
+        self.good_should_be_restoring = 0
+        self.good_should_be_corrupt = 0
+        self.good_should_be_destroyed = 0
+        self.repairing_should_be_good = 0
+        self.repairing_should_be_restoring = 0
+        self.repairing_should_be_corrupt = 0
+        self.repairing_should_be_destroyed = 0   # Repairing does not fix destroyed state - you need to restore
+        self.repairing = 0
+        self.restoring_should_be_good = 0
+        self.restoring_should_be_repairing = 0
+        self.restoring_should_be_corrupt = 0     # Not the optimal method (as repair will fix corruption)
+        self.restoring_should_be_destroyed = 0
+        self.restoring = 0
+        self.corrupt_should_be_good = 0
+        self.corrupt_should_be_repairing = 0
+        self.corrupt_should_be_restoring = 0
+        self.corrupt_should_be_destroyed = 0
+        self.corrupt = 0
+        self.destroyed_should_be_good = 0
+        self.destroyed_should_be_repairing = 0
+        self.destroyed_should_be_restoring = 0
+        self.destroyed_should_be_corrupt = 0
+        self.destroyed = 0
+        self.scanning = 0
         # IER status
         self.red_ier_running = 0
         self.green_ier_blocked = 0
@@ -55,5 +84,8 @@ class config_values_main(object):
         # 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
+        self.service_patching_duration = 0                    # The time taken to patch a service   
+        self.file_system_repairing_limit = 0				  # The time take to repair a file
+        self.file_system_restoring_limit = 0				  # The time take to restore a file
+        self.file_system_scanning_limit = 0                   # The time taken to scan the file system
 

PRIMAITE/common/enums.py

@@ -59,6 +59,16 @@ class NODE_POL_TYPE(Enum):
     OPERATING = 1
     OS = 2
     SERVICE = 3
+    FILE = 4
+
+class NODE_POL_INITIATOR(Enum):
+    """
+    Node Pattern of Life initiator enumeration
+    """
+
+    DIRECT = 1
+    IER = 2
+    SERVICE = 3
 
 class PROTOCOL(Enum):
     """
@@ -82,3 +92,13 @@ class ACTION_TYPE(Enum):
     NODE = 0
     ACL = 1
 
+class FILE_SYSTEM_STATE(Enum):
+    """
+    File System State
+    """
+
+    GOOD = 1
+    CORRUPT = 2
+    DESTROYED = 3
+    REPAIRING = 4
+    RESTORING = 5