segafilm: Check for memory allocation failures in segafilm demuxer.

Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
pull/2/head
Laurent Aimar 14 years ago committed by Michael Niedermayer
parent 3eb1e65f5f
commit 7cbe025758
  1. 6
      libavformat/segafilm.c

@ -176,6 +176,8 @@ static int film_read_header(AVFormatContext *s,
if(film->sample_count >= UINT_MAX / sizeof(film_sample)) if(film->sample_count >= UINT_MAX / sizeof(film_sample))
return -1; return -1;
film->sample_table = av_malloc(film->sample_count * sizeof(film_sample)); 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++) for(i=0; i<s->nb_streams; i++)
av_set_pts_info(s->streams[i], 33, 1, film->base_clock); av_set_pts_info(s->streams[i], 33, 1, film->base_clock);
@ -252,6 +254,10 @@ static int film_read_packet(AVFormatContext *s,
av_free(film->stereo_buffer); av_free(film->stereo_buffer);
film->stereo_buffer_size = sample->sample_size; film->stereo_buffer_size = sample->sample_size;
film->stereo_buffer = av_malloc(film->stereo_buffer_size); film->stereo_buffer = av_malloc(film->stereo_buffer_size);
if (!film->stereo_buffer) {
film->stereo_buffer_size = 0;
return AVERROR(ENOMEM);
}
} }
pkt->pos= avio_tell(pb); pkt->pos= avio_tell(pb);

Loading…
Cancel
Save