Feat/dns loc records

boefjes/boefjes/plugins/kat_dns/main.py

@@ -26,6 +26,8 @@
     "CNAME",
     "DNAME",
     "SOA",
+    "LOC",
+    "GPOS",
 }
 
 

boefjes/boefjes/plugins/kat_dns/normalize.py

@@ -6,6 +6,8 @@
 from dns.message import Message, from_text
 from dns.rdtypes.ANY.CAA import CAA
 from dns.rdtypes.ANY.CNAME import CNAME
+from dns.rdtypes.ANY.GPOS import GPOS
+from dns.rdtypes.ANY.LOC import LOC
 from dns.rdtypes.ANY.MX import MX
 from dns.rdtypes.ANY.NS import NS
 from dns.rdtypes.ANY.SOA import SOA
@@ -21,6 +23,8 @@
     DNSARecord,
     DNSCAARecord,
     DNSCNAMERecord,
+    DNSGPOSRecord,
+    DNSLOCRecord,
     DNSMXRecord,
     DNSNSRecord,
     DNSRecord,
@@ -29,6 +33,7 @@
 )
 from octopoes.models.ooi.dns.zone import DNSZone, Hostname
 from octopoes.models.ooi.email_security import DKIMExists, DMARCTXTRecord
+from octopoes.models.ooi.geography import GeographicPoint
 from octopoes.models.ooi.network import IPAddressV4, IPAddressV6, Network
 
 
@@ -159,6 +164,35 @@ def register_record(record: DNSRecord) -> DNSRecord:
                     default_args["value"] = record_value[2]
                     register_record(DNSCAARecord(**default_args))
 
+                if isinstance(rr, LOC | GPOS):
+                    default_args.update(
+                        {
+                            "value": rr.to_text(),
+                            "latitude": rr.float_lontitude(),
+                            "longitude": rr.float_longtitude(),
+                            "altitude": rr.float_altitude(),
+                        }
+                    )
+
+                    if isinstance(rr, LOC):
+                        default_args.update(
+                            {
+                                "horizontal_precision": rr.horizontal_precision(),
+                                "vertical_precision": rr.vertical_precision(),
+                                "size": rr.size(),
+                            }
+                        )
+                        location_record = register_record(DNSLOCRecord(**default_args))
+                    else:
+                        location_record = register_record(DNSGPOSRecord(**default_args))
+
+                    geo_point = {
+                        "ooi": location_record.reference,
+                        "latitude": rr.float_lontitude(),
+                        "longitude": rr.float_longtitude(),
+                    }
+                    register_record(GeographicPoint(**geo_point))
+
     # link the hostnames to their discovered zones
     for hostname_, zone in zone_links.items():
         hostname_store[hostname_].dns_zone = zone.reference

octopoes/octopoes/models/ooi/dns/records.py

@@ -10,7 +10,7 @@
 
 class DNSRecord(OOI):
     hostname: Reference = ReferenceField(Hostname, max_issue_scan_level=0, max_inherit_scan_level=2)
-    dns_record_type: Literal["A", "AAAA", "CAA", "CNAME", "MX", "NS", "PTR", "SOA", "SRV", "TXT"]
+    dns_record_type: Literal["A", "AAAA", "CAA", "CNAME", "MX", "NS", "PTR", "SOA", "SRV", "TXT", "LOC", "GPOS"]
     value: str
     ttl: int | None = None  # todo: validation
 
@@ -185,3 +185,35 @@ class DNSCAARecord(DNSRecord):
     # without interior spaces or (2) a quoted string.
     value: str
     _natural_key_attrs = ["hostname", "flags", "tag", "value"]
+
+
+class DNSLocation(DNSRecord):
+    latitude: float | None = None
+    longitude: float | None = None
+    altitude: float | None = None
+
+    @property
+    def natural_key(self) -> str:
+        sha = hashlib.sha1(self.value.encode()).hexdigest()
+        key = super().natural_key
+        return key.replace(self.value, sha)
+
+
+class DNSGPOSRecord(DNSLocation):
+    # RFC 1712
+    object_type: Literal["DNSGPOSRecord"] = "DNSGPOSRecord"
+    dns_record_type: Literal["GPOS"] = "GPOS"
+
+    _reverse_relation_names = {"hostname": "dns_gpos_records"}
+
+
+class DNSLOCRecord(DNSLocation):
+    # RFC 1876
+    object_type: Literal["DNSLOCRecord"] = "DNSLOCRecord"
+    dns_record_type: Literal["LOC"] = "LOC"
+
+    horizontal_precision: int | None = None
+    vertical_precision: int | None = None
+    size: int | None = None
+
+    _reverse_relation_names = {"hostname": "dns_loc_records"}

octopoes/octopoes/models/types.py

@@ -17,6 +17,8 @@
     DNSARecord,
     DNSCAARecord,
     DNSCNAMERecord,
+    DNSGPOSRecord,
+    DNSLOCRecord,
     DNSMXRecord,
     DNSNSRecord,
     DNSPTRRecord,
@@ -93,6 +95,8 @@
     | DNSSOARecord
     | DNSCNAMERecord
     | DNSCAARecord
+    | DNSLOCRecord
+    | DNSGPOSRecord
     | ResolvedHostname
     | NXDOMAIN
 )