From c4f15e7e75ad6392e9227731a9b3b612e1da5a99 Mon Sep 17 00:00:00 2001
From: shelld3v <59408894+shelld3v@users.noreply.github.com>
Date: Wed, 13 Nov 2024 21:55:40 +0700
Subject: [PATCH] Move warning to dirsearch.py

---
 dirsearch.py                 | 5 +++++
 lib/controller/controller.py | 5 -----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/dirsearch.py b/dirsearch.py
index dcd539b69..97635d60f 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 88be4dc57..bef0c7446 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: