1
0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-12-23 12:43:46 +02:00

mace: only calculate output buffer size once

This commit is contained in:
Justin Ruggles 2011-10-28 21:27:36 -04:00
parent ce33320b30
commit f50b6be57d

View File

@ -243,11 +243,14 @@ static int mace_decode_frame(AVCodecContext *avctx,
int16_t *samples = data;
MACEContext *ctx = avctx->priv_data;
int i, j, k, l;
int out_size;
int is_mace3 = (avctx->codec_id == CODEC_ID_MACE3);
if (*data_size < (3 * buf_size << (2-is_mace3))) {
av_log(avctx, AV_LOG_ERROR, "Output buffer too small!\n");
return -1;
out_size = 3 * (buf_size << (1 - is_mace3)) *
av_get_bytes_per_sample(avctx->sample_fmt);
if (*data_size < out_size) {
av_log(avctx, AV_LOG_ERROR, "Output buffer is too small\n");
return AVERROR(EINVAL);
}
for(i = 0; i < avctx->channels; i++) {
@ -274,7 +277,7 @@ static int mace_decode_frame(AVCodecContext *avctx,
}
}
*data_size = 3 * buf_size << (2-is_mace3);
*data_size = out_size;
return buf_size;
}