diff options
Diffstat (limited to 'fast/submitter.py')
| -rw-r--r-- | fast/submitter.py | 16 |
1 files changed, 11 insertions, 5 deletions
diff --git a/fast/submitter.py b/fast/submitter.py index dcaffe7..a5df7cb 100644 --- a/fast/submitter.py +++ b/fast/submitter.py @@ -1,7 +1,13 @@ -import requests +from pwn import * def submit(flags): - flag_responses = requests.post('http://example.ctf/flags', json=flags).json() - accepted_flags = { item['flag']: item['response'] for item in flag_responses if item['response'].endswith('OK') } - rejected_flags = { item['flag']: item['response'] for item in flag_responses if not item['response'].endswith('OK') } - return accepted_flags, rejected_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 |