Exemple #1
0
    def handle_request(self, listener_name, req, sock, addr):
        request_start = datetime.now()
        environ = {}
        resp = None
        try:
            self.cfg.pre_request(self, req)
            resp, environ = wsgi.create(req, sock, addr,
                    listener_name, self.cfg)
            environ["wsgi.multithread"] = True
            self.nr += 1
            if self.alive and self.nr >= self.max_requests:
                self.log.info("Autorestarting worker after current request.")
                resp.force_close()
                self.alive = False

            if not self.cfg.keepalive:
                resp.force_close()

            respiter = self.wsgi(environ, resp.start_response)
            if self.is_already_handled(respiter):
                return False
            try:
                if isinstance(respiter, environ['wsgi.file_wrapper']):
                    resp.write_file(respiter)
                else:
                    for item in respiter:
                        resp.write(item)
                resp.close()
                request_time = datetime.now() - request_start
                self.log.access(resp, req, environ, request_time)
            finally:
                if hasattr(respiter, "close"):
                    respiter.close()
            if resp.should_close():
                raise StopIteration()
        except StopIteration:
            raise
        except EnvironmentError:
            # If the original exception was a socket.error we delegate
            # handling it to the caller (where handle() might ignore it)
            six.reraise(*sys.exc_info())
        except Exception:
            if resp and resp.headers_sent:
                # If the requests have already been sent, we should close the
                # connection to indicate the error.
                self.log.exception("Error handling request")
                try:
                    sock.shutdown(socket.SHUT_RDWR)
                    sock.close()
                except EnvironmentError:
                    pass
                raise StopIteration()
            raise
        finally:
            try:
                self.cfg.post_request(self, req, environ, resp)
            except Exception:
                self.log.exception("Exception in post_request hook")
        return True
Exemple #2
0
 def handle(self,
            listener,
            client,
            addr,
            RequestParser=http.RequestParser,
            util_close=util.close):
     req = None
     try:
         parser = RequestParser(self.cfg, client)
         try:
             listener_name = listener.getsockname()
             if not self.cfg.keepalive:
                 req = next(parser)
                 self.handle_request(listener_name, req, client, addr)
             else:
                 # keepalive loop
                 proxy_protocol_info = {}
                 while True:
                     req = None
                     with self.timeout_ctx():
                         req = next(parser)
                     if not req:
                         break
                     if req.proxy_protocol_info:
                         proxy_protocol_info = req.proxy_protocol_info
                     else:
                         req.proxy_protocol_info = proxy_protocol_info
                     self.handle_request(listener_name, req, client, addr)
         except http.errors.NoMoreData as e:
             self.log.debug("Ignored premature client disconnection. %s", e)
         except StopIteration as e:
             self.log.debug("Closing connection. %s", e)
         except ssl.SSLError:
             # pass to next try-except level
             six.reraise(*sys.exc_info())
         except EnvironmentError:
             # pass to next try-except level
             six.reraise(*sys.exc_info())
         except Exception as e:
             self.handle_error(req, client, addr, e)
     except ssl.SSLError as e:
         if e.args[0] == ssl.SSL_ERROR_EOF:
             self.log.debug("ssl connection closed")
             client.close()
         else:
             self.log.debug("Error processing SSL request.")
             self.handle_error(req, client, addr, e)
     except EnvironmentError as e:
         if e.errno not in (errno.EPIPE, errno.ECONNRESET):
             self.log.exception("Socket error processing request.")
         else:
             if e.errno == errno.ECONNRESET:
                 self.log.debug("Ignoring connection reset")
             else:
                 self.log.debug("Ignoring EPIPE")
     except Exception as e:
         self.handle_error(req, client, addr, e)
     finally:
         util_close(client)
Exemple #3
0
def _wrap_error(exc, mapping, key):
    if key not in mapping:
        return
    new_err_cls = mapping[key]
    new_err = new_err_cls(*exc.args)

    # raise a new exception with the original traceback
    six.reraise(
        new_err_cls, new_err, exc.__traceback__ if hasattr(
            exc, '__traceback__') else sys.exc_info()[2])
Exemple #4
0
    def handle_request(self,
                       listener_name,
                       req,
                       sock,
                       addr,
                       ALREADY_HANDLED=ALREADY_HANDLED):
        environ = {}
        resp = None
        try:
            resp, environ = wsgi.create(req, sock, addr, listener_name,
                                        self.cfg)
            environ["wsgi.multithread"] = True
            self.nr += 1

            if not self.cfg.keepalive:
                resp.force_close()

            respiter = self.wsgi(environ, resp.start_response)
            if respiter == ALREADY_HANDLED:
                return False
            try:
                if isinstance(respiter, environ['wsgi.file_wrapper']):
                    resp.write_file(respiter)
                else:
                    for item in respiter:
                        resp.write(item)
                resp.close()
            finally:
                if hasattr(respiter, "close"):
                    respiter.close()
            if resp.should_close():
                raise StopIteration()
        except StopIteration:
            raise
        except EnvironmentError:
            # If the original exception was a socket.error we delegate
            # handling it to the caller (where handle() might ignore it)
            six.reraise(*sys.exc_info())
        except Exception:
            if resp and resp.headers_sent:
                # If the requests have already been sent, we should close the
                # connection to indicate the error.
                self.log.exception("Error handling request")
                try:
                    sock.shutdown(socket.SHUT_RDWR)
                    sock.close()
                except EnvironmentError:
                    pass
                raise StopIteration()
            raise

        return True
Exemple #5
0
    def start_response(self, status, headers, exc_info=None):
        if exc_info:
            try:
                if self.status and self.headers_sent:
                    reraise(exc_info[0], exc_info[1], exc_info[2])
            finally:
                exc_info = None
        elif self.status is not None:
            raise AssertionError("Response headers already set!")

        self.status = status

        # get the status code from the response here so we can use it to check
        # the need for the connection header later without parsing the string
        # each time.
        try:
            self.status_code = int(self.status.split()[0])
        except ValueError:
            self.status_code = None

        self.process_headers(headers)
        self.chunked = self.is_chunked()
        return self.write