llvm-project/lldb/examples/python/in_call_stack.py
Dave Lee 307f395d4f [lldb] Allow flexible importing of in_call_stack
Allow `in_call_stack` to be imported in either of the following ways:

```
command script import path/to/in_call_stack.py
command script import lldb.utils.in_call_stack
```

rdar://102249295

Differential Revision: https://reviews.llvm.org/D137860
2022-11-11 14:28:40 -08:00

25 lines
743 B
Python
Executable File

#!/usr/bin/env python
def __lldb_init_module(debugger, internal_dict):
debugger.HandleCommand(
f'command alias in_call_stack breakpoint command add --python-function {__name__}.in_call_stack -k name -v %1'
)
def in_call_stack(frame, bp_loc, arg_dict, _):
"""Only break if the given name is in the current call stack."""
name = arg_dict.GetValueForKey('name').GetStringValue(1000)
thread = frame.GetThread()
found = False
for frame in thread.frames:
# Check the symbol.
symbol = frame.GetSymbol()
if symbol and name in frame.GetSymbol().GetName():
return True
# Check the function.
function = frame.GetFunction()
if function and name in function.GetName():
return True
return False