hilda 2.0.0__tar.gz → 2.0.2__tar.gz

{hilda-2.0.0 → hilda-2.0.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hilda
-Version: 2.0.0
+Version: 2.0.2
 Summary: LLDB wrapped and empowered by iPython's features
 Author-email: doronz88 <doron88@gmail.com>, matan <matan1008@gmail.com>, netanel cohen <netanelc305@protonmail.com>
 Maintainer-email: doronz88 <doron88@gmail.com>, matan <matan1008@gmail.com>, netanel cohen <netanelc305@protonmail.com>
@@ -51,6 +51,7 @@ Requires-Dist: construct
 Requires-Dist: pymobiledevice3
 Requires-Dist: keystone-engine
 Requires-Dist: tabulate
+Requires-Dist: inquirer3
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 
@@ -370,6 +371,10 @@ Here is a gist of methods you can access from `p`:
     - Import & reload given python module (intended mainly for external snippets)
 - `unwind`
     - Unwind the stack (useful when get_evaluation_unwind() == False)
+- `set_selected_thread`
+    - sets the currently selected thread, which is used in other parts of the program, such as displaying disassembly or
+      checking registers.
+      This ensures the application focuses on the specified thread for these operations.
 
 ## Magic functions
 

{hilda-2.0.0 → hilda-2.0.2}/README.md

@@ -314,6 +314,10 @@ Here is a gist of methods you can access from `p`:
     - Import & reload given python module (intended mainly for external snippets)
 - `unwind`
     - Unwind the stack (useful when get_evaluation_unwind() == False)
+- `set_selected_thread`
+    - sets the currently selected thread, which is used in other parts of the program, such as displaying disassembly or
+      checking registers.
+      This ensures the application focuses on the specified thread for these operations.
 
 ## Magic functions
 

{hilda-2.0.0 → hilda-2.0.2}/hilda/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '2.0.0'
-__version_tuple__ = version_tuple = (2, 0, 0)
+__version__ = version = '2.0.2'
+__version_tuple__ = version_tuple = (2, 0, 2)

{hilda-2.0.0 → hilda-2.0.2}/hilda/cli.py

@@ -54,10 +54,10 @@ def attach(name: str, pid: int, startup_files: Optional[List[str]] = None) -> No
 @click.argument('exec_path')
 @click.option('--argv', multiple=True, help='Command line arguments to pass to the process')
 @click.option('--envp', multiple=True, callback=parse_envp, help='Environment variables in the form KEY=VALUE')
-@click.option('--stdin', type=Path, help='Redirect stdin from this file path')
-@click.option('--stdout', type=Path, help='Redirect stdout to this file path')
-@click.option('--stderr', type=Path, help='Redirect stderr to this file path')
-@click.option('--cwd', type=Path, help='Set the working directory for the process')
+@click.option('--stdin', type=str, help='Redirect stdin from this file path')
+@click.option('--stdout', type=str, help='Redirect stdout to this file path')
+@click.option('--stderr', type=str, help='Redirect stderr to this file path')
+@click.option('--cwd', type=str, help='Set the working directory for the process')
 @click.option('--flags', type=click.INT, default=0, help='Launch flags (bitmask)')
 @click.option('--stop-at-entry', is_flag=True, help='Stop the process at the entry point')
 @startup_files_option

hilda-2.0.2/hilda/common.py

@@ -0,0 +1,14 @@
+from datetime import datetime
+from typing import Any, List, Mapping, Tuple, Union
+
+import inquirer3
+from inquirer3.themes import GreenPassion
+
+CfSerializable = Union[
+    Mapping[str, Any], List, Tuple[Any, ...], str, bool, float, bytes, datetime, None]
+
+
+def selection_prompt(options_list: List):
+    question = [inquirer3.List('choice', message='choose device', choices=options_list, carousel=True)]
+    result = inquirer3.prompt(question, theme=GreenPassion(), raise_keyboard_interrupt=True)
+    return result['choice']

{hilda-2.0.0 → hilda-2.0.2}/hilda/exceptions.py

@@ -1,7 +1,7 @@
 __all__ = ['HildaException', 'SymbolAbsentError', 'EvaluatingExpressionError', 'CreatingObjectiveCSymbolError',
            'ConvertingToNsObjectError', 'ConvertingFromNSObjectError', 'DisableJetsamMemoryChecksError',
            'GettingObjectiveCClassError', 'AccessingRegisterError', 'AccessingMemoryError',
-           'BrokenLocalSymbolsJarError', 'AddingLldbSymbolError', 'LLDBException']
+           'BrokenLocalSymbolsJarError', 'AddingLldbSymbolError', 'LLDBException', 'InvalidThreadIndexError']
 
 
 class HildaException(Exception):
