avformat/mpegtsenc: check avformat_new_stream() return

Fixes CID1206645 Signed-off-by: Michael Niedermayer <[email protected]>
afedchin · Apr 26, 2014 · 66e30a2 · 66e30a2
1 parent 24725f8
commit 66e30a2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/libavformat/mpegtsenc.c b/libavformat/mpegtsenc.c
@@ -700,6 +700,10 @@ static int mpegts_write_header(AVFormatContext *s)
                 goto fail;
             }
             ast = avformat_new_stream(ts_st->amux, NULL);
+            if (!ast) {
+                ret = AVERROR(ENOMEM);
+                goto fail;
+            }
             ret = avcodec_copy_context(ast->codec, st->codec);
             if (ret != 0)
                 goto fail;