[FFmpeg-devel] [PATCH 2/2] avformat/mpegtsenc: first dts check needed

[lance.lmwang at gmail.com]

From: Limin Wang <lance.lmwang at gmail.com>

Signed-off-by: Limin Wang <lance.lmwang at gmail.com>
---
 libavformat/mpegtsenc.c | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/libavformat/mpegtsenc.c b/libavformat/mpegtsenc.c
index 383181d..acc8c32 100644
--- a/libavformat/mpegtsenc.c
+++ b/libavformat/mpegtsenc.c
@@ -233,6 +233,7 @@ typedef struct MpegTSWriteStream {
     int discontinuity;
     int payload_size;
     int first_pts_checked; ///< first pts check needed
+    int first_dts_checked; ///< first dts check needed
     int prev_payload_key;
     int64_t payload_pts;
     int64_t payload_dts;
@@ -1705,6 +1706,12 @@ static int mpegts_write_packet_internal(AVFormatContext *s, AVPacket *pkt)
     }
     ts_st->first_pts_checked = 1;
 
+    if (!ts_st->first_dts_checked && dts == AV_NOPTS_VALUE) {
+        av_log(s, AV_LOG_ERROR, "first dts value must be set\n");
+        return AVERROR_INVALIDDATA;
+    }
+    ts_st->first_dts_checked = 1;
+
     if (st->codecpar->codec_id == AV_CODEC_ID_H264) {
         const uint8_t *p = buf, *buf_end = p + size;
         uint32_t state = -1;
-- 
1.8.3.1