matrix-rust-sdk

"""

A mitmproxy script that blocks and removes well known Matrix server

information.

There are two ways a Matrix server can trigger the client to reconfigure the

homeserver URL:

    1. By responding to a `./well-known/matrix/client` request with a new

    homeserver URL.

    2. By including a new homeserver URL inside the `/login` response.

To run execute it with mitmproxy:

    >>> mitmproxy -s well-known-block.py`

"""

import json

from mitmproxy import http

def request(flow):

    if flow.request.path == "/.well-known/matrix/client":

        flow.response = http.HTTPResponse.make(

            404,  # (optional) status code

            b"Not found",  # (optional) content

            {"Content-Type": "text/html"}  # (optional) headers

        )

def response(flow: http.HTTPFlow):

    if flow.request.path == "/_matrix/client/r0/login":

        if flow.response.status_code == 200:

            body = json.loads(flow.response.content)

            body.pop("well_known", None)

            flow.response.text = json.dumps(body)