setup_lru_func setup_lru,
enum AVPixelFormat fmt)
{
- GetByteContext gb;
+ GetByteContext gb, gb_check;
GetBitContext gb2;
int nslices, slice, line = 0;
uint32_t off, slice_size;
+ uint64_t off_check;
uint8_t lru[3][8];
int ret;
if (ret < 0)
return ret;
+ off_check = off;
+ gb_check = gb;
+ for (slice = 0; slice < nslices; slice++) {
+ slice_size = bytestream2_get_le32(&gb_check);
+
+ if (slice_size <= 16 + (avctx->height * avctx->width / (8 * nslices)))
+ return AVERROR_INVALIDDATA;
+ off_check += slice_size;
+ }
+
+ if (off_check - avctx->discard_damaged_percentage*off_check/100 > src_size)
+ return AVERROR_INVALIDDATA;
+
avctx->pix_fmt = fmt;
if ((ret = ff_get_buffer(avctx, pic, 0)) < 0)
return ret;