[Lldb-commits] [lldb] r126769 - /lldb/trunk/test/python_api/process/TestProcessAPI.py
Johnny Chen
johnny.chen at apple.com
Tue Mar 1 10:51:48 PST 2011
Author: johnny
Date: Tue Mar 1 12:51:47 2011
New Revision: 126769
URL: http://llvm.org/viewvc/llvm-project?rev=126769&view=rev
Log:
Add test methods for SBProcess.WriteMemory() API to the TestProcessAPI.py file.
This makes the number of total tests equal to 201.
Modified:
lldb/trunk/test/python_api/process/TestProcessAPI.py
Modified: lldb/trunk/test/python_api/process/TestProcessAPI.py
URL: http://llvm.org/viewvc/llvm-project/lldb/trunk/test/python_api/process/TestProcessAPI.py?rev=126769&r1=126768&r2=126769&view=diff
==============================================================================
--- lldb/trunk/test/python_api/process/TestProcessAPI.py (original)
+++ lldb/trunk/test/python_api/process/TestProcessAPI.py Tue Mar 1 12:51:47 2011
@@ -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 @@
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_read_memory_with_dsym(self):
+ """Test Python SBProcess.ReadMemory() API."""
+ self.buildDsym()
+ self.read_memory()
+
+ @python_api_test
+ def test_read_memory_with_dwarf(self):
+ """Test Python SBProcess.ReadMemory() API."""
+ self.buildDwarf()
+ self.read_memory()
+ @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_write_memory_with_dsym(self):
+ """Test Python SBProcess.WriteMemory() API."""
self.buildDsym()
- self.process_api()
+ self.write_memory()
@python_api_test
- def test_with_dwarf_and_python_api(self):
- """Test Python process APIs."""
+ def test_write_memory_with_dwarf(self):
+ """Test Python SBProcess.WriteMemory() API."""
self.buildDwarf()
- self.process_api()
+ self.write_memory()
def setUp(self):
# Call super's setUp().
@@ -34,8 +44,8 @@
# 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 @@
# 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.
- #
-
- # Create the filespec by which to locate our a.out module.
- filespec = lldb.SBFileSpec(exe, False)
-
- module = target.FindModule(filespec)
- self.assertTrue(module.IsValid(), VALID_MODULE)
-
- # 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)
-
- # 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")
+ def write_memory(self):
+ """Test Python SBProcess.WriteMemory() API."""
+ exe = os.path.join(os.getcwd(), "a.out")
+ self.runCmd("file " + exe, CURRENT_EXECUTABLE_SET)
+
+ target = self.dbg.CreateTarget(exe)
+ self.assertTrue(target.IsValid(), VALID_TARGET)
+
+ breakpoint = target.BreakpointCreateByLocation("main.cpp", self.line)
+ self.assertTrue(breakpoint.IsValid(), VALID_BREAKPOINT)
+
+ # 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)
+
+ 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__':
More information about the lldb-commits
mailing list