From: Ralf Jung Date: Sun, 9 Nov 2014 17:44:54 +0000 (+0100) Subject: count serials; automagic TTLs X-Git-Url: https://git.ralfj.de/zonemaker.git/commitdiff_plain/d1f1548719487b0107be8befe090483401abbe52?ds=sidebyside count serials; automagic TTLs --- diff --git a/.gitignore b/.gitignore index 087ec42..e941f95 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,2 @@ __pycache__ -ralfj.de.py +ralfj.de.* diff --git a/zone-maker b/zone-maker index 2d52a78..57f8afe 100755 --- a/zone-maker +++ b/zone-maker @@ -15,7 +15,7 @@ def make_zone(filename: str) -> None: zonefile = load_module(os.path.basename(filename), filename) zones = zonefile.__zones__ # type: Sequence[Zone] for zone in zones: - zone.write(sys.stdout) + zone.write() if __name__ == "__main__": for name in sys.argv[1:]: diff --git a/zonemaker/zone.py b/zonemaker/zone.py index c30cab1..e5e6641 100644 --- a/zonemaker/zone.py +++ b/zonemaker/zone.py @@ -17,15 +17,6 @@ def check_hostname(name: str) -> str: return name raise Exception(name+" is not a valid hostname") -def abs_hostname(name: str, root: str = None) -> str: - if name.endswith('.'): - return name - if root is None: - raise Exception("Name {0} was expected to be absolute, but it is not".format(name)) - if not root.endswith('.'): - raise Exception("Root {0} was expected to be absolute, but it is not".format(name)) - return name+"."+root - def time(time: int) -> str: if time == 0: return "0" @@ -70,22 +61,15 @@ class Delegation(): def __init__(self, NS: str, DS: str = None) -> None: pass -class RR(): - def __init__(self, owner: str, TTL: int, recordType: str, data: str) -> None: - assert owner.endswith('.') # users have to make this absolute first - self._owner = owner - self._TTL = TTL - self._recordType = recordType - self._data = data - - def __str__(self) -> str: - return "{0}\t{1}\t{2}\t{3}".format(self._owner, self._TTL, self._recordType, self._data) - class Zone: - def __init__(self, name: str, mail: str, NS: List[str], + def __init__(self, name: str, serialfile: str, dbfile: str, mail: str, NS: List[str], secondary_refresh: int, secondary_retry: int, secondary_expire: int, NX_TTL: int = None, A_TTL: int = None, other_TTL: int = None, domains: Dict[str, Any] = {}) -> None: + self._serialfile = serialfile + self._dbfile = dbfile + + if not name.endswith('.'): raise Exception("Expected an absolute hostname") self._name = check_hostname(name) if not mail.endswith('.'): raise Exception("Mail must be absolute, end with a dot") atpos = mail.find('@') @@ -97,20 +81,67 @@ class Zone: self._retry = secondary_retry self._expire = secondary_expire - assert other_TTL is not None - self._NX_TTL = other_TTL if NX_TTL is None else NX_TTL - self._A_TTL = other_TTL if A_TTL is None else A_TTL + if other_TTL is None: raise Exception("Must give other_TTL") + self._NX_TTL = NX_TTL + self._A_TTL = A_TTL self._other_TTL = other_TTL + + self._domains = domains + + def RR(self, owner: str, recordType: str, data: str) -> str: + '''generate given RR, in textual representation''' + assert re.match(r'^[A-Z]+$', recordType), "got invalid record type" + # figure out TTL + attrname = "_"+recordType+"_TTL" + TTL = None # type: int + if hasattr(self, attrname): + TTL = getattr(self, attrname) + if TTL is None: + TTL = self._other_TTL + # be done + return "{0}\t{1}\t{2}\t{3}".format(self.abs_hostname(owner), TTL, recordType, data) + + def abs_hostname(self, name): + if name.endswith('.'): + return name + return name+"."+self._name - def generate_rrs(self) -> Iterator[RR]: - serial = -1 - yield (RR(abs_hostname(self._name), self._other_TTL, 'SOA', - '{NS} {mail} ({serial} {refresh} {retry} {expire} {NX_TTL})'.format( - NS=self._NS[0], mail=self._mail, serial=serial, - refresh=time(self._refresh), retry=time(self._retry), expire=time(self._expire), - NX_TTL=time(self._NX_TTL)) - )) + def inc_serial(self) -> int: + # get serial + cur_serial = 0 + try: + with open(self._serialfile) as f: + cur_serial = int(f.read()) + except FileNotFoundError: + pass + # increment serial + cur_serial += 1 + # save serial + with open(self._serialfile, 'w') as f: + f.write(str(cur_serial)) + # be done + return cur_serial + + def generate_rrs(self) -> Iterator: + # SOA record + serial = self.inc_serial() + yield self.RR(self._name, 'SOA', + ('{NS} {mail} ({serial} {refresh} {retry} {expire} {NX_TTL}) ; '+ + '(serial refresh retry expire NX_TTL)').format( + NS=self.abs_hostname(self._NS[0]), mail=self._mail, serial=serial, + refresh=time(self._refresh), retry=time(self._retry), expire=time(self._expire), + NX_TTL=time(self._NX_TTL)) + ) + # NS records + for ns in self._NS: + yield self.RR(self._name, 'NS', self.abs_hostname(ns)) + + # all the rest + for name in sorted(self._domains.keys(), key=lambda s: list(reversed(s.split('.')))): + print(name) + #for rr in self._domains[name].generate_rrs(self): + #yield rr - def write(self, file:Any) -> None: + def write(self) -> None: for rr in self.generate_rrs(): print(rr)