[FFmpeg-devel] [PATCH 01/30] avcodec/flashsvenc: Avoid allocation of buffer, fix memleak
Andreas Rheinhardt
andreas.rheinhardt at gmail.com
Fri Sep 18 05:23:07 EEST 2020
Andreas Rheinhardt:
> Up until now, the flashsv encoder tried to allocate two buffers in its
> init function; if only one of these allocations succeeds, the other
> buffer leaks. Fix this by making one of these buffers part of the
> context (its size is a compile-time constant).
>
> Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt at gmail.com>
> ---
> libavcodec/flashsvenc.c | 6 ++----
> 1 file changed, 2 insertions(+), 4 deletions(-)
>
> diff --git a/libavcodec/flashsvenc.c b/libavcodec/flashsvenc.c
> index f7f98efde3..4ac643c036 100644
> --- a/libavcodec/flashsvenc.c
> +++ b/libavcodec/flashsvenc.c
> @@ -59,11 +59,11 @@ typedef struct FlashSVContext {
> uint8_t *previous_frame;
> int image_width, image_height;
> int block_width, block_height;
> - uint8_t *tmpblock;
> uint8_t *encbuffer;
> int block_size;
> z_stream zstream;
> int last_key_frame;
> + uint8_t tmpblock[3 * 256 * 256];
> } FlashSVContext;
>
> static int copy_region_enc(uint8_t *sptr, uint8_t *dptr, int dx, int dy,
> @@ -96,7 +96,6 @@ static av_cold int flashsv_encode_end(AVCodecContext *avctx)
>
> av_freep(&s->encbuffer);
> av_freep(&s->previous_frame);
> - av_freep(&s->tmpblock);
>
> return 0;
> }
> @@ -121,10 +120,9 @@ static av_cold int flashsv_encode_init(AVCodecContext *avctx)
> s->image_width = avctx->width;
> s->image_height = avctx->height;
>
> - s->tmpblock = av_mallocz(3 * 256 * 256);
> s->encbuffer = av_mallocz(s->image_width * s->image_height * 3);
>
> - if (!s->tmpblock || !s->encbuffer) {
> + if (!s->encbuffer) {
> av_log(avctx, AV_LOG_ERROR, "Memory allocation failed.\n");
> return AVERROR(ENOMEM);
> }
>
Will apply the rest of this patchset tomorrow unless there are objections.
- Andreas
More information about the ffmpeg-devel
mailing list