mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-11-29 05:01:23 +00:00
videotestsrc: add mutex around cache buffer to prevent race condition
Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/6889>
This commit is contained in:
parent
1d2a0d75a0
commit
d5f3b77e50
2 changed files with 27 additions and 2 deletions
|
@ -131,6 +131,8 @@ G_DEFINE_TYPE (GstVideoTestSrc, gst_video_test_src, GST_TYPE_PUSH_SRC);
|
|||
GST_ELEMENT_REGISTER_DEFINE (videotestsrc, "videotestsrc",
|
||||
GST_RANK_NONE, GST_TYPE_VIDEO_TEST_SRC);
|
||||
|
||||
static void gst_video_test_src_finalize (GObject * object);
|
||||
|
||||
static void gst_video_test_src_set_pattern (GstVideoTestSrc * videotestsrc,
|
||||
int pattern_type);
|
||||
static void gst_video_test_src_set_property (GObject * object, guint prop_id,
|
||||
|
@ -265,6 +267,7 @@ gst_video_test_src_class_init (GstVideoTestSrcClass * klass)
|
|||
|
||||
gobject_class->set_property = gst_video_test_src_set_property;
|
||||
gobject_class->get_property = gst_video_test_src_get_property;
|
||||
gobject_class->finalize = gst_video_test_src_finalize;
|
||||
|
||||
g_object_class_install_property (gobject_class, PROP_PATTERN,
|
||||
g_param_spec_enum ("pattern", "Pattern",
|
||||
|
@ -423,6 +426,15 @@ gst_video_test_src_init (GstVideoTestSrc * src)
|
|||
src->motion_type = DEFAULT_MOTION_TYPE;
|
||||
src->flip = DEFAULT_FLIP;
|
||||
|
||||
g_mutex_init (&src->cache_lock);
|
||||
}
|
||||
|
||||
static void
|
||||
gst_video_test_src_finalize (GObject * object)
|
||||
{
|
||||
GstVideoTestSrc *src = GST_VIDEO_TEST_SRC (object);
|
||||
g_mutex_clear (&src->cache_lock);
|
||||
G_OBJECT_CLASS (parent_class)->finalize (object);
|
||||
}
|
||||
|
||||
static GstCaps *
|
||||
|
@ -762,9 +774,11 @@ gst_video_test_src_set_property (GObject * object, guint prop_id,
|
|||
}
|
||||
|
||||
if (invalidate) {
|
||||
g_mutex_lock (&src->cache_lock);
|
||||
/* Property change invalidated the current pattern - check if it's static now or not */
|
||||
src->have_static_pattern = gst_video_test_src_is_static_pattern (src);
|
||||
gst_clear_buffer (&src->cached);
|
||||
g_mutex_unlock (&src->cache_lock);
|
||||
}
|
||||
}
|
||||
|
||||
|
@ -1068,7 +1082,9 @@ gst_video_test_src_setcaps (GstBaseSrc * bsrc, GstCaps * caps)
|
|||
videotestsrc->running_time = 0;
|
||||
videotestsrc->n_frames = 0;
|
||||
|
||||
g_mutex_lock (&videotestsrc->cache_lock);
|
||||
gst_clear_buffer (&videotestsrc->cached);
|
||||
g_mutex_unlock (&videotestsrc->cache_lock);
|
||||
|
||||
GST_OBJECT_UNLOCK (videotestsrc);
|
||||
|
||||
|
@ -1304,12 +1320,15 @@ gst_video_test_src_fill (GstPushSrc * psrc, GstBuffer * buffer)
|
|||
if (src->have_static_pattern) {
|
||||
GstVideoFrame sframe, dframe;
|
||||
|
||||
g_mutex_lock (&src->cache_lock);
|
||||
if (src->cached == NULL) {
|
||||
src->cached = gst_buffer_new_allocate (NULL, src->info.size, NULL);
|
||||
|
||||
ret = fill_image (GST_PUSH_SRC (src), src->cached);
|
||||
if (G_UNLIKELY (ret != GST_FLOW_OK))
|
||||
if (G_UNLIKELY (ret != GST_FLOW_OK)) {
|
||||
g_mutex_unlock (&src->cache_lock);
|
||||
goto fill_failed;
|
||||
}
|
||||
} else {
|
||||
GST_LOG_OBJECT (src, "Reusing cached pattern buffer");
|
||||
}
|
||||
|
@ -1321,11 +1340,14 @@ gst_video_test_src_fill (GstPushSrc * psrc, GstBuffer * buffer)
|
|||
gst_video_frame_map (&sframe, &src->info, src->cached, GST_MAP_READ);
|
||||
gst_video_frame_map (&dframe, &src->info, buffer, GST_MAP_WRITE);
|
||||
|
||||
if (!gst_video_frame_copy (&dframe, &sframe))
|
||||
if (!gst_video_frame_copy (&dframe, &sframe)) {
|
||||
g_mutex_unlock (&src->cache_lock);
|
||||
goto copy_failed;
|
||||
}
|
||||
|
||||
gst_video_frame_unmap (&sframe);
|
||||
gst_video_frame_unmap (&dframe);
|
||||
g_mutex_unlock (&src->cache_lock);
|
||||
} else {
|
||||
ret = fill_image (GST_PUSH_SRC (src), buffer);
|
||||
if (G_UNLIKELY (ret != GST_FLOW_OK))
|
||||
|
@ -1425,7 +1447,9 @@ gst_video_test_src_stop (GstBaseSrc * basesrc)
|
|||
src->n_lines = 0;
|
||||
src->lines = NULL;
|
||||
|
||||
g_mutex_lock (&src->cache_lock);
|
||||
gst_clear_buffer (&src->cached);
|
||||
g_mutex_unlock (&src->cache_lock);
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
|
|
@ -210,6 +210,7 @@ struct _GstVideoTestSrc {
|
|||
|
||||
/* cached buffer used for static patterns that don't change */
|
||||
GstBuffer *cached;
|
||||
GMutex cache_lock;
|
||||
gboolean have_static_pattern;
|
||||
};
|
||||
|
||||
|
|
Loading…
Reference in a new issue