aboutsummaryrefslogtreecommitdiffstats
path: root/bin/get_catalog
blob: 772d5b4e4bf051c581586e8d8f79f506cbf0fdaf (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
#! /usr/bin/env python3
# vim:set noet sw=8 ts=8 sts=8:

import sys, socket
import dns.query
import dns.zone
import dns.rdatatype

def resolve(hostname):
	try:
		addrs = socket.getaddrinfo(
				hostname, 53, proto=socket.IPPROTO_UDP)
		return list(dict.fromkeys([addr[4][0] for addr in addrs]))
	except Exception as e:
		raise (Exception(f"resolving {hostname}: {e}"))

def axfr(zone, hostname):
	addrs = resolve(hostname)
	last_exc = None

	for addr in addrs:
		try:
			return dns.zone.from_xfr(dns.query.xfr(addr, zone))
		except Exception as e:
			last_exc = e
	raise (Exception(str(last_exc)))

def print_catalog(zone):
	for name, node in zone.nodes.items():
		for rdata in node.rdatasets:
			if rdata.rdtype == dns.rdatatype.PTR:
				for ptr in rdata:
					print(str(ptr.target).rstrip('.'))


def main():
	if len(sys.argv) != 3:
		print(f"Usage: {sys.argv[0]} <zone> <server>")
		return 1

	zone_name = sys.argv[1]
	server = sys.argv[2]

	try:
		zone = axfr(zone_name, server)
		print_catalog(zone)
	except Exception as e:
		print(f"Zone transfer failed: {e}")
		return 1

	return 0

if __name__ == "__main__":
	sys.exit(main())