Validate ransomware target directories

monkey/common/common_consts/validation_formats.py

@@ -1,3 +1,5 @@
 # Defined in UI on ValidationFormats.js
 IP_RANGE = "ip-range"
 IP = "ip"
+VALID_DIR_LINUX = "valid-directory-linux"
+VALID_DIR_WINDOWS = "valid-directory-windows"

monkey/monkey_island/cc/services/config_schema/ransomware.py

@@ -1,3 +1,5 @@
+from common.common_consts.validation_formats import VALID_DIR_LINUX, VALID_DIR_WINDOWS
+
 RANSOMWARE = {
     "title": "Ransomware",
     "type": "object",
@@ -20,6 +22,7 @@
                         "linux_target_dir": {
                             "title": "Linux target directory",
                             "type": "string",
+                            "format": VALID_DIR_LINUX,
                             "default": "",
                             "description": "A path to a directory on Linux systems that contains "
                             "files that you will allow Infection Monkey to encrypt. If no "
@@ -28,6 +31,7 @@
                         "windows_target_dir": {
                             "title": "Windows target directory",
                             "type": "string",
+                            "format": VALID_DIR_WINDOWS,
                             "default": "",
                             "description": "A path to a directory on Windows systems that contains "
                             "files that you will allow Infection Monkey to encrypt. If no "

monkey/monkey_island/cc/ui/src/components/configuration-components/ValidationErrorMessages.js

@@ -1,4 +1,6 @@
-import {IP, IP_RANGE} from './ValidationFormats';
+import {IP, IP_RANGE, VALID_DIR_LINUX, VALID_DIR_WINDOWS} from './ValidationFormats';
+
+let invalidDirMessage = 'Invalid directory. Path should be absolute or begin with an environment variable.';
 
 export default function transformErrors(errors) {
   return errors.map(error => {
@@ -8,6 +10,10 @@ export default function transformErrors(errors) {
       error.message = 'Invalid IP range, refer to description for valid examples.'
     } else if (error.name === 'format' && error.params.format === IP) {
       error.message = 'Invalid IP.'
+    } else if (error.name === 'format' && error.params.format === VALID_DIR_LINUX) {
+      error.message = invalidDirMessage
+    } else if (error.name === 'format' && error.params.format === VALID_DIR_WINDOWS) {
+      error.message = invalidDirMessage
     }
     return error;
   });

monkey/monkey_island/cc/ui/src/components/configuration-components/ValidationFormats.js

@@ -1,13 +1,22 @@
 const ipRegex = '((25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\\.){3}(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)'
 const cidrNotationRegex = '([0-9]|1[0-9]|2[0-9]|3[0-2])'
 const hostnameRegex = '^([A-Za-z0-9]*[A-Za-z]+[A-Za-z0-9]*.?)*([A-Za-z0-9]*[A-Za-z]+[A-Za-z0-9]*)$'
+// path starts with `/` OR `$`
+const linuxDirRegex = '^/|\\$'
+// path starts like `C:\` OR `C:/` OR `$` OR `%abc%`
+const windowsDirRegex = '^([A-Za-z]:(\\\\|\\/))|\\$|(%\\w*\\d*\\s*%)'
+
 
 export const IP_RANGE = 'ip-range';
 export const IP = 'ip';
+export const VALID_DIR_LINUX = 'valid-directory-linux'
+export const VALID_DIR_WINDOWS = 'valid-directory-windows'
 
 export const formValidationFormats = {
   [IP_RANGE]: buildIpRangeRegex(),
-  [IP]: buildIpRegex()
+  [IP]: buildIpRegex(),
+  [VALID_DIR_LINUX]: buildValidDirLinuxRegex(),
+  [VALID_DIR_WINDOWS]: buildValidDirWindowsRegex()
 };
 
 function buildIpRangeRegex(){
@@ -22,3 +31,11 @@ function buildIpRangeRegex(){
 function buildIpRegex(){
   return new RegExp('^'+ipRegex+'$')
 }
+
+function buildValidDirLinuxRegex() {
+  return new RegExp(linuxDirRegex)
+}
+
+function buildValidDirWindowsRegex() {
+  return new RegExp(windowsDirRegex)
+}