avcodec/tscc: 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.

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 730b8182b1
commit 54cb70594d

View File

@ -57,6 +57,7 @@ typedef struct TsccContext {
unsigned char* decomp_buf;
GetByteContext gb;
int height;
int zlib_init_ok;
z_stream zstream;
uint32_t pal[256];
@ -128,8 +129,6 @@ static av_cold int decode_init(AVCodecContext *avctx)
c->height = avctx->height;
// Needed if zlib unused or init aborted before inflateInit
memset(&c->zstream, 0, sizeof(z_stream));
switch(avctx->bits_per_coded_sample){
case 8: avctx->pix_fmt = AV_PIX_FMT_PAL8; break;
case 16: avctx->pix_fmt = AV_PIX_FMT_RGB555; break;
@ -160,6 +159,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;
c->frame = av_frame_alloc();
if (!c->frame)
@ -175,7 +175,8 @@ static av_cold int decode_end(AVCodecContext *avctx)
av_freep(&c->decomp_buf);
av_frame_free(&c->frame);
inflateEnd(&c->zstream);
if (c->zlib_init_ok)
inflateEnd(&c->zstream);
return 0;
}