mirror of
https://github.com/Grasscutters/mitmproxy.git
synced 2024-11-30 03:14:22 +00:00
55 lines
2.1 KiB
Python
55 lines
2.1 KiB
Python
"""
|
|
This script simply prints all received HTTP Trailers.
|
|
|
|
HTTP requests and responses can container trailing headers which are sent after
|
|
the body is fully transmitted. Such trailers need to be announced in the initial
|
|
headers by name, so the receiving endpoint can wait and read them after the
|
|
body.
|
|
"""
|
|
|
|
from mitmproxy import http
|
|
from mitmproxy.net.http import Headers
|
|
|
|
|
|
def request(flow: http.HTTPFlow):
|
|
if flow.request.trailers:
|
|
print("HTTP Trailers detected! Request contains:", flow.request.trailers)
|
|
|
|
if flow.request.path == "/inject_trailers":
|
|
if flow.request.is_http10:
|
|
# HTTP/1.0 doesn't support trailers
|
|
return
|
|
elif flow.request.is_http11:
|
|
if not flow.request.content:
|
|
# Avoid sending a body on GET requests or a 0 byte chunked body with trailers.
|
|
# Otherwise some servers return 400 Bad Request.
|
|
return
|
|
# HTTP 1.1 requires transfer-encoding: chunked to send trailers
|
|
flow.request.headers["transfer-encoding"] = "chunked"
|
|
# HTTP 2+ supports trailers on all requests/responses
|
|
|
|
flow.request.headers["trailer"] = "x-my-injected-trailer-header"
|
|
flow.request.trailers = Headers([
|
|
(b"x-my-injected-trailer-header", b"foobar")
|
|
])
|
|
print("Injected a new request trailer...", flow.request.headers["trailer"])
|
|
|
|
|
|
def response(flow: http.HTTPFlow):
|
|
if flow.response.trailers:
|
|
print("HTTP Trailers detected! Response contains:", flow.response.trailers)
|
|
|
|
if flow.request.path == "/inject_trailers":
|
|
if flow.request.is_http10:
|
|
return
|
|
elif flow.request.is_http11:
|
|
if not flow.response.content:
|
|
return
|
|
flow.response.headers["transfer-encoding"] = "chunked"
|
|
|
|
flow.response.headers["trailer"] = "x-my-injected-trailer-header"
|
|
flow.response.trailers = Headers([
|
|
(b"x-my-injected-trailer-header", b"foobar")
|
|
])
|
|
print("Injected a new response trailer...", flow.response.headers["trailer"])
|