tsdemux: Protect demux->segment_event with a mutex

Would otherwise cause weird issues when processing a delayed seek event

Part-of: <https://gitlab.freedesktop.org/gstreamer/gstreamer/-/merge_requests/980>
This commit is contained in:
Vivia Nikolaidou 2021-04-05 10:28:51 +03:00 committed by GStreamer Marge Bot
parent a102444c90
commit 5800ffea0f
2 changed files with 27 additions and 0 deletions

View file

@ -375,6 +375,16 @@ gst_ts_demux_dispose (GObject * object)
GST_CALL_PARENT (G_OBJECT_CLASS, dispose, (object)); GST_CALL_PARENT (G_OBJECT_CLASS, dispose, (object));
} }
static void
gst_ts_demux_finalize (GObject * object)
{
GstTSDemux *demux = GST_TS_DEMUX_CAST (object);
g_mutex_clear (&demux->lock);
GST_CALL_PARENT (G_OBJECT_CLASS, finalize, (object));
}
static void static void
gst_ts_demux_class_init (GstTSDemuxClass * klass) gst_ts_demux_class_init (GstTSDemuxClass * klass)
{ {
@ -386,6 +396,7 @@ gst_ts_demux_class_init (GstTSDemuxClass * klass)
gobject_class->set_property = gst_ts_demux_set_property; gobject_class->set_property = gst_ts_demux_set_property;
gobject_class->get_property = gst_ts_demux_get_property; gobject_class->get_property = gst_ts_demux_get_property;
gobject_class->dispose = gst_ts_demux_dispose; gobject_class->dispose = gst_ts_demux_dispose;
gobject_class->finalize = gst_ts_demux_finalize;
g_object_class_install_property (gobject_class, PROP_PROGRAM_NUMBER, g_object_class_install_property (gobject_class, PROP_PROGRAM_NUMBER,
g_param_spec_int ("program-number", "Program number", g_param_spec_int ("program-number", "Program number",
@ -459,10 +470,12 @@ gst_ts_demux_reset (MpegTSBase * base)
GstTSDemux *demux = (GstTSDemux *) base; GstTSDemux *demux = (GstTSDemux *) base;
demux->rate = 1.0; demux->rate = 1.0;
g_mutex_lock (&demux->lock);
if (demux->segment_event) { if (demux->segment_event) {
gst_event_unref (demux->segment_event); gst_event_unref (demux->segment_event);
demux->segment_event = NULL; demux->segment_event = NULL;
} }
g_mutex_unlock (&demux->lock);
if (demux->global_tags) { if (demux->global_tags) {
gst_tag_list_unref (demux->global_tags); gst_tag_list_unref (demux->global_tags);
@ -498,6 +511,8 @@ gst_ts_demux_init (GstTSDemux * demux)
demux->program_number = -1; demux->program_number = -1;
demux->latency = DEFAULT_LATENCY; demux->latency = DEFAULT_LATENCY;
gst_ts_demux_reset (base); gst_ts_demux_reset (base);
g_mutex_init (&demux->lock);
} }
@ -936,6 +951,7 @@ gst_ts_demux_do_seek (MpegTSBase * base, GstEvent * event)
&seeksegment); &seeksegment);
/* If the position actually changed, update == TRUE */ /* If the position actually changed, update == TRUE */
g_mutex_lock (&demux->lock);
if (update) { if (update) {
GstClockTime target = seeksegment.start; GstClockTime target = seeksegment.start;
if (target >= SEEK_TIMESTAMP_OFFSET) if (target >= SEEK_TIMESTAMP_OFFSET)
@ -982,6 +998,7 @@ gst_ts_demux_do_seek (MpegTSBase * base, GstEvent * event)
stream->need_newsegment = TRUE; stream->need_newsegment = TRUE;
} }
} }
g_mutex_unlock (&demux->lock);
/* Commit the new segment */ /* Commit the new segment */
memcpy (&base->out_segment, &seeksegment, sizeof (GstSegment)); memcpy (&base->out_segment, &seeksegment, sizeof (GstSegment));
@ -2218,10 +2235,12 @@ gst_ts_demux_program_started (MpegTSBase * base, MpegTSBaseProgram * program)
/* If this is not the initial program, we need to calculate /* If this is not the initial program, we need to calculate
* a new segment */ * a new segment */
g_mutex_lock (&demux->lock);
if (demux->segment_event) { if (demux->segment_event) {
gst_event_unref (demux->segment_event); gst_event_unref (demux->segment_event);
demux->segment_event = NULL; demux->segment_event = NULL;
} }
g_mutex_unlock (&demux->lock);
/* DRAIN ALL STREAMS FIRST ! */ /* DRAIN ALL STREAMS FIRST ! */
if (demux->previous_program) { if (demux->previous_program) {
@ -2769,12 +2788,14 @@ calculate_and_push_newsegment (GstTSDemux * demux, TSDemuxStream * stream,
GST_LOG_OBJECT (demux, "Output segment now %" GST_SEGMENT_FORMAT, GST_LOG_OBJECT (demux, "Output segment now %" GST_SEGMENT_FORMAT,
&base->out_segment); &base->out_segment);
g_mutex_lock (&demux->lock);
if (!demux->segment_event) { if (!demux->segment_event) {
demux->segment_event = gst_event_new_segment (&base->out_segment); demux->segment_event = gst_event_new_segment (&base->out_segment);
if (base->last_seek_seqnum != GST_SEQNUM_INVALID) if (base->last_seek_seqnum != GST_SEQNUM_INVALID)
gst_event_set_seqnum (demux->segment_event, base->last_seek_seqnum); gst_event_set_seqnum (demux->segment_event, base->last_seek_seqnum);
} }
g_mutex_unlock (&demux->lock);
push_new_segment: push_new_segment:
for (tmp = target_program->stream_list; tmp; tmp = tmp->next) { for (tmp = target_program->stream_list; tmp; tmp = tmp->next) {
@ -2782,12 +2803,14 @@ push_new_segment:
if (stream->pad == NULL) if (stream->pad == NULL)
continue; continue;
g_mutex_lock (&demux->lock);
if (demux->segment_event) { if (demux->segment_event) {
GST_DEBUG_OBJECT (stream->pad, "Pushing newsegment event"); GST_DEBUG_OBJECT (stream->pad, "Pushing newsegment event");
gst_event_ref (demux->segment_event); gst_event_ref (demux->segment_event);
gst_pad_push_event (stream->pad, demux->segment_event); gst_pad_push_event (stream->pad, demux->segment_event);
} }
g_mutex_unlock (&demux->lock);
if (demux->global_tags) { if (demux->global_tags) {
gst_pad_push_event (stream->pad, gst_pad_push_event (stream->pad,
@ -3531,10 +3554,12 @@ gst_ts_demux_flush (MpegTSBase * base, gboolean hard)
gst_ts_demux_flush_streams (demux, hard); gst_ts_demux_flush_streams (demux, hard);
g_mutex_lock (&demux->lock);
if (demux->segment_event) { if (demux->segment_event) {
gst_event_unref (demux->segment_event); gst_event_unref (demux->segment_event);
demux->segment_event = NULL; demux->segment_event = NULL;
} }
g_mutex_unlock (&demux->lock);
if (demux->global_tags) { if (demux->global_tags) {
gst_tag_list_unref (demux->global_tags); gst_tag_list_unref (demux->global_tags);
demux->global_tags = NULL; demux->global_tags = NULL;

View file

@ -107,6 +107,8 @@ struct _GstTSDemux
* in the MPEG time domain. This is used for potentially updating * in the MPEG time domain. This is used for potentially updating
* SCTE 35 sections' pts_adjustment further down the line (eg mpegtsmux) */ * SCTE 35 sections' pts_adjustment further down the line (eg mpegtsmux) */
guint64 mpeg_pts_offset; guint64 mpeg_pts_offset;
GMutex lock;
}; };
struct _GstTSDemuxClass struct _GstTSDemuxClass