huffyuv: Check memory allocation

7d2a6826 · Vittorio Giovara · Luca Barbato · 7fccc96d · 7d2a6826
Commit 7d2a6826 authored May 31, 2015 by Vittorio Giovara Committed by Luca Barbato May 31, 2015
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

huffyuvenc.c libavcodec/huffyuvenc.c +1 -1

No files found.
--- a/libavcodec/huffyuvenc.c
+++ b/libavcodec/huffyuvenc.c
@@ -155,7 +155,7 @@ static av_cold int encode_init(AVCodecContext *avctx)
    s->version = 2;

    avctx->coded_frame = av_frame_alloc();
-    if (!avctx->coded_frame)
+    if (!avctx->extradata || !avctx->stats_out || !avctx->coded_frame)
        return AVERROR(ENOMEM);

    avctx->coded_frame->pict_type = AV_PICTURE_TYPE_I;