Merge pull request #209 from stratosphereips/improve-env

netsecenv. Add check in actions to check if ips and nets is a real ip and nets
stratosphereips · Apr 25, 2024 · d67b377 · d67b377
2 parents 618ef5c + 9d42b13
commit d67b377
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/NetSecGameAgents b/NetSecGameAgents
diff --git a/env/game_components.py b/env/game_components.py
@@ -30,6 +30,15 @@ class IP():
     """
     ip: str
 
+    def __post_init__(self):
+        """
+        Check if the provided IP is valid
+        """
+        try:
+            ipaddress.ip_address(self.ip)
+        except ValueError:
+            raise ValueError("Invalid IP address provided")
+
     def __repr__(self):
         return self.ip
+4 −1		.gitignore
+5 −5		agents/agent_utils.py
+482 −0		agents/concepts_q_learning/conceptual_q_agent.py
+32 −35		agents/q_learning/q_agent.py