上传文件至 'src/lib/adafruit_wiznet5k'

src/lib/adafruit_wiznet5k/adafruit_wiznet5k_wsgiserver.py

@@ -0,0 +1,190 @@
+# Based on ESP32 code Copyright (c) 2019 Matt Costi for Adafruit Industries
+# SPDX-FileCopyrightText: Copyright (c) 2020 Patrick Van Oosterwijck
+#
+# SPDX-License-Identifier: MIT
+
+"""
+`adafruit_wiznet5k_wsgiserver`
+================================================================================
+
+A simple WSGI (Web Server Gateway Interface) server that interfaces with the W5500.
+Opens a listening port on the W5500 to listen for incoming HTTP Requests and
+Accepts an Application object that must be callable, which gets called
+whenever a new HTTP Request has been received.
+
+The Application MUST accept 2 ordered parameters:
+    1. environ object (incoming request data)
+    2. start_response function. Must be called before the Application
+        callable returns, in order to set the response status and headers.
+
+The Application MUST return strings in a list, which is the response data
+
+Requires update_poll being called in the applications main event loop.
+
+For more details about Python WSGI see:
+https://www.python.org/dev/peps/pep-0333/
+
+* Author(s): Matt Costi, Patrick Van Oosterwijck
+"""
+# pylint: disable=no-name-in-module
+
+import io
+import gc
+from micropython import const
+import adafruit_wiznet5k.adafruit_wiznet5k_socket as socket
+
+_the_interface = None  # pylint: disable=invalid-name
+
+
+def set_interface(iface):
+    """Helper to set the global internet interface"""
+    global _the_interface  # pylint: disable=global-statement, invalid-name
+    _the_interface = iface
+    socket.set_interface(iface)
+
+
+# Maximum number of sockets for the web server (number of connections we can hold)
+MAX_SOCK_NUM = const(6)
+
+# pylint: disable=invalid-name
+class WSGIServer:
+    """
+    A simple server that implements the WSGI interface
+    """
+
+    def __init__(self, port=80, debug=False, application=None):
+        self.application = application
+        self.port = port
+        self._timeout = 20
+        self._client_sock = []
+        self._debug = debug
+
+        self._response_status = None
+        self._response_headers = []
+
+    def start(self):
+        """
+        Starts the server and begins listening for incoming connections.
+        Call update_poll in the main loop for the application callable to be
+        invoked on receiving an incoming request.
+        """
+        for _ in range(MAX_SOCK_NUM):
+            new_sock = socket.socket()
+            new_sock.settimeout(self._timeout)
+            new_sock.bind((None, self.port))
+            new_sock.listen()
+            self._client_sock.append(new_sock)
+        if self._debug:
+            ip = _the_interface.pretty_ip(_the_interface.ip_address)
+            print("Server available at {0}:{1}".format(ip, self.port))
+
+    def update_poll(self):
+        """
+        Call this method inside your main event loop to get the server
+        check for new incoming client requests. When a request comes in,
+        the application callable will be invoked.
+        """
+        add_sock = []
+        for sock in self._client_sock:
+            if sock.available():
+                environ = self._get_environ(sock)
+                result = self.application(environ, self._start_response)
+                self.finish_response(result, sock)
+                self._client_sock.remove(sock)
+                new_sock = socket.socket()
+                new_sock.settimeout(self._timeout)
+                new_sock.bind((None, self.port))
+                new_sock.listen()
+                add_sock.append(new_sock)
+        self._client_sock.extend(add_sock)
+
+    def finish_response(self, result, client):
+        """
+        Called after the application callable returns result data to respond with.
+        Creates the HTTP Response payload from the response_headers and results data,
+        and sends it back to client.
+
+        :param string result: the data string to send back in the response to the client.
+        :param Socket client: the socket to send the response to.
+        """
+        try:
+            response = "HTTP/1.1 {0}\r\n".format(self._response_status)
+            for header in self._response_headers:
+                response += "{0}: {1}\r\n".format(*header)
+            response += "\r\n"
+            client.send(response.encode("utf-8"))
+            for data in result:
+                if isinstance(data, bytes):
+                    client.send(data)
+                else:
+                    client.send(data.encode("utf-8"))
+            gc.collect()
+        finally:
+            client.disconnect()
+            client.close()
+
+    def _start_response(self, status, response_headers):
+        """
+        The application callable will be given this method as the second param
+        This is to be called before the application callable returns, to signify
+        the response can be started with the given status and headers.
+
+        :param string status: a status string including the code and reason. ex: "200 OK"
+        :param list response_headers: a list of tuples to represent the headers.
+            ex ("header-name", "header value")
+        """
+        self._response_status = status
+        self._response_headers = [("Server", "w5kWSGIServer")] + response_headers
+
+    def _get_environ(self, client):
+        """
+        The application callable will be given the resulting environ dictionary.
+        It contains metadata about the incoming request and the request body ("wsgi.input")
+
+        :param Socket client: socket to read the request from
+        """
+        env = {}
+        line = str(client.readline(), "utf-8")
+        (method, path, ver) = line.rstrip("\r\n").split(None, 2)
+
+        env["wsgi.version"] = (1, 0)
+        env["wsgi.url_scheme"] = "http"
+        env["wsgi.multithread"] = False
+        env["wsgi.multiprocess"] = False
+        env["wsgi.run_once"] = False
+
+        env["REQUEST_METHOD"] = method
+        env["SCRIPT_NAME"] = ""
+        env["SERVER_NAME"] = _the_interface.pretty_ip(_the_interface.ip_address)
+        env["SERVER_PROTOCOL"] = ver
+        env["SERVER_PORT"] = self.port
+        if path.find("?") >= 0:
+            env["PATH_INFO"] = path.split("?")[0]
+            env["QUERY_STRING"] = path.split("?")[1]
+        else:
+            env["PATH_INFO"] = path
+
+        headers = {}
+        while True:
+            header = str(client.readline(), "utf-8")
+            if header == "":
+                break
+            title, content = header.split(": ", 1)
+            headers[title.lower()] = content
+
+        if "content-type" in headers:
+            env["CONTENT_TYPE"] = headers.get("content-type")
+        if "content-length" in headers:
+            env["CONTENT_LENGTH"] = headers.get("content-length")
+            body = client.recv(int(env["CONTENT_LENGTH"]))
+            env["wsgi.input"] = io.StringIO(body)
+        else:
+            body = client.recv()
+            env["wsgi.input"] = io.StringIO(body)
+        for name, value in headers.items():
+            key = "HTTP_" + name.replace("-", "_").upper()
+            if key in env:
+                value = "{0},{1}".format(env[key], value)
+            env[key] = value
+
+        return env