From d860084c50c310a243ece0dd0fce730d5b209003 Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Fri, 29 May 2015 00:33:18 +0200 Subject: [PATCH] avcodec/mpegvideo: Reset bitstream_buffer_size on allocation failure Signed-off-by: Michael Niedermayer --- libavcodec/mpeg4videodec.c | 4 +++- libavcodec/mpegvideo.c | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/libavcodec/mpeg4videodec.c b/libavcodec/mpeg4videodec.c index 9ef65ffc84..41caac6835 100644 --- a/libavcodec/mpeg4videodec.c +++ b/libavcodec/mpeg4videodec.c @@ -2670,8 +2670,10 @@ int ff_mpeg4_frame_end(AVCodecContext *avctx, const uint8_t *buf, int buf_size) av_fast_padded_malloc(&s->bitstream_buffer, &s->allocated_bitstream_buffer_size, buf_size - current_pos); - if (!s->bitstream_buffer) + if (!s->bitstream_buffer) { + s->bitstream_buffer_size = 0; return AVERROR(ENOMEM); + } memcpy(s->bitstream_buffer, buf + current_pos, buf_size - current_pos); s->bitstream_buffer_size = buf_size - current_pos; diff --git a/libavcodec/mpegvideo.c b/libavcodec/mpegvideo.c index 5fa4cb61c5..e798565c73 100644 --- a/libavcodec/mpegvideo.c +++ b/libavcodec/mpegvideo.c @@ -1049,8 +1049,10 @@ do {\ av_fast_malloc(&s->bitstream_buffer, &s->allocated_bitstream_buffer_size, s1->allocated_bitstream_buffer_size); - if (!s->bitstream_buffer) + if (!s->bitstream_buffer) { + s->bitstream_buffer_size = 0; return AVERROR(ENOMEM); + } } s->bitstream_buffer_size = s1->bitstream_buffer_size; memcpy(s->bitstream_buffer, s1->bitstream_buffer,