[FFmpeg-devel] [PATCH] libavfilter/dnn_backend_native: check mem allocation
Andreas Rheinhardt
andreas.rheinhardt at gmail.com
Tue Oct 13 06:16:58 EEST 2020
Chris Miceli:
> check that frame allocations return non-null.
> ---
> libavfilter/dnn/dnn_backend_native.c | 8 ++++++++
> 1 file changed, 8 insertions(+)
>
> diff --git a/libavfilter/dnn/dnn_backend_native.c b/libavfilter/dnn/dnn_backend_native.c
> index d45e211f0c..d1eb992335 100644
> --- a/libavfilter/dnn/dnn_backend_native.c
> +++ b/libavfilter/dnn/dnn_backend_native.c
> @@ -81,6 +81,14 @@ 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;
> + }
> +
> + if (!out_frame) {
> + return DNN_ERROR;
> + }
> in_frame->width = input_width;
> in_frame->height = input_height;
>
> If exactly one of the allocations fails, the other one will leak.
- Andreas
More information about the ffmpeg-devel
mailing list