avcodec/alsdec, mlz: Check allocation

Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
This commit is contained in:
Andreas Rheinhardt 2022-02-11 16:22:16 +01:00
parent e925d5453b
commit 3b08e046af
3 changed files with 10 additions and 3 deletions

View File

@ -2111,7 +2111,9 @@ static av_cold int decode_init(AVCodecContext *avctx)
return AVERROR(ENOMEM);
}
ff_mlz_init_dict(avctx, ctx->mlz);
ret = ff_mlz_init_dict(avctx, ctx->mlz);
if (ret < 0)
return ret;
ff_mlz_flush_dict(ctx->mlz);
for (c = 0; c < avctx->channels; ++c) {

View File

@ -20,8 +20,11 @@
#include "mlz.h"
av_cold void ff_mlz_init_dict(void* context, MLZ *mlz) {
av_cold int ff_mlz_init_dict(void *context, MLZ *mlz)
{
mlz->dict = av_mallocz(TABLE_SIZE * sizeof(*mlz->dict));
if (!mlz->dict)
return AVERROR(ENOMEM);
mlz->flush_code = FLUSH_CODE;
mlz->current_dic_index_max = DIC_INDEX_INIT;
@ -30,6 +33,8 @@ av_cold void ff_mlz_init_dict(void* context, MLZ *mlz) {
mlz->next_code = FIRST_CODE;
mlz->freeze_flag = 0;
mlz->context = context;
return 0;
}
av_cold void ff_mlz_flush_dict(MLZ *mlz) {

View File

@ -57,7 +57,7 @@ typedef struct MLZ {
/** Initialize the dictionary
*/
void ff_mlz_init_dict(void* context, MLZ *mlz);
int ff_mlz_init_dict(void *context, MLZ *mlz);
/** Flush the dictionary
*/