From 5ad6ad3fcf1ef1e809aadcb19ff4a9d2e9b45cdc Mon Sep 17 00:00:00 2001 From: ElJeffe Date: Tue, 14 Sep 2021 12:24:36 -0400 Subject: [PATCH] Default flag-format This checks to see if any flag-format has been provided and instead of failing with an exception, sets a standard default format, such as `FLAG{my_flag}`. --- katana/repl/__init__.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/katana/repl/__init__.py b/katana/repl/__init__.py index 813d2a2..84f4c18 100644 --- a/katana/repl/__init__.py +++ b/katana/repl/__init__.py @@ -343,6 +343,14 @@ def __init__(self, manager: Manager): # Register hook to update prompt self.register_cmdfinalization_hook(self.finalization_hook) + # Set a default flag-format if none supplied + if self.manager.flag_pattern is None: + flag_format = "S+{.*?}" + self.manager.set("manager", "flag-format", flag_format) + self.poutput( + f"[{Fore.BLUE}+{Style.RESET_ALL}] Using default flag-format `{flag_format}`" + ) + # Start the manager self.manager.start()