asv: Check memory allocation

10 years ago · e524f37356
parent 074a1b3732
commit e524f37356
1 changed files with 2 additions and 0 deletions
--- a/libavcodec/asvenc.c
+++ b/libavcodec/asvenc.c
@ -299,6 +299,8 @@ static av_cold int encode_init(AVCodecContext *avctx)
                     avctx->global_quality / 2) / avctx->global_quality;

    avctx->extradata                   = av_mallocz(8);
+    if (!avctx->extradata)
+        return AVERROR(ENOMEM);
    avctx->extradata_size              = 8;
    ((uint32_t *) avctx->extradata)[0] = av_le2ne32(a->inv_qscale);
    ((uint32_t *) avctx->extradata)[1] = av_le2ne32(AV_RL32("ASUS"));