apt_dht/HTTPServer.py

   1
   2 from twisted.python import log
   3 from twisted.internet import defer
   4 from twisted.web2 import server, http, resource, channel
   5 from twisted.web2 import static, http_headers, responsecode
   6
   7 class FileDownloader(static.File):
   8
   9     def __init__(self, path, manager, defaultType="text/plain", ignoredExts=(), processors=None, indexNames=None):
  10         self.manager = manager
  11         super(FileDownloader, self).__init__(path, defaultType, ignoredExts, processors, indexNames)
  12
  13     def renderHTTP(self, req):
  14         log.msg('Got request for %s from %s' % (req.uri, req.remoteAddr))
  15         resp = super(FileDownloader, self).renderHTTP(req)
  16         if isinstance(resp, defer.Deferred):
  17             resp.addCallback(self._renderHTTP_done, req)
  18         else:
  19             resp = self._renderHTTP_done(resp, req)
  20         return resp
  21
  22     def _renderHTTP_done(self, resp, req):
  23         log.msg('Initial response to %s: %r' % (req.uri, resp))
  24
  25         if self.manager:
  26             path = 'http:/' + req.uri
  27             if resp.code >= 200 and resp.code < 400:
  28                 return self.manager.check_freshness(req, path, resp.headers.getHeader('Last-Modified'), resp)
  29
  30             log.msg('Not found, trying other methods for %s' % req.uri)
  31             return self.manager.get_resp(req, path)
  32
  33         return resp
  34
  35     def createSimilarFile(self, path):
  36         return self.__class__(path, self.manager, self.defaultType, self.ignoredExts,
  37                               self.processors, self.indexNames[:])
  38
  39
  40 class TopLevel(resource.Resource):
  41     addSlash = True
  42
  43     def __init__(self, directory, manager):
  44         self.directory = directory
  45         self.manager = manager
  46         self.subdirs = {}
  47         self.factory = None
  48
  49     def getHTTPFactory(self):
  50         if self.factory is None:
  51             self.factory = channel.HTTPFactory(server.Site(self),
  52                                                **{'maxPipeline': 10,
  53                                                   'betweenRequestsTimeOut': 60})
  54         return self.factory
  55
  56     def setDirectories(self, dirs):
  57         self.subdirs = {}
  58         for k in dirs:
  59             # Don't allow empty subdirectory
  60             if k:
  61                 self.subdirs[k] = dirs[k]
  62         log.msg('new subdirectories initialized')
  63
  64     def render(self, ctx):
  65         return http.Response(
  66             200,
  67             {'content-type': http_headers.MimeType('text', 'html')},
  68             """<html><body>
  69             <h2>Statistics</h2>
  70             <p>TODO: eventually some stats will be shown here.</body></html>""")
  71
  72     def locateChild(self, request, segments):
  73         name = segments[0]
  74         if name in self.subdirs:
  75             log.msg('Sharing %s with %s' % (request.uri, request.remoteAddr))
  76             return static.File(self.subdirs[name].path), segments[1:]
  77
  78         if request.remoteAddr.host != "127.0.0.1":
  79             log.msg('Blocked illegal access to %s from %s' % (request.uri, request.remoteAddr))
  80             return None, ()
  81
  82         if len(name) > 1:
  83             return FileDownloader(self.directory.path, self.manager), segments[0:]
  84         else:
  85             return self, ()
  86
  87 if __name__ == '__builtin__':
  88     # Running from twistd -y
  89     t = TopLevel('/home', None)
  90     t.setDirectories({'~1': '/tmp', '~2': '/var/log'})
  91     factory = t.getHTTPFactory()
  92
  93     # Standard twisted application Boilerplate
  94     from twisted.application import service, strports
  95     application = service.Application("demoserver")
  96     s = strports.service('tcp:18080', factory)
  97     s.setServiceParent(application)