From c2028f15a435170b448e377aa3c06b869c884cb1 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Mon, 18 Jul 2022 15:00:06 -0400 Subject: [PATCH] BB: Add depth_1_a_test_configuration --- .../blackbox/test_configurations/__init__.py | 1 + .../blackbox/test_configurations/depth_1_a.py | 65 +++++++++++++++++++ 2 files changed, 66 insertions(+) create mode 100644 envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py diff --git a/envs/monkey_zoo/blackbox/test_configurations/__init__.py b/envs/monkey_zoo/blackbox/test_configurations/__init__.py index 022b20e35..0e2c06c53 100644 --- a/envs/monkey_zoo/blackbox/test_configurations/__init__.py +++ b/envs/monkey_zoo/blackbox/test_configurations/__init__.py @@ -1,2 +1,3 @@ from .test_configuration import TestConfiguration from .zerologon import zerologon_test_configuration +from .depth_1_a import depth_1_a_test_configuration diff --git a/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py b/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py new file mode 100644 index 000000000..1c5193fb6 --- /dev/null +++ b/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py @@ -0,0 +1,65 @@ +from common.configuration import AgentConfiguration, PluginConfiguration +from common.credentials import Credentials, Password, Username + +from .noop import noop_test_configuration +from .utils import ( + add_credential_collectors, + add_exploiters, + add_subnets, + replace_agent_configuration, + replace_propagation_credentials, +) + +credentials = ( + Credentials(Username("m0nk3y"), None), + Credentials(None, Password("Ivrrw5zEzs")), + Credentials(None, Password("Xk8VDTsC")), +) + + +def _add_exploiters(agent_configuration: AgentConfiguration) -> AgentConfiguration: + brute_force = [ + PluginConfiguration(name="HadoopExploiter", options={}), + PluginConfiguration(name="Log4ShellExploiter", options={}), + PluginConfiguration(name="MSSQLExploiter", options={}), + PluginConfiguration(name="SmbExploiter", options={}), + PluginConfiguration(name="SSHExploiter", options={}), + ] + vulnerability = [PluginConfiguration(name="ZerologonExploiter", options={})] + + return add_exploiters(agent_configuration, brute_force=brute_force, vulnerability=vulnerability) + + +def _add_subnets(agent_configuration: AgentConfiguration) -> AgentConfiguration: + subnets = [ + "10.2.2.2", + "10.2.2.3", + "10.2.3.55", + "10.2.3.56", + "10.2.3.49", + "10.2.3.50", + "10.2.3.51", + "10.2.3.52", + "10.2.2.16", + "10.2.2.14", + "10.2.2.15", + ] + return add_subnets(agent_configuration, subnets) + + +def _add_credential_collectors(agent_configuration: AgentConfiguration) -> AgentConfiguration: + return add_credential_collectors( + agent_configuration, [PluginConfiguration("MimikatzCollector", {})] + ) + + +agent_configuration = _add_exploiters( + _add_subnets(_add_credential_collectors(noop_test_configuration.agent_configuration)) +) + +depth_1_a_test_configuration = replace_agent_configuration( + noop_test_configuration, agent_configuration +) +depth_1_a_test_configuration = replace_propagation_credentials( + depth_1_a_test_configuration, credentials +)