[FFmpeg-devel,v1,5/8] avformat/mpeg: check av_strdup() return value

Submitted by Steven Liu on Oct. 10, 2019, 3:40 a.m.

Details

Message ID 20191010034011.1205-5-lq@chinaffmpeg.org
State Accepted
Commit f5263172de31a03b3d25b962c36047c904e73670
Headers show

Commit Message

Steven Liu Oct. 10, 2019, 3:40 a.m.
Signed-off-by: Steven Liu <lq@chinaffmpeg.org>
---
 libavformat/mpeg.c | 5 +++++
 1 file changed, 5 insertions(+)

Patch hide | download patch | download mbox

diff --git a/libavformat/mpeg.c b/libavformat/mpeg.c
index bd182e4429..df37692f12 100644
--- a/libavformat/mpeg.c
+++ b/libavformat/mpeg.c
@@ -898,6 +898,11 @@  static int vobsub_read_header(AVFormatContext *s)
     for (i = 0; i < s->nb_streams; i++) {
         AVStream *sub_st = s->streams[i];
         sub_st->codecpar->extradata      = av_strdup(header_str);
+        if (!sub_st->codecpar->extradata) {
+            ret = AVERROR(ENOMEM);
+            sub_st->codecpar->extradata_size = 0;
+            goto end;
+        }
         sub_st->codecpar->extradata_size = header.len;
     }
     av_free(header_str);