From ad6449507d41f28efe9083325ed8bae635aa1013 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Thu, 1 Sep 2022 16:59:58 +0530 Subject: [PATCH] BB: Fix configuration modification logic in wmi_mimikatz.py --- .../test_configurations/wmi_mimikatz.py | 26 +++++++------------ 1 file changed, 10 insertions(+), 16 deletions(-) diff --git a/envs/monkey_zoo/blackbox/test_configurations/wmi_mimikatz.py b/envs/monkey_zoo/blackbox/test_configurations/wmi_mimikatz.py index fe4a1414e..c525f5ca5 100644 --- a/envs/monkey_zoo/blackbox/test_configurations/wmi_mimikatz.py +++ b/envs/monkey_zoo/blackbox/test_configurations/wmi_mimikatz.py @@ -7,8 +7,6 @@ from .utils import ( add_exploiters, add_subnets, add_tcp_ports, - replace_agent_configuration, - replace_propagation_credentials, set_maximum_depth, ) @@ -40,17 +38,12 @@ def _add_tcp_ports(agent_configuration: AgentConfiguration) -> AgentConfiguratio return add_tcp_ports(agent_configuration, ports) -test_configuration = set_maximum_depth(noop_test_configuration.agent_configuration, 1) -test_configuration = _add_exploiters(test_configuration) -test_configuration = _add_subnets(test_configuration) -test_configuration = _add_credential_collectors(test_configuration) -test_configuration = _add_tcp_ports(test_configuration) -test_configuration = _add_credential_collectors(test_configuration) - -wmi_mimikatz_test_configuration = replace_agent_configuration( - noop_test_configuration, test_configuration -) - +test_agent_configuration = set_maximum_depth(noop_test_configuration.agent_configuration, 1) +test_agent_configuration = _add_exploiters(test_agent_configuration) +test_agent_configuration = _add_subnets(test_agent_configuration) +test_agent_configuration = _add_credential_collectors(test_agent_configuration) +test_agent_configuration = _add_tcp_ports(test_agent_configuration) +test_agent_configuration = _add_credential_collectors(test_agent_configuration) CREDENTIALS = ( Credentials(Username("Administrator"), None), @@ -59,6 +52,7 @@ CREDENTIALS = ( Credentials(None, Password("Ivrrw5zEzs")), Credentials(None, Password("Password1!")), ) -wmi_mimikatz_test_configuration = replace_propagation_credentials( - wmi_mimikatz_test_configuration, CREDENTIALS -) + +wmi_mimikatz_test_configuration = noop_test_configuration.copy() +wmi_mimikatz_test_configuration.agent_configuration = test_agent_configuration +wmi_mimikatz_test_configuration.propagation_credentials = CREDENTIALS