2015-07-12 20:36:46 +02:00
|
|
|
import logging
|
|
|
|
import time
|
|
|
|
import cgi
|
2015-09-13 23:17:13 +02:00
|
|
|
import socket
|
2015-12-28 00:19:58 +01:00
|
|
|
import sys
|
2017-01-23 12:54:47 +01:00
|
|
|
import gevent
|
2015-07-12 20:36:46 +02:00
|
|
|
|
2015-01-12 02:03:45 +01:00
|
|
|
from gevent.pywsgi import WSGIServer
|
|
|
|
from gevent.pywsgi import WSGIHandler
|
2019-03-15 21:06:59 +01:00
|
|
|
from geventwebsocket.handler import WebSocketHandler
|
2015-07-12 20:36:46 +02:00
|
|
|
|
2019-03-15 21:06:59 +01:00
|
|
|
from .UiRequest import UiRequest
|
2015-01-12 02:03:45 +01:00
|
|
|
from Site import SiteManager
|
|
|
|
from Config import config
|
2015-01-17 18:50:56 +01:00
|
|
|
from Debug import Debug
|
2019-03-15 21:06:59 +01:00
|
|
|
import importlib
|
2015-01-12 02:03:45 +01:00
|
|
|
|
version 0.2.7, plugin system, multiuser plugin for zeroproxies, reworked imports, cookie parse, stats moved to plugin, usermanager class, dont generate site auth on listing, multiline notifications, allow server side prompt from user, update script keep plugins disabled status
2015-03-24 01:33:09 +01:00
|
|
|
|
2015-01-12 02:03:45 +01:00
|
|
|
# Skip websocket handler if not necessary
|
|
|
|
class UiWSGIHandler(WSGIHandler):
|
2015-07-12 20:36:46 +02:00
|
|
|
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
|
|
self.server = args[2]
|
|
|
|
super(UiWSGIHandler, self).__init__(*args, **kwargs)
|
|
|
|
self.args = args
|
|
|
|
self.kwargs = kwargs
|
|
|
|
|
|
|
|
def run_application(self):
|
|
|
|
if "HTTP_UPGRADE" in self.environ: # Websocket request
|
|
|
|
try:
|
|
|
|
ws_handler = WebSocketHandler(*self.args, **self.kwargs)
|
|
|
|
ws_handler.__dict__ = self.__dict__ # Match class variables
|
|
|
|
ws_handler.run_application()
|
2019-03-15 21:06:59 +01:00
|
|
|
except Exception as err:
|
2015-07-12 20:36:46 +02:00
|
|
|
logging.error("UiWSGIHandler websocket error: %s" % Debug.formatException(err))
|
|
|
|
if config.debug: # Allow websocket errors to appear on /Debug
|
2019-04-15 12:31:33 +02:00
|
|
|
import main
|
|
|
|
main.DebugHook.handleError()
|
2015-07-12 20:36:46 +02:00
|
|
|
else: # Standard HTTP request
|
|
|
|
try:
|
|
|
|
super(UiWSGIHandler, self).run_application()
|
2019-03-15 21:06:59 +01:00
|
|
|
except Exception as err:
|
2015-07-12 20:36:46 +02:00
|
|
|
logging.error("UiWSGIHandler error: %s" % Debug.formatException(err))
|
|
|
|
if config.debug: # Allow websocket errors to appear on /Debug
|
2019-04-15 12:31:33 +02:00
|
|
|
import main
|
|
|
|
main.DebugHook.handleError()
|
2015-09-13 23:17:13 +02:00
|
|
|
|
|
|
|
def handle(self):
|
|
|
|
# Save socket to be able to close them properly on exit
|
|
|
|
self.server.sockets[self.client_address] = self.socket
|
|
|
|
super(UiWSGIHandler, self).handle()
|
|
|
|
del self.server.sockets[self.client_address]
|
2015-01-12 02:03:45 +01:00
|
|
|
|
|
|
|
|
|
|
|
class UiServer:
|
2015-02-20 01:37:12 +01:00
|
|
|
|
2015-07-12 20:36:46 +02:00
|
|
|
def __init__(self):
|
|
|
|
self.ip = config.ui_ip
|
|
|
|
self.port = config.ui_port
|
2019-04-15 15:45:07 +02:00
|
|
|
self.running = False
|
2015-07-12 20:36:46 +02:00
|
|
|
if self.ip == "*":
|
2017-06-15 19:42:38 +02:00
|
|
|
self.ip = "0.0.0.0" # Bind all
|
2017-06-15 19:48:01 +02:00
|
|
|
if config.ui_host:
|
|
|
|
self.allowed_hosts = set(config.ui_host)
|
|
|
|
elif config.ui_ip == "127.0.0.1":
|
2018-12-18 14:36:14 +01:00
|
|
|
# IP Addresses are inherently allowed as they are immune to DNS
|
|
|
|
# rebinding attacks.
|
|
|
|
self.allowed_hosts = set(["zero", "localhost:%s" % config.ui_port])
|
2018-06-02 12:24:00 +02:00
|
|
|
# "URI producers and normalizers should omit the port component and
|
|
|
|
# its ':' delimiter if port is empty or if its value would be the
|
|
|
|
# same as that of the scheme's default."
|
|
|
|
# Source: https://tools.ietf.org/html/rfc3986#section-3.2.3
|
|
|
|
# As a result, we need to support portless hosts if port 80 is in
|
|
|
|
# use.
|
|
|
|
if config.ui_port == 80:
|
2018-12-18 14:36:14 +01:00
|
|
|
self.allowed_hosts.update(["localhost"])
|
2017-06-15 19:48:01 +02:00
|
|
|
else:
|
|
|
|
self.allowed_hosts = set([])
|
2018-06-02 14:17:10 +02:00
|
|
|
self.allow_trans_proxy = config.ui_trans_proxy
|
2017-06-15 19:48:01 +02:00
|
|
|
|
2015-09-10 23:25:09 +02:00
|
|
|
self.wrapper_nonces = []
|
2017-07-15 01:32:15 +02:00
|
|
|
self.add_nonces = []
|
2017-11-05 23:40:59 +01:00
|
|
|
self.websockets = []
|
2017-02-11 18:21:18 +01:00
|
|
|
self.site_manager = SiteManager.site_manager
|
2015-07-12 20:36:46 +02:00
|
|
|
self.sites = SiteManager.site_manager.list()
|
|
|
|
self.log = logging.getLogger(__name__)
|
|
|
|
|
2016-03-06 02:17:58 +01:00
|
|
|
# After WebUI started
|
|
|
|
def afterStarted(self):
|
|
|
|
from util import Platform
|
|
|
|
Platform.setMaxfilesopened(config.max_files_opened)
|
|
|
|
|
2015-07-12 20:36:46 +02:00
|
|
|
# Handle WSGI request
|
|
|
|
def handleRequest(self, env, start_response):
|
2019-03-15 23:58:18 +01:00
|
|
|
path = bytes(env["PATH_INFO"], "raw-unicode-escape").decode("utf8")
|
2015-07-12 20:36:46 +02:00
|
|
|
if env.get("QUERY_STRING"):
|
|
|
|
get = dict(cgi.parse_qsl(env['QUERY_STRING']))
|
|
|
|
else:
|
|
|
|
get = {}
|
|
|
|
ui_request = UiRequest(self, get, env, start_response)
|
|
|
|
if config.debug: # Let the exception catched by werkezung
|
|
|
|
return ui_request.route(path)
|
|
|
|
else: # Catch and display the error
|
|
|
|
try:
|
|
|
|
return ui_request.route(path)
|
2019-03-15 21:06:59 +01:00
|
|
|
except Exception as err:
|
2015-07-12 20:36:46 +02:00
|
|
|
logging.debug("UiRequest error: %s" % Debug.formatException(err))
|
|
|
|
return ui_request.error500("Err: %s" % Debug.formatException(err))
|
|
|
|
|
|
|
|
# Reload the UiRequest class to prevent restarts in debug mode
|
|
|
|
def reload(self):
|
|
|
|
global UiRequest
|
|
|
|
import imp
|
|
|
|
import sys
|
2019-03-15 21:06:59 +01:00
|
|
|
importlib.reload(sys.modules["User.UserManager"])
|
|
|
|
importlib.reload(sys.modules["Ui.UiWebsocket"])
|
2015-07-12 20:36:46 +02:00
|
|
|
UiRequest = imp.load_source("UiRequest", "src/Ui/UiRequest.py").UiRequest
|
|
|
|
# UiRequest.reload()
|
|
|
|
|
|
|
|
# Bind and run the server
|
|
|
|
def start(self):
|
2019-04-15 15:45:07 +02:00
|
|
|
self.running = True
|
2015-07-12 20:36:46 +02:00
|
|
|
handler = self.handleRequest
|
|
|
|
|
|
|
|
if config.debug:
|
|
|
|
# Auto reload UiRequest on change
|
|
|
|
from Debug import DebugReloader
|
2019-03-16 02:42:43 +01:00
|
|
|
DebugReloader.watcher.addCallback(self.reload)
|
2015-07-12 20:36:46 +02:00
|
|
|
|
|
|
|
# Werkzeug Debugger
|
|
|
|
try:
|
|
|
|
from werkzeug.debug import DebuggedApplication
|
|
|
|
handler = DebuggedApplication(self.handleRequest, evalex=True)
|
2019-03-15 21:06:59 +01:00
|
|
|
except Exception as err:
|
2015-07-12 20:36:46 +02:00
|
|
|
self.log.info("%s: For debugging please download Werkzeug (http://werkzeug.pocoo.org/)" % err)
|
|
|
|
from Debug import DebugReloader
|
|
|
|
self.log.write = lambda msg: self.log.debug(msg.strip()) # For Wsgi access.log
|
|
|
|
self.log.info("--------------------------------------")
|
2019-04-03 13:56:14 +02:00
|
|
|
if ":" in config.ui_ip:
|
|
|
|
self.log.info("Web interface: http://[%s]:%s/" % (config.ui_ip, config.ui_port))
|
|
|
|
else:
|
|
|
|
self.log.info("Web interface: http://%s:%s/" % (config.ui_ip, config.ui_port))
|
2015-07-12 20:36:46 +02:00
|
|
|
self.log.info("--------------------------------------")
|
|
|
|
|
2018-07-10 03:37:38 +02:00
|
|
|
if config.open_browser and config.open_browser != "False":
|
2015-07-12 20:36:46 +02:00
|
|
|
logging.info("Opening browser: %s...", config.open_browser)
|
|
|
|
import webbrowser
|
2018-07-10 03:37:38 +02:00
|
|
|
try:
|
|
|
|
if config.open_browser == "default_browser":
|
|
|
|
browser = webbrowser.get()
|
|
|
|
else:
|
|
|
|
browser = webbrowser.get(config.open_browser)
|
|
|
|
url = "http://%s:%s/%s" % (config.ui_ip if config.ui_ip != "*" else "127.0.0.1", config.ui_port, config.homepage)
|
|
|
|
gevent.spawn_later(0.3, browser.open, url, new=2)
|
|
|
|
except Exception as err:
|
2019-03-15 21:06:59 +01:00
|
|
|
print("Error starting browser: %s" % err)
|
2015-07-12 20:36:46 +02:00
|
|
|
|
2017-06-15 19:43:23 +02:00
|
|
|
self.server = WSGIServer((self.ip, self.port), handler, handler_class=UiWSGIHandler, log=self.log)
|
2015-07-12 20:36:46 +02:00
|
|
|
self.server.sockets = {}
|
2016-03-06 02:17:58 +01:00
|
|
|
self.afterStarted()
|
2015-12-28 00:19:58 +01:00
|
|
|
try:
|
|
|
|
self.server.serve_forever()
|
2019-03-15 21:06:59 +01:00
|
|
|
except Exception as err:
|
2015-12-28 00:19:58 +01:00
|
|
|
self.log.error("Web interface bind error, must be running already, exiting.... %s" % err)
|
2019-04-15 12:31:33 +02:00
|
|
|
import main
|
|
|
|
main.file_server.stop()
|
2015-07-12 20:36:46 +02:00
|
|
|
self.log.debug("Stopped.")
|
|
|
|
|
|
|
|
def stop(self):
|
|
|
|
self.log.debug("Stopping...")
|
|
|
|
# Close WS sockets
|
|
|
|
if "clients" in dir(self.server):
|
2019-03-15 21:06:59 +01:00
|
|
|
for client in list(self.server.clients.values()):
|
2015-07-12 20:36:46 +02:00
|
|
|
client.ws.close()
|
|
|
|
# Close http sockets
|
|
|
|
sock_closed = 0
|
2019-03-15 21:06:59 +01:00
|
|
|
for sock in list(self.server.sockets.values()):
|
2015-07-12 20:36:46 +02:00
|
|
|
try:
|
2019-03-15 21:06:59 +01:00
|
|
|
sock.send(b"bye")
|
2015-09-13 23:17:13 +02:00
|
|
|
sock.shutdown(socket.SHUT_RDWR)
|
2015-10-25 23:08:25 +01:00
|
|
|
# sock._sock.close()
|
|
|
|
# sock.close()
|
2015-07-12 20:36:46 +02:00
|
|
|
sock_closed += 1
|
2019-03-15 21:06:59 +01:00
|
|
|
except Exception as err:
|
2015-09-13 23:17:13 +02:00
|
|
|
self.log.debug("Http connection close error: %s" % err)
|
2015-07-12 20:36:46 +02:00
|
|
|
self.log.debug("Socket closed: %s" % sock_closed)
|
2015-09-13 23:17:13 +02:00
|
|
|
time.sleep(0.1)
|
2019-04-10 14:56:10 +02:00
|
|
|
if config.debug:
|
|
|
|
from Debug import DebugReloader
|
|
|
|
DebugReloader.watcher.stop()
|
2015-07-12 20:36:46 +02:00
|
|
|
|
|
|
|
self.server.socket.close()
|
|
|
|
self.server.stop()
|
2019-04-15 15:45:07 +02:00
|
|
|
self.running = False
|
2017-11-05 23:42:14 +01:00
|
|
|
time.sleep(1)
|
|
|
|
|
|
|
|
def updateWebsocket(self, **kwargs):
|
|
|
|
for ws in self.websockets:
|
2019-04-10 14:56:10 +02:00
|
|
|
ws.event("serverChanged", kwargs)
|