xp-dev.com Git - nip.io/blob - src/backend.py

   1 #!/usr/bin/python
   2
   3 import ConfigParser
   4 import os
   5 import re
   6 import sys
   7
   8 DEBUG = 0
   9
  10
  11 def log(msg):
  12     sys.stderr.write('backend (%s): %s\n' % (os.getpid(), msg))
  13
  14
  15 def write(*l):
  16     args = len(l)
  17     c = 0
  18     for a in l:
  19         c += 1
  20         if DEBUG:
  21             log('writing: %s' % a)
  22         sys.stdout.write(a)
  23         if c < args:
  24             if DEBUG:
  25                 log('writetab')
  26             sys.stdout.write('\t')
  27     if DEBUG:
  28         log('writenewline')
  29     sys.stdout.write('\n')
  30     sys.stdout.flush()
  31
  32
  33 def get_next():
  34     if DEBUG:
  35         log('reading now')
  36     l = sys.stdin.readline()
  37     if DEBUG:
  38         log('read line: %s' % l)
  39     return l.strip().split('\t')
  40
  41
  42 class DynamicBackend:
  43     def __init__(self):
  44         self.id = ''
  45         self.soa = ''
  46         self.domain = ''
  47         self.ip_address = ''
  48         self.ttl = ''
  49         self.name_servers = {}
  50
  51     def configure(self):
  52         fname = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'backend.conf')
  53         if not os.path.exists(fname):
  54             log('%s does not exist' % fname)
  55             sys.exit(1)
  56
  57         fp = open(fname)
  58         config = ConfigParser.ConfigParser()
  59         config.readfp(fp)
  60         fp.close()
  61
  62         self.id = config.get('soa', 'id')
  63         self.soa = '%s %s %s' % (config.get('soa', 'ns'), config.get('soa', 'hostmaster'), self.id)
  64         self.domain = config.get('main', 'domain')
  65         self.ip_address = config.get('main', 'ipaddress')
  66         self.ttl = config.get('main', 'ttl')
  67
  68         for entry in config.items('nameservers'):
  69             self.name_servers[entry[0]] = entry[1]
  70
  71         log('Name servers: %s' % self.name_servers)
  72         log('ID: %s' % self.id)
  73         log('TTL %s' % self.ttl)
  74         log('SOA: %s' % self.soa)
  75         log('IP Address: %s' % self.ip_address)
  76         log('DOMAIN: %s' % self.domain)
  77
  78     def run(self):
  79         log('starting up')
  80         handshake = get_next()
  81         if handshake[1] != '1':
  82             log('Not version 1: %s' % handshake)
  83             sys.exit(1)
  84         write('OK', 'We are good')
  85         log('Done handshake')
  86
  87         while True:
  88             cmd = get_next()
  89             if DEBUG:
  90                 log(cmd)
  91
  92             if len(cmd) < 6:
  93                 log('did not understand: %s' % cmd)
  94                 write('FAIL')
  95                 continue
  96
  97             qname = cmd[1].lower()
  98             qtype = cmd[3]
  99
 100             if (qtype == 'A' or qtype == 'ANY') and qname.endswith(self.domain):
 101                 if qname == self.domain:
 102                     self.handle_self(self.domain)
 103                 elif qname in self.name_servers:
 104                     self.handle_nameservers(qname)
 105                 else:
 106                     self.handle_subdomains(qname)
 107             elif qtype == 'SOA' and qname.endswith(self.domain):
 108                 self.handle_soa(qname)
 109             else:
 110                 self.handle_unknown(qtype, qname)
 111
 112     def handle_self(self, name):
 113         write('DATA', name, 'IN', 'A', self.ttl, self.id, self.ip_address)
 114         self.write_name_servers(name)
 115         write('END')
 116
 117     def handle_subdomains(self, qname):
 118         subdomain = qname[0:qname.find(self.domain) - 1]
 119
 120         subparts = subdomain.split('.')
 121         if len(subparts) < 4:
 122             if DEBUG:
 123                 log('subparts less than 4')
 124             self.handle_self(qname)
 125             return
 126
 127         ipaddress = subparts[-4:]
 128         if DEBUG:
 129             log('ip: %s' % ipaddress)
 130         for part in ipaddress:
 131             if re.match('^\d{1,3}$', part) is None:
 132                 if DEBUG:
 133                     log('%s is not a number' % part)
 134                 self.handle_self(qname)
 135                 return
 136             parti = int(part)
 137             if parti < 0 or parti > 255:
 138                 if DEBUG:
 139                     log('%d is too big/small' % parti)
 140                 self.handle_self(qname)
 141                 return
 142
 143         write('DATA', qname, 'IN', 'A', self.ttl, self.id, '%s.%s.%s.%s' % (ipaddress[0], ipaddress[1], ipaddress[2], ipaddress[3]))
 144         self.write_name_servers(qname)
 145         write('END')
 146
 147     def handle_nameservers(self, qname):
 148         ip = self.name_servers[qname]
 149         write('DATA', qname, 'IN', 'A', self.ttl, self.id, ip)
 150         write('END')
 151
 152     def write_name_servers(self, qname):
 153         for nameServer in self.name_servers:
 154             write('DATA', qname, 'IN', 'NS', self.ttl, self.id, nameServer)
 155
 156     def handle_soa(self, qname):
 157         write('DATA', qname, 'IN', 'SOA', self.ttl, self.id, self.soa)
 158         write('END')
 159
 160     def handle_unknown(self, qtype, qname):
 161         write('LOG', 'Unknown type: %s, domain: %s' % (qtype, qname))
 162         write('END')
 163
 164
 165 if __name__ == '__main__':
 166     backend = DynamicBackend()
 167     backend.configure()
 168     backend.run()