def got_request_headers(self, stream, request):
""" Filter incoming HTTP requests """
auth = request["Authorization"]
if not auth:
return False
if auth not in self.negotiator.peers:
return False
return DASHServerSmpl.got_request_headers(self, stream, request)
def main(args):
""" Main function """
try:
options, arguments = getopt.getopt(args[1:], "6A:b:d:flnp:v")
except getopt.error:
sys.exit(USAGE)
if arguments:
sys.exit(USAGE)
prefer_ipv6 = 0
address = "127.0.0.1"
backend = "volatile"
datadir = None # means: pick the default
force = 0
listen = 0
negotiate = 1
port = 80
noisy = 0
for name, value in options:
if name == "-6":
prefer_ipv6 = 1
elif name == "-A":
address = value
elif name == "-b":
backend = value
elif name == "-d":
datadir = value
elif name == "-f":
force = 1
elif name == "-l":
listen = 1
elif name == "-n":
negotiate = 0
elif name == "-p":
port = int(value)
elif name == "-v":
noisy = 1
if noisy:
log.set_verbose()
conf = CONFIG.copy()
BACKEND.use_backend(backend)
BACKEND.datadir_init(None, datadir)
if listen:
if not negotiate:
server = DASHServerSmpl(POLLER)
server.configure(conf)
server.listen((address, port))
else:
# Code adapted from neubot/server.py
conf["http.server.rootdir"] = ""
server = ServerHTTP(POLLER)
server.configure(conf)
server.listen((address, port))
negotiate_server = NegotiateServer(POLLER)
negotiate_server.configure(conf)
server.register_child(negotiate_server, "/negotiate")
server.register_child(negotiate_server, "/collect")
dash_negotiate_server = DASHNegotiateServer()
negotiate_server.register_module("dash", dash_negotiate_server)
dash_server = DASHServerGlue(POLLER, dash_negotiate_server)
dash_server.configure(conf)
server.register_child(dash_server, "/dash")
elif not force:
result = runner_clnt.runner_client(CONFIG["agent.api.address"],
CONFIG["agent.api.port"], CONFIG["verbose"], "dash")
if result:
sys.exit(0)
logging.warning("dash: failed to contact Neubot; is Neubot running?")
sys.exit(1)
else:
if negotiate:
client = DASHNegotiateClient(POLLER)
else:
client = DASHClientSmpl(POLLER, None, DASH_RATES)
client.configure(conf)
client.connect((address, port))
POLLER.loop()
def __init__(self, poller, negotiator):
DASHServerSmpl.__init__(self, poller)
self.negotiator = negotiator
