lavc/vaapi_decode: Don't update buffer number if allocataion fail

Signed-off-by: Fei Wang <fei.w.wang@intel.com>
8 months ago · 8962e2b1aa
parent a8d9fab06b
commit 8962e2b1aa
1 changed files with 3 additions and 6 deletions
--- a/libavcodec/vaapi_decode.c
+++ b/libavcodec/vaapi_decode.c
@ -72,17 +72,14 @@ int ff_vaapi_decode_make_slice_buffer(AVCodecContext *avctx,

    av_assert0(pic->nb_slices <= pic->slices_allocated);
    if (pic->nb_slices == pic->slices_allocated) {
-        if (pic->slices_allocated > 0)
-            pic->slices_allocated *= 2;
-        else
-            pic->slices_allocated = 64;
-
        pic->slice_buffers =
            av_realloc_array(pic->slice_buffers,
-                             pic->slices_allocated,
+                             pic->slices_allocated ? pic->slices_allocated * 2 : 64,
                             2 * sizeof(*pic->slice_buffers));
        if (!pic->slice_buffers)
            return AVERROR(ENOMEM);
+
+        pic->slices_allocated = pic->slices_allocated ? pic->slices_allocated * 2 : 64;
    }
    av_assert0(pic->nb_slices + 1 <= pic->slices_allocated);