Function test_tun_mode

test/mitmproxy/proxy/test_mode_servers.py:429–454

(monkeypatch, caplog)

@skip_not_linux
async def test_tun_mode(monkeypatch, caplog):
    monkeypatch.setattr(ConnectionHandler, "handle_client", _echo_server)

    with taddons.context(Proxyserver()):
        inst = TunInstance.make(f"tun", MagicMock())
        assert inst.tun_name is None
        try:
            await inst.start()
        except RuntimeError as e:
            if "Operation not permitted" in str(e):
                return pytest.skip("tun mode test must be run as root")
            raise
        assert inst.tun_name
        assert inst.is_running
        assert "tun_name" in inst.to_json()

        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        s.setsockopt(socket.SOL_SOCKET, socket.SO_BINDTODEVICE, inst.tun_name.encode())
        await asyncio.get_running_loop().sock_connect(s, ("192.0.2.1", 1234))
        reader, writer = await asyncio.open_connection(sock=s)
        writer.write(b"hello")
        await writer.drain()
        assert await reader.readexactly(5) == b"HELLO"
        writer.close()
        await writer.wait_closed()
        await inst.stop()


async def test_tun_mode_mocked(monkeypatch):