rtspclientsink: fix deadlock on shutdown before preroll

Fixes https://gitlab.freedesktop.org/gstreamer/gst-rtsp-server/-/issues/130

Part-of: <https://gitlab.freedesktop.org/gstreamer/gst-rtsp-server/-/merge_requests/189>

1 files changed, 28 insertions, 1 deletions

diff --git a/gst/rtsp-sink/gstrtspclientsink.c b/gst/rtsp-sink/gstrtspclientsink.c
index 090a264..fa18e78 100644
--- a/gst/rtsp-sink/gstrtspclientsink.c
+++ b/gst/rtsp-sink/gstrtspclientsink.c
@@ -4372,6 +4372,18 @@ done:
   return res;
 }
 
+static gboolean
+gst_rtsp_client_sink_is_stopping (GstRTSPClientSink * sink)
+{
+  gboolean is_stopping;
+
+  GST_OBJECT_LOCK (sink);
+  is_stopping = sink->task == NULL;
+  GST_OBJECT_UNLOCK (sink);
+
+  return is_stopping;
+}
+
 static GstRTSPResult
 gst_rtsp_client_sink_record (GstRTSPClientSink * sink, gboolean async)
 {
@@ -4406,14 +4418,25 @@ gst_rtsp_client_sink_record (GstRTSPClientSink * sink, gboolean async)
    * parts yet. */
   gst_rtsp_client_sink_collect_streams (sink);
 
+  if (gst_rtsp_client_sink_is_stopping (sink)) {
+    GST_INFO_OBJECT (sink, "task stopped, shutting down");
+    return GST_RTSP_EINTR;
+  }
+
   g_mutex_lock (&sink->block_streams_lock);
   /* Wait for streams to be blocked */
-  while (sink->n_streams_blocked < g_list_length (sink->contexts)) {
+  while (sink->n_streams_blocked < g_list_length (sink->contexts)
+      && !gst_rtsp_client_sink_is_stopping (sink)) {
     GST_DEBUG_OBJECT (sink, "waiting for streams to be blocked");
     g_cond_wait (&sink->block_streams_cond, &sink->block_streams_lock);
   }
   g_mutex_unlock (&sink->block_streams_lock);
 
+  if (gst_rtsp_client_sink_is_stopping (sink)) {
+    GST_INFO_OBJECT (sink, "task stopped, shutting down");
+    return GST_RTSP_EINTR;
+  }
+
   /* Send announce, then setup for all streams */
   gst_sdp_message_init (&sink->cursdp);
   sdp = &sink->cursdp;
@@ -4955,6 +4978,10 @@ gst_rtsp_client_sink_stop (GstRTSPClientSink * sink)
 
     gst_task_stop (task);
 
+    g_mutex_lock (&sink->block_streams_lock);
+    g_cond_broadcast (&sink->block_streams_cond);
+    g_mutex_unlock (&sink->block_streams_lock);
+
     /* make sure it is not running */
     GST_RTSP_STREAM_LOCK (sink);
     GST_RTSP_STREAM_UNLOCK (sink);