paf: prevent invalid write

Closes #1631.

Signed-off-by: Paul B Mahol <onemda@gmail.com>
This commit is contained in:
Paul B Mahol 2012-08-13 12:26:38 +00:00
parent c3da2c19e4
commit bd70a52712
1 changed files with 4 additions and 2 deletions

View File

@ -164,14 +164,16 @@ static int decode_0(AVCodecContext *avctx, uint8_t code, uint8_t *pkt)
} while (--i); } while (--i);
} }
dst = c->frame[c->current_frame]; dst = c->frame[c->current_frame];
dend = c->frame[c->current_frame] + c->frame_size;
do { do {
a = bytestream2_get_byte(&c->gb); a = bytestream2_get_byte(&c->gb);
b = bytestream2_get_byte(&c->gb); b = bytestream2_get_byte(&c->gb);
p = (a & 0xC0) >> 6; p = (a & 0xC0) >> 6;
src = c->frame[p] + get_video_page_offset(avctx, a, b); src = c->frame[p] + get_video_page_offset(avctx, a, b);
send = c->frame[p] + c->frame_size; send = c->frame[p] + c->frame_size;
if (src + 3 * avctx->width + 4 > send) if ((src + 3 * avctx->width + 4 > send) ||
(dst + 3 * avctx->width + 4 > dend))
return AVERROR_INVALIDDATA; return AVERROR_INVALIDDATA;
copy_block4(dst, src, avctx->width, avctx->width, 4); copy_block4(dst, src, avctx->width, avctx->width, 4);
i++; i++;