diff options
Diffstat (limited to 'fast/submitter.py')
| -rw-r--r-- | fast/submitter.py | 30 |
1 files changed, 19 insertions, 11 deletions
diff --git a/fast/submitter.py b/fast/submitter.py index a5df7cb..1c7c21d 100644 --- a/fast/submitter.py +++ b/fast/submitter.py @@ -1,13 +1,21 @@ -from pwn import * +import requests + +TEAM_TOKEN = "0574cec7b21a0cba8f1b2efb5ca8fac5" + 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 + accepted_flags, rejected_flags = {}, {} + response = requests.put( + "http://10.10.0.1:8080/flags", + headers={"X-Team-Token": TEAM_TOKEN}, + json=flags, + ) + + responses = response.json() + + for flag in responses: + if flag["status"]: + accepted_flags[flag["flag"]] = flag["msg"] + else: + rejected_flags[flag["flag"]] = flag["msg"] + return accepted_flags, rejected_flags |