Fichero de zona DNS: comprobaciones y resolución de nombres

1import ipaddress
2import sys
3
4
5def main():
6    origen = "."
7    registros = []                      # (nombre absoluto, tipo, datos)
8    consultas = []
9    errores = []
10
11    def absoluto(nombre):
12        if nombre == "@":
13            return origen
14        return nombre if nombre.endswith(".") else f"{nombre}.{origen}"
15
16    for n, linea in enumerate(sys.stdin.read().split("\n"), start=1):
17        linea = linea.split(";")[0].strip()
18        if not linea:
19            continue
20        p = linea.split()
21        if p[0] == "$ORIGIN" and len(p) == 2:
22            origen = p[1] if p[1].endswith(".") else p[1] + "."
23            continue
24        if p[0] == "$TTL":
25            continue
26        if p[0] == "CONSULTA" and len(p) == 3:
27            consultas.append((absoluto(p[1]), p[2].upper()))
28            continue
29        nombre, resto = p[0], p[1:]
30        if resto and resto[0].isdigit():
31            resto = resto[1:]                       # TTL propio del registro
32        if resto and resto[0].upper() == "IN":
33            resto = resto[1:]
34        if not resto:
35            errores.append(f"línea {n}: no se entiende «{linea}»")
36            continue
37        tipo, datos = resto[0].upper(), resto[1:]
38        if tipo in ("A", "AAAA") and len(datos) == 1:
39            try:
40                ip = ipaddress.ip_address(datos[0])
41                if (tipo == "A") != (ip.version == 4):
42                    raise ValueError
43            except ValueError:
44                errores.append(f"línea {n}: «{datos[0]}» no es una dirección {'IPv4' if tipo == 'A' else 'IPv6'}")
45                continue
46            registros.append((absoluto(nombre), tipo, datos[0]))
47        elif tipo in ("CNAME", "NS") and len(datos) == 1:
48            registros.append((absoluto(nombre), tipo, absoluto(datos[0])))
49        elif tipo == "MX" and len(datos) == 2 and datos[0].isdigit():
50            registros.append((absoluto(nombre), tipo, (int(datos[0]), absoluto(datos[1]))))
51        elif tipo == "SOA" and len(datos) >= 3:
52            registros.append((absoluto(nombre), tipo, datos[2]))
53        elif tipo == "TXT" and datos:
54            registros.append((absoluto(nombre), tipo, " ".join(datos)))
55        else:
56            errores.append(f"línea {n}: no se entiende «{linea}»")
57
58    def de(nombre, tipo=None):
59        return [r for r in registros if r[0] == nombre and (tipo is None or r[1] == tipo)]
60
61    def en_zona(nombre):
62        return nombre == origen or nombre.endswith("." + origen)
63
64    soa = [r for r in registros if r[1] == "SOA"]
65    if len(soa) != 1:
66        errores.append(f"la zona debe tener exactamente un SOA y tiene {len(soa)}")
67    if not de(origen, "NS"):
68        errores.append(f"no hay registros NS para {origen}")
69    avisos = []
70    for nombre in dict.fromkeys(r[0] for r in registros):
71        tipos = [r[1] for r in de(nombre)]
72        if "CNAME" in tipos and len(tipos) > 1:
73            errores.append(f"{nombre} tiene un CNAME y otros registros ({', '.join(t for t in tipos if t != 'CNAME')})")
74    for _, tipo, dato in [r for r in registros if r[1] in ("NS", "MX")]:
75        destino = dato[1] if tipo == "MX" else dato
76        if de(destino, "CNAME"):
77            avisos.append(f"el {tipo} apunta a {destino}, que es un CNAME (debe apuntar a un nombre con A o AAAA)")
78        elif tipo == "NS" and en_zona(destino) and not (de(destino, "A") or de(destino, "AAAA")):
79            errores.append(f"falta el registro A de {destino}, que es servidor de nombres de la zona")
80    print(f"Zona {origen} (serie {soa[0][2] if len(soa) == 1 else '?'}, {len(registros)} registro{'' if len(registros) == 1 else 's'})")
81    print("Errores:" if errores else "Errores: ninguno")
82    for e in errores:
83        print(f"  - {e}")
84    for a in avisos:
85        print(f"Aviso: {a}")
86    for nombre, tipo in consultas:
87        print(f"? {nombre} {tipo}")
88        actual = nombre
89        vistos = set()
90        while True:
91            if not en_zona(actual):
92                print(f"  {actual} está fuera de la zona: lo resuelve otro servidor")
93                break
94            cname = de(actual, "CNAME")
95            if cname and tipo != "CNAME":
96                if actual in vistos:
97                    print("  bucle de CNAME")
98                    break
99                vistos.add(actual)
100                print(f"  {actual} CNAME {cname[0][2]}")
101                actual = cname[0][2]
102                continue
103            respuesta = de(actual, tipo)
104            if tipo == "MX":
105                respuesta = sorted(respuesta, key=lambda r: (r[2][0], r[2][1]))
106            for r in respuesta:
107                valor = f"{r[2][0]} {r[2][1]}" if tipo == "MX" else r[2]
108                print(f"  {actual} {tipo} {valor}")
109            if not respuesta:
110                print(f"  {actual}: sin registros {tipo} (el nombre existe)" if de(actual) else f"  {actual}: no existe (NXDOMAIN)")
111            break
112
113
114main()