saarctf

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