[FFmpeg-devel] [PATCH] libavfilter/dnn_backend_native: check mem allocation

[Andreas Rheinhardt]

Chris Miceli:
> check that frame allocations return non-null.
> ---
>  libavfilter/dnn/dnn_backend_native.c | 9 +++++++++
>  1 file changed, 9 insertions(+)
> 
> diff --git a/libavfilter/dnn/dnn_backend_native.c b/libavfilter/dnn/dnn_backend_native.c
> index d45e211f0c..cda857a9c5 100644
> --- a/libavfilter/dnn/dnn_backend_native.c
> +++ b/libavfilter/dnn/dnn_backend_native.c
> @@ -81,6 +81,15 @@ static DNNReturnType get_output_native(void *model, const char *input_name, int
>      NativeModel *native_model = (NativeModel *)model;
>      AVFrame *in_frame = av_frame_alloc();
>      AVFrame *out_frame = av_frame_alloc();
> +
> +    if (!in_frame)
> +        return DNN_ERROR;
> +

out_frame may leak here.

> +    if (!out_frame) {
> +        av_frame_free(&in_frame);
> +        return DNN_ERROR;
> +    }
> +
>      in_frame->width = input_width;
>      in_frame->height = input_height;
>  
>