1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117 |
- # -*- 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: 2007 Pallets
- :license: BSD-3-Clause
- """
- import io
- import os
- import signal
- import socket
- import sys
- from datetime import datetime as dt
- from datetime import timedelta
- from ._compat import PY2
- from ._compat import reraise
- from ._compat import WIN
- from ._compat import wsgi_encoding_dance
- from ._internal import _log
- from .exceptions import InternalServerError
- from .urls import uri_to_iri
- from .urls import url_parse
- from .urls import url_unquote
- try:
- import socketserver
- from http.server import BaseHTTPRequestHandler
- from http.server import HTTPServer
- except ImportError:
- import SocketServer as socketserver
- from BaseHTTPServer import HTTPServer
- from BaseHTTPServer import BaseHTTPRequestHandler
- try:
- import ssl
- except ImportError:
- class _SslDummy(object):
- def __getattr__(self, name):
- raise RuntimeError("SSL support unavailable")
- ssl = _SslDummy()
- try:
- import click
- except ImportError:
- click = None
- ThreadingMixIn = socketserver.ThreadingMixIn
- can_fork = hasattr(os, "fork")
- if can_fork:
- ForkingMixIn = socketserver.ForkingMixIn
- else:
- class ForkingMixIn(object):
- pass
- try:
- af_unix = socket.AF_UNIX
- except AttributeError:
- af_unix = None
- LISTEN_QUEUE = 128
- can_open_by_fd = not WIN and hasattr(socket, "fromfd")
- # On Python 3, ConnectionError represents the same errnos as
- # socket.error from Python 2, while socket.error is an alias for the
- # more generic OSError.
- if PY2:
- _ConnectionError = socket.error
- else:
- _ConnectionError = ConnectionError
- 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):
- from . import __version__
- return "Werkzeug/" + __version__
- def make_environ(self):
- request_url = url_parse(self.path)
- def shutdown_server():
- self.server.shutdown_signal = True
- url_scheme = "http" if self.server.ssl_context is None else "https"
- if not self.client_address:
- self.client_address = "<local>"
- if isinstance(self.client_address, str):
- self.client_address = (self.client_address, 0)
- else:
- pass
- # If there was no scheme but the path started with two slashes,
- # the first segment may have been incorrectly parsed as the
- # netloc, prepend it to the path again.
- if not request_url.scheme and request_url.netloc:
- path_info = "/%s%s" % (request_url.netloc, request_url.path)
- else:
- path_info = request_url.path
- path_info = url_unquote(path_info)
- 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),
- # Non-standard, added by mod_wsgi, uWSGI
- "REQUEST_URI": wsgi_encoding_dance(self.path),
- # Non-standard, added by gunicorn
- "RAW_URI": wsgi_encoding_dance(self.path),
- "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.get_header_items():
- key = key.upper().replace("-", "_")
- value = value.replace("\r\n", "")
- if key not in ("CONTENT_TYPE", "CONTENT_LENGTH"):
- key = "HTTP_" + key
- if key in environ:
- value = "{},{}".format(environ[key], value)
- environ[key] = value
- if environ.get("HTTP_TRANSFER_ENCODING", "").strip().lower() == "chunked":
- environ["wsgi.input_terminated"] = True
- environ["wsgi.input"] = DechunkedInput(environ["wsgi.input"])
- # Per RFC 2616, if the URL is absolute, use that as the host.
- # We're using "has a scheme" to indicate an absolute URL.
- if request_url.scheme and request_url.netloc:
- environ["HTTP_HOST"] = request_url.netloc
- try:
- # binary_form=False gives nicer information, but wouldn't be compatible with
- # what Nginx or Apache could return.
- peer_cert = self.connection.getpeercert(binary_form=True)
- if peer_cert is not None:
- # Nginx and Apache use PEM format.
- environ["SSL_CLIENT_CERT"] = ssl.DER_cert_to_PEM_cert(peer_cert)
- except ValueError:
- # SSL handshake hasn't finished.
- self.server.log("error", "Cannot fetch SSL peer certificate info")
- except AttributeError:
- # Not using TLS, the socket will not have getpeercert().
- pass
- 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"
- if data:
- # Only write data if there is any to avoid Python 3.5 SSL bug
- 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()
- try:
- execute(self.server.app)
- except (_ConnectionError, socket.timeout) as e:
- self.connection_dropped(e, environ)
- except Exception:
- if self.server.passthrough_errors:
- raise
- from .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."""
- try:
- BaseHTTPRequestHandler.handle(self)
- except (_ConnectionError, socket.timeout) as e:
- self.connection_dropped(e)
- except Exception as e:
- if self.server.ssl_context is None or not is_ssl_error(e):
- raise
- if self.server.shutdown_signal:
- self.initiate_shutdown()
- 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 is_running_from_reloader():
- 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"]
- elif not self.client_address:
- return "<local>"
- elif isinstance(self.client_address, str):
- return self.client_address
- else:
- return self.client_address[0]
- def port_integer(self):
- return self.client_address[1]
- def log_request(self, code="-", size="-"):
- try:
- path = uri_to_iri(self.path)
- msg = "%s %s %s" % (self.command, path, self.request_version)
- except AttributeError:
- # path isn't set if the requestline was bad
- msg = self.requestline
- code = str(code)
- if click:
- color = click.style
- if code[0] == "1": # 1xx - Informational
- msg = color(msg, bold=True)
- elif code[0] == "2": # 2xx - Success
- msg = color(msg, fg="white")
- elif code == "304": # 304 - Resource Not Modified
- msg = color(msg, fg="cyan")
- elif code[0] == "3": # 3xx - Redirection
- msg = color(msg, fg="green")
- elif code == "404": # 404 - Resource Not Found
- msg = color(msg, fg="yellow")
- elif code[0] == "4": # 4xx - Client Error
- msg = color(msg, fg="red", bold=True)
- else: # 5xx, or any other response
- msg = color(msg, fg="magenta", bold=True)
- 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),
- )
- def get_header_items(self):
- """
- Get an iterable list of key/value pairs representing headers.
- This function provides Python 2/3 compatibility as related to the
- parsing of request headers. Python 2.7 is not compliant with
- RFC 3875 Section 4.1.18 which requires multiple values for headers
- to be provided or RFC 2616 which allows for folding of multi-line
- headers. This function will return a matching list regardless
- of Python version. It can be removed once Python 2.7 support
- is dropped.
- :return: List of tuples containing header hey/value pairs
- """
- if PY2:
- # For Python 2, process the headers manually according to
- # W3C RFC 2616 Section 4.2.
- items = []
- for header in self.headers.headers:
- # Remove "\r\n" from the header and split on ":" to get
- # the field name and value.
- try:
- key, value = header[0:-2].split(":", 1)
- except ValueError:
- # If header could not be slit with : but starts with white
- # space and it follows an existing header, it's a folded
- # header.
- if header[0] in ("\t", " ") and items:
- # Pop off the last header
- key, value = items.pop()
- # Append the current header to the value of the last
- # header which will be placed back on the end of the
- # list
- value = value + header
- # Otherwise it's just a bad header and should error
- else:
- # Re-raise the value error
- raise
- # Add the key and the value once stripped of leading
- # white space. The specification allows for stripping
- # trailing white space but the Python 3 code does not
- # strip trailing white space. Therefore, trailing space
- # will be left as is to match the Python 3 behavior.
- items.append((key, value.lstrip()))
- else:
- items = self.headers.items()
- return items
- #: backwards compatible name if someone is subclassing it
- BaseRequestHandler = WSGIRequestHandler
- def generate_adhoc_ssl_pair(cn=None):
- try:
- from cryptography import x509
- from cryptography.x509.oid import NameOID
- from cryptography.hazmat.backends import default_backend
- from cryptography.hazmat.primitives import hashes
- from cryptography.hazmat.primitives.asymmetric import rsa
- except ImportError:
- raise TypeError("Using ad-hoc certificates requires the cryptography library.")
- pkey = rsa.generate_private_key(
- public_exponent=65537, key_size=2048, backend=default_backend()
- )
- # pretty damn sure that this is not actually accepted by anyone
- if cn is None:
- cn = u"*"
- subject = x509.Name(
- [
- x509.NameAttribute(NameOID.ORGANIZATION_NAME, u"Dummy Certificate"),
- x509.NameAttribute(NameOID.COMMON_NAME, cn),
- ]
- )
- cert = (
- x509.CertificateBuilder()
- .subject_name(subject)
- .issuer_name(subject)
- .public_key(pkey.public_key())
- .serial_number(x509.random_serial_number())
- .not_valid_before(dt.utcnow())
- .not_valid_after(dt.utcnow() + timedelta(days=365))
- .add_extension(x509.ExtendedKeyUsage([x509.OID_SERVER_AUTH]), critical=False)
- .add_extension(
- x509.SubjectAlternativeName([x509.DNSName(u"*")]), critical=False
- )
- .sign(pkey, hashes.SHA256(), default_backend())
- )
- 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.
- """
- if host is not None:
- cn = u"*.%s/CN=%s" % (host, host)
- cert, pkey = generate_adhoc_ssl_pair(cn=cn)
- from cryptography.hazmat.primitives import serialization
- cert_file = base_path + ".crt"
- pkey_file = base_path + ".key"
- with open(cert_file, "wb") as f:
- f.write(cert.public_bytes(serialization.Encoding.PEM))
- with open(pkey_file, "wb") as f:
- f.write(
- pkey.private_bytes(
- encoding=serialization.Encoding.PEM,
- format=serialization.PrivateFormat.TraditionalOpenSSL,
- encryption_algorithm=serialization.NoEncryption(),
- )
- )
- return cert_file, pkey_file
- def generate_adhoc_ssl_context():
- """Generates an adhoc SSL context for the development server."""
- import tempfile
- import atexit
- cert, pkey = generate_adhoc_ssl_pair()
- from cryptography.hazmat.primitives import serialization
- 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, cert.public_bytes(serialization.Encoding.PEM))
- os.write(
- pkey_handle,
- pkey.private_bytes(
- encoding=serialization.Encoding.PEM,
- format=serialization.PrivateFormat.TraditionalOpenSSL,
- encryption_algorithm=serialization.NoEncryption(),
- ),
- )
- 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:
- try:
- protocol = ssl.PROTOCOL_TLS_SERVER
- except AttributeError:
- # Python <= 3.5 compat
- 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."""
- if error is None:
- error = sys.exc_info()[1]
- return isinstance(error, ssl.SSLError)
- def select_address_family(host, port):
- """Return ``AF_INET4``, ``AF_INET6``, or ``AF_UNIX`` depending on
- the host and port."""
- # 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 host.startswith("unix://"):
- return socket.AF_UNIX
- elif ":" in host and hasattr(socket, "AF_INET6"):
- return socket.AF_INET6
- return socket.AF_INET
- def get_sockaddr(host, port, family):
- """Return a fully qualified socket address that can be passed to
- :func:`socket.bind`."""
- if family == af_unix:
- return host.split("://", 1)[1]
- try:
- res = socket.getaddrinfo(
- host, port, family, socket.SOCK_STREAM, socket.IPPROTO_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_address_family(host, port)
- if fd is not None:
- real_sock = socket.fromfd(fd, self.address_family, socket.SOCK_STREAM)
- port = 0
- server_address = get_sockaddr(host, int(port), self.address_family)
- # remove socket file if it already exists
- if self.address_family == af_unix and os.path.exists(server_address):
- os.unlink(server_address)
- HTTPServer.__init__(self, server_address, 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
- # Python 2 still causes a socket.error after the earlier
- # handling, so silence it here.
- if isinstance(sys.exc_info()[1], _ConnectionError):
- return
- 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.
- .. versionchanged:: 0.15
- Bind to a Unix socket by passing a path that starts with
- ``unix://`` as the ``hostname``.
- :param hostname: The host to bind to, for example ``'localhost'``.
- If the value is a path that starts with ``unix://`` it will bind
- to a Unix socket instead of a TCP socket..
- :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 .debug import DebuggedApplication
- application = DebuggedApplication(application, use_evalex)
- if static_files:
- from .middleware.shared_data import SharedDataMiddleware
- application = SharedDataMiddleware(application, static_files)
- def log_startup(sock):
- display_hostname = hostname if hostname not in ("", "*") else "localhost"
- quit_msg = "(Press CTRL+C to quit)"
- if sock.family == af_unix:
- _log("info", " * Running on %s %s", display_hostname, quit_msg)
- else:
- if ":" in display_hostname:
- display_hostname = "[%s]" % display_hostname
- port = sock.getsockname()[1]
- _log(
- "info",
- " * Running on %s://%s:%d/ %s",
- "http" if ssl_context is None else "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 not is_running_from_reloader():
- 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_address_family(hostname, port)
- server_address = get_sockaddr(hostname, port, address_family)
- s = socket.socket(address_family, socket.SOCK_STREAM)
- s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
- s.bind(server_address)
- 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()
- if address_family == af_unix:
- _log("info", "Unlinking %s" % server_address)
- os.unlink(server_address)
- # Do not use relative imports, otherwise "python -m werkzeug.serving"
- # breaks.
- from ._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 ._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 .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()
|