mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-11-26 19:01:44 +02:00
g726: pre-calculate the number of output samples.
Allows for checking output buffer size and simplification of decoding loop.
This commit is contained in:
parent
e61a670b53
commit
c8d36d254e
@ -377,16 +377,24 @@ static int g726_decode_frame(AVCodecContext *avctx,
|
|||||||
G726Context *c = avctx->priv_data;
|
G726Context *c = avctx->priv_data;
|
||||||
int16_t *samples = data;
|
int16_t *samples = data;
|
||||||
GetBitContext gb;
|
GetBitContext gb;
|
||||||
|
int out_samples, out_size;
|
||||||
|
|
||||||
|
out_samples = buf_size * 8 / c->code_size;
|
||||||
|
out_size = out_samples * 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);
|
||||||
|
}
|
||||||
|
|
||||||
init_get_bits(&gb, buf, buf_size * 8);
|
init_get_bits(&gb, buf, buf_size * 8);
|
||||||
|
|
||||||
while (get_bits_count(&gb) + c->code_size <= buf_size*8)
|
while (out_samples--)
|
||||||
*samples++ = g726_decode(c, get_bits(&gb, c->code_size));
|
*samples++ = g726_decode(c, get_bits(&gb, c->code_size));
|
||||||
|
|
||||||
if(buf_size*8 != get_bits_count(&gb))
|
if (get_bits_left(&gb) > 0)
|
||||||
av_log(avctx, AV_LOG_ERROR, "Frame invalidly split, missing parser?\n");
|
av_log(avctx, AV_LOG_ERROR, "Frame invalidly split, missing parser?\n");
|
||||||
|
|
||||||
*data_size = (uint8_t*)samples - (uint8_t*)data;
|
*data_size = out_size;
|
||||||
return buf_size;
|
return buf_size;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user