1"""
2Test SBSymbolContext APIs.
3"""
4
5
6
7import lldb
8from lldbsuite.test.decorators import *
9from lldbsuite.test.lldbtest import *
10from lldbsuite.test import lldbutil
11
12
13class SymbolContextTwoFilesTestCase(TestBase):
14
15    @expectedFailureAll(oslist=["windows"])
16    def test_lookup_by_address(self):
17        """Test lookup by address in a module with multiple compilation units"""
18        self.build()
19        exe = self.getBuildArtifact("a.out")
20        target = self.dbg.CreateTarget(exe)
21        self.assertTrue(target, VALID_TARGET)
22
23        module = target.GetModuleAtIndex(0)
24        self.assertTrue(module.IsValid())
25        for symbol_name in ["struct1::f()", "struct2::f()"]:
26            sc_list = module.FindFunctions(symbol_name, lldb.eSymbolTypeCode)
27            self.assertTrue(1, sc_list.GetSize())
28            symbol_address = sc_list.GetContextAtIndex(
29                0).GetSymbol().GetStartAddress()
30            self.assertTrue(symbol_address.IsValid())
31            sc_by_address = module.ResolveSymbolContextForAddress(
32                symbol_address, lldb.eSymbolContextFunction)
33            self.assertEqual(symbol_name,
34                             sc_by_address.GetFunction().GetName())
35
36    def test_ranges_in_multiple_compile_unit(self):
37        """This test verifies that we correctly handle the case when multiple
38        compile unit contains DW_AT_ranges and DW_AT_ranges_base attributes."""
39        self.build()
40        exe = self.getBuildArtifact("a.out")
41        target = self.dbg.CreateTarget(exe)
42        self.assertTrue(target, VALID_TARGET)
43
44        source1 = "file1.cpp"
45        line1 = line_number(source1, '// Break1')
46        breakpoint1 = target.BreakpointCreateByLocation(source1, line1)
47        self.assertIsNotNone(breakpoint1)
48        self.assertTrue(breakpoint1.IsValid())
49
50        source2 = "file2.cpp"
51        line2 = line_number(source2, '// Break2')
52        breakpoint2 = target.BreakpointCreateByLocation(source2, line2)
53        self.assertIsNotNone(breakpoint2)
54        self.assertTrue(breakpoint2.IsValid())
55
56        process = target.LaunchSimple(None, None, self.get_process_working_directory())
57        self.assertIsNotNone(process, PROCESS_IS_VALID)
58
59        threads = lldbutil.get_threads_stopped_at_breakpoint(
60            process, breakpoint2)
61        self.assertEqual(len(threads), 1)
62        frame = threads[0].GetFrameAtIndex(0)
63        value = frame.FindVariable("x")
64        self.assertTrue(value.IsValid())
65
66        process.Continue()
67
68        threads = lldbutil.get_threads_stopped_at_breakpoint(
69            process, breakpoint1)
70        self.assertEqual(len(threads), 1)
71        frame = threads[0].GetFrameAtIndex(0)
72        value = frame.FindVariable("x")
73        self.assertTrue(value.IsValid())
74
75        process.Continue()
76