forked from s434650/CatOrNot
863 lines
31 KiB
Python
863 lines
31 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
werkzeug.serving
|
||
|
|
~~~~~~~~~~~~~~~~
|
||
|
|
|
||
|
|
There are many ways to serve a WSGI application. While you're developing
|
||
|
|
it you usually don't want a full blown webserver like Apache but a simple
|
||
|
|
standalone one. From Python 2.5 onwards there is the `wsgiref`_ server in
|
||
|
|
the standard library. If you're using older versions of Python you can
|
||
|
|
download the package from the cheeseshop.
|
||
|
|
|
||
|
|
However there are some caveats. Sourcecode won't reload itself when
|
||
|
|
changed and each time you kill the server using ``^C`` you get an
|
||
|
|
`KeyboardInterrupt` error. While the latter is easy to solve the first
|
||
|
|
one can be a pain in the ass in some situations.
|
||
|
|
|
||
|
|
The easiest way is creating a small ``start-myproject.py`` that runs the
|
||
|
|
application::
|
||
|
|
|
||
|
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
from myproject import make_app
|
||
|
|
from werkzeug.serving import run_simple
|
||
|
|
|
||
|
|
app = make_app(...)
|
||
|
|
run_simple('localhost', 8080, app, use_reloader=True)
|
||
|
|
|
||
|
|
You can also pass it a `extra_files` keyword argument with a list of
|
||
|
|
additional files (like configuration files) you want to observe.
|
||
|
|
|
||
|
|
For bigger applications you should consider using `click`
|
||
|
|
(http://click.pocoo.org) instead of a simple start file.
|
||
|
|
|
||
|
|
|
||
|
|
:copyright: (c) 2014 by the Werkzeug Team, see AUTHORS for more details.
|
||
|
|
:license: BSD, see LICENSE for more details.
|
||
|
|
"""
|
||
|
|
from __future__ import with_statement
|
||
|
|
|
||
|
|
import io
|
||
|
|
import os
|
||
|
|
import socket
|
||
|
|
import sys
|
||
|
|
import signal
|
||
|
|
|
||
|
|
|
||
|
|
can_fork = hasattr(os, "fork")
|
||
|
|
|
||
|
|
|
||
|
|
try:
|
||
|
|
import termcolor
|
||
|
|
except ImportError:
|
||
|
|
termcolor = None
|
||
|
|
|
||
|
|
try:
|
||
|
|
import ssl
|
||
|
|
except ImportError:
|
||
|
|
class _SslDummy(object):
|
||
|
|
def __getattr__(self, name):
|
||
|
|
raise RuntimeError('SSL support unavailable')
|
||
|
|
ssl = _SslDummy()
|
||
|
|
|
||
|
|
|
||
|
|
def _get_openssl_crypto_module():
|
||
|
|
try:
|
||
|
|
from OpenSSL import crypto
|
||
|
|
except ImportError:
|
||
|
|
raise TypeError('Using ad-hoc certificates requires the pyOpenSSL '
|
||
|
|
'library.')
|
||
|
|
else:
|
||
|
|
return crypto
|
||
|
|
|
||
|
|
|
||
|
|
try:
|
||
|
|
import SocketServer as socketserver
|
||
|
|
from BaseHTTPServer import HTTPServer, BaseHTTPRequestHandler
|
||
|
|
except ImportError:
|
||
|
|
import socketserver
|
||
|
|
from http.server import HTTPServer, BaseHTTPRequestHandler
|
||
|
|
|
||
|
|
ThreadingMixIn = socketserver.ThreadingMixIn
|
||
|
|
|
||
|
|
if can_fork:
|
||
|
|
ForkingMixIn = socketserver.ForkingMixIn
|
||
|
|
else:
|
||
|
|
class ForkingMixIn(object):
|
||
|
|
pass
|
||
|
|
|
||
|
|
# important: do not use relative imports here or python -m will break
|
||
|
|
import werkzeug
|
||
|
|
from werkzeug._internal import _log
|
||
|
|
from werkzeug._compat import PY2, WIN, reraise, wsgi_encoding_dance
|
||
|
|
from werkzeug.urls import url_parse, url_unquote
|
||
|
|
from werkzeug.exceptions import InternalServerError
|
||
|
|
|
||
|
|
|
||
|
|
LISTEN_QUEUE = 128
|
||
|
|
can_open_by_fd = not WIN and hasattr(socket, 'fromfd')
|
||
|
|
|
||
|
|
|
||
|
|
class DechunkedInput(io.RawIOBase):
|
||
|
|
"""An input stream that handles Transfer-Encoding 'chunked'"""
|
||
|
|
|
||
|
|
def __init__(self, rfile):
|
||
|
|
self._rfile = rfile
|
||
|
|
self._done = False
|
||
|
|
self._len = 0
|
||
|
|
|
||
|
|
def readable(self):
|
||
|
|
return True
|
||
|
|
|
||
|
|
def read_chunk_len(self):
|
||
|
|
try:
|
||
|
|
line = self._rfile.readline().decode('latin1')
|
||
|
|
_len = int(line.strip(), 16)
|
||
|
|
except ValueError:
|
||
|
|
raise IOError('Invalid chunk header')
|
||
|
|
if _len < 0:
|
||
|
|
raise IOError('Negative chunk length not allowed')
|
||
|
|
return _len
|
||
|
|
|
||
|
|
def readinto(self, buf):
|
||
|
|
read = 0
|
||
|
|
while not self._done and read < len(buf):
|
||
|
|
if self._len == 0:
|
||
|
|
# This is the first chunk or we fully consumed the previous
|
||
|
|
# one. Read the next length of the next chunk
|
||
|
|
self._len = self.read_chunk_len()
|
||
|
|
|
||
|
|
if self._len == 0:
|
||
|
|
# Found the final chunk of size 0. The stream is now exhausted,
|
||
|
|
# but there is still a final newline that should be consumed
|
||
|
|
self._done = True
|
||
|
|
|
||
|
|
if self._len > 0:
|
||
|
|
# There is data (left) in this chunk, so append it to the
|
||
|
|
# buffer. If this operation fully consumes the chunk, this will
|
||
|
|
# reset self._len to 0.
|
||
|
|
n = min(len(buf), self._len)
|
||
|
|
buf[read:read + n] = self._rfile.read(n)
|
||
|
|
self._len -= n
|
||
|
|
read += n
|
||
|
|
|
||
|
|
if self._len == 0:
|
||
|
|
# Skip the terminating newline of a chunk that has been fully
|
||
|
|
# consumed. This also applies to the 0-sized final chunk
|
||
|
|
terminator = self._rfile.readline()
|
||
|
|
if terminator not in (b'\n', b'\r\n', b'\r'):
|
||
|
|
raise IOError('Missing chunk terminating newline')
|
||
|
|
|
||
|
|
return read
|
||
|
|
|
||
|
|
|
||
|
|
class WSGIRequestHandler(BaseHTTPRequestHandler, object):
|
||
|
|
|
||
|
|
"""A request handler that implements WSGI dispatching."""
|
||
|
|
|
||
|
|
@property
|
||
|
|
def server_version(self):
|
||
|
|
return 'Werkzeug/' + werkzeug.__version__
|
||
|
|
|
||
|
|
def make_environ(self):
|
||
|
|
request_url = url_parse(self.path)
|
||
|
|
|
||
|
|
def shutdown_server():
|
||
|
|
self.server.shutdown_signal = True
|
||
|
|
|
||
|
|
url_scheme = self.server.ssl_context is None and 'http' or 'https'
|
||
|
|
path_info = url_unquote(request_url.path)
|
||
|
|
|
||
|
|
environ = {
|
||
|
|
'wsgi.version': (1, 0),
|
||
|
|
'wsgi.url_scheme': url_scheme,
|
||
|
|
'wsgi.input': self.rfile,
|
||
|
|
'wsgi.errors': sys.stderr,
|
||
|
|
'wsgi.multithread': self.server.multithread,
|
||
|
|
'wsgi.multiprocess': self.server.multiprocess,
|
||
|
|
'wsgi.run_once': False,
|
||
|
|
'werkzeug.server.shutdown': shutdown_server,
|
||
|
|
'SERVER_SOFTWARE': self.server_version,
|
||
|
|
'REQUEST_METHOD': self.command,
|
||
|
|
'SCRIPT_NAME': '',
|
||
|
|
'PATH_INFO': wsgi_encoding_dance(path_info),
|
||
|
|
'QUERY_STRING': wsgi_encoding_dance(request_url.query),
|
||
|
|
'REMOTE_ADDR': self.address_string(),
|
||
|
|
'REMOTE_PORT': self.port_integer(),
|
||
|
|
'SERVER_NAME': self.server.server_address[0],
|
||
|
|
'SERVER_PORT': str(self.server.server_address[1]),
|
||
|
|
'SERVER_PROTOCOL': self.request_version
|
||
|
|
}
|
||
|
|
|
||
|
|
for key, value in self.headers.items():
|
||
|
|
key = key.upper().replace('-', '_')
|
||
|
|
if key not in ('CONTENT_TYPE', 'CONTENT_LENGTH'):
|
||
|
|
key = 'HTTP_' + key
|
||
|
|
environ[key] = value
|
||
|
|
|
||
|
|
if environ.get('HTTP_TRANSFER_ENCODING', '').strip().lower() == 'chunked':
|
||
|
|
environ['wsgi.input_terminated'] = True
|
||
|
|
environ['wsgi.input'] = DechunkedInput(environ['wsgi.input'])
|
||
|
|
|
||
|
|
if request_url.scheme and request_url.netloc:
|
||
|
|
environ['HTTP_HOST'] = request_url.netloc
|
||
|
|
|
||
|
|
return environ
|
||
|
|
|
||
|
|
def run_wsgi(self):
|
||
|
|
if self.headers.get('Expect', '').lower().strip() == '100-continue':
|
||
|
|
self.wfile.write(b'HTTP/1.1 100 Continue\r\n\r\n')
|
||
|
|
|
||
|
|
self.environ = environ = self.make_environ()
|
||
|
|
headers_set = []
|
||
|
|
headers_sent = []
|
||
|
|
|
||
|
|
def write(data):
|
||
|
|
assert headers_set, 'write() before start_response'
|
||
|
|
if not headers_sent:
|
||
|
|
status, response_headers = headers_sent[:] = headers_set
|
||
|
|
try:
|
||
|
|
code, msg = status.split(None, 1)
|
||
|
|
except ValueError:
|
||
|
|
code, msg = status, ""
|
||
|
|
code = int(code)
|
||
|
|
self.send_response(code, msg)
|
||
|
|
header_keys = set()
|
||
|
|
for key, value in response_headers:
|
||
|
|
self.send_header(key, value)
|
||
|
|
key = key.lower()
|
||
|
|
header_keys.add(key)
|
||
|
|
if not ('content-length' in header_keys or
|
||
|
|
environ['REQUEST_METHOD'] == 'HEAD' or
|
||
|
|
code < 200 or code in (204, 304)):
|
||
|
|
self.close_connection = True
|
||
|
|
self.send_header('Connection', 'close')
|
||
|
|
if 'server' not in header_keys:
|
||
|
|
self.send_header('Server', self.version_string())
|
||
|
|
if 'date' not in header_keys:
|
||
|
|
self.send_header('Date', self.date_time_string())
|
||
|
|
self.end_headers()
|
||
|
|
|
||
|
|
assert isinstance(data, bytes), 'applications must write bytes'
|
||
|
|
self.wfile.write(data)
|
||
|
|
self.wfile.flush()
|
||
|
|
|
||
|
|
def start_response(status, response_headers, exc_info=None):
|
||
|
|
if exc_info:
|
||
|
|
try:
|
||
|
|
if headers_sent:
|
||
|
|
reraise(*exc_info)
|
||
|
|
finally:
|
||
|
|
exc_info = None
|
||
|
|
elif headers_set:
|
||
|
|
raise AssertionError('Headers already set')
|
||
|
|
headers_set[:] = [status, response_headers]
|
||
|
|
return write
|
||
|
|
|
||
|
|
def execute(app):
|
||
|
|
application_iter = app(environ, start_response)
|
||
|
|
try:
|
||
|
|
for data in application_iter:
|
||
|
|
write(data)
|
||
|
|
if not headers_sent:
|
||
|
|
write(b'')
|
||
|
|
finally:
|
||
|
|
if hasattr(application_iter, 'close'):
|
||
|
|
application_iter.close()
|
||
|
|
application_iter = None
|
||
|
|
|
||
|
|
try:
|
||
|
|
execute(self.server.app)
|
||
|
|
except (socket.error, socket.timeout) as e:
|
||
|
|
self.connection_dropped(e, environ)
|
||
|
|
except Exception:
|
||
|
|
if self.server.passthrough_errors:
|
||
|
|
raise
|
||
|
|
from werkzeug.debug.tbtools import get_current_traceback
|
||
|
|
traceback = get_current_traceback(ignore_system_exceptions=True)
|
||
|
|
try:
|
||
|
|
# if we haven't yet sent the headers but they are set
|
||
|
|
# we roll back to be able to set them again.
|
||
|
|
if not headers_sent:
|
||
|
|
del headers_set[:]
|
||
|
|
execute(InternalServerError())
|
||
|
|
except Exception:
|
||
|
|
pass
|
||
|
|
self.server.log('error', 'Error on request:\n%s',
|
||
|
|
traceback.plaintext)
|
||
|
|
|
||
|
|
def handle(self):
|
||
|
|
"""Handles a request ignoring dropped connections."""
|
||
|
|
rv = None
|
||
|
|
try:
|
||
|
|
rv = BaseHTTPRequestHandler.handle(self)
|
||
|
|
except (socket.error, socket.timeout) as e:
|
||
|
|
self.connection_dropped(e)
|
||
|
|
except Exception:
|
||
|
|
if self.server.ssl_context is None or not is_ssl_error():
|
||
|
|
raise
|
||
|
|
if self.server.shutdown_signal:
|
||
|
|
self.initiate_shutdown()
|
||
|
|
return rv
|
||
|
|
|
||
|
|
def initiate_shutdown(self):
|
||
|
|
"""A horrible, horrible way to kill the server for Python 2.6 and
|
||
|
|
later. It's the best we can do.
|
||
|
|
"""
|
||
|
|
# Windows does not provide SIGKILL, go with SIGTERM then.
|
||
|
|
sig = getattr(signal, 'SIGKILL', signal.SIGTERM)
|
||
|
|
# reloader active
|
||
|
|
if os.environ.get('WERKZEUG_RUN_MAIN') == 'true':
|
||
|
|
os.kill(os.getpid(), sig)
|
||
|
|
# python 2.7
|
||
|
|
self.server._BaseServer__shutdown_request = True
|
||
|
|
# python 2.6
|
||
|
|
self.server._BaseServer__serving = False
|
||
|
|
|
||
|
|
def connection_dropped(self, error, environ=None):
|
||
|
|
"""Called if the connection was closed by the client. By default
|
||
|
|
nothing happens.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def handle_one_request(self):
|
||
|
|
"""Handle a single HTTP request."""
|
||
|
|
self.raw_requestline = self.rfile.readline()
|
||
|
|
if not self.raw_requestline:
|
||
|
|
self.close_connection = 1
|
||
|
|
elif self.parse_request():
|
||
|
|
return self.run_wsgi()
|
||
|
|
|
||
|
|
def send_response(self, code, message=None):
|
||
|
|
"""Send the response header and log the response code."""
|
||
|
|
self.log_request(code)
|
||
|
|
if message is None:
|
||
|
|
message = code in self.responses and self.responses[code][0] or ''
|
||
|
|
if self.request_version != 'HTTP/0.9':
|
||
|
|
hdr = "%s %d %s\r\n" % (self.protocol_version, code, message)
|
||
|
|
self.wfile.write(hdr.encode('ascii'))
|
||
|
|
|
||
|
|
def version_string(self):
|
||
|
|
return BaseHTTPRequestHandler.version_string(self).strip()
|
||
|
|
|
||
|
|
def address_string(self):
|
||
|
|
if getattr(self, 'environ', None):
|
||
|
|
return self.environ['REMOTE_ADDR']
|
||
|
|
else:
|
||
|
|
return self.client_address[0]
|
||
|
|
|
||
|
|
def port_integer(self):
|
||
|
|
return self.client_address[1]
|
||
|
|
|
||
|
|
def log_request(self, code='-', size='-'):
|
||
|
|
msg = self.requestline
|
||
|
|
code = str(code)
|
||
|
|
|
||
|
|
if termcolor:
|
||
|
|
color = termcolor.colored
|
||
|
|
|
||
|
|
if code[0] == '1': # 1xx - Informational
|
||
|
|
msg = color(msg, attrs=['bold'])
|
||
|
|
elif code[0] == '2': # 2xx - Success
|
||
|
|
msg = color(msg, color='white')
|
||
|
|
elif code == '304': # 304 - Resource Not Modified
|
||
|
|
msg = color(msg, color='cyan')
|
||
|
|
elif code[0] == '3': # 3xx - Redirection
|
||
|
|
msg = color(msg, color='green')
|
||
|
|
elif code == '404': # 404 - Resource Not Found
|
||
|
|
msg = color(msg, color='yellow')
|
||
|
|
elif code[0] == '4': # 4xx - Client Error
|
||
|
|
msg = color(msg, color='red', attrs=['bold'])
|
||
|
|
else: # 5xx, or any other response
|
||
|
|
msg = color(msg, color='magenta', attrs=['bold'])
|
||
|
|
|
||
|
|
self.log('info', '"%s" %s %s', msg, code, size)
|
||
|
|
|
||
|
|
def log_error(self, *args):
|
||
|
|
self.log('error', *args)
|
||
|
|
|
||
|
|
def log_message(self, format, *args):
|
||
|
|
self.log('info', format, *args)
|
||
|
|
|
||
|
|
def log(self, type, message, *args):
|
||
|
|
_log(type, '%s - - [%s] %s\n' % (self.address_string(),
|
||
|
|
self.log_date_time_string(),
|
||
|
|
message % args))
|
||
|
|
|
||
|
|
|
||
|
|
#: backwards compatible name if someone is subclassing it
|
||
|
|
BaseRequestHandler = WSGIRequestHandler
|
||
|
|
|
||
|
|
|
||
|
|
def generate_adhoc_ssl_pair(cn=None):
|
||
|
|
from random import random
|
||
|
|
crypto = _get_openssl_crypto_module()
|
||
|
|
|
||
|
|
# pretty damn sure that this is not actually accepted by anyone
|
||
|
|
if cn is None:
|
||
|
|
cn = '*'
|
||
|
|
|
||
|
|
cert = crypto.X509()
|
||
|
|
cert.set_serial_number(int(random() * sys.maxsize))
|
||
|
|
cert.gmtime_adj_notBefore(0)
|
||
|
|
cert.gmtime_adj_notAfter(60 * 60 * 24 * 365)
|
||
|
|
|
||
|
|
subject = cert.get_subject()
|
||
|
|
subject.CN = cn
|
||
|
|
subject.O = 'Dummy Certificate' # noqa: E741
|
||
|
|
|
||
|
|
issuer = cert.get_issuer()
|
||
|
|
issuer.CN = 'Untrusted Authority'
|
||
|
|
issuer.O = 'Self-Signed' # noqa: E741
|
||
|
|
|
||
|
|
pkey = crypto.PKey()
|
||
|
|
pkey.generate_key(crypto.TYPE_RSA, 2048)
|
||
|
|
cert.set_pubkey(pkey)
|
||
|
|
cert.sign(pkey, 'sha256')
|
||
|
|
|
||
|
|
return cert, pkey
|
||
|
|
|
||
|
|
|
||
|
|
def make_ssl_devcert(base_path, host=None, cn=None):
|
||
|
|
"""Creates an SSL key for development. This should be used instead of
|
||
|
|
the ``'adhoc'`` key which generates a new cert on each server start.
|
||
|
|
It accepts a path for where it should store the key and cert and
|
||
|
|
either a host or CN. If a host is given it will use the CN
|
||
|
|
``*.host/CN=host``.
|
||
|
|
|
||
|
|
For more information see :func:`run_simple`.
|
||
|
|
|
||
|
|
.. versionadded:: 0.9
|
||
|
|
|
||
|
|
:param base_path: the path to the certificate and key. The extension
|
||
|
|
``.crt`` is added for the certificate, ``.key`` is
|
||
|
|
added for the key.
|
||
|
|
:param host: the name of the host. This can be used as an alternative
|
||
|
|
for the `cn`.
|
||
|
|
:param cn: the `CN` to use.
|
||
|
|
"""
|
||
|
|
from OpenSSL import crypto
|
||
|
|
if host is not None:
|
||
|
|
cn = '*.%s/CN=%s' % (host, host)
|
||
|
|
cert, pkey = generate_adhoc_ssl_pair(cn=cn)
|
||
|
|
|
||
|
|
cert_file = base_path + '.crt'
|
||
|
|
pkey_file = base_path + '.key'
|
||
|
|
|
||
|
|
with open(cert_file, 'wb') as f:
|
||
|
|
f.write(crypto.dump_certificate(crypto.FILETYPE_PEM, cert))
|
||
|
|
with open(pkey_file, 'wb') as f:
|
||
|
|
f.write(crypto.dump_privatekey(crypto.FILETYPE_PEM, pkey))
|
||
|
|
|
||
|
|
return cert_file, pkey_file
|
||
|
|
|
||
|
|
|
||
|
|
def generate_adhoc_ssl_context():
|
||
|
|
"""Generates an adhoc SSL context for the development server."""
|
||
|
|
crypto = _get_openssl_crypto_module()
|
||
|
|
import tempfile
|
||
|
|
import atexit
|
||
|
|
|
||
|
|
cert, pkey = generate_adhoc_ssl_pair()
|
||
|
|
cert_handle, cert_file = tempfile.mkstemp()
|
||
|
|
pkey_handle, pkey_file = tempfile.mkstemp()
|
||
|
|
atexit.register(os.remove, pkey_file)
|
||
|
|
atexit.register(os.remove, cert_file)
|
||
|
|
|
||
|
|
os.write(cert_handle, crypto.dump_certificate(crypto.FILETYPE_PEM, cert))
|
||
|
|
os.write(pkey_handle, crypto.dump_privatekey(crypto.FILETYPE_PEM, pkey))
|
||
|
|
os.close(cert_handle)
|
||
|
|
os.close(pkey_handle)
|
||
|
|
ctx = load_ssl_context(cert_file, pkey_file)
|
||
|
|
return ctx
|
||
|
|
|
||
|
|
|
||
|
|
def load_ssl_context(cert_file, pkey_file=None, protocol=None):
|
||
|
|
"""Loads SSL context from cert/private key files and optional protocol.
|
||
|
|
Many parameters are directly taken from the API of
|
||
|
|
:py:class:`ssl.SSLContext`.
|
||
|
|
|
||
|
|
:param cert_file: Path of the certificate to use.
|
||
|
|
:param pkey_file: Path of the private key to use. If not given, the key
|
||
|
|
will be obtained from the certificate file.
|
||
|
|
:param protocol: One of the ``PROTOCOL_*`` constants in the stdlib ``ssl``
|
||
|
|
module. Defaults to ``PROTOCOL_SSLv23``.
|
||
|
|
"""
|
||
|
|
if protocol is None:
|
||
|
|
protocol = ssl.PROTOCOL_SSLv23
|
||
|
|
ctx = _SSLContext(protocol)
|
||
|
|
ctx.load_cert_chain(cert_file, pkey_file)
|
||
|
|
return ctx
|
||
|
|
|
||
|
|
|
||
|
|
class _SSLContext(object):
|
||
|
|
|
||
|
|
'''A dummy class with a small subset of Python3's ``ssl.SSLContext``, only
|
||
|
|
intended to be used with and by Werkzeug.'''
|
||
|
|
|
||
|
|
def __init__(self, protocol):
|
||
|
|
self._protocol = protocol
|
||
|
|
self._certfile = None
|
||
|
|
self._keyfile = None
|
||
|
|
self._password = None
|
||
|
|
|
||
|
|
def load_cert_chain(self, certfile, keyfile=None, password=None):
|
||
|
|
self._certfile = certfile
|
||
|
|
self._keyfile = keyfile or certfile
|
||
|
|
self._password = password
|
||
|
|
|
||
|
|
def wrap_socket(self, sock, **kwargs):
|
||
|
|
return ssl.wrap_socket(sock, keyfile=self._keyfile,
|
||
|
|
certfile=self._certfile,
|
||
|
|
ssl_version=self._protocol, **kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
def is_ssl_error(error=None):
|
||
|
|
"""Checks if the given error (or the current one) is an SSL error."""
|
||
|
|
exc_types = (ssl.SSLError,)
|
||
|
|
try:
|
||
|
|
from OpenSSL.SSL import Error
|
||
|
|
exc_types += (Error,)
|
||
|
|
except ImportError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
if error is None:
|
||
|
|
error = sys.exc_info()[1]
|
||
|
|
return isinstance(error, exc_types)
|
||
|
|
|
||
|
|
|
||
|
|
def select_ip_version(host, port):
|
||
|
|
"""Returns AF_INET4 or AF_INET6 depending on where to connect to."""
|
||
|
|
# disabled due to problems with current ipv6 implementations
|
||
|
|
# and various operating systems. Probably this code also is
|
||
|
|
# not supposed to work, but I can't come up with any other
|
||
|
|
# ways to implement this.
|
||
|
|
# try:
|
||
|
|
# info = socket.getaddrinfo(host, port, socket.AF_UNSPEC,
|
||
|
|
# socket.SOCK_STREAM, 0,
|
||
|
|
# socket.AI_PASSIVE)
|
||
|
|
# if info:
|
||
|
|
# return info[0][0]
|
||
|
|
# except socket.gaierror:
|
||
|
|
# pass
|
||
|
|
if ':' in host and hasattr(socket, 'AF_INET6'):
|
||
|
|
return socket.AF_INET6
|
||
|
|
return socket.AF_INET
|
||
|
|
|
||
|
|
|
||
|
|
def get_sockaddr(host, port, family):
|
||
|
|
"""Returns a fully qualified socket address, that can properly used by
|
||
|
|
socket.bind"""
|
||
|
|
try:
|
||
|
|
res = socket.getaddrinfo(host, port, family,
|
||
|
|
socket.SOCK_STREAM, socket.SOL_TCP)
|
||
|
|
except socket.gaierror:
|
||
|
|
return (host, port)
|
||
|
|
return res[0][4]
|
||
|
|
|
||
|
|
|
||
|
|
class BaseWSGIServer(HTTPServer, object):
|
||
|
|
|
||
|
|
"""Simple single-threaded, single-process WSGI server."""
|
||
|
|
multithread = False
|
||
|
|
multiprocess = False
|
||
|
|
request_queue_size = LISTEN_QUEUE
|
||
|
|
|
||
|
|
def __init__(self, host, port, app, handler=None,
|
||
|
|
passthrough_errors=False, ssl_context=None, fd=None):
|
||
|
|
if handler is None:
|
||
|
|
handler = WSGIRequestHandler
|
||
|
|
|
||
|
|
self.address_family = select_ip_version(host, port)
|
||
|
|
|
||
|
|
if fd is not None:
|
||
|
|
real_sock = socket.fromfd(fd, self.address_family,
|
||
|
|
socket.SOCK_STREAM)
|
||
|
|
port = 0
|
||
|
|
HTTPServer.__init__(self, get_sockaddr(host, int(port),
|
||
|
|
self.address_family), handler)
|
||
|
|
self.app = app
|
||
|
|
self.passthrough_errors = passthrough_errors
|
||
|
|
self.shutdown_signal = False
|
||
|
|
self.host = host
|
||
|
|
self.port = self.socket.getsockname()[1]
|
||
|
|
|
||
|
|
# Patch in the original socket.
|
||
|
|
if fd is not None:
|
||
|
|
self.socket.close()
|
||
|
|
self.socket = real_sock
|
||
|
|
self.server_address = self.socket.getsockname()
|
||
|
|
|
||
|
|
if ssl_context is not None:
|
||
|
|
if isinstance(ssl_context, tuple):
|
||
|
|
ssl_context = load_ssl_context(*ssl_context)
|
||
|
|
if ssl_context == 'adhoc':
|
||
|
|
ssl_context = generate_adhoc_ssl_context()
|
||
|
|
# If we are on Python 2 the return value from socket.fromfd
|
||
|
|
# is an internal socket object but what we need for ssl wrap
|
||
|
|
# is the wrapper around it :(
|
||
|
|
sock = self.socket
|
||
|
|
if PY2 and not isinstance(sock, socket.socket):
|
||
|
|
sock = socket.socket(sock.family, sock.type, sock.proto, sock)
|
||
|
|
self.socket = ssl_context.wrap_socket(sock, server_side=True)
|
||
|
|
self.ssl_context = ssl_context
|
||
|
|
else:
|
||
|
|
self.ssl_context = None
|
||
|
|
|
||
|
|
def log(self, type, message, *args):
|
||
|
|
_log(type, message, *args)
|
||
|
|
|
||
|
|
def serve_forever(self):
|
||
|
|
self.shutdown_signal = False
|
||
|
|
try:
|
||
|
|
HTTPServer.serve_forever(self)
|
||
|
|
except KeyboardInterrupt:
|
||
|
|
pass
|
||
|
|
finally:
|
||
|
|
self.server_close()
|
||
|
|
|
||
|
|
def handle_error(self, request, client_address):
|
||
|
|
if self.passthrough_errors:
|
||
|
|
raise
|
||
|
|
return HTTPServer.handle_error(self, request, client_address)
|
||
|
|
|
||
|
|
def get_request(self):
|
||
|
|
con, info = self.socket.accept()
|
||
|
|
return con, info
|
||
|
|
|
||
|
|
|
||
|
|
class ThreadedWSGIServer(ThreadingMixIn, BaseWSGIServer):
|
||
|
|
|
||
|
|
"""A WSGI server that does threading."""
|
||
|
|
multithread = True
|
||
|
|
daemon_threads = True
|
||
|
|
|
||
|
|
|
||
|
|
class ForkingWSGIServer(ForkingMixIn, BaseWSGIServer):
|
||
|
|
|
||
|
|
"""A WSGI server that does forking."""
|
||
|
|
multiprocess = True
|
||
|
|
|
||
|
|
def __init__(self, host, port, app, processes=40, handler=None,
|
||
|
|
passthrough_errors=False, ssl_context=None, fd=None):
|
||
|
|
if not can_fork:
|
||
|
|
raise ValueError('Your platform does not support forking.')
|
||
|
|
BaseWSGIServer.__init__(self, host, port, app, handler,
|
||
|
|
passthrough_errors, ssl_context, fd)
|
||
|
|
self.max_children = processes
|
||
|
|
|
||
|
|
|
||
|
|
def make_server(host=None, port=None, app=None, threaded=False, processes=1,
|
||
|
|
request_handler=None, passthrough_errors=False,
|
||
|
|
ssl_context=None, fd=None):
|
||
|
|
"""Create a new server instance that is either threaded, or forks
|
||
|
|
or just processes one request after another.
|
||
|
|
"""
|
||
|
|
if threaded and processes > 1:
|
||
|
|
raise ValueError("cannot have a multithreaded and "
|
||
|
|
"multi process server.")
|
||
|
|
elif threaded:
|
||
|
|
return ThreadedWSGIServer(host, port, app, request_handler,
|
||
|
|
passthrough_errors, ssl_context, fd=fd)
|
||
|
|
elif processes > 1:
|
||
|
|
return ForkingWSGIServer(host, port, app, processes, request_handler,
|
||
|
|
passthrough_errors, ssl_context, fd=fd)
|
||
|
|
else:
|
||
|
|
return BaseWSGIServer(host, port, app, request_handler,
|
||
|
|
passthrough_errors, ssl_context, fd=fd)
|
||
|
|
|
||
|
|
|
||
|
|
def is_running_from_reloader():
|
||
|
|
"""Checks if the application is running from within the Werkzeug
|
||
|
|
reloader subprocess.
|
||
|
|
|
||
|
|
.. versionadded:: 0.10
|
||
|
|
"""
|
||
|
|
return os.environ.get('WERKZEUG_RUN_MAIN') == 'true'
|
||
|
|
|
||
|
|
|
||
|
|
def run_simple(hostname, port, application, use_reloader=False,
|
||
|
|
use_debugger=False, use_evalex=True,
|
||
|
|
extra_files=None, reloader_interval=1,
|
||
|
|
reloader_type='auto', threaded=False,
|
||
|
|
processes=1, request_handler=None, static_files=None,
|
||
|
|
passthrough_errors=False, ssl_context=None):
|
||
|
|
"""Start a WSGI application. Optional features include a reloader,
|
||
|
|
multithreading and fork support.
|
||
|
|
|
||
|
|
This function has a command-line interface too::
|
||
|
|
|
||
|
|
python -m werkzeug.serving --help
|
||
|
|
|
||
|
|
.. versionadded:: 0.5
|
||
|
|
`static_files` was added to simplify serving of static files as well
|
||
|
|
as `passthrough_errors`.
|
||
|
|
|
||
|
|
.. versionadded:: 0.6
|
||
|
|
support for SSL was added.
|
||
|
|
|
||
|
|
.. versionadded:: 0.8
|
||
|
|
Added support for automatically loading a SSL context from certificate
|
||
|
|
file and private key.
|
||
|
|
|
||
|
|
.. versionadded:: 0.9
|
||
|
|
Added command-line interface.
|
||
|
|
|
||
|
|
.. versionadded:: 0.10
|
||
|
|
Improved the reloader and added support for changing the backend
|
||
|
|
through the `reloader_type` parameter. See :ref:`reloader`
|
||
|
|
for more information.
|
||
|
|
|
||
|
|
:param hostname: The host for the application. eg: ``'localhost'``
|
||
|
|
:param port: The port for the server. eg: ``8080``
|
||
|
|
:param application: the WSGI application to execute
|
||
|
|
:param use_reloader: should the server automatically restart the python
|
||
|
|
process if modules were changed?
|
||
|
|
:param use_debugger: should the werkzeug debugging system be used?
|
||
|
|
:param use_evalex: should the exception evaluation feature be enabled?
|
||
|
|
:param extra_files: a list of files the reloader should watch
|
||
|
|
additionally to the modules. For example configuration
|
||
|
|
files.
|
||
|
|
:param reloader_interval: the interval for the reloader in seconds.
|
||
|
|
:param reloader_type: the type of reloader to use. The default is
|
||
|
|
auto detection. Valid values are ``'stat'`` and
|
||
|
|
``'watchdog'``. See :ref:`reloader` for more
|
||
|
|
information.
|
||
|
|
:param threaded: should the process handle each request in a separate
|
||
|
|
thread?
|
||
|
|
:param processes: if greater than 1 then handle each request in a new process
|
||
|
|
up to this maximum number of concurrent processes.
|
||
|
|
:param request_handler: optional parameter that can be used to replace
|
||
|
|
the default one. You can use this to replace it
|
||
|
|
with a different
|
||
|
|
:class:`~BaseHTTPServer.BaseHTTPRequestHandler`
|
||
|
|
subclass.
|
||
|
|
:param static_files: a list or dict of paths for static files. This works
|
||
|
|
exactly like :class:`SharedDataMiddleware`, it's actually
|
||
|
|
just wrapping the application in that middleware before
|
||
|
|
serving.
|
||
|
|
:param passthrough_errors: set this to `True` to disable the error catching.
|
||
|
|
This means that the server will die on errors but
|
||
|
|
it can be useful to hook debuggers in (pdb etc.)
|
||
|
|
:param ssl_context: an SSL context for the connection. Either an
|
||
|
|
:class:`ssl.SSLContext`, a tuple in the form
|
||
|
|
``(cert_file, pkey_file)``, the string ``'adhoc'`` if
|
||
|
|
the server should automatically create one, or ``None``
|
||
|
|
to disable SSL (which is the default).
|
||
|
|
"""
|
||
|
|
if not isinstance(port, int):
|
||
|
|
raise TypeError('port must be an integer')
|
||
|
|
if use_debugger:
|
||
|
|
from werkzeug.debug import DebuggedApplication
|
||
|
|
application = DebuggedApplication(application, use_evalex)
|
||
|
|
if static_files:
|
||
|
|
from werkzeug.wsgi import SharedDataMiddleware
|
||
|
|
application = SharedDataMiddleware(application, static_files)
|
||
|
|
|
||
|
|
def log_startup(sock):
|
||
|
|
display_hostname = hostname not in ('', '*') and hostname or 'localhost'
|
||
|
|
if ':' in display_hostname:
|
||
|
|
display_hostname = '[%s]' % display_hostname
|
||
|
|
quit_msg = '(Press CTRL+C to quit)'
|
||
|
|
port = sock.getsockname()[1]
|
||
|
|
_log('info', ' * Running on %s://%s:%d/ %s',
|
||
|
|
ssl_context is None and 'http' or 'https',
|
||
|
|
display_hostname, port, quit_msg)
|
||
|
|
|
||
|
|
def inner():
|
||
|
|
try:
|
||
|
|
fd = int(os.environ['WERKZEUG_SERVER_FD'])
|
||
|
|
except (LookupError, ValueError):
|
||
|
|
fd = None
|
||
|
|
srv = make_server(hostname, port, application, threaded,
|
||
|
|
processes, request_handler,
|
||
|
|
passthrough_errors, ssl_context,
|
||
|
|
fd=fd)
|
||
|
|
if fd is None:
|
||
|
|
log_startup(srv.socket)
|
||
|
|
srv.serve_forever()
|
||
|
|
|
||
|
|
if use_reloader:
|
||
|
|
# If we're not running already in the subprocess that is the
|
||
|
|
# reloader we want to open up a socket early to make sure the
|
||
|
|
# port is actually available.
|
||
|
|
if os.environ.get('WERKZEUG_RUN_MAIN') != 'true':
|
||
|
|
if port == 0 and not can_open_by_fd:
|
||
|
|
raise ValueError('Cannot bind to a random port with enabled '
|
||
|
|
'reloader if the Python interpreter does '
|
||
|
|
'not support socket opening by fd.')
|
||
|
|
|
||
|
|
# Create and destroy a socket so that any exceptions are
|
||
|
|
# raised before we spawn a separate Python interpreter and
|
||
|
|
# lose this ability.
|
||
|
|
address_family = select_ip_version(hostname, port)
|
||
|
|
s = socket.socket(address_family, socket.SOCK_STREAM)
|
||
|
|
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||
|
|
s.bind(get_sockaddr(hostname, port, address_family))
|
||
|
|
if hasattr(s, 'set_inheritable'):
|
||
|
|
s.set_inheritable(True)
|
||
|
|
|
||
|
|
# If we can open the socket by file descriptor, then we can just
|
||
|
|
# reuse this one and our socket will survive the restarts.
|
||
|
|
if can_open_by_fd:
|
||
|
|
os.environ['WERKZEUG_SERVER_FD'] = str(s.fileno())
|
||
|
|
s.listen(LISTEN_QUEUE)
|
||
|
|
log_startup(s)
|
||
|
|
else:
|
||
|
|
s.close()
|
||
|
|
|
||
|
|
# Do not use relative imports, otherwise "python -m werkzeug.serving"
|
||
|
|
# breaks.
|
||
|
|
from werkzeug._reloader import run_with_reloader
|
||
|
|
run_with_reloader(inner, extra_files, reloader_interval,
|
||
|
|
reloader_type)
|
||
|
|
else:
|
||
|
|
inner()
|
||
|
|
|
||
|
|
|
||
|
|
def run_with_reloader(*args, **kwargs):
|
||
|
|
# People keep using undocumented APIs. Do not use this function
|
||
|
|
# please, we do not guarantee that it continues working.
|
||
|
|
from werkzeug._reloader import run_with_reloader
|
||
|
|
return run_with_reloader(*args, **kwargs)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
'''A simple command-line interface for :py:func:`run_simple`.'''
|
||
|
|
|
||
|
|
# in contrast to argparse, this works at least under Python < 2.7
|
||
|
|
import optparse
|
||
|
|
from werkzeug.utils import import_string
|
||
|
|
|
||
|
|
parser = optparse.OptionParser(
|
||
|
|
usage='Usage: %prog [options] app_module:app_object')
|
||
|
|
parser.add_option('-b', '--bind', dest='address',
|
||
|
|
help='The hostname:port the app should listen on.')
|
||
|
|
parser.add_option('-d', '--debug', dest='use_debugger',
|
||
|
|
action='store_true', default=False,
|
||
|
|
help='Use Werkzeug\'s debugger.')
|
||
|
|
parser.add_option('-r', '--reload', dest='use_reloader',
|
||
|
|
action='store_true', default=False,
|
||
|
|
help='Reload Python process if modules change.')
|
||
|
|
options, args = parser.parse_args()
|
||
|
|
|
||
|
|
hostname, port = None, None
|
||
|
|
if options.address:
|
||
|
|
address = options.address.split(':')
|
||
|
|
hostname = address[0]
|
||
|
|
if len(address) > 1:
|
||
|
|
port = address[1]
|
||
|
|
|
||
|
|
if len(args) != 1:
|
||
|
|
sys.stdout.write('No application supplied, or too much. See --help\n')
|
||
|
|
sys.exit(1)
|
||
|
|
app = import_string(args[0])
|
||
|
|
|
||
|
|
run_simple(
|
||
|
|
hostname=(hostname or '127.0.0.1'), port=int(port or 5000),
|
||
|
|
application=app, use_reloader=options.use_reloader,
|
||
|
|
use_debugger=options.use_debugger
|
||
|
|
)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|