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

pull/2/head
Justin Ruggles 13 years ago
parent b16960a8a5
commit e6923f683c
  1. 9
      libavcodec/tta.c

@ -285,15 +285,16 @@ static int tta_decode_frame(AVCodecContext *avctx,
int cur_chan = 0, framelen = s->frame_length; int cur_chan = 0, framelen = s->frame_length;
int32_t *p; 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 // FIXME: seeking
s->total_frames--; s->total_frames--;
if (!s->total_frames && s->last_frame_length) if (!s->total_frames && s->last_frame_length)
framelen = 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 // init per channel states
for (i = 0; i < s->channels; i++) { for (i = 0; i < s->channels; i++) {
s->ch_ctx[i].predictor = 0; s->ch_ctx[i].predictor = 0;

Loading…
Cancel
Save