Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-29 19:08:44 +00:00
Code Issues Projects Releases Wiki Activity

map local: fix mimetype detection

Browse Source
This commit is contained in:
Martin Plattner 2020-07-10 20:44:54 +02:00
parent 257c178bbe
commit 7781dcb15f
1 changed files with 5 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
mitmproxy/addons/maplocal.py
View File

@ -31,14 +31,6 @@ def parse_map_local_spec(option: str) -> MapLocalSpec:
return MapLocalSpec(filter, regex, path) return MapLocalSpec(filter, regex, path)
def get_mime_type(file_path: str) -> str:
mimetype = (
mimetypes.guess_type(file_path)[0]
or "application/octet-stream"
)
return mimetype
def _safe_path_join(root: Path, untrusted: str) -> Path: def _safe_path_join(root: Path, untrusted: str) -> Path:
"""Join a Path element with an untrusted str. """Join a Path element with an untrusted str.
@ -127,11 +119,15 @@ class MapLocal:
local_file = candidate local_file = candidate
break break
headers = {}
mimetype = mimetypes.guess_type(str(local_file))[0]
if mimetype:
headers = {"Content-Type": mimetype}
if local_file: if local_file:
flow.response = http.HTTPResponse.make( flow.response = http.HTTPResponse.make(
200, 200,
local_file.read_bytes(), local_file.read_bytes(),
{"Content-Type": get_mime_type(str(local_file))} headers
) )
# only set flow.response once, for the first matching rule # only set flow.response once, for the first matching rule
return return