import sys
+from zope.interface import implementer
+
import twisted
from twisted.internet import reactor
import twisted.internet.endpoints
from configparser import ConfigParser
from configparser import NoOptionError
+from functools import partial
+
import collections
import time
import codecs
+import traceback
import re as regexp
import hippotat.slip as slip
class DBG(twisted.python.constants.Names):
+ INIT = NamedConstant()
ROUTE = NamedConstant()
+ DROP = NamedConstant()
FLOW = NamedConstant()
HTTP = NamedConstant()
- HTTP_CTRL = NamedConstant()
- INIT = NamedConstant()
+ TWISTED = NamedConstant()
QUEUE = NamedConstant()
+ HTTP_CTRL = NamedConstant()
QUEUE_CTRL = NamedConstant()
+ HTTP_FULL = NamedConstant()
+ CTRL_DUMP = NamedConstant()
+ SLIP_FULL = NamedConstant()
+
+_hex_codec = codecs.getencoder('hex_codec')
-_hexcodec = codecs.getencoder('hex_codec')
+#---------- logging ----------
+
+org_stderr = sys.stderr
log = twisted.logger.Logger()
+debug_set = set()
+debug_def_detail = DBG.HTTP
+
def log_debug(dflag, msg, idof=None, d=None):
+ if dflag not in debug_set: return
+ #print('---------------->',repr((dflag, msg, idof, d)), file=sys.stderr)
if idof is not None:
- msg = '[%d] %s' % (id(idof), msg)
+ msg = '[%#x] %s' % (id(idof), msg)
if d is not None:
- d = d[0:64]
- d = _hex_codec(d)[0]
+ #d = d[0:64]
+ d = _hex_codec(d)[0].decode('ascii')
msg += ' ' + d
log.info('{dflag} {msgcore}', dflag=dflag, msgcore=msg)
+@implementer(twisted.logger.ILogFilterPredicate)
+class LogNotBoringTwisted:
+ def __call__(self, event):
+ yes = twisted.logger.PredicateResult.yes
+ no = twisted.logger.PredicateResult.no
+ return yes
+ try:
+ if event.get('log_level') != LogLevel.info:
+ return yes
+ try:
+ dflag = event.get('dflag')
+ except KeyError:
+ dflag = DBG.TWISTED
+ return yes if (dflag in debug_set) else no
+ except Exception:
+ print(traceback.format_exc(), file=org_stderr)
+ return yes
+
+#---------- default config ----------
+
defcfg = '''
[DEFAULT]
#[<client>] overrides
max_batch_down = 65536 # used by server, subject to [limits]
max_queue_time = 10 # used by server, subject to [limits]
-max_request_time = 54 # used by server, subject to [limits]
target_requests_outstanding = 3 # must match; subject to [limits] on server
+http_timeout = 30 # used by both } must be
+http_timeout_grace = 5 # used by both } compatible
max_requests_outstanding = 4 # used by client
max_batch_up = 4000 # used by client
-http_timeout = 30 # used by client
http_retry = 5 # used by client
#[server] or [<client>] overrides
[limits]
max_batch_down = 262144 # used by server
max_queue_time = 121 # used by server
-max_request_time = 121 # used by server
+http_timeout = 121 # used by server
target_requests_outstanding = 10 # used by server
'''
c = ConfigResults()
-def log_discard(packet, saddr, daddr, why):
-
- Print('Drop ', Saddr, Daddr, why)
-# syslog.syslog(syslog.LOG_DEBUG,
-# 'discarded packet %s -> %s (%s)' % (saddr, daddr, why))
+def log_discard(packet, iface, saddr, daddr, why):
+ log_debug(DBG.DROP,
+ 'discarded packet [%s] %s -> %s: %s' % (iface, saddr, daddr, why),
+ d=packet)
#---------- packet parsing ----------
#---------- ipif (SLIP) subprocess ----------
class SlipStreamDecoder():
- def __init__(self, on_packet):
- # we will call packet(<packet>)
+ def __init__(self, desc, on_packet):
self._buffer = b''
self._on_packet = on_packet
+ self._desc = desc
+ self._log('__init__')
+
+ def _log(self, msg, **kwargs):
+ log_debug(DBG.SLIP_FULL, 'slip %s: %s' % (self._desc, msg), **kwargs)
def inputdata(self, data):
- #print('SLIP-GOT ', repr(data))
- self._buffer += data
- packets = slip.decode(self._buffer)
+ self._log('inputdata', d=data)
+ packets = slip.decode(data)
+ packets[0] = self._buffer + packets[0]
self._buffer = packets.pop()
for packet in packets:
self._maybe_packet(packet)
+ self._log('bufremain', d=self._buffer)
def _maybe_packet(self, packet):
- if len(packet):
- self._on_packet(packet)
+ self._log('maybepacket', d=packet)
+ if len(packet):
+ self._on_packet(packet)
def flush(self):
+ self._log('flush')
self._maybe_packet(self._buffer)
self._buffer = b''
class _IpifProcessProtocol(twisted.internet.protocol.ProcessProtocol):
def __init__(self, router):
self._router = router
- self._decoder = SlipStreamDecoder(self.slip_on_packet)
+ self._decoder = SlipStreamDecoder('ipif', self.slip_on_packet)
def connectionMade(self): pass
def outReceived(self, data):
self._decoder.inputdata(data)
def slip_on_packet(self, packet):
(saddr, daddr) = packet_addrs(packet)
if saddr.is_link_local or daddr.is_link_local:
- log_discard(packet, saddr, daddr, 'link-local')
+ log_discard(packet, 'ipif', saddr, daddr, 'link-local')
return
self._router(packet, saddr, daddr)
def processEnded(self, status):
env=None)
def queue_inbound(packet):
+ log_debug(DBG.FLOW, "queue_inbound", d=packet)
ipif.transport.write(slip.delimiter)
ipif.transport.write(slip.encode(packet))
ipif.transport.write(slip.delimiter)
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)
+ def _log(self, dflag, msg, **kwargs):
log_debug(dflag, self._desc+' pq: '+msg, **kwargs)
def append(self, packet):
self._log(DBG.QUEUE_CTRL,
'process... (sofar=%d, max=%d) encoded' % (sofar, max_batch),
- d=encoded
+ d=encoded)
if sofar > 0:
if sofar + len(slip.delimiter) + len(encoded) > max_batch:
#---------- error handling ----------
+_crashing = False
+
def crash(err):
- print('CRASH ', err, file=sys.stderr)
+ global _crashing
+ _crashing = True
+ print('========== CRASH ==========', err,
+ '===========================', file=sys.stderr)
try: reactor.stop()
except twisted.internet.error.ReactorNotRunning: pass
except AttributeError: continue
setattr(c, d, v)
- print(repr(c))
+ #print(repr(c))
c.ipif_command = cfg.get(section,'ipif', vars=c.__dict__)
#---------- 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')
+
+ def ds_by_detail(od,os,detail_level,op):
+ global debug_set
+ debug_set = set([df for df in DBG.iterconstants() if df <= detail_level])
+
+ def ds_one(mutator,df, od,os,value,op):
+ mutator(df)
+
+ optparser.add_option('-D', '--debug',
+ default=debug_def_detail.name,
+ type='choice',
+ choices=[dl.name for dl in DBG.iterconstants()],
+ action='callback',
+ callback= ds_by_detail)
+
+ optparser.add_option('--no-debug',
+ nargs=0,
+ action='callback',
+ callback= partial(ds_by_detail,DBG.INIT))
+
+ for df in DBG.iterconstants():
+ optparser.add_option('--debug-'+df.name,
+ action='callback',
+ callback= partial(ds_one, debug_set.add, df))
+ optparser.add_option('--no-debug-'+df.name,
+ action='callback',
+ callback= partial(ds_one, debug_set.discard, df))
+
(opts, args) = optparser.parse_args()
if len(args): optparser.error('no non-option arguments please')
cfg.read_string(re.sub('', defcfg))
cfg.read(opts.configfile)
+ log_formatter = twisted.logger.formatEventAsClassicLogText
+ stdout_obs = twisted.logger.FileLogObserver(sys.stdout, log_formatter)
+ stderr_obs = twisted.logger.FileLogObserver(sys.stderr, log_formatter)
+ pred = twisted.logger.LogLevelFilterPredicate(LogLevel.error)
+ stdsomething_obs = twisted.logger.FilteringLogObserver(
+ stderr_obs, [pred], stdout_obs
+ )
+ log_observer = twisted.logger.FilteringLogObserver(
+ stdsomething_obs, [LogNotBoringTwisted()]
+ )
+ #log_observer = stdsomething_obs
+ twisted.logger.globalLogBeginner.beginLoggingTo(
+ [ log_observer, crash_on_critical ]
+ )
+
def common_run():
- log_debug(DBG.INIT, 'ready')
- reactor.run()
+ log_debug(DBG.INIT, 'entering reactor')
+ if not _crashing: reactor.run()
print('CRASHED (end)', file=sys.stderr)
~ian / hippotat.git / blobdiff