Fix quest flag display (#30)

* flag displays during win * fix * cleanup * cleanup * cleanup --------- Co-authored-by: daniel <daniel>
pwncollege · Dec 16, 2024 · 9ca25c1 · 9ca25c1
1 parent 9bea176
commit 9ca25c1
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 14 deletions.
diff --git a/reverse-engineering/cimg-quest-2/_0/quest.py b/reverse-engineering/cimg-quest-2/_0/quest.py
@@ -199,20 +199,26 @@ def game():
 
         # render everyone
         screen.blank()
+        correct_bytes = ''
         for rx,ry,r,g,b,c in revealed_bytes:
             screen.render_patch_monochrome([ c ], rx, ry, r=r, g=g, b=b)
+            correct_bytes += str(c.decode())
         if hidden_bytes:
             screen.render_patch_monochrome(
-                [ b"?" ], hidden_x, hidden_y,
+                [b"?"], hidden_x, hidden_y,
                 r=random.randrange(256), g=random.randrange(256), b=random.randrange(256)
             )
         else:
-            while True:
-                screen.animate_text("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!", 10, hidden_y)
-                screen.animate_text("!!! CONGRATULATIONS, YOU DID IT !!!", 10, hidden_y+1)
-                screen.animate_text("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!", 10, hidden_y+2)
-                screen.animate_text("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!", 10, hidden_y+1)
-
+            try:
+                while True:
+                    screen.animate_text("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!", 10, hidden_y)
+                    screen.animate_text("!!! CONGRATULATIONS, YOU DID IT !!!", 10, hidden_y + 1)
+                    screen.animate_text(correct_bytes, 10, hidden_y + 2)
+                    screen.animate_text("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!", 10, hidden_y + 1)
+            except KeyboardInterrupt:
+                print(flag.decode(), file=sys.stderr)  # Print decoded flag to stderr
+                break
+
         screen.render_patch_monochrome([ b"B" ], bomb_x, bomb_y)
         screen.render_sprite(our_sprite, x, y)
         screen.flush()

diff --git a/reverse-engineering/cimg-quest-3/_0/quest.py b/reverse-engineering/cimg-quest-3/_0/quest.py
@@ -199,20 +199,26 @@ def game():
 
         # render everyone
         screen.blank()
+        correct_bytes = ''
         for rx,ry,r,g,b,c in revealed_bytes:
             screen.render_patch_monochrome([ c ], rx, ry, r=r, g=g, b=b)
+            correct_bytes += str(c.decode())
         if hidden_bytes:
             screen.render_patch_monochrome(
-                [ b"?" ], hidden_x, hidden_y,
+                [b"?"], hidden_x, hidden_y,
                 r=random.randrange(256), g=random.randrange(256), b=random.randrange(256)
             )
         else:
-            while True:
-                screen.animate_text("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!", 10, hidden_y)
-                screen.animate_text("!!! CONGRATULATIONS, YOU DID IT !!!", 10, hidden_y+1)
-                screen.animate_text("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!", 10, hidden_y+2)
-                screen.animate_text("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!", 10, hidden_y+1)
-
+            try:
+                while True:
+                    screen.animate_text("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!", 10, hidden_y)
+                    screen.animate_text("!!! CONGRATULATIONS, YOU DID IT !!!", 10, hidden_y + 1)
+                    screen.animate_text(correct_bytes, 10, hidden_y + 2)
+                    screen.animate_text("!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!", 10, hidden_y + 1)
+            except KeyboardInterrupt:
+                print(flag.decode(), file=sys.stderr)  # Print decoded flag to stderr
+                break
+
         screen.render_patch_monochrome([ b"B" ], bomb_x, bomb_y)
         screen.render_sprite(our_sprite, x, y)
         screen.flush()