Chrome CDPをローカルで起動し、Dockerから接続する際にmitmproxyを使って解決

Chrome CDPをローカルで起動し、Dockerから接続する際にmitmproxyを使って解決 | GGAMI.NET
from mitmproxy import http
import json

TARGET_HOST = "127.0.0.1"
TARGET_PORT = 9222

PROXY_HOST = "host.docker.internal"
PROXY_PORT = 9223

def request(flow: http.HTTPFlow) -> None:
    """HTTP 요청을 수정하여 실제 Chrome 디버그 서버로 포워딩"""
    try:
        print("HTTP request intercepted")
        flow.request.headers["Host"] = "localhost"
        flow.request.host = TARGET_HOST
        flow.request.port = TARGET_PORT
    except Exception as e:
        print(f"Request modification error: {e}")

def websocket_handshake(flow: http.HTTPFlow) -> None:
    """WebSocket 핸드셰이크 요청을 수정하여 실제 Chrome 디버그 서버로 포워딩"""
    try:
        print("WebSocket handshake intercepted")
        flow.request.headers["Host"] = "localhost"
        flow.request.host = TARGET_HOST
        flow.request.port = TARGET_PORT
    except Exception as e:
        print(f"WebSocket handshake modification error: {e}")

def response(flow: http.HTTPFlow) -> None:
    """HTTP 응답을 수정하여 webSocketDebuggerUrl을 mitmproxy 포트으로 변경"""
    try:
        content_type = flow.response.headers.get("Content-Type", "")
        if "application/json" in content_type:
            data = json.loads(flow.response.text)
            if "webSocketDebuggerUrl" in data:
                original_ws_url = data["webSocketDebuggerUrl"]
                # 원래의 WS URL을 변경
                new_ws_url = original_ws_url.replace(
                    f"ws://{TARGET_HOST}:{TARGET_PORT}",
                    f"ws://{PROXY_HOST}:{PROXY_PORT}"
                )
                data["webSocketDebuggerUrl"] = new_ws_url
                flow.response.text = json.dumps(data)
                print(f"Modified webSocketDebuggerUrl: {new_ws_url}")
    except json.JSONDecodeError:
        print("Failed to decode JSON response")
    except Exception as e:
        print(f"Error modifying response: {e}")

"""
mitmproxy --mode reverse:http://127.0.0.1:9222/ --listen-port 9223 -s modify_host.py
"""
ほかの記事
