Repro #23

bug23_repro.py

@@ -0,0 +1,30 @@
+# %%
+from cyberbattle.simulation.environment_generation import create_random_environment
+import cyberbattle.simulation.commandcontrol as commandcontrol
+
+# %%
+print(commandcontrol.__file__)
+
+
+# %%
+env = create_random_environment('test', 10)
+# %%
+env
+# %%
+c2 = commandcontrol.CommandControl(env)
+# %%
+c2.print_all_attacks()
+
+# %%
+
+c2.list_local_attacks('0')
+# %%
+
+# # %%
+# !pip freeze
+
+# # %%
+# !python - -version
+# # %%
+
+# %%

cyberbattle/simulation/environment_generation.py

@@ -185,6 +185,8 @@ def create_random_environment(name: str, size: int) -> model.Environment:
         rand_os: str = os_types[random.randint(0, 1)]
         nodes[str(i)] = create_random_node(rand_os, potential_ports)
 
+    nodes['0'].agent_installed = True
+
     graph.add_nodes_from([(k, {'data': v}) for (k, v) in list(nodes.items())])
 
     return model.Environment(network=graph, vulnerability_library=local_vuln_lib, identifiers=ENV_IDENTIFIERS)