Function test_quic

test/mitmproxy/proxy/layers/test_modes.py:197–222

(tctx: Context, keep_host_header: bool)

@pytest.mark.parametrize("keep_host_header", [True, False])
def test_quic(tctx: Context, keep_host_header: bool):
    with taddons.context():
        tctx.options.keep_host_header = keep_host_header
        tctx.server.sni = "other.example.com"
        tctx.client.proxy_mode = ProxyMode.parse("reverse:quic://example.org:443")
        client_hello = Placeholder(bytes)

        def set_settings(data: quic.QuicTlsData):
            data.settings = quic.QuicTlsSettings()

        assert (
            Playbook(modes.ReverseProxy(tctx))
            << OpenConnection(tctx.server)
            >> reply(None)
            >> DataReceived(tctx.client, b"\x00")
            << NextLayerHook(Placeholder(NextLayer))
            >> reply_next_layer(layers.ServerQuicLayer)
            << quic.QuicStartServerHook(Placeholder(quic.QuicTlsData))
            >> reply(side_effect=set_settings)
            << SendData(tctx.server, client_hello)
            << RequestWakeup(Placeholder(float))
        )
        assert tctx.server.address == ("example.org", 443)
        assert quic.quic_parse_client_hello_from_datagrams([client_hello()]).sni == (
            "other.example.com" if keep_host_header else "example.org"
        )


def test_udp(tctx: Context):