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

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