feat(scanner): add timeout input for direct scanning methods

Author: ayanrajpoot10

bugscanx/modules/scanners/host_scanner.py

@@ -40,6 +40,7 @@ def get_input_direct(no302=False):
         return None, None, None
 
     port_list = get_input("Enter port(s)", "number", default="80").split(',')
+    timeout = get_input("Enter timeout (seconds)", "number", default="3")
     output, threads = get_common_inputs()
     method_list = get_input(
         "Select HTTP method(s)",
@@ -63,6 +64,7 @@ def get_input_direct(no302=False):
             cidr_ranges=cidr_ranges,
             port_list=port_list,
             no302=no302,
+            timeout=int(timeout),
             output_file=output
         )
     else:
@@ -72,6 +74,7 @@ def get_input_direct(no302=False):
             input_file=filename,
             port_list=port_list,
             no302=no302,
+            timeout=int(timeout),
             output_file=output
         )
 

bugscanx/modules/scanners/scanners/direct.py

@@ -18,16 +18,18 @@ def __init__(
         method_list=None,
         port_list=None,
         no302=False,
+        timeout=None,
         **kwargs
     ):
         super().__init__(**kwargs)
         self.method_list = method_list or []
         self.port_list = port_list or []
         self.no302 = no302
+        self.timeout = timeout or self.DEFAULT_TIMEOUT
 
     def request(self, method, url, **kwargs):
         method = method.upper()
-        kwargs['timeout'] = self.DEFAULT_TIMEOUT
+        kwargs['timeout'] = self.timeout
         max_attempts = self.DEFAULT_RETRY
 
         for attempt in range(max_attempts):
@@ -98,6 +100,7 @@ def __init__(
         no302=False,
         threads=50,
         output_file=None,
+        timeout=None,
         **kwargs
     ):
         super().__init__(
@@ -107,6 +110,7 @@ def __init__(
             threads=threads,
             is_cidr_input=False,
             output_file=output_file,
+            timeout=timeout,
             **kwargs
         )
         self.input_file = input_file
@@ -164,6 +168,7 @@ def __init__(
         no302=False,
         threads=50,
         output_file=None,
+        timeout=None,
         **kwargs
     ):
         super().__init__(
@@ -174,6 +179,7 @@ def __init__(
             is_cidr_input=True,
             cidr_ranges=cidr_ranges,
             output_file=output_file,
+            timeout=timeout,
             **kwargs
         )
         self.cidr_ranges = cidr_ranges or []