1*99451b44SJordan Rupprecht"""
2*99451b44SJordan RupprechtTest some lldb command abbreviations.
3*99451b44SJordan Rupprecht"""
4*99451b44SJordan Rupprechtfrom __future__ import print_function
5*99451b44SJordan Rupprecht
6*99451b44SJordan Rupprecht
7*99451b44SJordan Rupprechtimport lldb
8*99451b44SJordan Rupprechtfrom lldbsuite.test.decorators import *
9*99451b44SJordan Rupprechtfrom lldbsuite.test.lldbtest import *
10*99451b44SJordan Rupprechtfrom lldbsuite.test import lldbutil
11*99451b44SJordan Rupprecht
12*99451b44SJordan Rupprecht
13*99451b44SJordan Rupprechtclass ExecTestCase(TestBase):
14*99451b44SJordan Rupprecht
15*99451b44SJordan Rupprecht    NO_DEBUG_INFO_TESTCASE = True
16*99451b44SJordan Rupprecht
17*99451b44SJordan Rupprecht    mydir = TestBase.compute_mydir(__file__)
18*99451b44SJordan Rupprecht
19*99451b44SJordan Rupprecht    @expectedFailureAll(archs=['i386'], bugnumber="rdar://28656532")
20*99451b44SJordan Rupprecht    @expectedFailureAll(oslist=["ios", "tvos", "watchos", "bridgeos"], bugnumber="rdar://problem/34559552") # this exec test has problems on ios systems
21*99451b44SJordan Rupprecht    @expectedFailureNetBSD
22*99451b44SJordan Rupprecht    @skipIfAsan # rdar://problem/43756823
23*99451b44SJordan Rupprecht    @skipIfWindows
24*99451b44SJordan Rupprecht    def test_hitting_exec (self):
25*99451b44SJordan Rupprecht        self.do_test(False)
26*99451b44SJordan Rupprecht
27*99451b44SJordan Rupprecht    @expectedFailureAll(archs=['i386'], bugnumber="rdar://28656532")
28*99451b44SJordan Rupprecht    @expectedFailureAll(oslist=["ios", "tvos", "watchos", "bridgeos"], bugnumber="rdar://problem/34559552") # this exec test has problems on ios systems
29*99451b44SJordan Rupprecht    @expectedFailureNetBSD
30*99451b44SJordan Rupprecht    @skipIfAsan # rdar://problem/43756823
31*99451b44SJordan Rupprecht    @skipIfWindows
32*99451b44SJordan Rupprecht    def test_skipping_exec (self):
33*99451b44SJordan Rupprecht        self.do_test(True)
34*99451b44SJordan Rupprecht
35*99451b44SJordan Rupprecht    def do_test(self, skip_exec):
36*99451b44SJordan Rupprecht        self.build()
37*99451b44SJordan Rupprecht        exe = self.getBuildArtifact("a.out")
38*99451b44SJordan Rupprecht        secondprog = self.getBuildArtifact("secondprog")
39*99451b44SJordan Rupprecht
40*99451b44SJordan Rupprecht        # Create the target
41*99451b44SJordan Rupprecht        target = self.dbg.CreateTarget(exe)
42*99451b44SJordan Rupprecht
43*99451b44SJordan Rupprecht        # Create any breakpoints we need
44*99451b44SJordan Rupprecht        breakpoint1 = target.BreakpointCreateBySourceRegex(
45*99451b44SJordan Rupprecht            'Set breakpoint 1 here', lldb.SBFileSpec("main.cpp", False))
46*99451b44SJordan Rupprecht        self.assertTrue(breakpoint1, VALID_BREAKPOINT)
47*99451b44SJordan Rupprecht        breakpoint2 = target.BreakpointCreateBySourceRegex(
48*99451b44SJordan Rupprecht            'Set breakpoint 2 here', lldb.SBFileSpec("secondprog.cpp", False))
49*99451b44SJordan Rupprecht        self.assertTrue(breakpoint2, VALID_BREAKPOINT)
50*99451b44SJordan Rupprecht
51*99451b44SJordan Rupprecht        # Launch the process
52*99451b44SJordan Rupprecht        process = target.LaunchSimple(
53*99451b44SJordan Rupprecht            None, None, self.get_process_working_directory())
54*99451b44SJordan Rupprecht        self.assertTrue(process, PROCESS_IS_VALID)
55*99451b44SJordan Rupprecht
56*99451b44SJordan Rupprecht        if self.TraceOn():
57*99451b44SJordan Rupprecht            self.runCmd("settings show target.process.stop-on-exec", check=False)
58*99451b44SJordan Rupprecht        if skip_exec:
59*99451b44SJordan Rupprecht            self.dbg.HandleCommand("settings set target.process.stop-on-exec false")
60*99451b44SJordan Rupprecht            def cleanup():
61*99451b44SJordan Rupprecht                self.runCmd("settings set target.process.stop-on-exec false",
62*99451b44SJordan Rupprecht                            check=False)
63*99451b44SJordan Rupprecht
64*99451b44SJordan Rupprecht            # Execute the cleanup function during test case tear down.
65*99451b44SJordan Rupprecht            self.addTearDownHook(cleanup)
66*99451b44SJordan Rupprecht
67*99451b44SJordan Rupprecht        # The stop reason of the thread should be breakpoint.
68*99451b44SJordan Rupprecht        self.assertTrue(process.GetState() == lldb.eStateStopped,
69*99451b44SJordan Rupprecht                        STOPPED_DUE_TO_BREAKPOINT)
70*99451b44SJordan Rupprecht
71*99451b44SJordan Rupprecht        threads = lldbutil.get_threads_stopped_at_breakpoint(
72*99451b44SJordan Rupprecht        process, breakpoint1)
73*99451b44SJordan Rupprecht        self.assertTrue(len(threads) == 1)
74*99451b44SJordan Rupprecht
75*99451b44SJordan Rupprecht        # We had a deadlock tearing down the TypeSystemMap on exec, but only if some
76*99451b44SJordan Rupprecht        # expression had been evaluated.  So make sure we do that here so the teardown
77*99451b44SJordan Rupprecht        # is not trivial.
78*99451b44SJordan Rupprecht
79*99451b44SJordan Rupprecht        thread = threads[0]
80*99451b44SJordan Rupprecht        value = thread.frames[0].EvaluateExpression("1 + 2")
81*99451b44SJordan Rupprecht        self.assertTrue(
82*99451b44SJordan Rupprecht            value.IsValid(),
83*99451b44SJordan Rupprecht            "Expression evaluated successfully")
84*99451b44SJordan Rupprecht        int_value = value.GetValueAsSigned()
85*99451b44SJordan Rupprecht        self.assertTrue(int_value == 3, "Expression got the right result.")
86*99451b44SJordan Rupprecht
87*99451b44SJordan Rupprecht        # Run and we should stop due to exec
88*99451b44SJordan Rupprecht        process.Continue()
89*99451b44SJordan Rupprecht
90*99451b44SJordan Rupprecht        if not skip_exec:
91*99451b44SJordan Rupprecht            self.assertFalse(process.GetState() == lldb.eStateExited,
92*99451b44SJordan Rupprecht                             "Process should not have exited!")
93*99451b44SJordan Rupprecht            self.assertTrue(process.GetState() == lldb.eStateStopped,
94*99451b44SJordan Rupprecht                            "Process should be stopped at __dyld_start")
95*99451b44SJordan Rupprecht
96*99451b44SJordan Rupprecht            threads = lldbutil.get_stopped_threads(
97*99451b44SJordan Rupprecht                process, lldb.eStopReasonExec)
98*99451b44SJordan Rupprecht            self.assertTrue(
99*99451b44SJordan Rupprecht                len(threads) == 1,
100*99451b44SJordan Rupprecht                "We got a thread stopped for exec.")
101*99451b44SJordan Rupprecht
102*99451b44SJordan Rupprecht            # Run and we should stop at breakpoint in main after exec
103*99451b44SJordan Rupprecht            process.Continue()
104*99451b44SJordan Rupprecht
105*99451b44SJordan Rupprecht        threads = lldbutil.get_threads_stopped_at_breakpoint(
106*99451b44SJordan Rupprecht            process, breakpoint2)
107*99451b44SJordan Rupprecht        if self.TraceOn():
108*99451b44SJordan Rupprecht            for t in process.threads:
109*99451b44SJordan Rupprecht                print(t)
110*99451b44SJordan Rupprecht                if t.GetStopReason() != lldb.eStopReasonBreakpoint:
111*99451b44SJordan Rupprecht                    self.runCmd("bt")
112*99451b44SJordan Rupprecht        self.assertTrue(len(threads) == 1,
113*99451b44SJordan Rupprecht                        "Stopped at breakpoint in exec'ed process.")
114