Forward many ports

simple-mitm.py

import asyncio

TARGET_IP = '1.2.3.4'

async def print_lines(port, process):
    while True:
        line = await process.stdout.readline()
        if not line:
            print(f"[{port}]: Died?")
            break
        print(f"[{port}]: {line.decode().strip()}")


async def main():
    procs = []
    async with asyncio.TaskGroup() as tg:
        try:
            for port in [8080, 8081, 8082]:
                cmd = f'socat -v TCP-LISTEN:{port},fork,reuseaddr TCP:{TARGET_IP}:{port}'
                procs.append(await asyncio.create_subprocess_shell(
                    cmd, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.STDOUT
                ))
                tg.create_task(print_lines(port, procs[-1]))
        except KeyboardInterrupt:
            print("Stopping")
            for proc in procs:
                proc.terminate()


if __name__ == '__main__':
    asyncio.run(main())