From 133bf757740d10b856395e459ae4c958bbd4a4c7 Mon Sep 17 00:00:00 2001
From: Ilija Lazoroski <ilija.la@live.com>
Date: Wed, 3 Aug 2022 17:00:08 +0200
Subject: [PATCH] Island: Register deployment convention
---
monkey/monkey_island/cc/services/initialize.py | 18 ++++++++++++++++++
1 file changed, 18 insertions(+)
diff --git a/monkey/monkey_island/cc/services/initialize.py b/monkey/monkey_island/cc/services/initialize.py
index 2d2678ac3..7bb1063be 100644
--- a/monkey/monkey_island/cc/services/initialize.py
+++ b/monkey/monkey_island/cc/services/initialize.py
@@ -1,3 +1,4 @@
+import json
import logging
from pathlib import Path
@@ -33,6 +34,7 @@ from monkey_island.cc.repository import (
from monkey_island.cc.server_utils.consts import MONKEY_ISLAND_ABS_PATH
from monkey_island.cc.server_utils.encryption import ILockableEncryptor, RepositoryEncryptor
from monkey_island.cc.server_utils.island_logger import get_log_file_path
+from monkey_island.cc.deployment import Deployment
from monkey_island.cc.services import AWSService, IslandModeService, RepositoryService
from monkey_island.cc.services.attack.technique_reports.T1003 import T1003, T1003GetReportData
from monkey_island.cc.services.run_local_monkey import LocalMonkeyRunService
@@ -50,6 +52,7 @@ from .reporting.report import ReportService
logger = logging.getLogger(__name__)
AGENT_BINARIES_PATH = Path(MONKEY_ISLAND_ABS_PATH) / "cc" / "binaries"
+DEPLOYMENT_FILE_PATH = Path(MONKEY_ISLAND_ABS_PATH) / "cc" / "deployment.json"
REPOSITORY_KEY_FILE_NAME = "repository_key.bin"
@@ -89,6 +92,7 @@ def _register_conventions(container: DIContainer, data_dir: Path):
DEFAULT_RANSOMWARE_AGENT_CONFIGURATION,
)
container.register_convention(Path, "island_log_file_path", get_log_file_path(data_dir))
+ container.register_convention(Deployment, "deployment", _get_depyloyment_from_file(DEPLOYMENT_FILE_PATH))
def _register_repositories(container: DIContainer, data_dir: Path):
@@ -145,6 +149,20 @@ def _log_agent_binary_hashes(agent_binary_repository: IAgentBinaryRepository):
for os, binary_sha256_hash in agent_hashes.items():
logger.info(f"{os} agent: SHA-256 hash: {binary_sha256_hash}")
+def _get_depyloyment_from_file(file_path: Path) -> Deployment:
+ try:
+ with open(file_path, "r") as deployment_info_file:
+ deployment_info = json.load(deployment_info_file)
+ return Deployment[deployment_info["deployment"].upper()]
+ except FileNotFoundError as ex:
+ logger.debug(f"Deployment file {file_path} is not found. Exception: {ex}")
+ except KeyError as ex:
+ logger.debug(f"Invalid key in the deployment file. Exception: {ex}")
+ except json.JSONDecodeError as ex:
+ logger.debug(f"Invalid deployment info file. Exception: {ex}")
+ except Exception as ex:
+ logger.debug(f"Couldn't get deployment info from {file_path}. Exception: {ex}.")
+
def _register_services(container: DIContainer):
container.register_instance(AWSService, container.resolve(AWSService))