Exemple #1
0
 def __init__(self, socket, client_address, server_address, debug=False):
     self.debug = debug
     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 = Parser()
     self.start_response_called = False
     self.log = logging.getLogger(__name__)
Exemple #2
0
 def __init__(self, socket, client_address, server_address, conf):
     self.debug = conf['debug']
     self.conf = conf
     self._sock = socket
 
     self.client_address = client_address
     self.server_address = server_address
     self.response_status = None
     self.response_headers = []
     self._version = 11
     self.parser = Parser.parse_request()
     self.start_response_called = False
     self.log = logging.getLogger(__name__)
     self.response_chunked = False
Exemple #3
0
class Request(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, debug=False):
        self.debug = debug
        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 = Parser()
        self.start_response_called = False
        self.log = logging.getLogger(__name__)

    def read(self):
        environ = {}
        headers = []
        buf = ""
        buf = read_partial(self.socket, CHUNK_SIZE)
        i = self.parser.filter_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.filter_headers(headers, buf)
                if i != -1:
                    break

        self.log.debug("%s", self.parser.status)
        self.log.debug("Headers:\n%s" % headers)

        if self.parser.headers_dict.get("Expect", "").lower() == "100-continue":
            self.socket.send("100 Continue\n")

        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[i:])

        if self.debug:
            # This value should evaluate true if an equivalent application
            # object may be simultaneously invoked by another process, and
            # should evaluate false otherwise. In debug mode we fall to one
            # worker so we comply to pylons and other paster app.
            wsgi_multiprocess = False
        else:
            wsgi_multiprocess = True

        environ = {
            "wsgi.url_scheme": "http",
            "wsgi.input": wsgi_input,
            "wsgi.errors": sys.stderr,
            "wsgi.version": (1, 0),
            "wsgi.multithread": False,
            "wsgi.multiprocess": wsgi_multiprocess,
            "wsgi.run_once": False,
            "SCRIPT_NAME": "",
            "SERVER_SOFTWARE": self.SERVER_VERSION,
            "REQUEST_METHOD": self.parser.method,
            "PATH_INFO": unquote(self.parser.path),
            "QUERY_STRING": self.parser.query_string,
            "RAW_URI": self.parser.raw_path,
            "CONTENT_TYPE": self.parser.headers_dict.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.raw_version,
        }

        for key, value in self.parser.headers:
            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, exc_info=None):
        if exc_info:
            try:
                if self.start_response_called:
                    raise exc_info[0], exc_info[1], exc_info[2]
            finally:
                exc_info = None
        elif self.start_response_called:
            raise AssertionError("Response headers already set!")

        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