b2e1a5c998
method names of all the lldb container objects and returns an iterator object when
passed an eligible lldb container object.
Example:
from lldb_util import smart_iter
for thread in smart_iter(process):
ID = thread.GetThreadID()
if thread.GetStopReason() == lldb.eStopReasonBreakpoint:
stopped_due_to_breakpoint = True
for frame in smart_iter(thread):
self.assertTrue(frame.GetThread().GetThreadID() == ID)
...
Add a test case for lldb.smart_iter().
llvm-svn: 130332
135 lines
5.0 KiB
Python
135 lines
5.0 KiB
Python
"""
|
|
Test lldb_iter/smart_iter() which returns an iterator object for lldb container
|
|
objects.
|
|
"""
|
|
|
|
import os, time
|
|
import re
|
|
import unittest2
|
|
import lldb
|
|
from lldbtest import *
|
|
|
|
class LLDBIteratorTestCase(TestBase):
|
|
|
|
mydir = "python_api/lldbutil"
|
|
|
|
def setUp(self):
|
|
# Call super's setUp().
|
|
TestBase.setUp(self)
|
|
# Find the line numbers to break inside main().
|
|
self.line1 = line_number('main.cpp', '// Set break point at this line.')
|
|
self.line2 = line_number('main.cpp', '// And that line.')
|
|
|
|
def test_lldb_iter_1(self):
|
|
"""Test lldb_iter works correctly for SBTarget -> SBModule."""
|
|
self.buildDefault()
|
|
self.lldb_iter_1()
|
|
|
|
def test_lldb_iter_2(self):
|
|
"""Test lldb_iter works correctly for SBTarget -> SBBreakpoint."""
|
|
self.buildDefault()
|
|
self.lldb_iter_2()
|
|
|
|
def test_smart_iter_1(self):
|
|
"""Test smart_iter works correctly for SBProcess->SBThread->SBFrame."""
|
|
self.buildDefault()
|
|
self.smart_iter_1()
|
|
|
|
def lldb_iter_1(self):
|
|
exe = os.path.join(os.getcwd(), "a.out")
|
|
|
|
target = self.dbg.CreateTarget(exe)
|
|
self.assertTrue(target.IsValid(), VALID_TARGET)
|
|
|
|
breakpoint = target.BreakpointCreateByLocation("main.cpp", self.line1)
|
|
self.assertTrue(breakpoint.IsValid(), VALID_BREAKPOINT)
|
|
|
|
# Now launch the process, and do not stop at entry point.
|
|
rc = lldb.SBError()
|
|
self.process = target.Launch (self.dbg.GetListener(), None, None, os.ctermid(), os.ctermid(), os.ctermid(), None, 0, False, rc)
|
|
|
|
if not rc.Success() or not self.process.IsValid():
|
|
self.fail("SBTarget.LaunchProcess() failed")
|
|
|
|
from lldbutil import lldb_iter, get_description
|
|
yours = []
|
|
for i in range(target.GetNumModules()):
|
|
yours.append(target.GetModuleAtIndex(i))
|
|
mine = []
|
|
for m in lldb_iter(target, 'GetNumModules', 'GetModuleAtIndex'):
|
|
mine.append(m)
|
|
|
|
self.assertTrue(len(yours) == len(mine))
|
|
for i in range(len(yours)):
|
|
if self.TraceOn():
|
|
print "yours[%d]='%s'" % (i, get_description(yours[i]))
|
|
print "mine[%d]='%s'" % (i, get_description(mine[i]))
|
|
self.assertTrue(yours[i].GetUUIDString() == mine[i].GetUUIDString(),
|
|
"UUID of yours[{0}] and mine[{0}] matches".format(i))
|
|
|
|
def lldb_iter_2(self):
|
|
exe = os.path.join(os.getcwd(), "a.out")
|
|
|
|
target = self.dbg.CreateTarget(exe)
|
|
self.assertTrue(target.IsValid(), VALID_TARGET)
|
|
|
|
breakpoint = target.BreakpointCreateByLocation("main.cpp", self.line1)
|
|
self.assertTrue(breakpoint.IsValid(), VALID_BREAKPOINT)
|
|
breakpoint = target.BreakpointCreateByLocation("main.cpp", self.line2)
|
|
self.assertTrue(breakpoint.IsValid(), VALID_BREAKPOINT)
|
|
|
|
self.assertTrue(target.GetNumBreakpoints() == 2)
|
|
|
|
from lldbutil import lldb_iter, get_description
|
|
yours = []
|
|
for i in range(target.GetNumBreakpoints()):
|
|
yours.append(target.GetBreakpointAtIndex(i))
|
|
mine = []
|
|
for m in lldb_iter(target, 'GetNumBreakpoints', 'GetBreakpointAtIndex'):
|
|
mine.append(m)
|
|
|
|
self.assertTrue(len(yours) == len(mine))
|
|
for i in range(len(yours)):
|
|
if self.TraceOn():
|
|
print "yours[%d]='%s'" % (i, get_description(yours[i]))
|
|
print "mine[%d]='%s'" % (i, get_description(mine[i]))
|
|
self.assertTrue(yours[i].GetID() == mine[i].GetID(),
|
|
"ID of yours[{0}] and mine[{0}] matches".format(i))
|
|
|
|
def smart_iter_1(self):
|
|
exe = os.path.join(os.getcwd(), "a.out")
|
|
|
|
target = self.dbg.CreateTarget(exe)
|
|
self.assertTrue(target.IsValid(), VALID_TARGET)
|
|
|
|
breakpoint = target.BreakpointCreateByLocation("main.cpp", self.line1)
|
|
self.assertTrue(breakpoint.IsValid(), VALID_BREAKPOINT)
|
|
|
|
# Now launch the process, and do not stop at entry point.
|
|
rc = lldb.SBError()
|
|
self.process = target.Launch (self.dbg.GetListener(), None, None, os.ctermid(), os.ctermid(), os.ctermid(), None, 0, False, rc)
|
|
|
|
if not rc.Success() or not self.process.IsValid():
|
|
self.fail("SBTarget.LaunchProcess() failed")
|
|
|
|
from lldbutil import smart_iter, print_stacktrace
|
|
stopped_due_to_breakpoint = False
|
|
for thread in smart_iter(self.process):
|
|
if self.TraceOn():
|
|
print_stacktrace(thread)
|
|
ID = thread.GetThreadID()
|
|
if thread.GetStopReason() == lldb.eStopReasonBreakpoint:
|
|
stopped_due_to_breakpoint = True
|
|
for frame in smart_iter(thread):
|
|
self.assertTrue(frame.GetThread().GetThreadID() == ID)
|
|
if self.TraceOn():
|
|
print frame
|
|
|
|
self.assertTrue(stopped_due_to_breakpoint)
|
|
|
|
if __name__ == '__main__':
|
|
import atexit
|
|
lldb.SBDebugger.Initialize()
|
|
atexit.register(lambda: lldb.SBDebugger.Terminate())
|
|
unittest2.main()
|