mdw@git.distorted.org.uk Git - hippotat/blob - server

   1 #!/usr/bin/python2
   2
   3 from twisted.web.server import Site
   4 from twisted.web.resource import Resource
   5 from twisted.web.server import NOT_DONE_YET
   6 from twisted.internet import reactor
   7
   8 import ConfigParser
   9 import ipaddress
  10
  11 clients = { }
  12
  13 def ipaddress(input):
  14   try:
  15     r = ipaddress.IPv4Address(input)
  16   except AddressValueError:
  17     r = ipaddress.IPv6Address(input)
  18   return r
  19
  20 def ipnetwork(input):
  21   try:
  22     r = ipaddress.IPv4Network(input)
  23   except NetworkValueError:
  24     r = ipaddress.IPv6Network(input)
  25   return r
  26
  27 defcfg = u'''
  28 [default]
  29 max_batch_down: 65536
  30 max_queue_time: 10
  31 max_request_time: 54
  32
  33 [global]
  34 max_batch_down: 262144
  35 max_queue_time: 121
  36 max_request_time: 121
  37 '''
  38
  39 class Client():
  40   def __init__(ip, cs):
  41     # instance data members
  42     self._ip = ip
  43     self._cs = cs
  44     self.pw = cfg.get(cs, 'password')
  45     # plus:
  46     #  .cfg[<config-key>]
  47     self.cfg = { }
  48     for k in ('max_batch_down','max_queue_time','max_request_time'):
  49       req = cfg.getint(cs, k)
  50       limit = cfg.getint('global',k)
  51       self.cfg[k] = min(req, limit)
  52
  53     def process_arriving_data(d):
  54
  55
  56 def process_cfg():
  57   global network
  58   global ourself
  59
  60   network = ipnetwork(cfg.get('virtual','network'))
  61   try:
  62     ourself = cfg.get('virtual','server')
  63   except ConfigParser.NoOptionError:
  64     ourself = network.hosts().next()
  65
  66   for cs in cfg.sections():
  67     if not (':' in cs or '.' in cs): continue
  68     ci = ipaddress(cs)
  69     if ci not in network:
  70       raise ValueError('client %s not in network' % ci)
  71     if ci in clients:
  72       raise ValueError('multiple client cfg sections for %s' % ci)
  73     clients[ci] = Client(ci, cs)
  74
  75 class FormPage(Resource):
  76   def render_POST(self, request):
  77     # find client, update config, etc.
  78     ci = ipaddress(request.args['i'])
  79     c = clients[ci]
  80     pw = request.args['pw']
  81     if pw != c.pw: raise ValueError('bad password')
  82
  83     # update config
  84     for r, w in (('mbd', 'max_batch_down'),
  85                  ('mqt', 'max_queue_time'),
  86                  ('mrt', 'max_request_time')):
  87       try: v = request.args[r]
  88       except KeyError: continue
  89       v = int(v)
  90       c.cfg[w] = v
  91
  92     try: d = request.args['d']
  93     except KeyError: d = ''
  94
  95     c.process_arriving_data(d)
  96
  97     reactor.