54 lines
2.2 KiB
Python
54 lines
2.2 KiB
Python
|
from __future__ import print_function
|
||
|
|
from twisted.internet import reactor, endpoints
|
||
|
|
from twisted.application import service, internet
|
||
|
|
from twisted.web import server, static, resource
|
||
|
|
from ..util.endpoint_service import ServerEndpointService
|
||
|
|
from .. import __version__
|
||
|
|
from ..database import get_db
|
||
|
|
from .relay import Relay
|
||
|
|
from .transit_server import Transit
|
||
|
|
|
||
|
|
SECONDS = 1.0
|
||
|
|
MINUTE = 60*SECONDS
|
||
|
|
HOUR = 60*MINUTE
|
||
|
|
EXPIRATION_CHECK_PERIOD = 2*HOUR
|
||
|
|
|
||
|
|
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 RelayServer(service.MultiService):
|
||
|
|
def __init__(self, relayport, transitport, advertise_version,
|
||
|
|
db_url=":memory:"):
|
||
|
|
service.MultiService.__init__(self)
|
||
|
|
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 = server.Site(self.root)
|
||
|
|
r = endpoints.serverFromString(reactor, relayport)
|
||
|
|
self.relayport_service = ServerEndpointService(r, site)
|
||
|
|
self.relayport_service.setServiceParent(self)
|
||
|
|
self.relay = Relay(self.db, welcome) # accessible from tests
|
||
|
|
self.root.putChild(b"wormhole-relay", self.relay)
|
||
|
|
t = internet.TimerService(EXPIRATION_CHECK_PERIOD,
|
||
|
|
self.relay.prune_old_channels)
|
||
|
|
t.setServiceParent(self)
|
||
|
|
if transitport:
|
||
|
|
self.transit = Transit()
|
||
|
|
self.transit.setServiceParent(self) # for the timer
|
||
|
|
t = endpoints.serverFromString(reactor, transitport)
|
||
|
|
self.transport_service = ServerEndpointService(t, self.transit)
|
||
|
|
self.transport_service.setServiceParent(self)
|