hilda_client: add wait_for_module()

README.md

@@ -248,6 +248,8 @@ Here is a gist of methods you can access from `p`:
   - 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.
+- `wait_for_module`
+  - Wait for a module to be loaded (`dlopen`) by checking if given expression is contained within its filename
 
 All these methods are available from the global `p` within the newly created IPython shell. In addition, you may invoke any of the exported APIs described in the [Python API](#python-api)
 

hilda/hilda_client.py

@@ -1043,6 +1043,24 @@ def add_lldb_symbol(self, symbol: lldb.SBSymbol) -> Symbol:
 
         return value
 
+    def wait_for_module(self, expression: str) -> None:
+        """ Wait for a module to be loaded using `dlopen` by matching given expression """
+        self.log_info(f'Waiting for module name containing "{expression}" to be loaded')
+
+        def bp(client: HildaClient, frame, bp_loc, options) -> None:
+            loading_module_name = client.evaluate_expression('$arg1').peek_str()
+            client.log_info(f'Loading module: {loading_module_name}')
+            if expression not in loading_module_name:
+                client.cont()
+                return
+            client.finish()
+            client.log_info(f'Desired module has been loaded: {expression}. Process remains stopped')
+            bp = bp_loc.GetBreakpoint()
+            client.remove_hilda_breakpoint(bp.id)
+
+        self.bp('dlopen', bp)
+        self.cont()
+
     def interact(self, additional_namespace: Optional[typing.Mapping] = None,
                  startup_files: Optional[List[str]] = None) -> None:
         """ Start an interactive Hilda shell """