+#---------- packet queue ----------
+
+class PacketQueue():
+ def __init__(self, desc, max_queue_time):
+ self._desc = desc
+ assert(desc + '')
+ self._max_queue_time = max_queue_time
+ self._pq = collections.deque() # packets
+
+ def _log(self, dflag, msg, **kwargs):
+ log_debug(dflag, self._desc+' pq: '+msg, **kwargs)
+
+ def append(self, packet):
+ self._log(DBG.QUEUE, 'append', d=packet)
+ self._pq.append((time.monotonic(), packet))
+
+ def nonempty(self):
+ self._log(DBG.QUEUE, 'nonempty ?')
+ while True:
+ try: (queuetime, packet) = self._pq[0]
+ except IndexError:
+ self._log(DBG.QUEUE, 'nonempty ? empty.')
+ return False
+
+ age = time.monotonic() - queuetime
+ if age > self._max_queue_time:
+ # strip old packets off the front
+ self._log(DBG.QUEUE, 'dropping (old)', d=packet)
+ self._pq.popleft()
+ continue
+
+ self._log(DBG.QUEUE, 'nonempty ? nonempty.')
+ return True
+
+ def process(self, sizequery, moredata, max_batch):
+ # sizequery() should return size of batch so far
+ # moredata(s) should add s to batch
+ self._log(DBG.QUEUE, 'process...')
+ while True:
+ try: (dummy, packet) = self._pq[0]
+ except IndexError:
+ self._log(DBG.QUEUE, 'process... empty')
+ break
+
+ self._log(DBG.QUEUE_CTRL, 'process... packet', d=packet)
+
+ encoded = slip.encode(packet)
+ sofar = sizequery()
+
+ self._log(DBG.QUEUE_CTRL,
+ 'process... (sofar=%d, max=%d) encoded' % (sofar, max_batch),
+ d=encoded)
+
+ if sofar > 0:
+ if sofar + len(slip.delimiter) + len(encoded) > max_batch:
+ self._log(DBG.QUEUE_CTRL, 'process... overflow')
+ break
+ moredata(slip.delimiter)
+
+ moredata(encoded)
+ self._pq.popleft()
+
+#---------- error handling ----------
+
+_crashing = False
+
+def crash(err):
+ global _crashing
+ _crashing = True
+ print('CRASH ', err, file=sys.stderr)
+ try: reactor.stop()
+ except twisted.internet.error.ReactorNotRunning: pass
+
+def crash_on_defer(defer):
+ defer.addErrback(lambda err: crash(err))
+
+def crash_on_critical(event):
+ if event.get('log_level') >= LogLevel.critical:
+ crash(twisted.logger.formatEvent(event))
+
+#---------- config processing ----------
+
+def process_cfg_common_always():
+ global mtu
+ c.mtu = cfg.get('virtual','mtu')
+
+def process_cfg_ipif(section, varmap):
+ for d, s in varmap:
+ try: v = getattr(c, s)
+ except AttributeError: continue
+ setattr(c, d, v)
+
+ #print(repr(c))
+
+ c.ipif_command = cfg.get(section,'ipif', vars=c.__dict__)
+
+def process_cfg_network():
+ c.network = ipnetwork(cfg.get('virtual','network'))
+ if c.network.num_addresses < 3 + 2:
+ raise ValueError('network needs at least 2^3 addresses')
+
+def process_cfg_server():
+ try:
+ c.server = cfg.get('virtual','server')
+ except NoOptionError:
+ process_cfg_network()
+ c.server = next(c.network.hosts())
+
+class ServerAddr():
+ def __init__(self, port, addrspec):
+ self.port = port
+ # also self.addr
+ try:
+ self.addr = ipaddress.IPv4Address(addrspec)
+ self._endpointfactory = twisted.internet.endpoints.TCP4ServerEndpoint
+ self._inurl = b'%s'
+ except AddressValueError:
+ self.addr = ipaddress.IPv6Address(addrspec)
+ self._endpointfactory = twisted.internet.endpoints.TCP6ServerEndpoint
+ self._inurl = b'[%s]'
+ def make_endpoint(self):
+ return self._endpointfactory(reactor, self.port, self.addr)
+ def url(self):
+ url = b'http://' + (self._inurl % str(self.addr).encode('ascii'))
+ if self.port != 80: url += b':%d' % self.port
+ url += b'/'
+ return url
+
+def process_cfg_saddrs():
+ try: port = cfg.getint('server','port')
+ except NoOptionError: port = 80
+
+ c.saddrs = [ ]
+ for addrspec in cfg.get('server','addrs').split():
+ sa = ServerAddr(port, addrspec)
+ c.saddrs.append(sa)
+
+def process_cfg_clients(constructor):
+ c.clients = [ ]
+ for cs in cfg.sections():
+ if not (':' in cs or '.' in cs): continue
+ ci = ipaddr(cs)
+ pw = cfg.get(cs, 'password')
+ pw = pw.encode('utf-8')
+ constructor(ci,cs,pw)
+
+#---------- startup ----------
+
+def common_startup():
+ log_formatter = twisted.logger.formatEventAsClassicLogText
+ log_observer = twisted.logger.FileLogObserver(sys.stderr, log_formatter)
+ twisted.logger.globalLogBeginner.beginLoggingTo(
+ [ log_observer, crash_on_critical ]
+ )
+
+ optparser.add_option('-c', '--config', dest='configfile',
+ default='/etc/hippotat/config')
+ (opts, args) = optparser.parse_args()
+ if len(args): optparser.error('no non-option arguments please')
+
+ re = regexp.compile('#.*')
+ cfg.read_string(re.sub('', defcfg))
+ cfg.read(opts.configfile)
+
+def common_run():
+ log_debug(DBG.INIT, 'entering reactor')
+ if not _crashing: reactor.run()
+ print('CRASHED (end)', file=sys.stderr)