feat: improve flag format

codegen/generate.py

@@ -47,17 +47,18 @@ def get_check_code(flag_part, user_part, c_mangler):
 
 def main():
     if len(sys.argv) != 3:
-        eprint(f"Usage: {sys.argv[0]} <flag_prefix> <flag_length>")
+        eprint(f"Usage: {sys.argv[0]} <flag_format> <flag_length>")
         exit(1)
 
     # Flag prefix is not used, currently the flag is just used as-is
-    flag_prefix = sys.argv[1]
+    flag_format = sys.argv[1]
     flag_len = int(sys.argv[2])
 
-    eprint(f"Generating flag with prefix {flag_prefix} and length {flag_len}")
-
     # Generate a secret
-    secret = secrets.token_hex(flag_len)
+    secret = secrets.token_urlsafe(flag_len)
+    secret = flag_format.replace("FLAG_GOES_HERE", secret)
+
+    eprint(f"Generated Flag: {secret}")
 
     # Change the bytes of the secret
     mangled_secret, c_manglers = mangle(secret)