[Spice-commits] src/channel-display-gst.c

GitLab Mirror gitlab-mirror at kemper.freedesktop.org
Thu Jul 5 14:04:23 UTC 2018


 src/channel-display-gst.c |   82 +++++++++++++++++++++++++++++-----------------
 1 file changed, 53 insertions(+), 29 deletions(-)

New commits:
commit 8835e757922cb0342b6236997f5ef2ab61bac1a9
Author: Frediano Ziglio <fziglio at redhat.com>
Date:   Fri Apr 13 21:14:56 2018 +0100

    channel-display-gst: Prevent accumulating output queue
    
    display_queue is queued with decoded frames ready to be displayed.
    However current code can insert a timeout before displaying and
    removing the queued frames. As the frames are not compressed the
    usage of memory by this queue can became in some cases quite
    huge (in the order of multiple gigabytes).
    To prevent this do not queue all the frames but just one and count the
    pending frames.
    This way GStreamer will stop queuing after a while.
    
    Signed-off-by: Frediano Ziglio <fziglio at redhat.com>
    Acked-by: Snir Sheriber <ssheribe at redhat.com>

diff --git a/src/channel-display-gst.c b/src/channel-display-gst.c
index 093258e..29476cb 100644
--- a/src/channel-display-gst.c
+++ b/src/channel-display-gst.c
@@ -29,6 +29,8 @@
 #include <gst/video/gstvideometa.h>
 
 