@@ -70,3 +70,8 @@ class BrokenLocalSymbolsJarError(HildaException):
 class AddingLldbSymbolError(HildaException):
     """ Raise when failing to convert a LLDB symbol to Hilda's symbol. """
     pass
+
+
+class InvalidThreadIndexError(HildaException):
+    """ Raise when thread idx invalid """
+    pass

{hilda-2.0.0 → hilda-2.0.2}/hilda/hilda_client.py

@@ -30,10 +30,11 @@ from tqdm import tqdm
 from traitlets.config import Config
 
 from hilda import objective_c_class
-from hilda.common import CfSerializable
+from hilda.common import CfSerializable, selection_prompt
 from hilda.exceptions import AccessingMemoryError, AccessingRegisterError, AddingLldbSymbolError, \
     BrokenLocalSymbolsJarError, ConvertingFromNSObjectError, ConvertingToNsObjectError, CreatingObjectiveCSymbolError, \
-    DisableJetsamMemoryChecksError, EvaluatingExpressionError, HildaException, SymbolAbsentError
+    DisableJetsamMemoryChecksError, EvaluatingExpressionError, HildaException, InvalidThreadIndexError, \
+    SymbolAbsentError
 from hilda.lldb_importer import lldb
 from hilda.objective_c_symbol import ObjectiveCSymbol
 from hilda.registers import Registers
@@ -340,8 +341,6 @@ class HildaClient:
 
         if not self.process.Stop().Success():
             self.log_critical('failed to stop process')
-        else:
-            self.log_info('Process Stopped')
 
     def cont(self, *args) -> None:
         """ Continue process. """
@@ -357,8 +356,6 @@ class HildaClient:
 
         if not self.process.Continue().Success():
             self.log_critical('failed to continue process')
