tsdemux: more correctly determine expected PES packet payload size

... particularly considering that packet_length == 0 is possible.

1 files changed, 10 insertions, 2 deletions

diff --git a/gst/mpegtsdemux/tsdemux.c b/gst/mpegtsdemux/tsdemux.c
index 3290c24cd..0fbae586d 100644
--- a/gst/mpegtsdemux/tsdemux.c
+++ b/gst/mpegtsdemux/tsdemux.c
@@ -1322,8 +1322,16 @@ gst_ts_demux_parse_pes_header (GstTSDemux * demux, TSDemuxStream * stream,
   /* Remove PES headers */
   GST_DEBUG ("Moving data forward by %d bytes (packet_size:%d, have:%d)",
       header.header_size, header.packet_length, length);
-  g_assert (header.packet_length >= header.header_size);
-  stream->expected_size = header.packet_length - header.header_size;
+  stream->expected_size = header.packet_length;
+  if (stream->expected_size) {
+    if (G_LIKELY (stream->expected_size > header.header_size)) {
+      stream->expected_size -= header.header_size;
+    } else {
+      /* next packet will have to complete this one */
+      GST_ERROR ("invalid header and packet size combination");
+      stream->expected_size = 0;
+    }
+  }
   data += header.header_size;
   length -= header.header_size;