def get_network(self, nM, nS, **params):
generator = ScenarioGenerator()
scenario_dict = generator.generate(nM,
nS,
r_sensitive=self.r_sensitive,
r_user=self.r_user,
**params)
scenario = Scenario(scenario_dict)
network = Network(scenario)
return network
def _construct_scenario(self):
scenario_dict = dict()
scenario_dict[u.SUBNETS] = self.subnets
scenario_dict[u.TOPOLOGY] = self.topology
scenario_dict[u.SERVICES] = self.services
scenario_dict[u.SENSITIVE_HOSTS] = self.sensitive_hosts
scenario_dict[u.EXPLOITS] = self.exploits
scenario_dict[u.SERVICE_SCAN_COST] = self.service_scan_cost
scenario_dict[u.OS_SCAN_COST] = self.os_scan_cost
scenario_dict[u.SUBNET_SCAN_COST] = self.subnet_scan_cost
scenario_dict[u.FIREWALL] = self.firewall
scenario_dict[u.HOSTS] = self.hosts
return Scenario(scenario_dict)
def _construct_scenario(self):
scenario_dict = dict()
scenario_dict[u.SUBNETS] = self.subnets
scenario_dict[u.TOPOLOGY] = self.topology
scenario_dict[u.OS] = self.os
scenario_dict[u.SERVICES] = self.services
scenario_dict[u.PROCESSES] = self.processes
scenario_dict[u.SENSITIVE_HOSTS] = self.sensitive_hosts
scenario_dict[u.EXPLOITS] = self.exploits
scenario_dict[u.PRIVESCS] = self.privescs
scenario_dict[u.OS_SCAN_COST] = self.os_scan_cost
scenario_dict[u.SERVICE_SCAN_COST] = self.service_scan_cost
scenario_dict[u.SUBNET_SCAN_COST] = self.subnet_scan_cost
scenario_dict[u.PROCESS_SCAN_COST] = self.process_scan_cost
scenario_dict[u.FIREWALL] = self.firewall
scenario_dict[u.HOSTS] = self.hosts
scenario_dict[u.STEP_LIMIT] = self.step_limit
return Scenario(scenario_dict, name=self.name, generated=False)