loco: check the last plane too

Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
Michael Niedermayer 2013-02-24 13:03:13 +01:00
parent 714ff44858
commit 33796645dc
1 changed files with 5 additions and 1 deletions

View File

@ -240,10 +240,14 @@ static int decode_frame(AVCodecContext *avctx,
break;
}
if (decoded < 0 || decoded > buf_size)
goto buf_too_small;
buf_size -= decoded;
*got_frame = 1;
*(AVFrame*)data = l->pic;
return buf_size < 0 ? -1 : avpkt->size - buf_size;
return avpkt->size - buf_size;
buf_too_small:
av_log(avctx, AV_LOG_ERROR, "Input data too small.\n");
return AVERROR(EINVAL);