Commit d860084c authored by Michael Niedermayer's avatar Michael Niedermayer

avcodec/mpegvideo: Reset bitstream_buffer_size on allocation failure

Signed-off-by: 's avatarMichael Niedermayer <michaelni@gmx.at>
parent e5d1152c
...@@ -2670,8 +2670,10 @@ int ff_mpeg4_frame_end(AVCodecContext *avctx, const uint8_t *buf, int buf_size) ...@@ -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, av_fast_padded_malloc(&s->bitstream_buffer,
&s->allocated_bitstream_buffer_size, &s->allocated_bitstream_buffer_size,
buf_size - current_pos); buf_size - current_pos);
if (!s->bitstream_buffer) if (!s->bitstream_buffer) {
s->bitstream_buffer_size = 0;
return AVERROR(ENOMEM); return AVERROR(ENOMEM);
}
memcpy(s->bitstream_buffer, buf + current_pos, memcpy(s->bitstream_buffer, buf + current_pos,
buf_size - current_pos); buf_size - current_pos);
s->bitstream_buffer_size = buf_size - current_pos; s->bitstream_buffer_size = buf_size - current_pos;
......
...@@ -1049,8 +1049,10 @@ do {\ ...@@ -1049,8 +1049,10 @@ do {\
av_fast_malloc(&s->bitstream_buffer, av_fast_malloc(&s->bitstream_buffer,
&s->allocated_bitstream_buffer_size, &s->allocated_bitstream_buffer_size,
s1->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); return AVERROR(ENOMEM);
}
} }
s->bitstream_buffer_size = s1->bitstream_buffer_size; s->bitstream_buffer_size = s1->bitstream_buffer_size;
memcpy(s->bitstream_buffer, s1->bitstream_buffer, memcpy(s->bitstream_buffer, s1->bitstream_buffer,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment