Commit 81fe316a authored by Michael Niedermayer's avatar Michael Niedermayer

avcodec/hapdec: Check tex_size more strictly and before using it

Fixes: OOM
Fixes: 20774/clusterfuzz-testcase-minimized-ffmpeg_AV_CODEC_ID_HAP_fuzzer-5678608951803904
Fixes: 20956/clusterfuzz-testcase-minimized-ffmpeg_AV_CODEC_ID_HAP_fuzzer-5713643025203200

Found-by: continuous fuzzing process https://github.com/google/oss-fuzz/tree/master/projects/ffmpegSigned-off-by: 's avatarMichael Niedermayer <michael@niedermayer.cc>
parent 9624b76f
...@@ -305,7 +305,6 @@ static int hap_decode(AVCodecContext *avctx, void *data, ...@@ -305,7 +305,6 @@ static int hap_decode(AVCodecContext *avctx, void *data,
HapContext *ctx = avctx->priv_data; HapContext *ctx = avctx->priv_data;
ThreadFrame tframe; ThreadFrame tframe;
int ret, i, t; int ret, i, t;
int tex_size;
int section_size; int section_size;
enum HapSectionType section_type; enum HapSectionType section_type;
int start_texture_section = 0; int start_texture_section = 0;
...@@ -342,6 +341,13 @@ static int hap_decode(AVCodecContext *avctx, void *data, ...@@ -342,6 +341,13 @@ static int hap_decode(AVCodecContext *avctx, void *data,
if (ret < 0) if (ret < 0)
return ret; return ret;
if (ctx->tex_size != (avctx->coded_width / TEXTURE_BLOCK_W)
*(avctx->coded_height / TEXTURE_BLOCK_H)
*tex_rat[t]) {
av_log(avctx, AV_LOG_ERROR, "uncompressed size mismatches\n");
return AVERROR_INVALIDDATA;
}
start_texture_section += ctx->texture_section_size + 4; start_texture_section += ctx->texture_section_size + 4;
if (avctx->codec->update_thread_context) if (avctx->codec->update_thread_context)
...@@ -349,9 +355,16 @@ static int hap_decode(AVCodecContext *avctx, void *data, ...@@ -349,9 +355,16 @@ static int hap_decode(AVCodecContext *avctx, void *data,
/* Unpack the DXT texture */ /* Unpack the DXT texture */
if (hap_can_use_tex_in_place(ctx)) { if (hap_can_use_tex_in_place(ctx)) {
int tex_size;
/* Only DXTC texture compression in a contiguous block */ /* Only DXTC texture compression in a contiguous block */
ctx->tex_data = ctx->gbc.buffer; ctx->tex_data = ctx->gbc.buffer;
tex_size = FFMIN(ctx->texture_section_size, bytestream2_get_bytes_left(&ctx->gbc)); tex_size = FFMIN(ctx->texture_section_size, bytestream2_get_bytes_left(&ctx->gbc));
if (tex_size < (avctx->coded_width / TEXTURE_BLOCK_W)
*(avctx->coded_height / TEXTURE_BLOCK_H)
*tex_rat[t]) {
av_log(avctx, AV_LOG_ERROR, "Insufficient data\n");
return AVERROR_INVALIDDATA;
}
} else { } else {
/* Perform the second-stage decompression */ /* Perform the second-stage decompression */
ret = av_reallocp(&ctx->tex_buf, ctx->tex_size); ret = av_reallocp(&ctx->tex_buf, ctx->tex_size);
...@@ -367,14 +380,6 @@ static int hap_decode(AVCodecContext *avctx, void *data, ...@@ -367,14 +380,6 @@ static int hap_decode(AVCodecContext *avctx, void *data,
} }
ctx->tex_data = ctx->tex_buf; ctx->tex_data = ctx->tex_buf;
tex_size = ctx->tex_size;
}
if (tex_size < (avctx->coded_width / TEXTURE_BLOCK_W)
*(avctx->coded_height / TEXTURE_BLOCK_H)
*tex_rat[t]) {
av_log(avctx, AV_LOG_ERROR, "Insufficient data\n");
return AVERROR_INVALIDDATA;
} }
/* Use the decompress function on the texture, one block per thread */ /* Use the decompress function on the texture, one block per thread */
......
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