mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-27 02:24:18 +00:00
44ac64aa72
This commit introduces MultiDict, a multi-dictionary similar to ODict, but with improved semantics (as in the Headers class). MultiDict fixes a few issues that were present in the Request/Response API. In particular, `request.cookies["foo"] = "bar"` has previously been a no-op, as the cookies property returned a mutable _copy_ of the cookies.
132 lines
4.0 KiB
Python
132 lines
4.0 KiB
Python
from __future__ import absolute_import, print_function, division
|
|
|
|
import warnings
|
|
from email.utils import parsedate_tz, formatdate, mktime_tz
|
|
import time
|
|
|
|
from . import cookies
|
|
from .headers import Headers
|
|
from .message import Message, _native, _always_bytes, MessageData
|
|
from .. import utils
|
|
from ..odict import ODict
|
|
|
|
|
|
class ResponseData(MessageData):
|
|
def __init__(self, http_version, status_code, reason=None, headers=None, content=None,
|
|
timestamp_start=None, timestamp_end=None):
|
|
if not isinstance(headers, Headers):
|
|
headers = Headers(headers)
|
|
|
|
self.http_version = http_version
|
|
self.status_code = status_code
|
|
self.reason = reason
|
|
self.headers = headers
|
|
self.content = content
|
|
self.timestamp_start = timestamp_start
|
|
self.timestamp_end = timestamp_end
|
|
|
|
|
|
class Response(Message):
|
|
"""
|
|
An HTTP response.
|
|
"""
|
|
def __init__(self, *args, **kwargs):
|
|
self.data = ResponseData(*args, **kwargs)
|
|
|
|
def __repr__(self):
|
|
if self.content:
|
|
details = "{}, {}".format(
|
|
self.headers.get("content-type", "unknown content type"),
|
|
utils.pretty_size(len(self.content))
|
|
)
|
|
else:
|
|
details = "no content"
|
|
return "Response({status_code} {reason}, {details})".format(
|
|
status_code=self.status_code,
|
|
reason=self.reason,
|
|
details=details
|
|
)
|
|
|
|
@property
|
|
def status_code(self):
|
|
"""
|
|
HTTP Status Code, e.g. ``200``.
|
|
"""
|
|
return self.data.status_code
|
|
|
|
@status_code.setter
|
|
def status_code(self, status_code):
|
|
self.data.status_code = status_code
|
|
|
|
@property
|
|
def reason(self):
|
|
"""
|
|
HTTP Reason Phrase, e.g. "Not Found".
|
|
This is always :py:obj:`None` for HTTP2 requests, because HTTP2 responses do not contain a reason phrase.
|
|
"""
|
|
return _native(self.data.reason)
|
|
|
|
@reason.setter
|
|
def reason(self, reason):
|
|
self.data.reason = _always_bytes(reason)
|
|
|
|
# FIXME
|
|
@property
|
|
def cookies(self):
|
|
"""
|
|
Get the contents of all Set-Cookie headers.
|
|
|
|
A possibly empty :py:class:`ODict`, where keys are cookie name strings,
|
|
and values are [value, attr] lists. Value is a string, and attr is
|
|
an ODictCaseless containing cookie attributes. Within attrs, unary
|
|
attributes (e.g. HTTPOnly) are indicated by a Null value.
|
|
"""
|
|
ret = []
|
|
for header in self.headers.get_all("set-cookie"):
|
|
v = cookies.parse_set_cookie_header(header)
|
|
if v:
|
|
name, value, attrs = v
|
|
ret.append([name, [value, attrs]])
|
|
return ODict(ret)
|
|
|
|
# FIXME
|
|
@cookies.setter
|
|
def cookies(self, odict):
|
|
values = []
|
|
for i in odict.lst:
|
|
header = cookies.format_set_cookie_header(i[0], i[1][0], i[1][1])
|
|
values.append(header)
|
|
self.headers.set_all("set-cookie", values)
|
|
|
|
def refresh(self, now=None):
|
|
"""
|
|
This fairly complex and heuristic function refreshes a server
|
|
response for replay.
|
|
|
|
- It adjusts date, expires and last-modified headers.
|
|
- It adjusts cookie expiration.
|
|
"""
|
|
if not now:
|
|
now = time.time()
|
|
delta = now - self.timestamp_start
|
|
refresh_headers = [
|
|
"date",
|
|
"expires",
|
|
"last-modified",
|
|
]
|
|
for i in refresh_headers:
|
|
if i in self.headers:
|
|
d = parsedate_tz(self.headers[i])
|
|
if d:
|
|
new = mktime_tz(d) + delta
|
|
self.headers[i] = formatdate(new)
|
|
c = []
|
|
for set_cookie_header in self.headers.get_all("set-cookie"):
|
|
try:
|
|
refreshed = cookies.refresh_set_cookie_header(set_cookie_header, delta)
|
|
except ValueError:
|
|
refreshed = set_cookie_header
|
|
c.append(refreshed)
|
|
if c:
|
|
self.headers.set_all("set-cookie", c)
|