From c0f0d35f0b531be606b5c230504466cdabd3a07b Mon Sep 17 00:00:00 2001
From: vakarisz <vakarisz@yahoo.com>
Date: Mon, 20 Jun 2022 12:18:03 +0300
Subject: [PATCH] Island: Use OperatingSystems enum in config.py

---
 monkey/monkey_island/cc/services/config.py | 21 ++++++++++-----------
 1 file changed, 10 insertions(+), 11 deletions(-)

diff --git a/monkey/monkey_island/cc/services/config.py b/monkey/monkey_island/cc/services/config.py
index 684801ac2..86541c33d 100644
--- a/monkey/monkey_island/cc/services/config.py
+++ b/monkey/monkey_island/cc/services/config.py
@@ -8,6 +8,7 @@ from typing import Any, Dict, List
 
 from jsonschema import Draft4Validator, validators
 
+from common import OperatingSystems
 from common.config_value_paths import (
     LM_HASH_LIST_PATH,
     NTLM_HASH_LIST_PATH,
@@ -533,9 +534,7 @@ class ConfigService:
         formatted_scan_targets_config[flat_local_network_scan_field] = config[
             flat_local_network_scan_field
         ]
-        formatted_scan_targets_config["subnets"] = config[
-            flat_subnet_scan_list_field
-        ]
+        formatted_scan_targets_config["subnets"] = config[flat_subnet_scan_list_field]
 
         config.pop(flat_blocked_ips_field, None)
         config.pop(flat_inaccessible_subnets_field, None)
@@ -600,14 +599,14 @@ class ConfigService:
         formatted_config: Dict,
     ) -> Dict[str, List[Dict[str, Any]]]:
         supported_os = {
-            "HadoopExploiter": ["LINUX", "WINDOWS"],
-            "Log4ShellExploiter": ["LINUX", "WINDOWS"],
-            "MSSQLExploiter": ["WINDOWS"],
-            "PowerShellExploiter": ["WINDOWS"],
-            "SSHExploiter": ["LINUX"],
-            "SmbExploiter": ["WINDOWS"],
-            "WmiExploiter": ["WINDOWS"],
-            "ZerologonExploiter": ["WINDOWS"],
+            "HadoopExploiter": [OperatingSystems.LINUX, OperatingSystems.WINDOWS],
+            "Log4ShellExploiter": [OperatingSystems.LINUX, OperatingSystems.WINDOWS],
+            "MSSQLExploiter": [OperatingSystems.WINDOWS],
+            "PowerShellExploiter": [OperatingSystems.WINDOWS],
+            "SSHExploiter": [OperatingSystems.LINUX],
+            "SmbExploiter": [OperatingSystems.WINDOWS],
+            "WmiExploiter": [OperatingSystems.WINDOWS],
+            "ZerologonExploiter": [OperatingSystems.WINDOWS],
         }
         new_config = copy.deepcopy(formatted_config)
         for exploiter in chain(new_config["brute_force"], new_config["vulnerability"]):