caio/clang-p2996
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add test methods for SBProcess.WriteMemory() API to the TestProcessAPI.py file.

Browse Source 
This makes the number of total tests equal to 201.

llvm-svn: 126769
This commit is contained in:
Johnny Chen
2011-03-01 18:51:47 +00:00
parent 5e1af659e9
commit 90aa594c5e
1 changed files with 77 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

115
lldb/test/python_api/process/TestProcessAPI.py
View File

@@ -1,5 +1,5 @@
"""
Test symbol table access for main.m.
Test SBProcess APIs, including ReadMemory(), WriteMemory(), and others.
"""
import os, time
@@ -12,21 +12,31 @@ class ProcessAPITestCase(TestBase):
mydir = os.path.join("python_api", "process")
symbols_list = ['main',
'my_char'
]
@unittest2.skipUnless(sys.platform.startswith("darwin"), "requires Darwin")
@python_api_test
def test_with_dsym_and_python_api(self):
"""Test Python process APIs."""
def test_read_memory_with_dsym(self):
"""Test Python SBProcess.ReadMemory() API."""
self.buildDsym()
self.process_api()
self.read_memory()
@python_api_test
def test_with_dwarf_and_python_api(self):
"""Test Python process APIs."""
def test_read_memory_with_dwarf(self):
"""Test Python SBProcess.ReadMemory() API."""
self.buildDwarf()
self.process_api()
self.read_memory()
@unittest2.skipUnless(sys.platform.startswith("darwin"), "requires Darwin")
@python_api_test
def test_write_memory_with_dsym(self):
"""Test Python SBProcess.WriteMemory() API."""
self.buildDsym()
self.write_memory()
@python_api_test
def test_write_memory_with_dwarf(self):
"""Test Python SBProcess.WriteMemory() API."""
self.buildDwarf()
self.write_memory()
def setUp(self):
# Call super's setUp().
@@ -34,8 +44,8 @@ class ProcessAPITestCase(TestBase):
# Find the line number to break inside main().
self.line = line_number("main.cpp", "// Set break point at this line and check variable 'my_char'.")
def process_api(self):
"""Test Python process APIs."""
def read_memory(self):
"""Test Python SBProcess.ReadMemory() API."""
exe = os.path.join(os.getcwd(), "a.out")
self.runCmd("file " + exe, CURRENT_EXECUTABLE_SET)
@@ -54,45 +64,74 @@ class ProcessAPITestCase(TestBase):
# Get the SBValue for the global variable 'my_char'.
val = frame.FindValue("my_char", lldb.eValueTypeVariableGlobal)
location = int(val.GetLocation(frame), 16)
self.DebugSBValue(frame, val)
# If the variable does not have a load address, there's no sense continuing.
if not val.GetLocation(frame).startswith("0x"):
return
# OK, let's get the hex location of the variable.
location = int(val.GetLocation(frame), 16)
# Due to the typemap magic (see lldb.swig), we pass in 1 to ReadMemory and
# expect to get a Python string as the result object!
content = self.process.ReadMemory(location, 1, error)
print "content:", content
if not error.Success():
self.fail("SBProcess.ReadMemory() failed")
print "memory content:", content
self.expect(content, "Result from SBProcess.ReadMemory() matches our expected output: 'x'",
exe=False,
startstr = 'x')
#
# Exercise Python APIs to access the symbol table entries.
#
def write_memory(self):
"""Test Python SBProcess.WriteMemory() API."""
exe = os.path.join(os.getcwd(), "a.out")
self.runCmd("file " + exe, CURRENT_EXECUTABLE_SET)
# Create the filespec by which to locate our a.out module.
filespec = lldb.SBFileSpec(exe, False)
target = self.dbg.CreateTarget(exe)
self.assertTrue(target.IsValid(), VALID_TARGET)
module = target.FindModule(filespec)
self.assertTrue(module.IsValid(), VALID_MODULE)
breakpoint = target.BreakpointCreateByLocation("main.cpp", self.line)
self.assertTrue(breakpoint.IsValid(), VALID_BREAKPOINT)
# Create the set of known symbols. As we iterate through the symbol
# table, remove the symbol from the set if it is a known symbol.
expected_symbols = set(self.symbols_list)
from lldbutil import lldb_iter
for symbol in lldb_iter(module, 'GetNumSymbols', 'GetSymbolAtIndex'):
self.assertTrue(symbol.IsValid(), VALID_SYMBOL)
#print "symbol:", symbol
name = symbol.GetName()
if name in expected_symbols:
#print "Removing %s from known_symbols %s" % (name, expected_symbols)
expected_symbols.remove(name)
# Launch the process, and do not stop at the entry point.
error = lldb.SBError()
self.process = target.Launch (self.dbg.GetListener(), None, None, os.ctermid(), os.ctermid(), os.ctermid(), None, 0, False, error)
# At this point, the known_symbols set should have become an empty set.
# If not, raise an error.
#print "symbols unaccounted for:", expected_symbols
self.assertTrue(len(expected_symbols) == 0,
"All the known symbols are accounted for")
thread = self.process.GetThreadAtIndex(0);
frame = thread.GetFrameAtIndex(0);
# Get the SBValue for the global variable 'my_char'.
val = frame.FindValue("my_char", lldb.eValueTypeVariableGlobal)
self.DebugSBValue(frame, val)
# If the variable does not have a load address, there's no sense continuing.
if not val.GetLocation(frame).startswith("0x"):
return
# OK, let's get the hex location of the variable.
location = int(val.GetLocation(frame), 16)
# The program logic makes the 'my_char' variable to have memory content as 'x'.
# But we want to use the WriteMemory() API to assign 'a' to the variable.
# Now use WriteMemory() API to write 'a' into the global variable.
result = self.process.WriteMemory(location, 'a', error)
if not error.Success() or result != 1:
self.fail("SBProcess.WriteMemory() failed")
# Read from the memory location. This time it should be 'a'.
# Due to the typemap magic (see lldb.swig), we pass in 1 to ReadMemory and
# expect to get a Python string as the result object!
content = self.process.ReadMemory(location, 1, error)
if not error.Success():
self.fail("SBProcess.ReadMemory() failed")
print "memory content:", content
self.expect(content, "Result from SBProcess.ReadMemory() matches our expected output: 'a'",
exe=False,
startstr = 'a')
if __name__ == '__main__':