BB: Fix configuration modification logic in powershell_credentials_reuse.py
This commit is contained in:
parent
15fd1f9f72
commit
e564bd1072
|
@ -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:
|
||||||
|
@ -30,11 +24,10 @@ def _add_tcp_ports(agent_configuration: AgentConfiguration) -> AgentConfiguratio
|
||||||
return add_tcp_ports(agent_configuration, ports)
|
return add_tcp_ports(agent_configuration, ports)
|
||||||
|
|
||||||
|
|
||||||
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_subnets(test_configuration)
|
test_agent_configuration = _add_subnets(test_agent_configuration)
|
||||||
test_configuration = _add_tcp_ports(test_configuration)
|
test_agent_configuration = _add_tcp_ports(test_agent_configuration)
|
||||||
|
|
||||||
powershell_credentials_reuse_test_configuration = replace_agent_configuration(
|
powershell_credentials_reuse_test_configuration = noop_test_configuration.copy()
|
||||||
noop_test_configuration, test_configuration
|
powershell_credentials_reuse_test_configuration.agent_configuration = test_agent_configuration
|
||||||
)
|
|
||||||
|
|
Loading…
Reference in New Issue