feat(pydantic): Add Port and PortRange

example_rules.py

@@ -30,14 +30,14 @@ filter:
     - src: mgmt
       protocols:
         tcp:
-          dport: "22,240..242"
+          dport: [22, 240..242]
       verdict: accept
     - src: backbone
       protocols:
         ospf: true
         vrrp: true
         tcp:
-            dport: 179
+            dport: [179]
       verdict: accept
     - protocols:
         icmp: true
@@ -50,7 +50,7 @@ filter:
     - src: users-internet-allowed
       protocols:
         tcp:
-          dport: 25
+          dport: [25]
       verdict: drop
     - src: users-internet-allowed
       dest:

nftables.py

@@ -9,6 +9,8 @@ from pydantic import (
     FilePath,
     IPvAnyAddress,
     IPvAnyNetwork,
+    conint,
+    parse_obj_as,
     validator,
     root_validator,
 )
@@ -19,18 +21,27 @@ class RestrictiveBaseModel(BaseModel, extra=Extra.forbid):
     pass
 
 
-def parse_range_string(s):
+# Ports
-    parts = s.split(",")
+Port = conint(ge=0, le=2**16)
-    values = []
 
-    for part in parts:
-        if ".." in part:
-            start, end = part.split("..")
-            values.append(range(int(start), int(end) + 1))
-        else:
-            values.append(int(part))
 
-    return values
+class PortRange(str):
+    @classmethod
+    def __get_validators__(cls):
+        yield cls.validate
+
+    @classmethod
+    def validate(cls, v):
+        try:
+            start, end = v.split("..")
+        except ValueError:
+            raise ValueError("invalid port range: must be in the form start..end")
+
+        start, end = parse_obj_as(Port, start), parse_obj_as(Port, end)
+        if start > end:
+            raise ValueError("invalid port range: start must be less than end")
+
+        return range(start, end)
 
 
 # Zones
@@ -73,21 +84,13 @@ class Verdict(str, Enum):
 
 
 class TcpProtocol(RestrictiveBaseModel):
-    dport: str | None
+    dport: list[Port | PortRange] | None
-    sport: str | None
+    sport: list[Port | PortRange] | None
-
-    @validator("dport", "sport")
-    def parse_range(cls, v):
-        return parse_range_string(v)
 
 
 class UdpProtocol(RestrictiveBaseModel):
-    dport: str | None
+    dport: list[Port | PortRange] | None
-    sport: str | None
+    sport: list[Port | PortRange] | None
-
-    @validator("dport", "sport")
-    def parse_range(cls, v):
-        return parse_range_string(v)
 
 
 class Protocols(RestrictiveBaseModel):