From 5a9e80c01b4b49c6c7630a6d08b600114f38c0db Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Dr=C3=B6ge?= Date: Fri, 27 Sep 2024 15:50:54 +0300 Subject: [PATCH] qtdemux: Check sizes of stsc/stco/stts before trying to merge entries Thanks to Antonio Morales for finding and reporting the issue. Fixes GHSL-2024-246 Fixes https://gitlab.freedesktop.org/gstreamer/gstreamer/-/issues/3854 Part-of: --- .../gst-plugins-good/gst/isomp4/qtdemux.c | 22 +++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/subprojects/gst-plugins-good/gst/isomp4/qtdemux.c b/subprojects/gst-plugins-good/gst/isomp4/qtdemux.c index 52d5a35c0b..dbd42817c0 100644 --- a/subprojects/gst-plugins-good/gst/isomp4/qtdemux.c +++ b/subprojects/gst-plugins-good/gst/isomp4/qtdemux.c @@ -10040,6 +10040,21 @@ qtdemux_merge_sample_table (GstQTDemux * qtdemux, QtDemuxStream * stream) return; } + if (gst_byte_reader_get_remaining (&stream->stts) < 8) { + GST_DEBUG_OBJECT (qtdemux, "Too small stts"); + return; + } + + if (stream->stco.size < 8) { + GST_DEBUG_OBJECT (qtdemux, "Too small stco"); + return; + } + + if (stream->n_samples_per_chunk == 0) { + GST_DEBUG_OBJECT (qtdemux, "No samples per chunk"); + return; + } + /* Parse the stts to get the sample duration and number of samples */ gst_byte_reader_skip_unchecked (&stream->stts, 4); stts_duration = gst_byte_reader_get_uint32_be_unchecked (&stream->stts); @@ -10051,6 +10066,13 @@ qtdemux_merge_sample_table (GstQTDemux * qtdemux, QtDemuxStream * stream) GST_DEBUG_OBJECT (qtdemux, "sample_duration %d, num_chunks %u", stts_duration, num_chunks); + if (gst_byte_reader_get_remaining (&stream->stsc) < + stream->n_samples_per_chunk * 3 * 4 + + (stream->n_samples_per_chunk - 1) * 4) { + GST_DEBUG_OBJECT (qtdemux, "Too small stsc"); + return; + } + /* Now parse stsc, convert chunks into single samples and generate a * new stsc, stts and stsz from this information */ gst_byte_writer_init (&stsc);