diff --git a/AIDojoCoordinator/worlds/NSEGameCoordinator.py b/AIDojoCoordinator/worlds/NSEGameCoordinator.py index 954ad0b5..0917b99a 100644 --- a/AIDojoCoordinator/worlds/NSEGameCoordinator.py +++ b/AIDojoCoordinator/worlds/NSEGameCoordinator.py @@ -142,8 +142,8 @@ def _create_state_from_view(self, view:dict, add_neighboring_nets:bool=True)->Ga for controlled_host in controlled_hosts: for net in self._get_networks_from_host(controlled_host): #TODO net_obj = netaddr.IPNetwork(str(net)) + known_networks.add(net) if net_obj.ip.is_private(): #TODO - known_networks.add(net) net_obj.value += 256 if net_obj.ip.is_private(): ip = Network(str(net_obj.ip), net_obj.prefixlen) @@ -156,6 +156,10 @@ def _create_state_from_view(self, view:dict, add_neighboring_nets:bool=True)->Ga known_networks.add(ip) #return value back to the original net_obj.value += 256 + else: + for controlled_host in controlled_hosts: + for net in self._get_networks_from_host(controlled_host): #TODO + known_networks.add(net) # parse known services known_services = self._get_services_from_view(view["known_services"]) # parse known data