diff --git a/libavcodec/flacenc.c b/libavcodec/flacenc.c index 9877a838c4..31332a6d51 100644 --- a/libavcodec/flacenc.c +++ b/libavcodec/flacenc.c @@ -404,6 +404,8 @@ static av_cold int flac_encode_init(AVCodecContext *avctx) av_md5_init(s->md5ctx); streaminfo = av_malloc(FLAC_STREAMINFO_SIZE); + if (!streaminfo) + return AVERROR(ENOMEM); write_streaminfo(s, streaminfo); avctx->extradata = streaminfo; avctx->extradata_size = FLAC_STREAMINFO_SIZE; @@ -412,6 +414,8 @@ static av_cold int flac_encode_init(AVCodecContext *avctx) s->min_framesize = s->max_framesize; avctx->coded_frame = avcodec_alloc_frame(); + if (!avctx->coded_frame) + return AVERROR(ENOMEM); avctx->coded_frame->key_frame = 1; return 0;