Test case for changing ValueObjects with SBValue::SetValueFromCString.

llvm-svn: 137543

3 files changed, 181 insertions, 0 deletions

diff --git a/lldb/test/python_api/value/change_values/Makefile b/lldb/test/python_api/value/change_values/Makefile
new file mode 100644
index 00000000000..b09a579159d
--- /dev/null
+++ b/lldb/test/python_api/value/change_values/Makefile
@@ -0,0 +1,5 @@
+LEVEL = ../../../make
+
+C_SOURCES := main.c
+
+include $(LEVEL)/Makefile.rules
diff --git a/lldb/test/python_api/value/change_values/TestValueAPI.py b/lldb/test/python_api/value/change_values/TestValueAPI.py
new file mode 100644
index 00000000000..b8aa7646ca8
--- /dev/null
+++ b/lldb/test/python_api/value/change_values/TestValueAPI.py
@@ -0,0 +1,147 @@
+"""
+Test some SBValue APIs.
+"""
+
+import os, time
+import re
+import unittest2
+import lldb, lldbutil
+from lldbtest import *
+
+class ChangeValueAPITestCase(TestBase):
+
+    mydir = os.path.join("python_api", "value", "change_values")
+
+    @unittest2.skipUnless(sys.platform.startswith("darwin"), "requires Darwin")
+    @python_api_test
+    def test_change_value_with_dsym(self):
+        """Exercise the SBValue::SetValueFromCString API."""
+        d = {'EXE': self.exe_name}
+        self.buildDsym(dictionary=d)
+        self.setTearDownCleanup(dictionary=d)
+        self.change_value_api(self.exe_name)
+
+    @python_api_test
+    def test_change_value_with_dwarf(self):
+        """Exercise the SBValue::SetValueFromCString API."""
+        d = {'EXE': self.exe_name}
+        self.buildDwarf(dictionary=d)
+        self.setTearDownCleanup(dictionary=d)
+        self.change_value_api(self.exe_name)
+
+    def setUp(self):
+        # Call super's setUp().
+        TestBase.setUp(self)
+        # We'll use the test method name as the exe_name.
+        self.exe_name = self.testMethodName
+        # Find the line number to of function 'c'.
+        self.line = line_number('main.c', '// Stop here and set values')
+        self.end_line = line_number ('main.c', '// Set a breakpoint here at the end')
+
+    def change_value_api(self, exe_name):
+        """Exercise some SBValue APIs."""
+        exe = os.path.join(os.getcwd(), exe_name)
+
+        # Create a target by the debugger.
+        target = self.dbg.CreateTarget(exe)
+        self.assertTrue(target, VALID_TARGET)
+
+        # Create the breakpoint inside function 'main'.
+        breakpoint = target.BreakpointCreateByLocation('main.c', self.line)
+        self.assertTrue(breakpoint, VALID_BREAKPOINT)
+
+        # Create the breakpoint inside function 'main'.
+        end_breakpoint = target.BreakpointCreateByLocation('main.c', self.end_line)
+        self.assertTrue(end_breakpoint, VALID_BREAKPOINT)
+
+        # Now launch the process, and do not stop at entry point.
+        process = target.LaunchSimple(None, None, os.getcwd())
+        self.assertTrue(process, PROCESS_IS_VALID)
+
+        # Get Frame #0.
+        self.assertTrue(process.GetState() == lldb.eStateStopped)
+        thread = lldbutil.get_stopped_thread(process, lldb.eStopReasonBreakpoint)
+        self.assertTrue(thread != None, "There should be a thread stopped due to breakpoint condition")
+        frame0 = thread.GetFrameAtIndex(0)
+        self.assertTrue (frame0.IsValid(), "Got a valid frame.")
+
+        # Get the val variable and change it:
+        error = lldb.SBError()
+
+        val_value = frame0.FindVariable ("val")
+        self.assertTrue (val_value.IsValid(), "Got the SBValue for val")
+        actual_value = val_value.GetValueAsSigned (error, 0);
+        self.assertTrue (error.Success(), "Got a value from val")
+        self.assertTrue (actual_value == 100, "Got the right value from val")
+        
+        result = val_value.SetValueFromCString ("12345")
+        self.assertTrue (result, "Setting val returned True.")
+        actual_value = val_value.GetValueAsSigned (error, 0);
+        self.assertTrue (error.Success(), "Got a changed value from val")
+        self.assertTrue (actual_value == 12345, "Got the right changed value from val")
+        
+        # Now check that we can set a structure element:
+
+        mine_value = frame0.FindVariable ("mine")
+        self.assertTrue (mine_value.IsValid(), "Got the SBValue for mine")
+        
+        mine_second_value = mine_value.GetChildMemberWithName ("second_val")
+        self.assertTrue (mine_second_value.IsValid(), "Got second_val from mine")
+        actual_value = mine_second_value.GetValueAsUnsigned (error, 0)
+        self.assertTrue (error.Success(), "Got an unsigned value for second_val")
+        self.assertTrue (actual_value == 5555)
+
+        result = mine_second_value.SetValueFromCString ("98765")
+        self.assertTrue (result, "Success setting mine.second_value.")
+        actual_value = mine_second_value.GetValueAsSigned (error, 0);
+        self.assertTrue (error.Success(), "Got a changed value from mine.second_val")
+        self.assertTrue (actual_value == 98765, "Got the right changed value from mine.second_val")
+        
+        # Next do the same thing with the pointer version.
+        ptr_value = frame0.FindVariable ("ptr")
+        self.assertTrue (ptr_value.IsValid(), "Got the SBValue for ptr")
+        
+        ptr_second_value = ptr_value.GetChildMemberWithName ("second_val")
+        self.assertTrue (ptr_second_value.IsValid(), "Got second_val from ptr")
+        actual_value = ptr_second_value.GetValueAsUnsigned (error, 0)
+        self.assertTrue (error.Success(), "Got an unsigned value for ptr->second_val")
+        self.assertTrue (actual_value == 6666)
+
+        result = ptr_second_value.SetValueFromCString ("98765")
+        self.assertTrue (result, "Success setting ptr->second_value.")
+        actual_value = ptr_second_value.GetValueAsSigned (error, 0);
+        self.assertTrue (error.Success(), "Got a changed value from ptr->second_val")
+        self.assertTrue (actual_value == 98765, "Got the right changed value from ptr->second_val")
+        
+        # Now step, grab the stdout and make sure we changed the real values as well...
+        thread.StepOver()
+        expected_value = "Val - 12345 Mine - 55, 98765, 55555555. Ptr - 66, 98765, 66666666"
+        stdout = process.GetSTDOUT(1000)
+        print stdout
+        self.assertTrue (expected_value in stdout, "STDOUT showed changed values.")
+
+        # Finally, change the stack pointer to 0, and we should not make it to our end breakpoint.
+        frame0 = thread.GetFrameAtIndex(0)
+        self.assertTrue (frame0.IsValid(), "Second time: got a valid frame.")
+        sp_value = frame0.FindValue ("sp", lldb.eValueTypeRegister);
+        self.assertTrue (sp_value.IsValid(), "Got a stack pointer value")
+        result = sp_value.SetValueFromCString("1")
+        self.assertTrue (result, "Setting sp returned true.")
+        # GetValueAsUnsigned doesn't work for register value objects yet. 
+        # actual_value = sp_value.GetValueAsUnsigned (error, 0)
+        # self.assertTrue (error.Success(), "Got a changed value for sp")
+        # self.assertTrue (actual_value == 1, "Got the right changed value for sp.")
+        
+        process.Continue()
+
+        self.assertTrue(process.GetState() == lldb.eStateStopped)
+        thread = lldbutil.get_stopped_thread(process, lldb.eStopReasonBreakpoint)
+        self.assertTrue(thread == None, "We should not have managed to hit our second breakpoint with sp == 1")
+        
+        process.Kill()
+
+if __name__ == '__main__':
+    import atexit
+    lldb.SBDebugger.Initialize()
+    atexit.register(lambda: lldb.SBDebugger.Terminate())
+    unittest2.main()
diff --git a/lldb/test/python_api/value/change_values/main.c b/lldb/test/python_api/value/change_values/main.c
new file mode 100644
index 00000000000..a9cff8ec231
--- /dev/null
+++ b/lldb/test/python_api/value/change_values/main.c
@@ -0,0 +1,29 @@
+#include <stdio.h>
+#include <stdint.h>
+#include <stdlib.h>
+
+struct foo
+{
+  uint8_t   first_val;
+  uint32_t  second_val;
+  uint64_t  third_val;
+};
+  
+int main ()
+{
+  int val = 100;
+  struct foo mine = {55, 5555, 55555555};
+  struct foo *ptr = (struct foo *) malloc (sizeof (struct foo));
+  ptr->first_val = 66;
+  ptr->second_val = 6666;
+  ptr->third_val = 66666666;
+
+  // Stop here and set values
+  printf ("Val - %d Mine - %d, %d, %llu. Ptr - %d, %d, %llu\n", 
+          val, 
+          mine.first_val, mine.second_val, mine.third_val,
+          ptr->first_val, ptr->second_val, ptr->third_val); 
+
+  printf ("This is just another call which we won't make it over %d.", val);
+  return 0; // Set a breakpoint here at the end
+}