From f1d9ea64e5a62f8ce4bb22fc9675782d4e5d2904 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Mon, 18 Jul 2022 15:15:10 -0400 Subject: [PATCH] BB: Add tcp ports to depth_1_a_test_configuration --- envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py b/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py index ffd9ed882..d29451568 100644 --- a/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py +++ b/envs/monkey_zoo/blackbox/test_configurations/depth_1_a.py @@ -6,6 +6,7 @@ from .utils import ( add_credential_collectors, add_exploiters, add_subnets, + add_tcp_ports, replace_agent_configuration, replace_propagation_credentials, set_maximum_depth, @@ -54,6 +55,11 @@ def _add_credential_collectors(agent_configuration: AgentConfiguration) -> Agent ) +def _add_tcp_ports(agent_configuration: AgentConfiguration) -> AgentConfiguration: + ports = [445, 8088, 22] + return add_tcp_ports(agent_configuration, ports) + + agent_configuration = set_maximum_depth(noop_test_configuration.agent_configuration, 1) agent_configuration = _add_exploiters(agent_configuration) agent_configuration = _add_subnets(agent_configuration)