fix win message

pwncollege · Nov 5, 2024 · 8b659f5 · 8b659f5
1 parent 6876180
commit 8b659f5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pwnshop/challenge.py b/pwnshop/challenge.py
@@ -63,7 +63,7 @@ class Challenge:
     vbuf_in_constructor = False
     print_greeting = True
     constant_goodbye = True
-    win_message = "You win! Here is your flag:\n"
+    win_message = "You win! Here is your flag:"
     static_win_function_variables = True
 
     build_image = None

diff --git a/pwnshop/templates/base/base.c b/pwnshop/templates/base/base.c
@@ -72,7 +72,7 @@
       if (token != 0x1337) return;
     {% endif %}
 
-    {% if challenge.win_message %}printf({{win_message}});{% endif %}
+    {% if challenge.win_message %}puts("{{challenge.win_message}}");{% endif %}
     flag_fd = open("/flag", 0);
     if (flag_fd < 0) {
         printf("\n  ERROR: Failed to open the flag -- %s!\n", strerror(errno));