aacpsy: Check memory allocation

This commit is contained in:
Vittorio Giovara 2015-05-31 14:45:54 +02:00 committed by Luca Barbato
parent 149fa0b7ac
commit 074a1b3732

View File

@ -298,6 +298,8 @@ static av_cold int psy_3gpp_init(FFPsyContext *ctx) {
const float num_bark = calc_bark((float)bandwidth);
ctx->model_priv_data = av_mallocz(sizeof(AacPsyContext));
if (!ctx->model_priv_data)
return AVERROR(ENOMEM);
pctx = (AacPsyContext*) ctx->model_priv_data;
pctx->chan_bitrate = chan_bitrate;
@ -349,6 +351,10 @@ static av_cold int psy_3gpp_init(FFPsyContext *ctx) {
}
pctx->ch = av_mallocz(sizeof(AacPsyChannel) * ctx->avctx->channels);
if (!pctx->ch) {
av_freep(&pctx);
return AVERROR(ENOMEM);
}
lame_window_init(pctx, ctx->avctx);