Added new option --time-limit for running with a time limit in seco…

…nds (e.g. 3600)

doc/CHANGELOG.md

@@ -1,3 +1,6 @@
+## Version 4.0 (TBA)
+* Added: New option `--time-limit` for running with a time limit in seconds (e.g. 3600).
+
 ## Version 3.9 (2024-01-19)
 * Fixed: Multiple bug-fixes regarding several reported unhandled exceptions.
 * Revised: Minor improvement regarding logging user-supplied command(s) (i.e. `--os-cmd` option) to a file.

setup.py

@@ -17,7 +17,7 @@
 
 setup(
       name='commix',
-      version='3.9-stable',
+      version='4.0-dev',
       description='Automated All-in-One OS Command Injection Exploitation Tool',
       long_description=open('README.md').read(),
       long_description_content_type='text/markdown',

src/core/requests/headers.py

@@ -150,8 +150,9 @@ def https_open(self, req):
 
   opener = _urllib.request.build_opener(connection_handler())
 
-  # if len(settings.HTTP_METHOD) != 0:
-  #   request.get_method = lambda: settings.HTTP_METHOD
+  # Time limit mechanism.
+  if menu.options.time_limit and (time.time() - settings.START_TIME > menu.options.time_limit):
+    raise SystemExit()
 
   _ = False
   response = False

src/utils/menu.py

@@ -103,6 +103,11 @@ def banner():
                 default=None,
                 help="Log all HTTP traffic into a textual file.")
 
+general.add_option("--time-limit",
+                dest="time_limit",
+                type=float,
+                help="Run with a time limit in seconds (e.g. 3600).")
+
 general.add_option("--batch",
                 action="store_true",
                 dest="batch",

src/utils/settings.py

@@ -246,9 +246,9 @@ def sys_argv_errors():
 DESCRIPTION_FULL = "Automated All-in-One OS Command Injection Exploitation Tool"
 DESCRIPTION = "The command injection exploiter"
 AUTHOR  = "Anastasios Stasinopoulos"
-VERSION_NUM = "3.9"
-REVISION = "44"
-STABLE_RELEASE = True
+VERSION_NUM = "4.0"
+REVISION = "1"
+STABLE_RELEASE = False
 VERSION = "v"
 if STABLE_RELEASE:
   VERSION = VERSION + VERSION_NUM + "-stable"
@@ -273,6 +273,8 @@ def sys_argv_errors():
 # Random string generator
 RANDOM_STRING_GENERATOR = ''.join(random.choice(string.ascii_uppercase + string.digits + string.ascii_lowercase) for _ in range(10))
 
+START_TIME = time.time()
+
 # Readline
 READLINE_ERROR = False