Change SBProcess::ReadCStringFromMemory() back to returning

an empty Python string object when it reads a 0-length 
string out of memory (and a successful SBError object).

<rdar://problem/26186692> 

llvm-svn: 321338

3 files changed, 81 insertions, 0 deletions

diff --git a/lldb/packages/Python/lldbsuite/test/python_api/process/read-mem-cstring/Makefile b/lldb/packages/Python/lldbsuite/test/python_api/process/read-mem-cstring/Makefile
new file mode 100644
index 00000000000..f3414925f32
--- /dev/null
+++ b/lldb/packages/Python/lldbsuite/test/python_api/process/read-mem-cstring/Makefile
@@ -0,0 +1,6 @@
+LEVEL = ../../../make
+
+C_SOURCES := main.c
+EXE := read-mem-cstring
+
+include $(LEVEL)/Makefile.rules
diff --git a/lldb/packages/Python/lldbsuite/test/python_api/process/read-mem-cstring/TestReadMemCString.py b/lldb/packages/Python/lldbsuite/test/python_api/process/read-mem-cstring/TestReadMemCString.py
new file mode 100644
index 00000000000..e99779a5789
--- /dev/null
+++ b/lldb/packages/Python/lldbsuite/test/python_api/process/read-mem-cstring/TestReadMemCString.py
@@ -0,0 +1,62 @@
+"""Test reading c-strings from memory via SB API."""
+
+from __future__ import print_function
+
+import os
+import lldb
+from lldbsuite.test.decorators import *
+from lldbsuite.test.lldbtest import *
+from lldbsuite.test import lldbutil
+
+
+class TestReadMemCString(TestBase):
+
+    mydir = TestBase.compute_mydir(__file__)
+
+    def setUp(self):
+        TestBase.setUp(self)
+
+    # Need to have a char* pointer that points to unmapped memory to run
+    # this test on other platforms -- Darwin only for now.
+    @skipUnlessDarwin
+    def test_read_memory_c_string(self):
+        """Test corner case behavior of SBProcess::ReadCStringFromMemory"""
+        self.build()
+	self.dbg.SetAsync(False)
+
+        self.main_source = "main.c"
+	self.main_source_spec = lldb.SBFileSpec(self.main_source)
+	self.exe = os.path.join(os.getcwd(), "read-mem-cstring")
+
+        (target, process, thread, bkpt) = lldbutil.run_to_source_breakpoint(
+            self, 'breakpoint here', self.main_source_spec, None, self.exe)
+
+	frame = thread.GetFrameAtIndex(0)
+        
+        err = lldb.SBError()
+
+        empty_str_addr = frame.FindVariable("empty_string").GetValueAsUnsigned(err)
+        self.assertTrue(err.Success())
+        self.assertTrue(empty_str_addr != lldb.LLDB_INVALID_ADDRESS)
+
+        one_letter_str_addr = frame.FindVariable("one_letter_string").GetValueAsUnsigned(err)
+        self.assertTrue(err.Success())
+        self.assertTrue(one_letter_str_addr != lldb.LLDB_INVALID_ADDRESS)
+
+        invalid_memory_str_addr = frame.FindVariable("invalid_memory_string").GetValueAsUnsigned(err)
+        self.assertTrue(err.Success())
+        self.assertTrue(invalid_memory_str_addr != lldb.LLDB_INVALID_ADDRESS)
+
+        # Important:  An empty (0-length) c-string must come back as a Python string, not a
+        # None object.
+        empty_str = process.ReadCStringFromMemory(empty_str_addr, 2048, err)
+        self.assertTrue(err.Success())
+        self.assertTrue(empty_str == "")
+
+        one_letter_string = process.ReadCStringFromMemory(one_letter_str_addr, 2048, err)
+        self.assertTrue(err.Success())
+        self.assertTrue(one_letter_string == "1")
+
+        invalid_memory_string = process.ReadCStringFromMemory(invalid_memory_str_addr, 2048, err)
+        self.assertTrue(err.Fail())
+        self.assertTrue(invalid_memory_string == "" or invalid_memory_string == None)
diff --git a/lldb/packages/Python/lldbsuite/test/python_api/process/read-mem-cstring/main.c b/lldb/packages/Python/lldbsuite/test/python_api/process/read-mem-cstring/main.c
new file mode 100644
index 00000000000..94916997ad3
--- /dev/null
+++ b/lldb/packages/Python/lldbsuite/test/python_api/process/read-mem-cstring/main.c
@@ -0,0 +1,13 @@
+#include <stdlib.h>
+int main ()
+{
+   const char *empty_string = "";
+   const char *one_letter_string = "1";
+#if defined (__APPLE__)
+   const char *invalid_memory_string = (char*)0x100; // lower 4k is always PAGEZERO & unreadable on darwin
+#else
+   const char *invalid_memory_string = -1ULL; // maybe an invalid address on other platforms?
+#endif
+
+   return empty_string[0] + one_letter_string[0]; // breakpoint here
+}