from pwn import * def submit(flags): accepted_flags, rejected_flags = {}, {} r = remote('submission.ctf.saarland', 31337) for flag in flags: r.sendline(flag.encode()) response = r.recvline().decode().strip() if response.startswith('[OK]'): accepted_flags[flag] = response else: rejected_flags[flag] = response return accepted_flags, rejected_flags