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

   1 #!/usr/bin/python3
   2
   3 from hippotat import *
   4
   5 client_cs = None
   6
   7 def set_client(ci,cs,pw):
   8   global client_cs
   9   global password
  10   assert(client_cs is None)
  11   client_cs = cs
  12   c.client = ci
  13   c.max_outstanding = cfg.getint(cs, 'max_requests_outstanding')
  14   c.target_outstanding = cfg.getint(cs, 'target_requests_outstanding')
  15   password = pw
  16
  17 def process_cfg():
  18   global url
  19   global max_requests_outstanding
  20
  21   process_cfg_common_always()
  22   process_cfg_server()
  23
  24   try:
  25     c.url = cfg.get('server','url')
  26   except NoOptionError:
  27     process_cfg_saddrs()
  28     sa = c.saddrs[0].url()
  29
  30   process_cfg_clients(set_client)
  31
  32   c.routes = cfg.get('virtual','routes')
  33   c.max_queue_time = cfg.getint(client_cs, 'max_queue_time')
  34   c.max_batch_up   = cfg.getint(client_cs, 'max_batch_up')
  35
  36   process_cfg_ipif(client_cs,
  37                    (('local', 'client'),
  38                     ('peer',  'server'),
  39                     ('rnets', 'routes')))
  40
  41 outstanding = 0
  42
  43 def start_client():
  44   global queue
  45   global agent
  46   queue = PacketQueue(c.max_queue_time)
  47   agent = twisted.web.client.Agent(reactor, connectTimeout = c.http_timeout)
  48
  49 def outbound(packet, saddr, daddr):
  50   #print('OUT ', saddr, daddr, repr(packet))
  51   queue.append(packet)
  52   check_outbound()
  53
  54 def req_ok(resp):
  55
  56
  57 def req_err(err):
  58   print(err, >>sys.stderr)
  59
  60 def req_fin(*args):
  61   outstanding--
  62
  63 def check_outbound():
  64   while True:
  65     if                         outstanding >= c.max_outstanding   : break
  66     if not queue.nonempty() && outstanding >= c.target_outstanding: break
  67
  68     d = b''
  69     queue.process((lambda: len(d)),
  70                   (lambda s: d += s),
  71                   c.max_batch_up)
  72     assert(len(d))
  73
  74     crlf = b'\r\n'
  75     mime = (b'--b'                                      + crlf +
  76             b'Content-Disposition: form-data; name="m"' + crlf +
  77             password                                    + crlf +
  78             c.client                                    + crlf +
  79             c.target_outstanding                        + crlf +
  80             b'--b'                                      + crlf +
  81             b'Content-Disposition: form-data; name="d"' + crlf +
  82             mime_translate(d)                           + crlf +
  83             b'--b--'                                    + crlf)
  84
  85     hh = { 'User-Agent': ['hippotat'],
  86            'Content-Type': ['multipart/form-data; boundary="b"'] }
  87     req = agent.request(b'POST',
  88                         c.url,
  89                         twisted.web.client.Headers(hh))
  90     req.addTimeout(c.http_timeout,
  91     req.addCallbacks(req_ok, req_err)
  92     req.addBoth(req_fin)
  93     outstanding++
  94
  95 common_startup()
  96 process_cfg()
  97 start_client()
  98 start_ipif(c.ipif_command, outbound)
  99 common_run()