From 6b750928236adb7d3589413522d2ac095b63e54e Mon Sep 17 00:00:00 2001
From: Shreya <shreya.malviya@gmail.com>
Date: Wed, 24 Jun 2020 15:15:45 +0530
Subject: [PATCH] Small fixes after rebasing

---
 monkey/monkey_island/cc/services/attack/attack_report.py | 2 +-
 monkey/monkey_island/cc/services/attack/attack_schema.py | 1 +
 monkey/monkey_island/cc/services/config_schema.py        | 6 +++++-
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/monkey/monkey_island/cc/services/attack/attack_report.py b/monkey/monkey_island/cc/services/attack/attack_report.py
index 882cb2b32..16e6e51a1 100644
--- a/monkey/monkey_island/cc/services/attack/attack_report.py
+++ b/monkey/monkey_island/cc/services/attack/attack_report.py
@@ -39,7 +39,7 @@ TECHNIQUES = {'T1210': T1210.T1210,
               'T1064': T1064.T1064,
               'T1136': T1136.T1136,
               'T1156': T1156.T1156,
-              'T1504': T1504.T1504
+              'T1504': T1504.T1504,
               'T1158': T1158.T1158
               }
 
diff --git a/monkey/monkey_island/cc/services/attack/attack_schema.py b/monkey/monkey_island/cc/services/attack/attack_schema.py
index 3df8078da..04c63a850 100644
--- a/monkey/monkey_island/cc/services/attack/attack_schema.py
+++ b/monkey/monkey_island/cc/services/attack/attack_schema.py
@@ -99,6 +99,7 @@ SCHEMA = {
                     "description": "Adversaries may gain persistence and elevate privileges "
                                    "in certain situations by abusing PowerShell profiles which "
                                    "are scripts that run when PowerShell starts."
+                },
                 "T1158": {
                     "title": "Hidden Files and Directories",
                     "type": "bool",
diff --git a/monkey/monkey_island/cc/services/config_schema.py b/monkey/monkey_island/cc/services/config_schema.py
index 92b842757..30425dc1d 100644
--- a/monkey/monkey_island/cc/services/config_schema.py
+++ b/monkey/monkey_island/cc/services/config_schema.py
@@ -167,6 +167,10 @@ SCHEMA = {
                     ],
                     "title": "Modify shell startup files",
                     "attack_techniques": ["T1156", "T1504"]
+                },
+                {
+                    "type": "string",
+                    "enum": [
                         "HiddenFiles"
                     ],
                     "title": "Hidden files and directories",
@@ -392,7 +396,7 @@ SCHEMA = {
                             "default": [
                                 "BackdoorUser",
                                 "CommunicateAsNewUser",
-                                "ModifyShellStartupFiles"
+                                "ModifyShellStartupFiles",
                                 "HiddenFiles"
                             ],
                             "description": "List of actions the Monkey will run post breach"