Skip to main content

mitmproxy addon client for ProxyLens Server

Project description

ProxyLens for mitmproxy

The PyPI package is proxylens-mitmproxy and the Python import package is proxylens_mitmproxy. It captures HTTP and WebSocket traffic as normalized ProxyLens Server events.

It does three main things:

  • propagates X-ProxyLens-HopChain
  • generates a fresh X-ProxyLens-RequestId per observed hop
  • submits request, response, body, trailer, websocket, and error events to a server client

It can also optionally limit how many requests are allowed through the proxy at the same time, which is useful when you want more deterministic capture order from applications that cannot propagate ProxyLens headers themselves.

Setup

This subproject is self-contained under mitmproxy_addon/.

cd mitmproxy_addon
uv sync --dev

Run tests with:

uv run pytest

Basic Addon Usage

ProxyLens() now defaults to a real HTTP client for ProxyLens Server.

Current built-in clients:

  • RecordingProxyLensServerClient: test/dummy client that records uploads and events in memory
  • ProxyLensServerClient: real HTTP client for the server write API

Minimal example:

from proxylens_mitmproxy import ProxyLens

addon = ProxyLens(
    node_name="proxy-a",
    max_concurrent_requests_per_host=1,
)

addons = [addon]

If you do not pass node_name, the addon reads it from PROXYLENS_NODE_NAME.

export PROXYLENS_NODE_NAME=proxy-a

If you do not inject a client, the addon uses ProxyLensServerClient, which resolves the server base URL from PROXYLENS_SERVER_BASE_URL and otherwise defaults to http://127.0.0.1:8000.

export PROXYLENS_SERVER_BASE_URL=http://127.0.0.1:8000

You can still inject a custom client explicitly:

from proxylens_mitmproxy import ProxyLens, ProxyLensServerClient

addon = ProxyLens(
    client=ProxyLensServerClient(base_url="http://127.0.0.1:8000"),
    node_name="proxy-a",
)

max_concurrent_requests_per_host=None keeps the default unlimited behavior. Set it to 1 to force strict per-host serialization, or to a larger integer to allow a small bounded number of in-flight requests per host. WebSocket upgrade requests do not count toward this limit.

If you do not pass max_concurrent_requests_per_host, the addon reads it from PROXYLENS_MAX_CONCURRENT_REQUESTS_PER_HOST when set.

export PROXYLENS_MAX_CONCURRENT_REQUESTS_PER_HOST=1

Running In mitmproxy

Create a small loader script, for example run_proxylens_mitmproxy.py:

from proxylens_mitmproxy import ProxyLens

addons = [
    ProxyLens(
        node_name="proxy-a",
    )
]

Then run mitmproxy or mitmdump with that script:

cd mitmproxy_addon
uv run mitmdump -s run_proxylens_mitmproxy.py

Behavior

For each request seen by the current mitmproxy process, the addon:

  1. reads any inbound X-ProxyLens-HopChain
  2. appends the current node name or starts a new trace using a trace id extracted from traceparent, B3, or Jaeger headers when available
  3. replaces any inbound X-ProxyLens-RequestId with a fresh ULID
  4. emits normalized capture events in request-local event_index order
  5. uploads binary body chunks and binary websocket payloads before emitting referencing events

The addon currently captures:

  • request metadata
  • request body chunks
  • request trailers when exposed by mitmproxy
  • request completion
  • response metadata
  • response body chunks
  • response trailers when exposed by mitmproxy
  • response completion
  • websocket start, message, and end events
  • request error events

Dependency Injection

ProxyLens accepts injectable collaborators for deterministic tests and custom runtime behavior:

ProxyLens(
    client=...,
    node_name="proxy-a",
    server_base_url=...,
    trace_id_generator=...,
    request_id_generator=...,
    blob_id_generator=...,
    flow_filter=...,
    max_concurrent_requests_per_host=...,
)

flow_filter(flow) can be used to skip capture for selected flows. server_base_url is only used when client is not injected. If it is unset and PROXYLENS_SERVER_BASE_URL is also unset, the addon disables itself and becomes an early no-op. max_concurrent_requests_per_host limits how many flows can be active at once for each destination host; excess flows for that host are queued inside mitmproxy until a slot becomes available. WebSocket upgrade flows are excluded from that accounting.

In-Process Test Harness

The package includes a mitmproxy-native test harness for fast integration tests:

from mitmproxy import http
from proxylens_mitmproxy import (
    ProxyLens,
    RecordingProxyLensServerClient,
    TestMitmProxy,
)


def handler(flow: http.HTTPFlow) -> None:
    flow.response = http.Response.make(200, b"ok")


client = RecordingProxyLensServerClient()
addon = ProxyLens(client=client, node_name="proxy-a")

with TestMitmProxy(proxy_lens=addon, handler=handler) as proxy:
    flow = proxy.request("GET", "https://example.test/")

assert flow.response is not None
assert client.events[0]["type"] == "http_request_started"

TestMitmProxy provides:

  • request(...) / send(...) for sync tests
  • arequest(...) / asend(...) for async tests
  • real mitmproxy.http.HTTPFlow objects
  • responder exception surfacing instead of silent logging

Package Layout

mitmproxy_addon/
├── pyproject.toml
├── README.md
├── src/proxylens_mitmproxy/
└── tests/

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

proxylens_mitmproxy-0.4.1.tar.gz (29.6 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

proxylens_mitmproxy-0.4.1-py3-none-any.whl (22.5 kB view details)

Uploaded Python 3

File details

Details for the file proxylens_mitmproxy-0.4.1.tar.gz.

File metadata

  • Download URL: proxylens_mitmproxy-0.4.1.tar.gz
  • Upload date:
  • Size: 29.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.14.3

File hashes

Hashes for proxylens_mitmproxy-0.4.1.tar.gz
Algorithm Hash digest
SHA256 edb6c1a9fe16a31ae4888f99dd37dd4d43a8ad8fde4a74e3a4fcbd483c354d4e
MD5 4c7596c167775e90ffb9b2116dfa671f
BLAKE2b-256 b91ca562a0a9be39c8691aab43b8c703ef0cc5be0690a0610fbe897280e681da

See more details on using hashes here.

File details

Details for the file proxylens_mitmproxy-0.4.1-py3-none-any.whl.

File metadata

File hashes

Hashes for proxylens_mitmproxy-0.4.1-py3-none-any.whl
Algorithm Hash digest
SHA256 e976e4736d4bedeaed3a38f6f567d9e5c89042539824026f199ceb46adfd4e92
MD5 5b8aad5ff28a089db6ed9790488e3360
BLAKE2b-256 c151c822db673bfddc837c7d19d66061333151a1f68a55be2f402e7f3ed5159b

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page