Initial commit

receiver/challenges.old/Hirnfick.py

@@ -0,0 +1,42 @@
+import requests
+from base64 import b64decode
+
+from .Challenge import Challenge
+
+
+class Hirnfick(Challenge):
+    flag_location = 'flags/hirnfick.txt'
+    history_location = 'history/hirnfick.txt'
+
+    def distribute(self, flag):
+        try:
+            with open(self.flag_location, 'w') as f:
+                f.write(flag)
+
+            with open(self.history_location, 'a') as f:
+                f.write(flag + '\n')
+
+            self.logger.info(f'Flag {flag} written to {self.flag_location}')
+            return True
+
+        except Exception as e:
+            self.logger.error(
+                f'Could not write flag to {self.flag_location}: {e}')
+            return False
+
+    def check(self):
+        try:
+            res = requests.post(
+                f"http://localhost:{self.port}/api/run",
+                timeout=5,
+                json={
+                    "code":
+                    "+[-->-[>>+>-----<<]<--<---]>-.>>>+.>>..+++[.>]<<<<.+++.------.<<-.>>>>+."
+                })
+
+            assert b64decode(res.json()["output"]) == b"HirnFick 1.0\nHello, World!"
+
+            return True
+        except Exception as e:
+            self.logger.error(f'Could not check hirnfick: {e}')
+            return False