94 lines
3.7 KiB
Python
94 lines
3.7 KiB
Python
# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence.
|
|
"""The config class."""
|
|
|
|
|
|
class ConfigValuesMain(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
|
|
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
|
|
)
|
|
|
|
# Reward values
|
|
# Generic
|
|
self.all_ok = 0
|
|
# Node Hardware 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
|
|
self.resetting = 0
|
|
# Node Software 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.patching = 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
|
|
# 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
|
|
|
|
# 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.node_booting_duration = 0 # The Time taken to turn on the node
|
|
self.node_shutdown_duration = 0 # The time taken to turn off the node
|
|
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
|
|
|