156 lines
4.0 KiB
Python
156 lines
4.0 KiB
Python
import json
|
|
|
|
from sentry_sdk.hub import Hub, _should_send_default_pii
|
|
from sentry_sdk.utils import AnnotatedValue
|
|
from sentry_sdk._compat import text_type, iteritems
|
|
|
|
from sentry_sdk._types import MYPY
|
|
|
|
if MYPY:
|
|
from typing import Any
|
|
from typing import Dict
|
|
from typing import Optional
|
|
from typing import Union
|
|
|
|
|
|
SENSITIVE_ENV_KEYS = (
|
|
"REMOTE_ADDR",
|
|
"HTTP_X_FORWARDED_FOR",
|
|
"HTTP_SET_COOKIE",
|
|
"HTTP_COOKIE",
|
|
"HTTP_AUTHORIZATION",
|
|
"HTTP_X_FORWARDED_FOR",
|
|
"HTTP_X_REAL_IP",
|
|
)
|
|
|
|
SENSITIVE_HEADERS = tuple(
|
|
x[len("HTTP_") :] for x in SENSITIVE_ENV_KEYS if x.startswith("HTTP_")
|
|
)
|
|
|
|
|
|
class RequestExtractor(object):
|
|
def __init__(self, request):
|
|
# type: (Any) -> None
|
|
self.request = request
|
|
|
|
def extract_into_event(self, event):
|
|
# type: (Dict[str, Any]) -> None
|
|
client = Hub.current.client
|
|
if client is None:
|
|
return
|
|
|
|
data = None # type: Optional[Union[AnnotatedValue, Dict[str, Any]]]
|
|
|
|
content_length = self.content_length()
|
|
request_info = event.setdefault("request", {})
|
|
|
|
if _should_send_default_pii():
|
|
request_info["cookies"] = dict(self.cookies())
|
|
|
|
bodies = client.options["request_bodies"]
|
|
if (
|
|
bodies == "never"
|
|
or (bodies == "small" and content_length > 10 ** 3)
|
|
or (bodies == "medium" and content_length > 10 ** 4)
|
|
):
|
|
data = AnnotatedValue(
|
|
"",
|
|
{"rem": [["!config", "x", 0, content_length]], "len": content_length},
|
|
)
|
|
else:
|
|
parsed_body = self.parsed_body()
|
|
if parsed_body is not None:
|
|
data = parsed_body
|
|
elif self.raw_data():
|
|
data = AnnotatedValue(
|
|
"",
|
|
{"rem": [["!raw", "x", 0, content_length]], "len": content_length},
|
|
)
|
|
else:
|
|
return
|
|
|
|
request_info["data"] = data
|
|
|
|
def content_length(self):
|
|
# type: () -> int
|
|
try:
|
|
return int(self.env().get("CONTENT_LENGTH", 0))
|
|
except ValueError:
|
|
return 0
|
|
|
|
def cookies(self):
|
|
raise NotImplementedError()
|
|
|
|
def raw_data(self):
|
|
raise NotImplementedError()
|
|
|
|
def form(self):
|
|
raise NotImplementedError()
|
|
|
|
def parsed_body(self):
|
|
# type: () -> Optional[Dict[str, Any]]
|
|
form = self.form()
|
|
files = self.files()
|
|
if form or files:
|
|
data = dict(iteritems(form))
|
|
for k, v in iteritems(files):
|
|
size = self.size_of_file(v)
|
|
data[k] = AnnotatedValue(
|
|
"", {"len": size, "rem": [["!raw", "x", 0, size]]}
|
|
)
|
|
|
|
return data
|
|
|
|
return self.json()
|
|
|
|
def is_json(self):
|
|
# type: () -> bool
|
|
return _is_json_content_type(self.env().get("CONTENT_TYPE"))
|
|
|
|
def json(self):
|
|
# type: () -> Optional[Any]
|
|
try:
|
|
if self.is_json():
|
|
raw_data = self.raw_data()
|
|
if not isinstance(raw_data, text_type):
|
|
raw_data = raw_data.decode("utf-8")
|
|
return json.loads(raw_data)
|
|
except ValueError:
|
|
pass
|
|
|
|
return None
|
|
|
|
def files(self):
|
|
raise NotImplementedError()
|
|
|
|
def size_of_file(self, file):
|
|
raise NotImplementedError()
|
|
|
|
def env(self):
|
|
raise NotImplementedError()
|
|
|
|
|
|
def _is_json_content_type(ct):
|
|
# type: (str) -> bool
|
|
mt = (ct or "").split(";", 1)[0]
|
|
return (
|
|
mt == "application/json"
|
|
or (mt.startswith("application/"))
|
|
and mt.endswith("+json")
|
|
)
|
|
|
|
|
|
def _filter_headers(headers):
|
|
# type: (Dict[str, str]) -> Dict[str, str]
|
|
if _should_send_default_pii():
|
|
return headers
|
|
|
|
return {
|
|
k: (
|
|
v
|
|
if k.upper().replace("-", "_") not in SENSITIVE_HEADERS
|
|
else AnnotatedValue("", {"rem": [["!config", "x", 0, len(v)]]})
|
|
)
|
|
for k, v in iteritems(headers)
|
|
}
|