Commit 0a8c9020 authored by Michael Niedermayer's avatar Michael Niedermayer

avcodec/utvideoenc: fix slice_bits size

Fixes assertion failure
Signed-off-by: 's avatarMichael Niedermayer <michaelni@gmx.at>
parent 0c9dd939
...@@ -494,7 +494,7 @@ static int encode_plane(AVCodecContext *avctx, uint8_t *src, ...@@ -494,7 +494,7 @@ static int encode_plane(AVCodecContext *avctx, uint8_t *src,
* get the offset in bits and convert to bytes. * get the offset in bits and convert to bytes.
*/ */
offset += write_huff_codes(dst + sstart * width, c->slice_bits, offset += write_huff_codes(dst + sstart * width, c->slice_bits,
width * (send - sstart), width, width * height + 4, width,
send - sstart, he) >> 3; send - sstart, he) >> 3;
slice_len = offset - slice_len; slice_len = offset - slice_len;
...@@ -551,7 +551,7 @@ static int utvideo_encode_frame(AVCodecContext *avctx, AVPacket *pkt, ...@@ -551,7 +551,7 @@ static int utvideo_encode_frame(AVCodecContext *avctx, AVPacket *pkt,
bytestream2_init_writer(&pb, dst, pkt->size); bytestream2_init_writer(&pb, dst, pkt->size);
av_fast_padded_malloc(&c->slice_bits, &c->slice_bits_size, width * height); av_fast_padded_malloc(&c->slice_bits, &c->slice_bits_size, width * height + 4);
if (!c->slice_bits) { if (!c->slice_bits) {
av_log(avctx, AV_LOG_ERROR, "Cannot allocate temporary buffer 2.\n"); av_log(avctx, AV_LOG_ERROR, "Cannot allocate temporary buffer 2.\n");
......
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