tta: check output buffer size after adjusting frame length for last frame

This commit is contained in:
Justin Ruggles 2011-09-22 13:09:05 -04:00
parent b16960a8a5
commit e6923f683c
1 changed files with 5 additions and 4 deletions

View File

@ -285,15 +285,16 @@ static int tta_decode_frame(AVCodecContext *avctx,
int cur_chan = 0, framelen = s->frame_length;
int32_t *p;
if (*data_size < (framelen * s->channels * 2)) {
av_log(avctx, AV_LOG_ERROR, "Output buffer size is too small.\n");
return -1;
}
// FIXME: seeking
s->total_frames--;
if (!s->total_frames && s->last_frame_length)
framelen = s->last_frame_length;
if (*data_size < (framelen * s->channels * 2)) {
av_log(avctx, AV_LOG_ERROR, "Output buffer size is too small.\n");
return -1;
}
// init per channel states
for (i = 0; i < s->channels; i++) {
s->ch_ctx[i].predictor = 0;