refactoring run tool objection

Author: ByteSnipers

CHANGELOG.md

@@ -1,5 +1,8 @@
 # ChangeLog
 
+### 0.7.35
+- CHANGE: 	tool objection
+
 ### 0.7.34
 - BUGFIX: 	fixed issue with special chars in apk filename 
 - ADDED: 	new pentest folder structure

mpt/console.py

@@ -15,7 +15,7 @@
 
 from mpt import functions
 
-__version__ = '0.7.34'
+__version__ = '0.7.35'
 
 from mpt import settings, logger
 from mpt.config import Config
@@ -500,6 +500,35 @@ def print_run_tool_usage():
     print("  if you pass the argument to file location please use absolute path !!!")
 
 
+def run_tool_objection():
+    log.info("Execute the followings steps")
+    log.info(f"1) Please run 'frida-ps -U' to find the app name")
+    log.info(f"2) start objection with the following command: 'objection --gadget \"<APP-NAME>\" explore'")
+
+    conf = Config()
+    package = conf.load_config('package-name')
+    application_label = conf.load_config('application-label')
+
+    app = f"\"{application_label}\" ({package})"
+    options = [f"run objection for app {app}", "run objection"]
+    terminal_menu = TerminalMenu(options, title="Please select an option:")
+    menu_entry_index = terminal_menu.show()
+
+    command = settings.ANDROID_TOOLS['objection']['bin']
+    if menu_entry_index == 0:
+        objection_command = f"objection --gadget \\\"{application_label}\\\" explore"
+        command = command.rsplit(";", 1)[0] + f"; {objection_command}; exec $SHELL"
+        log_info = f"Running command in {Style.BRIGHT + Fore.CYAN}kitty terminal{Style.RESET_ALL} {objection_command}"
+    if menu_entry_index == 1:
+        # replace last objection mit exec $SHELL
+        command = command.rsplit(";", 1)[0] + "; objection --help; exec $SHELL"
+        log_info = f"Running command in {Style.BRIGHT + Fore.CYAN}kitty terminal{Style.RESET_ALL} ..."
+
+    terminal_with_command = f"{settings.ANDROID_TOOLS['kitty']['bin']} -- bash -c \"{command}\""
+    log.info(log_info)
+    functions.run_command(terminal_with_command, return_output=False)
+
+
 def run_tool(tool_with_args):
     """
     Accepts only two parameters tool + <arg>
@@ -549,18 +578,7 @@ def run_tool(tool_with_args):
                     command = command + " " + tool_args
 
                 if tool == 'objection':
-
-                    log.info("Execute the followings steps")
-                    log.info(f"1) Please run 'frida-ps -U' to find the app name")
-                    log.info(f"2) start objection with the following command: 'objection --gadget \"<APP-NAME>\" explore'")
-
-                    command = settings.ANDROID_TOOLS[tool]['bin']
-                    # replace last objection mit exec $SHELL
-                    command = command.rsplit(";", 1)[0] + "; exec $SHELL"
-
-                    terminal_with_command = f"{settings.ANDROID_TOOLS['kitty']['bin']} -- bash -c \"{command}\""
-                    log.info(f"Running command in {Style.BRIGHT + Fore.CYAN}kitty terminal{Style.RESET_ALL} ...")
-                    functions.run_command(terminal_with_command, return_output=False)
+                    run_tool_objection()
 
                 # default case
                 else: