avfilter/scale_npp: check for buffer allocation failure

[ffmpeg] / libavfilter / vf_scale_npp.c

diff --git a/libavfilter/vf_scale_npp.c b/libavfilter/vf_scale_npp.c
index f64ab956c7b17c4ac52df2c05f5079204d02a50c..3fe5bbe5008c66ec05574c68edba0b50417d8129 100644 (file)
--- a/libavfilter/vf_scale_npp.c
+++ b/libavfilter/vf_scale_npp.c
@@ -319,13 +319,7 @@ static int init_processing_chain(AVFilterContext *ctx, int in_width, int in_heig
         last_stage = i;
     }
 
-    if (last_stage < 0)
-    {
-        ctx->outputs[0]->hw_frames_ctx = av_buffer_ref(ctx->inputs[0]->hw_frames_ctx);
-        return 0;
-    }
-
-    ctx->outputs[0]->hw_frames_ctx = av_buffer_ref(s->stages[last_stage].frames_ctx);
+    ctx->outputs[0]->hw_frames_ctx = av_buffer_ref(s->stages[last_stage > 0 ? last_stage : 0].frames_ctx);
     if (!ctx->outputs[0]->hw_frames_ctx)
         return AVERROR(ENOMEM);