More strict type checking.

Tests/test_0_single_rule.py

@@ -15,6 +15,10 @@ def test_type_checking_init(self):
             rule.Rule("IPCIDR", "114514")
         with raises(ValueError):
             rule.Rule("IPCIDR6", "1919810")
+        with raises(ValueError):
+            rule.Rule("IPCIDR", "fc00:114::514")
+        with raises(ValueError):
+            rule.Rule("IPCIDR6", "1.14.5.14")
 
     def test_type_checking_runtime(self):
         test_rule = rule.Rule()
@@ -37,6 +41,14 @@ def test_type_checking_runtime(self):
         with raises(ValueError):
             test_rule.set_payload("1919810")
 
+        test_rule.set_type("IPCIDR")
+        with raises(ValueError):
+            test_rule.set_payload("fc00:114::514")
+
+        test_rule.set_type("IPCIDR6")
+        with raises(ValueError):
+            test_rule.set_payload("1.14.5.14")
+
     def test_to_str(self):
         test_rule = rule.Rule("DomainSuffix", "example.com", "TEST")
         assert str(test_rule) == 'Type: "DomainSuffix", Payload: "example.com", Tag: TEST'

Utils/rule.py

@@ -1,4 +1,4 @@
-from ipaddress import ip_network
+from ipaddress import ip_network, IPv4Network, IPv6Network
 
 
 class Rule:
@@ -37,9 +37,13 @@ def set_payload(self, payload: str):
                 raise ValueError(f"Invalid domain: {payload}")
         elif "IP" in self.Type:
             try:
-                ip_network(payload)
+                ip_type = ip_network(payload)
             except ValueError:
                 raise ValueError(f"Invalid IP address: {payload}")
+            if self.Type == "IPCIDR6" and type(ip_type) is IPv4Network:
+                raise ValueError(f"IPv4 address stored in IPv6 type: {payload}")
+            elif self.Type == "IPCIDR" and type(ip_type) is IPv6Network:
+                raise ValueError(f"IPv6 address stored in IPv4 type: {payload}")
         self.Payload = payload
 
     def set_tag(self, tag: str = ""):