p34709852
/
monkey
forked from p15670423/monkey
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

BB: Set maximum depth in zerologon_test_configuration

This commit is contained in:
Mike Salvatore 2022-07-18 15:09:50 -04:00
parent 5a1a40a515
commit 365b4098e4
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
envs/monkey_zoo/blackbox/test_configurations/zerologon.py
View File

@ -1,7 +1,13 @@
from common.configuration import AgentConfiguration, PluginConfiguration
from .noop import noop_test_configuration
from .utils import add_exploiters, add_subnets, add_tcp_ports, replace_agent_configuration
from .utils import (
add_exploiters,
add_subnets,
add_tcp_ports,
replace_agent_configuration,
set_maximum_depth,
)
def _add_exploiters(agent_configuration: AgentConfiguration) -> AgentConfiguration:
@ -21,9 +27,11 @@ def _add_subnets(agent_configuration: AgentConfiguration) -> AgentConfiguration:
return add_subnets(agent_configuration, subnets)
agent_configuration = _add_exploiters(
_add_tcp_ports(_add_subnets(noop_test_configuration.agent_configuration))
)
agent_configuration = set_maximum_depth(noop_test_configuration.agent_configuration, 1)
agent_configuration = _add_exploiters(agent_configuration)
agent_configuration = _add_tcp_ports(agent_configuration)
agent_configuration = _add_subnets(agent_configuration)
zerologon_test_configuration = replace_agent_configuration(
noop_test_configuration, agent_configuration
)