diff --git a/lldb/test/API/functionalities/interactive_scripted_process/TestInteractiveScriptedProcess.py b/lldb/test/API/functionalities/interactive_scripted_process/TestInteractiveScriptedProcess.py --- a/lldb/test/API/functionalities/interactive_scripted_process/TestInteractiveScriptedProcess.py +++ b/lldb/test/API/functionalities/interactive_scripted_process/TestInteractiveScriptedProcess.py @@ -12,15 +12,185 @@ NO_DEBUG_INFO_TESTCASE = True - def test_passthrough_launch(self): - """Test a simple pass-through process launch""" + def setUp(self): + # Call super's setUp(). + TestBase.setUp(self) + # Build and load test program self.build() self.runCmd("file " + self.getBuildArtifact("a.out"), CURRENT_EXECUTABLE_SET) self.main_source_file = lldb.SBFileSpec("main.cpp") self.script_module = "interactive_scripted_process" self.script_file = self.script_module + ".py" + + def test_passthrough_launch(self): + """Test a simple pass-through process launch""" self.passthrough_launch() + lldbutil.run_break_set_by_source_regexp(self, "also break here") + self.assertEqual(self.mux_target.GetNumBreakpoints(), 2) + error = self.mux_process.Continue() + self.assertSuccess(error, "Resuming multiplexer scripted process") + self.assertTrue(self.mux_process.IsValid(), "Got a valid process") + + event = lldbutil.fetch_next_event( + self, self.mux_process_listener, self.mux_process.GetBroadcaster() + ) + self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateRunning) + event = lldbutil.fetch_next_event( + self, self.mux_process_listener, self.mux_process.GetBroadcaster() + ) + self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateStopped) + + def test_multiplexed_launch(self): + """Test a multiple interactive scripted process debugging""" + self.passthrough_launch() + self.assertEqual(self.dbg.GetNumTargets(), 2) + + driving_target = self.mux_process.GetScriptedImplementation().driving_target + self.assertTrue(driving_target.IsValid(), "Driving target is invalid") + + # Create a target for the multiplexed even scripted process + even_target = self.duplicate_target(driving_target) + self.assertTrue( + even_target.IsValid(), + "Couldn't duplicate driving target to launch multiplexed even scripted process", + ) + + class_name = f"{self.script_module}.MultiplexedScriptedProcess" + dictionary = {"driving_target_idx": self.dbg.GetIndexOfTarget(self.mux_target)} + + dictionary["parity"] = 0 + muxed_launch_info = self.get_launch_info(class_name, dictionary) + + # Launch Even Child Scripted Process + error = lldb.SBError() + even_process = even_target.Launch(muxed_launch_info, error) + self.assertTrue( + even_process, "Couldn't launch multiplexed even scripted process" + ) + self.multiplex(even_process) + + # Check that the even process started running + event = lldbutil.fetch_next_event( + self, self.dbg.GetListener(), even_process.GetBroadcaster() + ) + self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateRunning) + # Check that the even process stopped + event = lldbutil.fetch_next_event( + self, self.dbg.GetListener(), even_process.GetBroadcaster() + ) + self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateStopped) + + self.assertTrue(even_process.IsValid(), "Got a valid process") + self.assertState( + even_process.GetState(), lldb.eStateStopped, "Process is stopped" + ) + + # Create a target for the multiplexed odd scripted process + odd_target = self.duplicate_target(driving_target) + self.assertTrue( + odd_target.IsValid(), + "Couldn't duplicate driving target to launch multiplexed odd scripted process", + ) + + dictionary["parity"] = 1 + muxed_launch_info = self.get_launch_info(class_name, dictionary) + + # Launch Odd Child Scripted Process + error = lldb.SBError() + odd_process = odd_target.Launch(muxed_launch_info, error) + self.assertTrue(odd_process, "Couldn't launch multiplexed odd scripted process") + self.multiplex(odd_process) + + # Check that the odd process started running + event = lldbutil.fetch_next_event( + self, self.dbg.GetListener(), odd_process.GetBroadcaster() + ) + self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateRunning) + # Check that the odd process stopped + event = lldbutil.fetch_next_event( + self, self.dbg.GetListener(), odd_process.GetBroadcaster() + ) + self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateStopped) + + self.assertTrue(odd_process.IsValid(), "Got a valid process") + self.assertState( + odd_process.GetState(), lldb.eStateStopped, "Process is stopped" + ) + + # Set a breakpoint on the odd child process + bkpt = odd_target.BreakpointCreateBySourceRegex( + "also break here", self.main_source_file + ) + self.assertEqual(odd_target.GetNumBreakpoints(), 1) + self.assertTrue(bkpt, "Second breakpoint set on child scripted process") + self.assertEqual(bkpt.GetNumLocations(), 1, "Second breakpoint has 1 location") + + # Verify that the breakpoint was also set on the multiplexer & real target + self.assertEqual(self.mux_target.GetNumBreakpoints(), 2) + bkpt = self.mux_target.GetBreakpointAtIndex(1) + self.assertEqual( + bkpt.GetNumLocations(), 1, "Second breakpoint set on mux scripted process" + ) + self.assertTrue(bkpt.MatchesName("multiplexed_scripted_process_421")) + + self.assertGreater(driving_target.GetNumBreakpoints(), 1) + + # Resume execution on child process + error = odd_process.Continue() + self.assertSuccess(error, "Resuming odd child scripted process") + self.assertTrue(odd_process.IsValid(), "Got a valid process") + + # Since all the execution is asynchronous, the order in which events + # arrive is non-deterministic, so we need a data structure to make sure + # we received both the running and stopped event for each target. + + # Initialize the execution event "bingo book", that maps a process index + # to a dictionary that contains flags that are not set for the process + # events that we care about (running & stopped) + + execution_events = { + 1: {lldb.eStateRunning: False, lldb.eStateStopped: False}, + 2: {lldb.eStateRunning: False, lldb.eStateStopped: False}, + 3: {lldb.eStateRunning: False, lldb.eStateStopped: False}, + } + + def fetch_process_event(self, execution_events): + event = lldbutil.fetch_next_event( + self, + self.dbg.GetListener(), + lldb.SBProcess.GetBroadcasterClass(), + match_class=True, + ) + state = lldb.SBProcess.GetStateFromEvent(event) + self.assertIn(state, [lldb.eStateRunning, lldb.eStateStopped]) + event_process = lldb.SBProcess.GetProcessFromEvent(event) + self.assertTrue(event_process.IsValid()) + event_target = event_process.GetTarget() + event_target_idx = self.dbg.GetIndexOfTarget(event_target) + self.assertFalse( + execution_events[event_target_idx][state], + "Event already received for this process", + ) + execution_events[event_target_idx][state] = True + + event = lldbutil.fetch_next_event( + self, self.mux_process_listener, self.mux_process.GetBroadcaster() + ) + self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateRunning) + + event = lldbutil.fetch_next_event( + self, self.mux_process_listener, self.mux_process.GetBroadcaster() + ) + self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateStopped) + + for _ in range((self.dbg.GetNumTargets() - 1) * 2): + fetch_process_event(self, execution_events) + + for target_index, event_states in execution_events.items(): + for state, is_set in event_states.items(): + self.assertTrue(is_set, f"Target {target_index} has state {state} set") + def duplicate_target(self, driving_target): exe = driving_target.executable.fullpath triple = driving_target.triple @@ -36,6 +206,14 @@ launch_info.SetScriptedProcessDictionary(structured_data) return launch_info + def multiplex(self, muxed_process): + muxed_process.GetScriptedImplementation().multiplexer = ( + self.mux_process.GetScriptedImplementation() + ) + self.mux_process.GetScriptedImplementation().multiplexed_processes[ + muxed_process.GetProcessID() + ] = muxed_process + def passthrough_launch(self): """Test that a simple passthrough wrapper functions correctly""" # First build the real target: @@ -49,42 +227,42 @@ script_path = os.path.join(self.getSourceDir(), self.script_file) self.runCmd(f"command script import '{script_path}'") - mux_target = self.duplicate_target(real_target) - self.assertTrue(mux_target.IsValid(), "duplicate target succeeded") + self.mux_target = self.duplicate_target(real_target) + self.assertTrue(self.mux_target.IsValid(), "duplicate target succeeded") mux_class = f"{self.script_module}.MultiplexerScriptedProcess" script_dict = {"driving_target_idx": real_target_id} mux_launch_info = self.get_launch_info(mux_class, script_dict) - mux_process_listener = lldb.SBListener( + self.mux_process_listener = lldb.SBListener( "lldb.test.interactive-scripted-process.listener" ) - mux_launch_info.SetPassthroughListener(mux_process_listener) + mux_launch_info.SetShadowListener(self.mux_process_listener) self.dbg.SetAsync(True) error = lldb.SBError() - mux_process = mux_target.Launch(mux_launch_info, error) + self.mux_process = self.mux_target.Launch(mux_launch_info, error) self.assertSuccess(error, "Launched multiplexer scripted process") - self.assertTrue(mux_process.IsValid(), "Got a valid process") + self.assertTrue(self.mux_process.IsValid(), "Got a valid process") # Check that the mux process started running event = lldbutil.fetch_next_event( - self, mux_process_listener, mux_process.GetBroadcaster(), timeout=60 * 5 + self, self.mux_process_listener, self.mux_process.GetBroadcaster() ) self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateRunning) # Check that the real process started running event = lldbutil.fetch_next_event( - self, self.dbg.GetListener(), mux_process.GetBroadcaster() + self, self.dbg.GetListener(), self.mux_process.GetBroadcaster() ) self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateRunning) # Check that the real process stopped event = lldbutil.fetch_next_event( - self, self.dbg.GetListener(), mux_process.GetBroadcaster(), timeout=60 * 5 + self, self.dbg.GetListener(), self.mux_process.GetBroadcaster() ) self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateStopped) # Check that the mux process stopped event = lldbutil.fetch_next_event( - self, mux_process_listener, mux_process.GetBroadcaster(), timeout=60 * 5 + self, self.mux_process_listener, self.mux_process.GetBroadcaster() ) self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateStopped) @@ -98,25 +276,10 @@ # Check that we got the right threads: self.assertEqual( len(real_process.threads), - len(mux_process.threads), + len(self.mux_process.threads), "Same number of threads", ) for id in range(len(real_process.threads)): real_pc = real_process.threads[id].frame[0].pc - mux_pc = mux_process.threads[id].frame[0].pc + mux_pc = self.mux_process.threads[id].frame[0].pc self.assertEqual(real_pc, mux_pc, f"PC's equal for {id}") - - lldbutil.run_break_set_by_source_regexp(self, "also break here") - self.assertEqual(mux_target.GetNumBreakpoints(), 2) - error = mux_process.Continue() - self.assertSuccess(error, "Resuming multiplexer scripted process") - self.assertTrue(mux_process.IsValid(), "Got a valid process") - - event = lldbutil.fetch_next_event( - self, mux_process_listener, mux_process.GetBroadcaster() - ) - self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateRunning) - event = lldbutil.fetch_next_event( - self, mux_process_listener, mux_process.GetBroadcaster() - ) - self.assertState(lldb.SBProcess.GetStateFromEvent(event), lldb.eStateStopped)