mirror of https://git.ffmpeg.org/ffmpeg.git
avcodec/eatgq: : Check index increments in tgq_decode_block()
Fixes: out of array access
Fixes: 48567/clusterfuzz-testcase-minimized-ffmpeg_AV_CODEC_ID_EATGQ_fuzzer-6743211456724992
Found-by: continuous fuzzing process https://github.com/google/oss-fuzz/tree/master/projects/ffmpeg
Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
(cherry picked from commit e7755b433e
)
Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>
This commit is contained in:
parent
5cc68bb176
commit
102384cd7d
|
@ -58,7 +58,7 @@ static av_cold int tgq_decode_init(AVCodecContext *avctx)
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void tgq_decode_block(TgqContext *s, int16_t block[64], GetBitContext *gb)
|
static int tgq_decode_block(TgqContext *s, int16_t block[64], GetBitContext *gb)
|
||||||
{
|
{
|
||||||
uint8_t *perm = s->scantable.permutated;
|
uint8_t *perm = s->scantable.permutated;
|
||||||
int i, j, value;
|
int i, j, value;
|
||||||
|
@ -66,6 +66,8 @@ static void tgq_decode_block(TgqContext *s, int16_t block[64], GetBitContext *gb
|
||||||
for (i = 1; i < 64;) {
|
for (i = 1; i < 64;) {
|
||||||
switch (show_bits(gb, 3)) {
|
switch (show_bits(gb, 3)) {
|
||||||
case 4:
|
case 4:
|
||||||
|
if (i >= 63)
|
||||||
|
return AVERROR_INVALIDDATA;
|
||||||
block[perm[i++]] = 0;
|
block[perm[i++]] = 0;
|
||||||
case 0:
|
case 0:
|
||||||
block[perm[i++]] = 0;
|
block[perm[i++]] = 0;
|
||||||
|
@ -75,6 +77,8 @@ static void tgq_decode_block(TgqContext *s, int16_t block[64], GetBitContext *gb
|
||||||
case 1:
|
case 1:
|
||||||
skip_bits(gb, 2);
|
skip_bits(gb, 2);
|
||||||
value = get_bits(gb, 6);
|
value = get_bits(gb, 6);
|
||||||
|
if (value > 64 - i)
|
||||||
|
return AVERROR_INVALIDDATA;
|
||||||
for (j = 0; j < value; j++)
|
for (j = 0; j < value; j++)
|
||||||
block[perm[i++]] = 0;
|
block[perm[i++]] = 0;
|
||||||
break;
|
break;
|
||||||
|
@ -102,6 +106,7 @@ static void tgq_decode_block(TgqContext *s, int16_t block[64], GetBitContext *gb
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
block[0] += 128 << 4;
|
block[0] += 128 << 4;
|
||||||
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void tgq_idct_put_mb(TgqContext *s, int16_t (*block)[64], AVFrame *frame,
|
static void tgq_idct_put_mb(TgqContext *s, int16_t (*block)[64], AVFrame *frame,
|
||||||
|
@ -161,8 +166,11 @@ static int tgq_decode_mb(TgqContext *s, AVFrame *frame, int mb_y, int mb_x)
|
||||||
if (ret < 0)
|
if (ret < 0)
|
||||||
return ret;
|
return ret;
|
||||||
|
|
||||||
for (i = 0; i < 6; i++)
|
for (i = 0; i < 6; i++) {
|
||||||
tgq_decode_block(s, s->block[i], &gb);
|
int ret = tgq_decode_block(s, s->block[i], &gb);
|
||||||
|
if (ret < 0)
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
tgq_idct_put_mb(s, s->block, frame, mb_x, mb_y);
|
tgq_idct_put_mb(s, s->block, frame, mb_x, mb_y);
|
||||||
bytestream2_skip(&s->gb, mode);
|
bytestream2_skip(&s->gb, mode);
|
||||||
} else {
|
} else {
|
||||||
|
|
Loading…
Reference in New Issue