adelorenzo/nip.io
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add support for optional cnames

Browse Source
This commit is contained in:
Rene Moser 2019-09-16 12:44:39 +02:00
parent b945468ee2
commit cea40c90a7
2 changed files with 29 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/backend.conf.example
View File

@ -24,3 +24,6 @@ ns=ns1.example.com
ns1.example.com=127.0.0.1
ns2.example.com=127.0.0.1
[additional_cnames]
_acme-challenge=xyz.auth.example.com.
www=business.example.com

28
src/nip.py
View File

@ -62,6 +62,7 @@ class DynamicBackend:
self.ip_address = ''
self.ttl = ''
self.name_servers = {}
self.additional_cnames = {}
def configure(self):
fname = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'backend.conf')
@ -81,7 +82,19 @@ class DynamicBackend:
self.ttl = config.get('main', 'ttl')
for entry in config.items('nameservers'):
self.name_servers[entry[0]] = entry[1]
if not entry[0].endswith(self.domain):
name = entry[0] + '.' + self.domain
else:
name = entry[0]
self.name_servers[name] = entry[1]
for entry in config.items('additional_cnames') or []:
if not entry[0].endswith(self.domain):
name = entry[0] + '.' + self.domain
else:
name = entry[0]
self.additional_cnames[name] = entry[1]
log('add static CNAME: %s = %s' % (entry[0], entry[1]))
log('Name servers: %s' % self.name_servers)
log('ID: %s' % self.id)
@ -112,11 +125,13 @@ class DynamicBackend:
qname = cmd[1].lower()
qtype = cmd[3]
if (qtype == 'A' or qtype == 'ANY') and qname.endswith(self.domain):
if qtype in ('CNAME', 'A', 'ANY') and qname.endswith(self.domain):
if qname == self.domain:
self.handle_self(self.domain)
elif qname in self.name_servers:
self.handle_nameservers(qname)
elif qname in self.additional_cnames:
self.handle_additional_cnames(qname)
else:
self.handle_subdomains(qname)
elif qtype == 'SOA' and qname.endswith(self.domain):
@ -159,7 +174,16 @@ class DynamicBackend:
self.write_name_servers(qname)
write('END')
def handle_additional_cnames(self, qname):
if DEBUG:
log('Found CNAME: %s' % qname)
cname=self.additional_cnames[qname]
write('DATA', qname, 'IN', 'CNAME', self.ttl, self.id, cname)
write('END')
def handle_nameservers(self, qname):
if DEBUG:
log('Found name server: %s' % qname)
ip = self.name_servers[qname]
write('DATA', qname, 'IN', 'A', self.ttl, self.id, ip)
write('END')