Skip to content

Commit

Permalink
[lldb/test] Fix TestProgressReporting.py race issue with the event li…
Browse files Browse the repository at this point in the history
…stener

This patch is a follow-up of D120100 to address some feedbacks from
@labath.

This should mainly fix the race issue with the even listener by moving
the listener setup to the main thread.

This also changes the SBDebugger::GetProgressFromEvent SWIG binding
arguments to be output only, so the user don't have to provide them.

Finally, this updates the test to check it the out arguments are returned
in a tuple and re-enables the test on all platforms.

Differential Revision: https://reviews.llvm.org/D120284

Signed-off-by: Med Ismail Bennani <medismail.bennani@gmail.com>
  • Loading branch information
medismailben committed Feb 24, 2022
1 parent 6a38336 commit 3e3e79a
Show file tree
Hide file tree
Showing 2 changed files with 23 additions and 25 deletions.
11 changes: 4 additions & 7 deletions lldb/bindings/interface/SBDebugger.i
Expand Up @@ -123,14 +123,11 @@ public:
};


%apply uint64_t& INOUT { uint64_t& progress_id };
%apply uint64_t& INOUT { uint64_t& completed };
%apply uint64_t& INOUT { uint64_t& total };
%apply bool& INOUT { bool& is_debugger_specific };
static const char *GetProgressFromEvent(const lldb::SBEvent &event,
uint64_t &progress_id,
uint64_t &completed, uint64_t &total,
bool &is_debugger_specific);
uint64_t &OUTPUT,
uint64_t &OUTPUT,
uint64_t &OUTPUT,
bool &OUTPUT);

SBBroadcaster GetBroadcaster();

Expand Down
Expand Up @@ -17,41 +17,42 @@ def setUp(self):
TestBase.setUp(self)
self.progress_events = []

def fetch_events(self, test_broadcaster):
listener = lldb.SBListener("lldb.progress.listener")
listener.StartListeningForEvents(test_broadcaster,
self.eBroadcastBitStopProgressThread)

progress_broadcaster = self.dbg.GetBroadcaster()
progress_broadcaster.AddListener(listener, lldb.SBDebugger.eBroadcastBitProgress)

def fetch_events(self):
event = lldb.SBEvent()

done = False
while not done:
if listener.WaitForEvent(1, event):
if self.listener.WaitForEvent(1, event):
event_mask = event.GetType();
if event.BroadcasterMatchesRef(test_broadcaster):
if event.BroadcasterMatchesRef(self.test_broadcaster):
if event_mask & self.eBroadcastBitStopProgressThread:
done = True;
elif event.BroadcasterMatchesRef(progress_broadcaster):
message = lldb.SBDebugger().GetProgressFromEvent(event, 0, 0, 0, False);
elif event.BroadcasterMatchesRef(self.progress_broadcaster):
ret_args = lldb.SBDebugger().GetProgressFromEvent(event);
self.assertGreater(len(ret_args), 1)

message = ret_args[0]
if message:
self.progress_events.append((message, event))

@skipUnlessDarwin
def test_dwarf_symbol_loading_progress_report(self):
"""Test that we are able to fetch dwarf symbol loading progress events"""
self.build()

test_broadcaster = lldb.SBBroadcaster('lldb.broadcaster.test')
listener_thread = threading.Thread(target=self.fetch_events,
args=[test_broadcaster])
self.listener = lldb.SBListener("lldb.progress.listener")
self.test_broadcaster = lldb.SBBroadcaster('lldb.broadcaster.test')
self.listener.StartListeningForEvents(self.test_broadcaster,
self.eBroadcastBitStopProgressThread)

self.progress_broadcaster = self.dbg.GetBroadcaster()
self.progress_broadcaster.AddListener(self.listener, lldb.SBDebugger.eBroadcastBitProgress)

listener_thread = threading.Thread(target=self.fetch_events)
listener_thread.start()

lldbutil.run_to_source_breakpoint(self, 'break here', lldb.SBFileSpec('main.c'))

test_broadcaster.BroadcastEventByType(self.eBroadcastBitStopProgressThread)
self.test_broadcaster.BroadcastEventByType(self.eBroadcastBitStopProgressThread)
listener_thread.join()

self.assertTrue(len(self.progress_events) > 0)
self.assertGreater(len(self.progress_events), 0)

0 comments on commit 3e3e79a

Please sign in to comment.