From 8e28cb9e60ae0be4ea4393c08e02fd11812d65b2 Mon Sep 17 00:00:00 2001 From: Ondrej Lukas Date: Fri, 8 Aug 2025 14:23:36 +0200 Subject: [PATCH] Fix adding known networks from controlled hosts --- AIDojoCoordinator/worlds/NSEGameCoordinator.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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