[FFmpeg-cvslog] avformat/matroskaenc: add missing allocation failure checks for stream durations

[James Almer]

ffmpeg | branch: release/3.4 | James Almer <jamrial at gmail.com> | Tue Nov 28 01:23:23 2017 -0300| [2940b3e17cd352fc22152909843867771922b4dd] | committer: James Almer

avformat/matroskaenc: add missing allocation failure checks for stream durations

Signed-off-by: James Almer <jamrial at gmail.com>
(cherry picked from commit 5f67073b4cb12abf21ed840948b271903d61bf3b)

> http://git.videolan.org/gitweb.cgi/ffmpeg.git/?a=commit;h=2940b3e17cd352fc22152909843867771922b4dd
---

 libavformat/matroskaenc.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/libavformat/matroskaenc.c b/libavformat/matroskaenc.c
index 3c6bb6816a..3073fd8f77 100644
--- a/libavformat/matroskaenc.c
+++ b/libavformat/matroskaenc.c
@@ -1982,6 +1982,10 @@ static int mkv_write_header(AVFormatContext *s)
     // initialize stream_duration fields
     mkv->stream_durations = av_mallocz(s->nb_streams * sizeof(int64_t));
     mkv->stream_duration_offsets = av_mallocz(s->nb_streams * sizeof(int64_t));
+    if (!mkv->stream_durations || !mkv->stream_duration_offsets) {
+        ret = AVERROR(ENOMEM);
+        goto fail;
+    }
 
     ret = mkv_write_tracks(s);
     if (ret < 0)