+typedef struct SpiceGstFrame SpiceGstFrame;
+
 /* GStreamer decoder implementation */
 
 typedef struct SpiceGstDecoder {
@@ -49,8 +51,9 @@ typedef struct SpiceGstDecoder {
 
     GMutex queues_mutex;
     GQueue *decoding_queue;
-    GQueue *display_queue;
+    SpiceGstFrame *display_frame;
     guint timer_id;
+    guint pending_samples;
 } SpiceGstDecoder;
 
 #define VALID_VIDEO_CODEC_TYPE(codec) \
@@ -76,11 +79,11 @@ typedef enum {
 
 /* ---------- SpiceGstFrame ---------- */
 
-typedef struct SpiceGstFrame {
+struct SpiceGstFrame {
     GstClockTime timestamp;
     SpiceFrame *frame;
     GstSample *sample;
-} SpiceGstFrame;
+};
 
 static SpiceGstFrame *create_gst_frame(GstBuffer *buffer, SpiceFrame *frame)
 {
@@ -104,6 +107,7 @@ static void free_gst_frame(SpiceGstFrame *gstframe)
 /* ---------- GStreamer pipeline ---------- */
 
 static void schedule_frame(SpiceGstDecoder *decoder);
+static void fetch_pending_sample(SpiceGstDecoder *decoder);
 
 static int spice_gst_buffer_get_stride(GstBuffer *buffer)
 {
@@ -124,7 +128,8 @@ static gboolean display_frame(gpointer video_decoder)
 
     g_mutex_lock(&decoder->queues_mutex);
     decoder->timer_id = 0;
-    gstframe = g_queue_pop_head(decoder->display_queue);
+    gstframe = decoder->display_frame;
+    decoder->display_frame = NULL;
     g_mutex_unlock(&decoder->queues_mutex);
     /* If the queue is empty we don't even need to reschedule */
     g_return_val_if_fail(gstframe, G_SOURCE_REMOVE);
@@ -170,7 +175,11 @@ static void schedule_frame(SpiceGstDecoder *decoder)
     g_mutex_lock(&decoder->queues_mutex);
 
     while (!decoder->timer_id) {
-        SpiceGstFrame *gstframe = g_queue_peek_head(decoder->display_queue);
+        while (decoder->display_frame == NULL && decoder->pending_samples) {
+            fetch_pending_sample(decoder);
+        }
+
+        SpiceGstFrame *gstframe = decoder->display_frame;
         if (!gstframe) {
             break;
         }
@@ -178,7 +187,7 @@ static void schedule_frame(SpiceGstDecoder *decoder)
         if (spice_mmtime_diff(now, gstframe->frame->mm_time) < 0) {
             decoder->timer_id = g_timeout_add(gstframe->frame->mm_time - now,
                                               display_frame, decoder);
-        } else if (g_queue_get_length(decoder->display_queue) == 1) {
+        } else if (decoder->display_frame && !decoder->pending_samples) {
             /* Still attempt to display the least out of date frame so the
              * video is not completely frozen for an extended period of time.
              */
@@ -188,7 +197,7 @@ static void schedule_frame(SpiceGstDecoder *decoder)
                         __FUNCTION__, now - gstframe->frame->mm_time,
                         gstframe->frame->mm_time, now);
             stream_dropped_frame_on_playback(decoder->base.stream);
-            g_queue_pop_head(decoder->display_queue);
+            decoder->display_frame = NULL;
             free_gst_frame(gstframe);
         }
     }
@@ -196,22 +205,14 @@ static void schedule_frame(SpiceGstDecoder *decoder)
     g_mutex_unlock(&decoder->queues_mutex);
 }
 
-/* GStreamer thread
- *
- * We cannot use GStreamer's signals because they are not always run in
- * the main context. So use a callback (lower overhead) and have it pull
- * the sample to avoid a race with free_pipeline(). This means queuing the
- * decoded frames outside GStreamer. So while we're at it, also schedule
- * the frame display ourselves in schedule_frame().
- */
-static GstFlowReturn new_sample(GstAppSink *gstappsink, gpointer video_decoder)
+static void fetch_pending_sample(SpiceGstDecoder *decoder)
 {
-    SpiceGstDecoder *decoder = video_decoder;
-
     GstSample *sample = gst_app_sink_pull_sample(decoder->appsink);
     if (sample) {
+        // account for the fetched sample
+        decoder->pending_samples--;
+
         GstBuffer *buffer = gst_sample_get_buffer(sample);
-        g_mutex_lock(&decoder->queues_mutex);
 
         /* gst_app_sink_pull_sample() sometimes returns the same buffer twice
          * or buffers that have a modified, and thus unrecognizable, PTS.
@@ -227,7 +228,7 @@ static GstFlowReturn new_sample(GstAppSink *gstappsink, gpointer video_decoder)
             if (gstframe->timestamp == GST_BUFFER_PTS(buffer)) {
                 /* The frame is now ready for display */
                 gstframe->sample = sample;
-                g_queue_push_tail(decoder->display_queue, gstframe);
+                decoder->display_frame = gstframe;
 
                 /* Now that we know there is a match, remove it and the older
                  * frames from the decoding queue.
@@ -250,12 +251,35 @@ static GstFlowReturn new_sample(GstAppSink *gstappsink, gpointer video_decoder)
             spice_warning("got an unexpected decoded buffer!");
             gst_sample_unref(sample);
         }
-
-        g_mutex_unlock(&decoder->queues_mutex);
-        schedule_frame(decoder);
     } else {
+        // no more samples to get, possibly some sample was dropped
+        decoder->pending_samples = 0;
         spice_warning("GStreamer error: could not pull sample");
     }
+}
+
+/* GStreamer thread
+ *
+ * We cannot use GStreamer's signals because they are not always run in
+ * the main context. So use a callback (lower overhead) and have it pull
+ * the sample to avoid a race with free_pipeline(). This means queuing the
+ * decoded frames outside GStreamer. So while we're at it, also schedule
+ * the frame display ourselves in schedule_frame().
+ */
+static GstFlowReturn new_sample(GstAppSink *gstappsink, gpointer video_decoder)
+{
+    SpiceGstDecoder *decoder = video_decoder;
+
+    g_mutex_lock(&decoder->queues_mutex);
+    decoder->pending_samples++;
+    if (decoder->timer_id && decoder->display_frame) {
+        g_mutex_unlock(&decoder->queues_mutex);
+        return GST_FLOW_OK;
+    }
+    g_mutex_unlock(&decoder->queues_mutex);
+
+    schedule_frame(decoder);
+
     return GST_FLOW_OK;
 }
 
@@ -464,6 +488,8 @@ static gboolean create_pipeline(SpiceGstDecoder *decoder)
         GstAppSinkCallbacks appsink_cbs = { NULL };
         appsink_cbs.new_sample = new_sample;
         gst_app_sink_set_callbacks(decoder->appsink, &appsink_cbs, decoder, NULL);
+        gst_app_sink_set_max_buffers(decoder->appsink, 2);
+        gst_app_sink_set_drop(decoder->appsink, FALSE);
     }
     bus = gst_pipeline_get_bus(GST_PIPELINE(decoder->pipeline));
     gst_bus_add_watch(bus, handle_pipeline_message, decoder);
@@ -522,10 +548,9 @@ static void spice_gst_decoder_destroy(VideoDecoder *video_decoder)
         free_gst_frame(gstframe);
     }
     g_queue_free(decoder->decoding_queue);
-    while ((gstframe = g_queue_pop_head(decoder->display_queue))) {
-        free_gst_frame(gstframe);
+    if (decoder->display_frame) {
+        free_gst_frame(decoder->display_frame);
     }
-    g_queue_free(decoder->display_queue);
 
     g_free(decoder);
 
@@ -551,11 +576,11 @@ static void spice_gst_decoder_destroy(VideoDecoder *video_decoder)
  *    the decoding queue using the GStreamer timestamp information to deal with
  *    dropped frames. The SpiceGstFrame is popped from the decoding_queue.
  * 6) new_sample() then attaches the decompressed frame to the SpiceGstFrame,
- *    pushes it to the display_queue and calls schedule_frame().
+ *    set into display_frame and calls schedule_frame().
  * 7) schedule_frame() then uses gstframe->frame->mm_time to arrange for
  *    display_frame() to be called, in the main thread, at the right time for
  *    the next frame.
- * 8) display_frame() pops the first SpiceGstFrame from the display_queue and
+ * 8) display_frame() use SpiceGstFrame from display_frame and
  *    calls stream_display_frame().
  * 9) display_frame() then frees the SpiceGstFrame, which frees the SpiceFrame
  *    and decompressed frame with it.
@@ -664,7 +689,6 @@ VideoDecoder* create_gstreamer_decoder(int codec_type, display_stream *stream)
         decoder->base.stream = stream;
         g_mutex_init(&decoder->queues_mutex);
         decoder->decoding_queue = g_queue_new();
-        decoder->display_queue = g_queue_new();
 
         if (!create_pipeline(decoder)) {
             decoder->base.destroy((VideoDecoder*)decoder);


More information about the Spice-commits mailing list