Grasscutters/mitmproxy
3
0
Fork 0
mirror of https://github.com/Grasscutters/mitmproxy.git synced 2024-11-26 18:18:25 +00:00
Code Issues Projects Releases Wiki Activity

maplocal addon: add mimetype guessing

Browse Source
This commit is contained in:
Martin Plattner 2020-07-07 23:19:29 +02:00
parent 1b6b3cd96c
commit 7022a54737
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
mitmproxy/addons/maplocal.py
View File

@ -1,3 +1,4 @@
import mimetypes
import re
import typing
import urllib
@ -77,6 +78,14 @@ class MapLocal:
return candidates
def get_mime_type(self, file_path):
mimetype = (
mimetypes.guess_type(file_path)[0]
or "text/plain"
)
return mimetype
def request(self, flow: http.HTTPFlow) -> None:
if flow.reply and flow.reply.has_message:
return
@ -84,8 +93,10 @@ class MapLocal:
req = flow.request
url = req.pretty_url
base_path = Path(spec.replacement_str)
if spec.matches(flow) and re.search(spec.subject, url.encode("utf8", "surrogateescape")):
file_candidates = self.file_candidates(url, spec)
for file_candidate in file_candidates:
file_candidate = Path(file_candidate)
if self.sanitize_candidate_path(file_candidate, base_path):
@ -97,8 +108,7 @@ class MapLocal:
return
flow.response = http.HTTPResponse.make(
200, # (optional) status code
replacement, # (optional) content
# todo: guess mime type
{"Content-Type": "image/jpeg"} # (optional) headers
200,
replacement,
{"Content-Type": self.get_mime_type(str(file_candidate))}
)