Fix first part of #46

microsoft · May 24, 2022 · ddb99f9 · ddb99f9
1 parent caec5f2
commit ddb99f9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cyberbattle/simulation/generate_network.py b/cyberbattle/simulation/generate_network.py
@@ -161,10 +161,13 @@ def create_cached_credential(node: NodeID, port: PortName) -> CredentialID:
 
     def add_leak_neighbors_vulnerability(
             node_id: m.NodeID,
-            library: m.VulnerabilityLibrary = {}) -> m.VulnerabilityLibrary:
+            library: Optional[m.VulnerabilityLibrary] = None) -> m.VulnerabilityLibrary:
         """Create random vulnerabilities
         that reveals immediate traffic neighbors from a given node"""
 
+        if not library:
+            library = {}
+
         rdp_neighbors = traffic_targets(node_id, 'RDP')
 
         if len(rdp_neighbors) > 0: