chiark / gitweb /
hippotatd: --ownsource-local etc. options
[hippotat.git] / hippotatd
index 3adf9b313f27a89f81fd122facc5dfb491f5364b..0ef6369559c295b90d69e45aa78e0ce4129e7886 100755 (executable)
--- a/hippotatd
+++ b/hippotatd
@@ -1,17 +1,53 @@
 #!/usr/bin/python3
+#
+# Hippotat - Asinine IP Over HTTP program
+# ./hippotatd - server main program
+#
+# Copyright 2017 Ian Jackson
+#
+# AGPLv3+ + CAFv2+
+#
+#    This program is free software: you can redistribute it and/or
+#    modify it under the terms of the GNU Affero General Public
+#    License as published by the Free Software Foundation, either
+#    version 3 of the License, or (at your option) any later version,
+#    with the "CAF Login Exception" as published by Ian Jackson
+#    (version 2, or at your option any later version) as an Additional
+#    Permission.
+#
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+#    Affero General Public License for more details.
+#
+#    You should have received a copy of the GNU Affero General Public
+#    License and the CAF Login Exception along with this program, in
+#    the file AGPLv3+CAFv2.  If not, email Ian Jackson
+#    <ijackson@chiark.greenend.org.uk>.
+
 
 from hippotatlib import *
 
 import os
+import tempfile
+import atexit
+import shutil
 
 import twisted.internet
 from twisted.web.server import NOT_DONE_YET
 
+import twisted.web.static
+
+import hippotatlib.ownsource
+from hippotatlib.ownsource import SourceShipmentPreparer
+
 #import twisted.web.server import Site
 #from twisted.web.resource import Resource
 
 import syslog
 
+cleanups = [ ]
+
 clients = { }
 
 #---------- "router" ----------
@@ -198,26 +234,65 @@ class IphttpResource(NotStupidResource):
     log_debug(DBG.HTTP_CTRL, '...', idof=id(request))
     return NOT_DONE_YET
 
-  def render_GET(self, request):
-    log_debug(DBG.HTTP, 'GET request')
-    return b'<html><body>hippotat</body></html>'
+  # instantiator should set
+  # self.hippotat_sources = (source_names[0], source_names[1])
+  def __init__(self):
+    self.hippotat_sources = [None, None]
+    super().__init__()
 
-class SourceResource(NotStupidResource):
   def render_GET(self, request):
-    return b'<html><body>SUBDIR</body></html>'
+    log_debug(DBG.HTTP, 'GET request')
+    s = '<html><body>hippotat\n'
+    (s0,s1) = self.hippotat_sources
+    if s0:
+      s += '<p><a href="%s">source</a>\n' % s0
+      if self.hippotat_sources[1]:
+        s += ('(and that of dependency <a href="%s">packages</a>)\n' % s1)
+      s += 'available'
+    else:
+      s += 'TESTING'
+    s += '</body></html>'
+    return s.encode('utf-8')
 
 def start_http():
   resource = IphttpResource()
-  resource.putChild(b'source',SourceResource())
   site = twisted.web.server.Site(resource)
+
   for sa in c.saddrs:
     ep = sa.make_endpoint()
     crash_on_defer(ep.listen(site))
     log_debug(DBG.INIT, 'listening on %s' % sa)
 
+  td = tempfile.mkdtemp()
+
+  def cleanup():
+    try: shutil.rmtree(td)
+    except FileNotFoundError: pass
+    
+  cleanups.append(cleanup)
+
+  ssp = SourceShipmentPreparer(td)
+  ssp.logger = partial(log_debug, DBG.OWNSOURCE)
+  if DBG.OWNSOURCE in debug_set: ssp.stream_debug = sys.stdout
+  ssp.download_packages = opts.ownsource >= 2
+  if opts.ownsource >= 1: ssp.generate()
+
+  for ix in (0,1):
+    bn = ssp.output_names[ix]
+    op = ssp.output_paths[ix]
+    if op is None: continue
+    resource.hippotat_sources[ix] = bn
+    subresource =twisted.web.static.File(op)
+    resource.putChild(bn.encode('utf-8'), subresource)
+
+  reactor.callLater(0.1, (lambda: log.info('hippotatd started', dflag=False)))
+
 #---------- config and setup ----------
-        
-def process_cfg(putative_servers, putative_clients):
+
+def process_cfg(_opts, putative_servers, putative_clients):
+  global opts
+  opts = _opts
+
   global c
   c = ConfigResults()
   c.server = cfg.get('SERVER','server')
@@ -249,7 +324,37 @@ def process_cfg(putative_servers, putative_clients):
                     ('peer', 'vrelay'),
                     ('rnets','vnetwork')))
 
+def catch_termination():
+  def run_cleanups():
+    for cleanup in cleanups:
+      cleanup()
+
+  atexit.register(run_cleanups)
+
+  def signal_handler(name, sig, *args):
+    signal.signal(sig, signal.SIG_DFL)
+    print('exiting due to %s' % name, file=sys.stderr)
+    run_cleanups()
+    os.kill(os.getpid(), sig)
+    raise RuntimeError('did not die due to signal %s !' % name)
+
+  for sig in (signal.SIGINT, signal.SIGTERM):
+    signal.signal(sig, partial(signal_handler, sig.name))
+
+optparser.add_option('--ownsource', default=2,
+                     action='store_const', dest='ownsource', const=2,
+                     help='source download fully enabled (default)')
+
+optparser.add_option('--ownsource-local',
+                     action='store_const', dest='ownsource', const=1,
+                     help='source download is local source code only')
+
+optparser.add_option('--no-ownsource',
+                     action='store_const', dest='ownsource', const=0,
+                     help='source download disabled (for testing only)')
+
 common_startup(process_cfg)
+catch_termination()
 ipif = start_ipif(c.ipif_command, (lambda p,s,d: route(p,"[ipif]",s,d)))
 start_http()
 common_run()