如何在mitmproxy插件中处理3XX重定向且不告知客户端？

要实现客户端无感知的后端重定向处理，核心思路是在mitmproxy收到后端的3XX响应后，自动跟进完整的重定向链，直到获取最终的非3XX响应，再将该响应返回给客户端。以下是具体实现方式：

核心逻辑 #

• 用response钩子拦截后端返回的3XX响应
• 基于重定向目标地址构造新请求，继承原请求的上下文（如Cookie、请求头等）
• 循环或递归跟进重定向，直到拿到最终非3XX响应
• 替换当前请求流的响应为最终结果，客户端全程无感知

代码示例 #

from mitmproxy import ctx, http
from urllib.parse import urljoin

MAX_REDIRECTS = 5  # 限制最大重定向次数，避免死循环

def follow_redirect(flow: http.HTTPFlow, redirect_count: int = 0) -> http.HTTPResponse:
    if redirect_count >= MAX_REDIRECTS:
        ctx.log.error(f"达到最大重定向次数({MAX_REDIRECTS})，终止跟进")
        return flow.response

    # 判断是否为需要处理的3XX重定向
    if 300 <= flow.response.status_code < 400 and flow.response.headers.get("Location"):
        redirect_url = urljoin(flow.request.url, flow.response.headers["Location"])
        ctx.log.info(f"跟进重定向: {flow.request.url} -> {redirect_url}")

        # 构造新请求，复用原请求的关键属性
        new_request = http.HTTPRequest(
            method=flow.request.method,
            url=redirect_url,
            headers=flow.request.headers,
            content=flow.request.content,
            trailers=flow.request.trailers,
            timestamp_start=flow.request.timestamp_start
        )

        # 创建新请求流并发起请求
        new_flow = http.HTTPFlow(flow.client_conn, http.ServerConnection.make(new_request.host))
        new_flow.request = new_request
        try:
            # 通过mitmproxy内部命令发送请求
            ctx.master.commands.call("replay.client", [new_flow])
            # 递归处理后续可能的重定向
            return follow_redirect(new_flow, redirect_count + 1)
        except Exception as e:
            ctx.log.error(f"跟进重定向失败: {str(e)}")
            return flow.response
    else:
        # 返回最终非3XX响应
        return flow.response

def response(flow: http.HTTPFlow) -> None:
    # 跳过mitmproxy内部发起的重定向请求，避免重复处理
    if hasattr(flow, "_is_internal_redirect"):
        return

    final_response = follow_redirect(flow)
    # 替换当前流的响应为最终结果
    flow.response = final_response
    # 标记当前流已处理重定向
    setattr(flow, "_is_internal_redirect", True)

关键细节说明 #

• 重定向次数限制：设置MAX_REDIRECTS防止循环重定向导致的无限递归，保障进程稳定。
• 请求上下文复用：新请求继承原请求的Cookie、请求头等信息，确保重定向请求的合法性（比如携带登录状态）。
• 内部请求标记：通过_is_internal_redirect属性区分mitmproxy内部发起的重定向请求，避免在response钩子中重复处理。
• URL合法性处理：用urljoin处理相对路径形式的Location头，保证重定向地址的正确性。

该方案可自动处理任意次数的后端重定向，客户端只会收到最终的非3XX响应，完全感知不到中间的重定向过程。

内容的提问来源于stack exchange，提问作者Mr. Curious