mirror of
https://github.com/frappe/gunicorn.git
synced 2026-01-14 11:09:11 +08:00
138 lines
4.2 KiB
Python
138 lines
4.2 KiB
Python
# -*- coding: utf-8 -
|
|
#
|
|
# This file is part of gunicorn. See the NOTICE file included
|
|
# with this release for more information.
|
|
|
|
|
|
import errno
|
|
from ctypes import *
|
|
import re
|
|
import StringIO
|
|
import socket
|
|
import sys
|
|
from urllib import unquote
|
|
import array
|
|
import logging
|
|
|
|
from gunicorn import __version__
|
|
from gunicorn.http.http_parser import HttpParser
|
|
from gunicorn.http.tee import TeeInput
|
|
from gunicorn.util import CHUNK_SIZE, read_partial
|
|
|
|
|
|
NORMALIZE_SPACE = re.compile(r'(?:\r\n)?[ \t]+')
|
|
|
|
|
|
|
|
def _normalize_name(name):
|
|
return "-".join([w.lower().capitalize() for w in name.split("-")])
|
|
|
|
class RequestError(Exception):
|
|
""" raised when something wrong happend"""
|
|
|
|
|
|
class HTTPRequest(object):
|
|
|
|
SERVER_VERSION = "gunicorn/%s" % __version__
|
|
|
|
DEFAULTS = {
|
|
"wsgi.url_scheme": 'http',
|
|
"wsgi.input": StringIO.StringIO(),
|
|
"wsgi.errors": sys.stderr,
|
|
"wsgi.version": (1, 0),
|
|
"wsgi.multithread": False,
|
|
"wsgi.multiprocess": True,
|
|
"wsgi.run_once": False,
|
|
"SCRIPT_NAME": "",
|
|
"SERVER_SOFTWARE": "gunicorn/%s" % __version__
|
|
}
|
|
|
|
|
|
def __init__(self, socket, client_address, server_address):
|
|
self.socket = socket
|
|
self.client_address = client_address
|
|
self.server_address = server_address
|
|
self.response_status = None
|
|
self.response_headers = {}
|
|
self._version = 11
|
|
self.parser = HttpParser()
|
|
self.start_response_called = False
|
|
|
|
self.log = logging.getLogger(__name__)
|
|
|
|
def read(self):
|
|
environ = {}
|
|
headers = {}
|
|
remain = CHUNK_SIZE
|
|
buf = ""
|
|
buf = read_partial(self.socket, CHUNK_SIZE)
|
|
i = self.parser.headers(headers, buf)
|
|
if i == -1 and buf:
|
|
while True:
|
|
data = read_partial(self.socket, CHUNK_SIZE)
|
|
if not data: break
|
|
buf += data
|
|
i = self.parser.headers(headers, buf)
|
|
if i != -1: break
|
|
|
|
if not headers:
|
|
environ.update(self.DEFAULTS)
|
|
return environ
|
|
|
|
buf = buf[i:]
|
|
|
|
|
|
self.log.info("Got headers:\n%s" % headers)
|
|
|
|
if headers.get('Except', '').lower() == "100-continue":
|
|
self.socket.send("100 Continue\n")
|
|
|
|
if "?" in self.parser.path:
|
|
path_info, query = self.parser.path.split('?', 1)
|
|
else:
|
|
path_info = self.parser.path
|
|
query = ""
|
|
|
|
if not self.parser.content_len and not self.parser.is_chunked:
|
|
wsgi_input = StringIO.StringIO()
|
|
else:
|
|
wsgi_input = TeeInput(self.socket, self.parser, buf)
|
|
|
|
environ = {
|
|
"wsgi.url_scheme": 'http',
|
|
"wsgi.input": wsgi_input,
|
|
"wsgi.errors": sys.stderr,
|
|
"wsgi.version": (1, 0),
|
|
"wsgi.multithread": False,
|
|
"wsgi.multiprocess": True,
|
|
"wsgi.run_once": False,
|
|
"SCRIPT_NAME": "",
|
|
"SERVER_SOFTWARE": self.SERVER_VERSION,
|
|
"REQUEST_METHOD": self.parser.method,
|
|
"PATH_INFO": unquote(path_info),
|
|
"QUERY_STRING": query,
|
|
"RAW_URI": self.parser.path,
|
|
"CONTENT_TYPE": headers.get('Content-Type', ''),
|
|
"CONTENT_LENGTH": str(wsgi_input.len),
|
|
"REMOTE_ADDR": self.client_address[0],
|
|
"REMOTE_PORT": self.client_address[1],
|
|
"SERVER_NAME": self.server_address[0],
|
|
"SERVER_PORT": self.server_address[1],
|
|
"SERVER_PROTOCOL": self.parser.version
|
|
}
|
|
|
|
for key, value in headers.items():
|
|
key = 'HTTP_' + key.upper().replace('-', '_')
|
|
if key not in ('HTTP_CONTENT_TYPE', 'HTTP_CONTENT_LENGTH'):
|
|
environ[key] = value
|
|
return environ
|
|
|
|
def start_response(self, status, response_headers):
|
|
self.response_status = status
|
|
for name, value in response_headers:
|
|
name = _normalize_name(name)
|
|
if not isinstance(value, basestring):
|
|
value = str(value)
|
|
self.response_headers[name] = value.strip()
|
|
self.start_response_called = True
|