199451b44SJordan Rupprecht"""Test reading c-strings from memory via SB API."""
299451b44SJordan Rupprecht
399451b44SJordan Rupprecht
499451b44SJordan Rupprechtimport os
599451b44SJordan Rupprechtimport lldb
699451b44SJordan Rupprechtfrom lldbsuite.test.decorators import *
799451b44SJordan Rupprechtfrom lldbsuite.test.lldbtest import *
899451b44SJordan Rupprechtfrom lldbsuite.test import lldbutil
999451b44SJordan Rupprecht
1099451b44SJordan Rupprecht
1199451b44SJordan Rupprechtclass TestReadMemCString(TestBase):
1299451b44SJordan Rupprecht
1399451b44SJordan Rupprecht    mydir = TestBase.compute_mydir(__file__)
1499451b44SJordan Rupprecht    NO_DEBUG_INFO_TESTCASE = True
1599451b44SJordan Rupprecht
167606a543SJonas Devlieghere    @skipIfReproducer # SBProcess::ReadCStringFromMemory is not instrumented.
1799451b44SJordan Rupprecht    def test_read_memory_c_string(self):
1899451b44SJordan Rupprecht        """Test corner case behavior of SBProcess::ReadCStringFromMemory"""
1999451b44SJordan Rupprecht        self.build()
2099451b44SJordan Rupprecht        self.dbg.SetAsync(False)
2199451b44SJordan Rupprecht
2299451b44SJordan Rupprecht        self.main_source = "main.c"
2399451b44SJordan Rupprecht        self.main_source_path = os.path.join(self.getSourceDir(),
2499451b44SJordan Rupprecht                                             self.main_source)
2599451b44SJordan Rupprecht        self.main_source_spec = lldb.SBFileSpec(self.main_source_path)
2699451b44SJordan Rupprecht        self.exe = self.getBuildArtifact("read-mem-cstring")
2799451b44SJordan Rupprecht
2899451b44SJordan Rupprecht        (target, process, thread, bkpt) = lldbutil.run_to_source_breakpoint(
2999451b44SJordan Rupprecht            self, 'breakpoint here', self.main_source_spec, None, self.exe)
3099451b44SJordan Rupprecht
3199451b44SJordan Rupprecht        frame = thread.GetFrameAtIndex(0)
3299451b44SJordan Rupprecht
3399451b44SJordan Rupprecht        err = lldb.SBError()
3499451b44SJordan Rupprecht
3599451b44SJordan Rupprecht        empty_str_addr = frame.FindVariable("empty_string").GetValueAsUnsigned(err)
3699451b44SJordan Rupprecht        self.assertTrue(err.Success())
3799451b44SJordan Rupprecht        self.assertTrue(empty_str_addr != lldb.LLDB_INVALID_ADDRESS)
3899451b44SJordan Rupprecht
3999451b44SJordan Rupprecht        one_letter_str_addr = frame.FindVariable("one_letter_string").GetValueAsUnsigned(err)
4099451b44SJordan Rupprecht        self.assertTrue(err.Success())
4199451b44SJordan Rupprecht        self.assertTrue(one_letter_str_addr != lldb.LLDB_INVALID_ADDRESS)
4299451b44SJordan Rupprecht
4399451b44SJordan Rupprecht        invalid_memory_str_addr = frame.FindVariable("invalid_memory_string").GetValueAsUnsigned(err)
4499451b44SJordan Rupprecht        self.assertTrue(err.Success())
4599451b44SJordan Rupprecht        self.assertTrue(invalid_memory_str_addr != lldb.LLDB_INVALID_ADDRESS)
4699451b44SJordan Rupprecht
4799451b44SJordan Rupprecht        # Important:  An empty (0-length) c-string must come back as a Python string, not a
4899451b44SJordan Rupprecht        # None object.
4999451b44SJordan Rupprecht        empty_str = process.ReadCStringFromMemory(empty_str_addr, 2048, err)
5099451b44SJordan Rupprecht        self.assertTrue(err.Success())
51*619e2e09SDave Lee        self.assertEqual(empty_str, "")
5299451b44SJordan Rupprecht
5399451b44SJordan Rupprecht        one_letter_string = process.ReadCStringFromMemory(one_letter_str_addr, 2048, err)
5499451b44SJordan Rupprecht        self.assertTrue(err.Success())
55*619e2e09SDave Lee        self.assertEqual(one_letter_string, "1")
5699451b44SJordan Rupprecht
5799451b44SJordan Rupprecht        invalid_memory_string = process.ReadCStringFromMemory(invalid_memory_str_addr, 2048, err)
5899451b44SJordan Rupprecht        self.assertTrue(err.Fail())
5999451b44SJordan Rupprecht        self.assertTrue(invalid_memory_string == "" or invalid_memory_string == None)
60