forked from OSchip/llvm-project
140 lines
4.6 KiB
Python
140 lines
4.6 KiB
Python
"""
|
|
Test some SBModule and SBSection APIs.
|
|
"""
|
|
|
|
import os, time
|
|
import re
|
|
import unittest2
|
|
import lldb
|
|
from lldbtest import *
|
|
from lldbutil import symbol_type_to_str
|
|
|
|
class ModuleAndSectionAPIsTestCase(TestBase):
|
|
|
|
mydir = TestBase.compute_mydir(__file__)
|
|
|
|
@python_api_test
|
|
def test_module_and_section(self):
|
|
"""Test module and section APIs."""
|
|
self.buildDefault()
|
|
self.module_and_section()
|
|
|
|
@python_api_test
|
|
def test_module_and_section_boundary_condition(self):
|
|
"""Test module and section APIs by passing None when it expects a Python string."""
|
|
self.buildDefault()
|
|
self.module_and_section_boundary_condition()
|
|
|
|
@python_api_test
|
|
def test_module_compile_unit_iter(self):
|
|
"""Test module's compile unit iterator APIs."""
|
|
self.buildDefault()
|
|
self.module_compile_unit_iter()
|
|
|
|
def module_and_section(self):
|
|
exe = os.path.join(os.getcwd(), "a.out")
|
|
|
|
target = self.dbg.CreateTarget(exe)
|
|
self.assertTrue(target, VALID_TARGET)
|
|
self.assertTrue(target.GetNumModules() > 0)
|
|
|
|
# Hide stdout if not running with '-t' option.
|
|
if not self.TraceOn():
|
|
self.HideStdout()
|
|
|
|
print "Number of modules for the target: %d" % target.GetNumModules()
|
|
for module in target.module_iter():
|
|
print module
|
|
|
|
# Get the executable module at index 0.
|
|
exe_module = target.GetModuleAtIndex(0)
|
|
|
|
print "Exe module: %s" % str(exe_module)
|
|
print "Number of sections: %d" % exe_module.GetNumSections()
|
|
INDENT = ' ' * 4
|
|
INDENT2 = INDENT * 2
|
|
for sec in exe_module.section_iter():
|
|
print sec
|
|
print INDENT + "Number of subsections: %d" % sec.GetNumSubSections()
|
|
if sec.GetNumSubSections() == 0:
|
|
for sym in exe_module.symbol_in_section_iter(sec):
|
|
print INDENT + str(sym)
|
|
print INDENT + "symbol type: %s" % symbol_type_to_str(sym.GetType())
|
|
else:
|
|
for subsec in sec:
|
|
print INDENT + str(subsec)
|
|
# Now print the symbols belonging to the subsection....
|
|
for sym in exe_module.symbol_in_section_iter(subsec):
|
|
print INDENT2 + str(sym)
|
|
print INDENT2 + "symbol type: %s" % symbol_type_to_str(sym.GetType())
|
|
|
|
def module_and_section_boundary_condition(self):
|
|
exe = os.path.join(os.getcwd(), "a.out")
|
|
|
|
target = self.dbg.CreateTarget(exe)
|
|
self.assertTrue(target, VALID_TARGET)
|
|
self.assertTrue(target.GetNumModules() > 0)
|
|
|
|
# Hide stdout if not running with '-t' option.
|
|
if not self.TraceOn():
|
|
self.HideStdout()
|
|
|
|
print "Number of modules for the target: %d" % target.GetNumModules()
|
|
for module in target.module_iter():
|
|
print module
|
|
|
|
# Get the executable module at index 0.
|
|
exe_module = target.GetModuleAtIndex(0)
|
|
|
|
print "Exe module: %s" % str(exe_module)
|
|
print "Number of sections: %d" % exe_module.GetNumSections()
|
|
|
|
# Boundary condition testings. Should not crash lldb!
|
|
exe_module.FindFirstType(None)
|
|
exe_module.FindTypes(None)
|
|
exe_module.FindGlobalVariables(target, None, 1)
|
|
exe_module.FindFunctions(None, 0)
|
|
exe_module.FindSection(None)
|
|
|
|
# Get the section at index 1.
|
|
if exe_module.GetNumSections() > 1:
|
|
sec1 = exe_module.GetSectionAtIndex(1)
|
|
print sec1
|
|
else:
|
|
sec1 = None
|
|
|
|
if sec1:
|
|
sec1.FindSubSection(None)
|
|
|
|
def module_compile_unit_iter(self):
|
|
exe = os.path.join(os.getcwd(), "a.out")
|
|
|
|
target = self.dbg.CreateTarget(exe)
|
|
self.assertTrue(target, VALID_TARGET)
|
|
self.assertTrue(target.GetNumModules() > 0)
|
|
|
|
# Hide stdout if not running with '-t' option.
|
|
if not self.TraceOn():
|
|
self.HideStdout()
|
|
|
|
print "Number of modules for the target: %d" % target.GetNumModules()
|
|
for module in target.module_iter():
|
|
print module
|
|
|
|
# Get the executable module at index 0.
|
|
exe_module = target.GetModuleAtIndex(0)
|
|
|
|
print "Exe module: %s" % str(exe_module)
|
|
print "Number of compile units: %d" % exe_module.GetNumCompileUnits()
|
|
INDENT = ' ' * 4
|
|
INDENT2 = INDENT * 2
|
|
for cu in exe_module.compile_unit_iter():
|
|
print cu
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import atexit
|
|
lldb.SBDebugger.Initialize()
|
|
atexit.register(lambda: lldb.SBDebugger.Terminate())
|
|
unittest2.main()
|