-        else:
-            self.log_info('Process Continued')
 
     def detach(self):
         """
@@ -514,7 +511,7 @@ class HildaClient:
             if options.get('name', False):
                 name = options['name']
 
-            log_message = f'🚨 #{bp.id} 0x{symbol:x} {name}'
+            log_message = f'🚨 #{bp.id} 0x{symbol:x} {name} - Thread #{self.thread.idx}:{hex(self.thread.id)}'
 
             if 'regs' in options:
                 log_message += '\nregs:'
@@ -528,7 +525,7 @@ class HildaClient:
                     value = hilda.symbol(hilda.evaluate_expression(name))
                     log_message += f'\n\t{name} = {hilda._monitor_format_value(fmt, value)}'
 
-            if options.get('force_return', False):
+            if options.get('force_return', None) is not None:
                 hilda.force_return(options['force_return'])
                 log_message += f'\nforced return: {options["force_return"]}'
 
@@ -537,7 +534,7 @@ class HildaClient:
                 hilda.finish()
                 hilda.bt()
 
-            if options.get('retval', False):
+            if options.get('retval', None) is not None:
                 # return from function
                 hilda.finish()
                 value = hilda.evaluate_expression('$arg1')
@@ -884,6 +881,16 @@ class HildaClient:
         spec.loader.exec_module(m)
         return m
 
+    def set_selected_thread(self, idx: Optional[int] = None) -> None:
+        if idx is None:
+            thread = selection_prompt(self.process.threads)
+        else:
+            try:
+                thread = [t for t in self.process.threads if t.idx == idx][0]
+            except IndexError:
+                raise InvalidThreadIndexError()
+        self.process.SetSelectedThread(thread)
+
     def unwind(self) -> bool:
         """ Unwind the stack (useful when get_evaluation_unwind() == False) """
         return self.thread.UnwindInnermostExpression().Success()

{hilda-2.0.0 → hilda-2.0.2}/hilda/ipython_extensions/keybindings.py

@@ -8,8 +8,8 @@ def load_ipython_extension(ipython):
         hilda = ipython.user_ns['p']
         keys_mapping = {Keys.F7: hilda.step_into,
                         Keys.F8: hilda.step_over,
-                        Keys.F9: hilda.cont,
-                        Keys.F10: hilda.stop}
+                        Keys.F9: lambda _: (hilda.log_info('Sending continue'), hilda.cont()),
+                        Keys.F10: lambda _: (hilda.log_info('Sending stop'), hilda.stop())}
 
         insert_mode = ViInsertMode() | EmacsInsertMode()
         registry = ipython.pt_app.key_bindings

{hilda-2.0.0 → hilda-2.0.2}/hilda/launch_lldb.py

@@ -70,9 +70,22 @@ class LLDBListenerThread(Thread, ABC):
                 logger.debug(f'Process Exited with status {self.process.GetExitStatus()}')
                 self.should_quit = True
             elif state == lldb.eStateRunning and last_state == lldb.eStateStopped:
-                logger.debug("Process Continued")
+                logger.debug('Process Continued')
             elif state == lldb.eStateStopped and last_state == lldb.eStateRunning:
                 logger.debug('Process Stopped')
+                for thread in self.process:
+                    frame = thread.GetFrameAtIndex(0)
+                    stop_reason = thread.GetStopReason()
+                    logger.debug(f'tid = {hex(thread.GetThreadID())} pc = {frame.GetPC()}')
+                    if stop_reason not in [lldb.eStopReasonSignal, lldb.eStopReasonException,
+                                           lldb.eStopReasonBreakpoint,
+                                           lldb.eStopReasonWatchpoint, lldb.eStopReasonPlanComplete,
+                                           lldb.eStopReasonTrace,
+                                           lldb.eStopReasonSignal]:
+                        continue
+                    self.process.SetSelectedThread(thread)
+                    break
+
             last_state = state
 
 
@@ -114,7 +127,7 @@ class LLDBAttachName(LLDBListenerThread):
         return self.debugger.CreateTargetWithFileAndArch(None, None)
 
     def _create_process(self) -> lldb.SBProcess:
-        logger.debug(f'Attaching to {self.name}')
+        logger.debug(f'Attaching to {self.proc_name}')
         return self.target.AttachToProcessWithName(self.listener, self.proc_name, self.wait_for, self.error)
 
 

{hilda-2.0.0 → hilda-2.0.2}/hilda.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hilda
-Version: 2.0.0
+Version: 2.0.2
 Summary: LLDB wrapped and empowered by iPython's features
 Author-email: doronz88 <doron88@gmail.com>, matan <matan1008@gmail.com>, netanel cohen <netanelc305@protonmail.com>
 Maintainer-email: doronz88 <doron88@gmail.com>, matan <matan1008@gmail.com>, netanel cohen <netanelc305@protonmail.com>
@@ -51,6 +51,7 @@ Requires-Dist: construct
 Requires-Dist: pymobiledevice3
 Requires-Dist: keystone-engine
 Requires-Dist: tabulate
+Requires-Dist: inquirer3
 Provides-Extra: test
 Requires-Dist: pytest; extra == "test"
 
@@ -370,6 +371,10 @@ Here is a gist of methods you can access from `p`:
     - Import & reload given python module (intended mainly for external snippets)
 - `unwind`
     - Unwind the stack (useful when get_evaluation_unwind() == False)
+- `set_selected_thread`
+    - sets the currently selected thread, which is used in other parts of the program, such as displaying disassembly or
+      checking registers.
+      This ensures the application focuses on the specified thread for these operations.
 
 ## Magic functions
 

{hilda-2.0.0 → hilda-2.0.2}/hilda.egg-info/requires.txt

@@ -9,6 +9,7 @@ construct
 pymobiledevice3
 keystone-engine
 tabulate
+inquirer3
 
 [test]
 pytest

{hilda-2.0.0 → hilda-2.0.2}/requirements.txt

@@ -8,4 +8,5 @@ objc_types_decoder
 construct
 pymobiledevice3
 keystone-engine
-tabulate
+tabulate
+inquirer3

hilda-2.0.0/hilda/common.py

@@ -1,5 +0,0 @@
-from datetime import datetime
-from typing import Any, List, Mapping, Tuple, Union
-
-CfSerializable = Union[
-    Mapping[str, Any], List, Tuple[Any, ...], str, bool, float, bytes, datetime, None]