66 lines
2.8 KiB
Python
66 lines
2.8 KiB
Python
from __future__ import print_function
|
|
from twisted.python import log
|
|
from twisted.internet import reactor, endpoints
|
|
from twisted.application import service
|
|
from twisted.web import server, static, resource
|
|
from .endpoint_service import ServerEndpointService
|
|
from wormhole import __version__
|
|
from .database import get_db
|
|
from .relay_server import Relay
|
|
from .transit_server import Transit
|
|
|
|
class Root(resource.Resource):
|
|
# child_FOO is a nevow thing, not a twisted.web.resource thing
|
|
def __init__(self):
|
|
resource.Resource.__init__(self)
|
|
self.putChild(b"", static.Data(b"Wormhole Relay\n", "text/plain"))
|
|
|
|
class PrivacyEnhancedSite(server.Site):
|
|
logRequests = True
|
|
def log(self, request):
|
|
if self.logRequests:
|
|
return server.Site.log(self, request)
|
|
|
|
class RelayServer(service.MultiService):
|
|
def __init__(self, relayport, transitport, advertise_version,
|
|
db_url=":memory:", blur_usage=None):
|
|
service.MultiService.__init__(self)
|
|
self._blur_usage = blur_usage
|
|
self.db = get_db(db_url)
|
|
welcome = {
|
|
"current_version": __version__,
|
|
# adding .motd will cause all clients to display the message,
|
|
# then keep running normally
|
|
#"motd": "Welcome to the public relay.\nPlease enjoy this service.",
|
|
#
|
|
# adding .error will cause all clients to fail, with this message
|
|
#"error": "This server has been disabled, see URL for details.",
|
|
}
|
|
if advertise_version:
|
|
welcome["current_version"] = advertise_version
|
|
self.root = Root()
|
|
site = PrivacyEnhancedSite(self.root)
|
|
if blur_usage:
|
|
site.logRequests = False
|
|
r = endpoints.serverFromString(reactor, relayport)
|
|
self.relayport_service = ServerEndpointService(r, site)
|
|
self.relayport_service.setServiceParent(self)
|
|
self.relay = Relay(self.db, welcome, blur_usage) # accessible from tests
|
|
self.relay.setServiceParent(self) # for the pruning timer
|
|
self.root.putChild(b"wormhole-relay", self.relay)
|
|
if transitport:
|
|
self.transit = Transit(self.db, blur_usage)
|
|
self.transit.setServiceParent(self) # for the timer
|
|
t = endpoints.serverFromString(reactor, transitport)
|
|
self.transport_service = ServerEndpointService(t, self.transit)
|
|
self.transport_service.setServiceParent(self)
|
|
|
|
def startService(self):
|
|
service.MultiService.startService(self)
|
|
log.msg("Wormhole relay server (Rendezvous and Transit) running")
|
|
if self._blur_usage:
|
|
log.msg("blurring access times to %d seconds" % self._blur_usage)
|
|
log.msg("not logging HTTP requests")
|
|
else:
|
|
log.msg("not blurring access times")
|