#!/usr/bin/python2 from twisted.web.server import Site from twisted.web.resource import Resource from twisted.web.server import NOT_DONE_YET from twisted.internet import reactor import ConfigParser import ipaddress clients = { } def ipaddress(input): try: r = ipaddress.IPv4Address(input) except AddressValueError: r = ipaddress.IPv6Address(input) return r def ipnetwork(input): try: r = ipaddress.IPv4Network(input) except NetworkValueError: r = ipaddress.IPv6Network(input) return r defcfg = u''' [default] max_batch_down: 65536 max_queue_time: 10 max_request_time: 54 [global] max_batch_down: 262144 max_queue_time: 121 max_request_time: 121 ''' class Client(): def __init__(ip, cs): # instance data members self._ip = ip self._cs = cs self.pw = cfg.get(cs, 'password') # plus: # .cfg[] self.cfg = { } for k in ('max_batch_down','max_queue_time','max_request_time'): req = cfg.getint(cs, k) limit = cfg.getint('global',k) self.cfg[k] = min(req, limit) def process_arriving_data(d): def process_cfg(): global network global ourself network = ipnetwork(cfg.get('virtual','network')) try: ourself = cfg.get('virtual','server') except ConfigParser.NoOptionError: ourself = network.hosts().next() for cs in cfg.sections(): if not (':' in cs or '.' in cs): continue ci = ipaddress(cs) if ci not in network: raise ValueError('client %s not in network' % ci) if ci in clients: raise ValueError('multiple client cfg sections for %s' % ci) clients[ci] = Client(ci, cs) class FormPage(Resource): def render_POST(self, request): # find client, update config, etc. ci = ipaddress(request.args['i']) c = clients[ci] pw = request.args['pw'] if pw != c.pw: raise ValueError('bad password') # update config for r, w in (('mbd', 'max_batch_down'), ('mqt', 'max_queue_time'), ('mrt', 'max_request_time')): try: v = request.args[r] except KeyError: continue v = int(v) c.cfg[w] = v try: d = request.args['d'] except KeyError: d = '' c.process_arriving_data(d) reactor.