diff --git a/dirsearch.py b/dirsearch.py index dcd539b6..97635d60 100755 --- a/dirsearch.py +++ b/dirsearch.py @@ -66,6 +66,11 @@ def main(): options.update(parse_options()) + if options["session_file"]: + print("WARNING: Running an untrusted session file might lead to unwanted code execution!") + if input("[c]ontinue / [q]uit: ") != "c": + exit(1) + from lib.controller.controller import Controller Controller() diff --git a/lib/controller/controller.py b/lib/controller/controller.py index 88be4dc5..bef0c744 100755 --- a/lib/controller/controller.py +++ b/lib/controller/controller.py @@ -72,11 +72,6 @@ class Controller: def __init__(self) -> None: if options["session_file"]: - print("WARNING: Running an untrusted session file might lead to unwanted code execution!") - interface.in_line("[c]continue / [q]uit: ") - if input() != "c": - exit(1) - self._import(options["session_file"]) self.old_session = True else: