Move warning to dirsearch.py

maurosoria · Nov 13, 2024 · c4f15e7 · c4f15e7
1 parent f1a1d30
commit c4f15e7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git 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
@@ -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: