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/config/config_main.yaml

@@ -5,13 +5,19 @@
 # "STABLE_BASELINES3_PPO"
 # "STABLE_BASELINES3_A2C"
 # "GENERIC"
-agentIdentifier: STABLE_BASELINES3_PPO
-# Maximum number of episodes to run per training session
+agentIdentifier: STABLE_BASELINES3_A2C
+# Number of episodes to run per session
 numEpisodes: 10
 # Time delay between steps (for generic agents)
 timeDelay: 10
 # Filename of the scenario / laydown
-configFilename: config_2_DDOS_BASIC.yaml
+configFilename: config_5_DATA_MANIPULATION.yaml
+# Type of session to be run (TRAINING or EVALUATION)
+sessionType: TRAINING
+# Determine whether to load an agent from file
+loadAgent: False
+# File path and file name of agent if you're loading one in
+agentLoadFile: C:\[Path]\[agent_saved_filename.zip]
 
 # Environment config values
 # The high value for the observation space
@@ -27,6 +33,7 @@ onShouldBeOff: -2
 onShouldBeResetting: -5
 resettingShouldBeOn: -5
 resettingShouldBeOff: -2 
+resetting: -3
 # Node O/S or Service State
 goodShouldBePatching: 2
 goodShouldBeCompromised: 5
@@ -34,6 +41,7 @@ goodShouldBeOverwhelmed: 5
 patchingShouldBeGood: -5
 patchingShouldBeCompromised: 2
 patchingShouldBeOverwhelmed: 2
+patching: -3
 compromisedShouldBeGood: -20
 compromisedShouldBePatching: -20
 compromisedShouldBeOverwhelmed: -20
@@ -42,11 +50,40 @@ overwhelmedShouldBeGood: -20
 overwhelmedShouldBePatching: -20
 overwhelmedShouldBeCompromised: -20
 overwhelmed: -20
+# Node File System State
+goodShouldBeRepairing: 2
+goodShouldBeRestoring: 2
+goodShouldBeCorrupt: 5
+goodShouldBeDestroyed: 10
+repairingShouldBeGood: -5
+repairingShouldBeRestoring: 2
+repairingShouldBeCorrupt: 2
+repairingShouldBeDestroyed: 0
+repairing: -3
+restoringShouldBeGood: -10
+restoringShouldBeRepairing: -2
+restoringShouldBeCorrupt: 1     
+restoringShouldBeDestroyed: 2
+restoring: -6
+corruptShouldBeGood: -10
+corruptShouldBeRepairing: -10
+corruptShouldBeRestoring: -10
+corruptShouldBeDestroyed: 2
+corrupt: -10
+destroyedShouldBeGood: -20
+destroyedShouldBeRepairing: -20
+destroyedShouldBeRestoring: -20
+destroyedShouldBeCorrupt: -20
+destroyed: -20
+scanning: -2
 # IER status
 redIerRunning: -5
 greenIerBlocked: -10
 
 # Patching / Reset durations
-osPatchingDuration: 5                         # The time taken to patch the OS
-nodeResetDuration: 5                          # The time taken to reset a node (hardware)
-servicePatchingDuration: 5                    # The time taken to patch a service
+osPatchingDuration: 5            # The time taken to patch the OS
+nodeResetDuration: 5             # The time taken to reset a node (hardware)
+servicePatchingDuration: 5       # The time taken to patch a service
+fileSystemRepairingLimit: 5      # The time take to repair the file system
+fileSystemRestoringLimit: 5      # The time take to restore the file system
+fileSystemScanningLimit: 5       # The time taken to scan the file system