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

BB: Fix configuration modification logic in zerologon.py

This commit is contained in:
Shreya Malviya 2022-09-01 17:01:05 +05:30
parent ad6449507d
commit 2352bb0d5e
1 changed files with 7 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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