Initial commit

receiver/challenges/Niko.py

@@ -0,0 +1,93 @@
+from .Challenge import Challenge
+
+import io
+import requests
+import random
+import subprocess
+import json
+
+class Niko(Challenge):
+    flag_location = 'flags/niko.txt'
+    history_location = 'history/niko.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):
+        error_messages = [
+            "あなたはどんなオタクですか",
+            "冗談じゃないよ！",
+            "ｔｃｈ なんだよ こいつ",
+            "どうしてそんなことが可能でしょうか…不可能です",
+            "本当のあなたは何ですか？",
+            "うわー、ごめんなさい",
+            "御心のままに、主よ",
+            "もういいよ、やめて！",
+            "時間です",
+            "悪くないよ。"
+        ]
+        try:
+            # Step 1: Check if the flag still exists and matches the one in the container
+            with open(self.flag_location, 'r') as f:
+                host_flag = f.read().strip()
+
+            container_flag = subprocess.run(
+                ["docker", "exec", "niko_container", "cat", "/flag.txt"],
+                capture_output=True,
+                text=True
+            ).stdout.strip()                    
+
+            assert host_flag == container_flag, 'Flag mismatch between host and container'
+            
+            self.logger.info('Flag check passed for niko')
+
+            # Step 2: Check if can access flag
+            container_key = subprocess.run(
+                ["docker", "exec", "niko_container", "cat", "/opt/flag"],
+                capture_output=True,
+                text=True
+            ).stdout.strip()   
+            urlFlag = f'http://localhost:{self.port}/api/getFlag?flag={container_key}'
+            expected_output = container_flag
+            response2 = requests.get(urlFlag)
+            assert response2.text.strip() == expected_output, 'Get Flag check does not work'
+            self.logger.info('Get flag endpoint check passed for niko')
+            
+            # Step 3: Check if the webpage can be accessed
+            url = f'http://localhost:{self.port}/'
+            response = requests.get(url)
+            status_code = response.status_code
+            assert status_code == 200, 'Webpage is not accessible'
+            self.logger.info('Webpage accessibility check passed for niko')
+
+            # Step 4: Check if the output of the specific URL equals the expected string
+            urlFlag = f'http://localhost:{self.port}/api/getFlag?flag=1'
+            expected_output = "(⋟﹏⋞) 私をバカにしようとしているのか (´ ͡༎ຶ ͜ʖ ͡༎ຶ `)︵‿︵"
+            response2 = requests.get(urlFlag)
+            assert response2.text.strip() == expected_output, 'Webpage output does not match expected output'
+            self.logger.info('Webpage output check passed for niko')
+            
+            # Step 5: Check if the chat endpoint is working
+            urlChat = f'http://localhost:{self.port}/api/chat'
+            data = 'test'
+            response3 = requests.post(urlChat, data=data)
+            assert json.loads(response3.text.strip()).get("output") in error_messages, 'Api endpoint is not accessible'
+            self.logger.info('Webpage delay check passed for niko')
+            
+            return True
+
+        except Exception as e:
+            self.logger.error(f'Could not check niko: {e}')
+            return False