[FFmpeg-cvslog] segafilm: Check for memory allocation failures in segafilm demuxer.

[Laurent Aimar]

ffmpeg | branch: release/0.5 | Laurent Aimar <fenrir at videolan.org> | Mon Sep 12 20:58:35 2011 +0200| [d6f8b65417f2170829a774ce07df7fdd34b32367] | committer: Michael Niedermayer

segafilm: Check for memory allocation failures in segafilm demuxer.

Signed-off-by: Michael Niedermayer <michaelni at gmx.at>
(cherry picked from commit 7cbe02575868e7d25acf3d319ece664702700f0a)

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

 libavformat/segafilm.c |    6 ++++++
 1 files changed, 6 insertions(+), 0 deletions(-)

diff --git a/libavformat/segafilm.c b/libavformat/segafilm.c
index ae1263c..b802b33 100644
--- a/libavformat/segafilm.c
+++ b/libavformat/segafilm.c
@@ -167,6 +167,8 @@ static int film_read_header(AVFormatContext *s,
     if(film->sample_count >= UINT_MAX / sizeof(film_sample))
         return -1;
     film->sample_table = av_malloc(film->sample_count * sizeof(film_sample));
+    if (!film->sample_table)
+        return AVERROR(ENOMEM);
 
     for(i=0; i<s->nb_streams; i++)
         av_set_pts_info(s->streams[i], 33, 1, film->base_clock);
@@ -238,6 +240,10 @@ static int film_read_packet(AVFormatContext *s,
             av_free(film->stereo_buffer);
             film->stereo_buffer_size = sample->sample_size;
             film->stereo_buffer = av_malloc(film->stereo_buffer_size);
+            if (!film->stereo_buffer) {
+                film->stereo_buffer_size = 0;
+                return AVERROR(ENOMEM);
+            }
         }
 
         pkt->pos= url_ftell(pb);