From 5f41ce54c6627c30419418dcc77eeecc36bf8677 Mon Sep 17 00:00:00 2001
From: VakarisZ <vakarisz@yahoo.com>
Date: Mon, 8 Mar 2021 17:13:35 +0200
Subject: [PATCH] Added config file generation script, which can generate
 config files from config templates.

---
 envs/monkey_zoo/blackbox/utils/README.md      |  9 +++
 .../utils/config_generation_script.py         | 69 +++++++++++++++++++
 .../utils/generated_configs/.gitignore        |  1 +
 3 files changed, 79 insertions(+)
 create mode 100644 envs/monkey_zoo/blackbox/utils/README.md
 create mode 100644 envs/monkey_zoo/blackbox/utils/config_generation_script.py
 create mode 100644 envs/monkey_zoo/blackbox/utils/generated_configs/.gitignore

diff --git a/envs/monkey_zoo/blackbox/utils/README.md b/envs/monkey_zoo/blackbox/utils/README.md
new file mode 100644
index 000000000..b5d8e9125
--- /dev/null
+++ b/envs/monkey_zoo/blackbox/utils/README.md
@@ -0,0 +1,9 @@
+# BlackBox utility scripts
+
+## Config generation script
+
+This script is used to generate config files for manual tests.
+Config file will be generated according to the templates in `envs/monkey_zoo/blackbox/config_templates`.
+
+Run `envs/monkey_zoo/blackbox/utils/config_generation_script.py` to populate 
+`envs/monkey_zoo/blackbox/utils/generated_configs` directory with configuration files.
diff --git a/envs/monkey_zoo/blackbox/utils/config_generation_script.py b/envs/monkey_zoo/blackbox/utils/config_generation_script.py
new file mode 100644
index 000000000..7ecee6bd7
--- /dev/null
+++ b/envs/monkey_zoo/blackbox/utils/config_generation_script.py
@@ -0,0 +1,69 @@
+import argparse
+import pathlib
+from typing import Type
+
+from envs.monkey_zoo.blackbox.config_templates.config_template import ConfigTemplate
+from envs.monkey_zoo.blackbox.config_templates.elastic import Elastic
+from envs.monkey_zoo.blackbox.config_templates.hadoop import Hadoop
+from envs.monkey_zoo.blackbox.config_templates.mssql import Mssql
+from envs.monkey_zoo.blackbox.config_templates.performance import Performance
+from envs.monkey_zoo.blackbox.config_templates.shellshock import ShellShock
+from envs.monkey_zoo.blackbox.config_templates.smb_mimikatz import SmbMimikatz
+from envs.monkey_zoo.blackbox.config_templates.smb_pth import SmbPth
+from envs.monkey_zoo.blackbox.config_templates.ssh import Ssh
+from envs.monkey_zoo.blackbox.config_templates.struts2 import Struts2
+from envs.monkey_zoo.blackbox.config_templates.tunneling import Tunneling
+from envs.monkey_zoo.blackbox.config_templates.weblogic import Weblogic
+from envs.monkey_zoo.blackbox.config_templates.wmi_mimikatz import WmiMimikatz
+from envs.monkey_zoo.blackbox.config_templates.wmi_pth import WmiPth
+from envs.monkey_zoo.blackbox.config_templates.zerologon import Zerologon
+from envs.monkey_zoo.blackbox.island_client.island_config_parser import IslandConfigParser
+from envs.monkey_zoo.blackbox.island_client.monkey_island_client import MonkeyIslandClient
+
+DST_DIR_NAME = 'generated_configs'
+DST_DIR_PATH = pathlib.Path(pathlib.Path(__file__).parent.absolute(), DST_DIR_NAME)
+
+parser = argparse.ArgumentParser(description='Generate config files.')
+parser.add_argument('island_ip', metavar='IP:PORT',
+                    help='Island endpoint. Example: 123.123.123.123:5000')
+
+args = parser.parse_args()
+island_client = MonkeyIslandClient(args.island_ip)
+
+
+CONFIG_TEMPLATES = [
+    Elastic,
+    Hadoop,
+    Mssql,
+    Performance,
+    ShellShock,
+    SmbMimikatz,
+    SmbPth,
+    Ssh,
+    Struts2,
+    Tunneling,
+    Weblogic,
+    WmiMimikatz,
+    WmiPth,
+    Zerologon
+]
+
+
+def generate_templates():
+    for template in CONFIG_TEMPLATES:
+        save_template_as_config(template)
+
+
+def save_template_as_config(template: Type[ConfigTemplate]):
+    file_path = pathlib.Path(DST_DIR_PATH, f"{template.__name__}.conf")
+    file_contents = IslandConfigParser.get_raw_config(template, island_client)
+    save_to_file(file_path, file_contents)
+
+
+def save_to_file(file_path, contents):
+    with open(file_path, 'w') as file:
+        file.write(contents)
+
+
+if __name__ == '__main__':
+    generate_templates()
diff --git a/envs/monkey_zoo/blackbox/utils/generated_configs/.gitignore b/envs/monkey_zoo/blackbox/utils/generated_configs/.gitignore
new file mode 100644
index 000000000..9c558e357
--- /dev/null
+++ b/envs/monkey_zoo/blackbox/utils/generated_configs/.gitignore
@@ -0,0 +1 @@
+.