avcodec/cfhdenc: free alpha buffer on closing

diff --git a/libavcodec/cfhdenc.c b/libavcodec/cfhdenc.c
index 8030413548cba70c60d0b88b0d3f46b0fd1dcf90..d53604ae110d5475b818c3aa8ca646ef8d23656b 100644 (file)
--- a/libavcodec/cfhdenc.c
+++ b/libavcodec/cfhdenc.c
@@ -356,6 +356,9 @@ static av_cold int cfhd_encode_init(AVCodecContext *avctx)
             s->lut[i] = last;
     }
 
+    if (s->planes != 4)
+        return 0;
+
     s->alpha = av_calloc(avctx->width * avctx->height, sizeof(*s->alpha));
     if (!s->alpha)
         return AVERROR(ENOMEM);
@@ -864,6 +867,8 @@ static av_cold int cfhd_encode_close(AVCodecContext *avctx)
             s->plane[i].l_h[j] = NULL;
     }
 
+    av_freep(&s->alpha);
+
     return 0;
 }