TestSymbolAPI.py revision 919ee60577297be31eb69f14a5901e7c6170a9ca 
1"""
2Test newly added SBSymbol and SBAddress APIs.
3"""
4
5import os, time
6import re
7import unittest2
8import lldb, lldbutil
9from lldbtest import *
10
11class SymbolAPITestCase(TestBase):
12
13    mydir = os.path.join("python_api", "function_symbol")
14
15    @unittest2.skipUnless(sys.platform.startswith("darwin"), "requires Darwin")
16    @python_api_test
17    def test_with_dsym(self):
18        """Exercise some SBSymbol and SBAddress APIs."""
19        self.buildDsym()
20        self.symbol_and_address_api()
21
22    @python_api_test
23    def test_with_dwarf(self):
24        """Exercise some SBSymbol and SBAddress APIs."""
25        self.buildDwarf()
26        self.symbol_and_address_api()
27
28    def setUp(self):
29        # Call super's setUp().
30        TestBase.setUp(self)
31        # Find the line number to of function 'c'.
32        self.line1 = line_number('main.c', '// Find the line number for breakpoint 1 here.')
33        self.line2 = line_number('main.c', '// Find the line number for breakpoint 2 here.')
34
35    def symbol_and_address_api(self):
36        """Exercise some SBSymbol and SBAddress APIs."""
37        exe = os.path.join(os.getcwd(), "a.out")
38
39        # Create a target by the debugger.
40        target = self.dbg.CreateTarget(exe)
41        self.assertTrue(target.IsValid(), VALID_TARGET)
42
43        # Now create the two breakpoints inside function 'a'.
44        breakpoint1 = target.BreakpointCreateByLocation('main.c', self.line1)
45        breakpoint2 = target.BreakpointCreateByLocation('main.c', self.line2)
46        #print "breakpoint1:", breakpoint1
47        #print "breakpoint2:", breakpoint2
48        self.assertTrue(breakpoint1.IsValid() and
49                        breakpoint1.GetNumLocations() == 1,
50                        VALID_BREAKPOINT)
51        self.assertTrue(breakpoint2.IsValid() and
52                        breakpoint2.GetNumLocations() == 1,
53                        VALID_BREAKPOINT)
54
55        # Now launch the process, and do not stop at entry point.
56        error = lldb.SBError()
57        self.process = target.LaunchSimple (None, None, os.getcwd())
58
59        self.process = target.GetProcess()
60        self.assertTrue(self.process.IsValid(), PROCESS_IS_VALID)
61
62        # Frame #0 should be on self.line1.
63        self.assertTrue(self.process.GetState() == lldb.eStateStopped)
64        thread = lldbutil.get_stopped_thread(self.process, lldb.eStopReasonBreakpoint)
65        self.assertTrue(thread != None, "There should be a thread stopped due to breakpoint condition")
66        frame0 = thread.GetFrameAtIndex(0)
67        symbol_line1 = frame0.GetSymbol()
68        # We should have a symbol type of code.
69        self.assertTrue(symbol_line1.GetType() == lldb.eSymbolTypeCode)
70        addr_line1 = symbol_line1.GetStartAddress()
71        # And a section type of code, too.
72        self.assertTrue(addr_line1.GetSectionType() == lldb.eSectionTypeCode)
73
74        # Continue the inferior, the breakpoint 2 should be hit.
75        self.process.Continue()
76        self.assertTrue(self.process.GetState() == lldb.eStateStopped)
77        thread = lldbutil.get_stopped_thread(self.process, lldb.eStopReasonBreakpoint)
78        self.assertTrue(thread != None, "There should be a thread stopped due to breakpoint condition")
79        frame0 = thread.GetFrameAtIndex(0)
80        symbol_line2 = frame0.GetSymbol()
81        # We should have a symbol type of code.
82        self.assertTrue(symbol_line2.GetType() == lldb.eSymbolTypeCode)
83        addr_line2 = symbol_line2.GetStartAddress()
84        # And a section type of code, too.
85        self.assertTrue(addr_line2.GetSectionType() == lldb.eSectionTypeCode)
86
87        # Now verify that both addresses point to the same module.
88        print "UUID:", addr_line1.GetModule().GetUUIDString()
89        self.assertTrue(addr_line1.GetModule().GetUUIDString() == addr_line2.GetModule().GetUUIDString())
90
91
92if __name__ == '__main__':
93    import atexit
94    lldb.SBDebugger.Initialize()
95    atexit.register(lambda: lldb.SBDebugger.Terminate())
96    unittest2.main()
97