Merge pull request #341 from stratosphereips/ondra-fix-empty-network-list

Fix adding known networks from controlled hosts

Author: ondrej-lukas

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