If one encodes MJPEG with a single slice and uses input with AV_FRAME_DATA_ICC_PROFILE side data, the current allocation code in ff_mpv_encode_picture() will always increase the size of the temporary buffer used for allocating packets by the size needed for to write the ICC chunk even when the current buffer is actually large enough. This commit fixes this.
Unified · Split
| @@ -1684,9 +1684,8 @@ int ff_mpv_encode_picture(AVCodecContext *avctx, AVPacket *pkt, · 1683 unmodified lines | ||
| 1684 | 1684 | /* output? */ |
| 1685 | 1685 | if (s->new_picture->data[0]) { |
| 1686 | 1686 | int growing_buffer = context_count == 1 && !pkt->data && !s->data_partitioning; |
| 1687 | − size_t pkt_size = growing_buffer ? FFMAX(s->mb_width*s->mb_height*64+10000, avctx->internal->byte_buffer_size) - AV_INPUT_BUFFER_PADDING_SIZE | |
| 1688 | − : | |
| 1689 | − s->mb_width*s->mb_height*(MAX_MB_BYTES+100)+10000; | |
| 1687 | + size_t pkt_size = 10000 + s->mb_width * s->mb_height * | |
| 1688 | + (growing_buffer ? 64 : (MAX_MB_BYTES + 100)); | |
| 1690 | 1689 | if (CONFIG_MJPEG_ENCODER && avctx->codec_id == AV_CODEC_ID_MJPEG) { |
| 1691 | 1690 | ret = ff_mjpeg_add_icc_profile_size(avctx, s->new_picture, &pkt_size); |
| 1692 | 1691 | if (ret < 0) |
| @@ -1694,6 +1693,7 @@ int ff_mpv_encode_picture(AVCodecContext *avctx, AVPacket *pkt, · 1 unmodified line | ||
| 1694 | 1693 | } |
| 1695 | 1694 | if ((ret = ff_alloc_packet(avctx, pkt, pkt_size)) < 0) |
| 1696 | 1695 | return ret; |
| 1696 | + pkt->size = avctx->internal->byte_buffer_size - AV_INPUT_BUFFER_PADDING_SIZE; | |
| 1697 | 1697 | if (s->mb_info) { |
| 1698 | 1698 | s->mb_info_ptr = av_packet_new_side_data(pkt, |
| 1699 | 1699 | AV_PKT_DATA_H263_MB_INFO, |