1441ffe6a6
Update examples and docs to demonstrate using `__lldb_init_module` instead of
the idiom that checks for `lldb.debugger` at the top-level.
```
if __name__ == '__main__':
...
elif lldb.debugger:
...
```
Is replaced with:
```
if __name__ == '__main__':
...
def __lldb_init_module(debugger, internal_dict):
...
```
This change is for two reasons. First, it's generally encouraged not to only
use the convenience singletons (`lldb.{debugger,process,target,etc}`)
interactively from the `script` command. Second, there's a bug where
registering a python class as a command (using `command script add -c ...`),
result in the command not being runnable. Note that registering function-backed
commands does not have this bug.
Differential Revision: https://reviews.llvm.org/D117237
70 lines
2.6 KiB
Python
Executable File
70 lines
2.6 KiB
Python
Executable File
#!/usr/bin/env python
|
|
from __future__ import print_function
|
|
import lldb
|
|
import optparse
|
|
import shlex
|
|
|
|
|
|
def stack_frames(debugger, command, result, dict):
|
|
command_args = shlex.split(command)
|
|
usage = "usage: %prog [options] <PATH> [PATH ...]"
|
|
description = '''This command will enumerate all stack frames, print the stack size for each, and print an aggregation of which functions have the largest stack frame sizes at the end.'''
|
|
parser = optparse.OptionParser(
|
|
description=description, prog='ls', usage=usage)
|
|
parser.add_option(
|
|
'-v',
|
|
'--verbose',
|
|
action='store_true',
|
|
dest='verbose',
|
|
help='display verbose debug info',
|
|
default=False)
|
|
try:
|
|
(options, args) = parser.parse_args(command_args)
|
|
except:
|
|
return
|
|
|
|
target = debugger.GetSelectedTarget()
|
|
process = target.GetProcess()
|
|
|
|
frame_info = {}
|
|
for thread in process:
|
|
last_frame = None
|
|
print("thread %u" % (thread.id))
|
|
for frame in thread.frames:
|
|
if last_frame:
|
|
frame_size = 0
|
|
if frame.idx == 1:
|
|
if frame.fp == last_frame.fp:
|
|
# No frame one the first frame (might be right at the
|
|
# entry point)
|
|
first_frame_size = 0
|
|
frame_size = frame.fp - frame.sp
|
|
else:
|
|
# First frame that has a valid size
|
|
first_frame_size = last_frame.fp - last_frame.sp
|
|
print("<%#7x> %s" % (first_frame_size, last_frame))
|
|
if first_frame_size:
|
|
name = last_frame.name
|
|
if name not in frame_info:
|
|
frame_info[name] = first_frame_size
|
|
else:
|
|
frame_info[name] += first_frame_size
|
|
else:
|
|
# Second or higher frame
|
|
frame_size = frame.fp - last_frame.fp
|
|
print("<%#7x> %s" % (frame_size, frame))
|
|
if frame_size > 0:
|
|
name = frame.name
|
|
if name not in frame_info:
|
|
frame_info[name] = frame_size
|
|
else:
|
|
frame_info[name] += frame_size
|
|
last_frame = frame
|
|
print(frame_info)
|
|
|
|
|
|
def __lldb_init_module(debugger, internal_dict):
|
|
debugger.HandleCommand(
|
|
"command script add -f stacks.stack_frames stack_frames")
|
|
print("A new command called 'stack_frames' was added, type 'stack_frames --help' for more information.")
|