1
0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-08-10 06:10:52 +02:00

avcodec/zmbv: Don't free uninitialized z_stream

It is not documented to be safe to call inflateEnd() on a z_stream
that has not been successfully initialized via inflateInit(); so
record whether it has been successfully initialized.

Reviewed-by: Tomas Härdin <tjoppen@acc.umu.se>
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
This commit is contained in:
Andreas Rheinhardt
2021-05-05 20:36:44 +02:00
parent 14a6025384
commit 9d752ab634

View File

@@ -56,6 +56,7 @@ enum ZmbvFormat {
typedef struct ZmbvContext {
AVCodecContext *avctx;
int zlib_init_ok;
int bpp;
int alloc_bpp;
unsigned int decomp_size;
@@ -611,9 +612,6 @@ static av_cold int decode_init(AVCodecContext *avctx)
c->bpp = avctx->bits_per_coded_sample;
// Needed if zlib unused or init aborted before inflateInit
memset(&c->zstream, 0, sizeof(z_stream));
if ((avctx->width + 255ULL) * (avctx->height + 64ULL) > FFMIN(avctx->max_pixels, INT_MAX / 4) ) {
av_log(avctx, AV_LOG_ERROR, "Internal buffer (decomp_size) larger than max_pixels or too large\n");
return AVERROR_INVALIDDATA;
@@ -637,6 +635,7 @@ static av_cold int decode_init(AVCodecContext *avctx)
av_log(avctx, AV_LOG_ERROR, "Inflate init error: %d\n", zret);
return AVERROR_UNKNOWN;
}
c->zlib_init_ok = 1;
return 0;
}
@@ -647,9 +646,10 @@ static av_cold int decode_end(AVCodecContext *avctx)
av_freep(&c->decomp_buf);
inflateEnd(&c->zstream);
av_freep(&c->cur);
av_freep(&c->prev);
if (c->zlib_init_ok)
inflateEnd(&c->zstream);
return 0;
}