[FFmpeg-devel] [PATCH] libavfilter/dnn_backend_native: check mem allocation
Guo, Yejun
yejun.guo at intel.com
Tue Oct 13 11:08:59 EEST 2020
> -----Original Message-----
> From: ffmpeg-devel <ffmpeg-devel-bounces at ffmpeg.org> On Behalf Of Chris
> Miceli
> Sent: 2020年10月13日 13:16
> To: ffmpeg-devel at ffmpeg.org
> Cc: Chris Miceli <chris at miceli.net.au>
> Subject: [FFmpeg-devel] [PATCH] libavfilter/dnn_backend_native: check mem
> allocation
>
> check that frame allocations return non-null.
> ---
> libavfilter/dnn/dnn_backend_native.c | 13 ++++++++++++-
> 1 file changed, 12 insertions(+), 1 deletion(-)
>
> diff --git a/libavfilter/dnn/dnn_backend_native.c
> b/libavfilter/dnn/dnn_backend_native.c
> index d45e211f0c..7958ceb4f7 100644
> --- a/libavfilter/dnn/dnn_backend_native.c
> +++ b/libavfilter/dnn/dnn_backend_native.c
> @@ -80,7 +80,18 @@ static DNNReturnType get_output_native(void *model,
> const char *input_name, int
> DNNReturnType ret;
> NativeModel *native_model = (NativeModel *)model;
> AVFrame *in_frame = av_frame_alloc();
> - AVFrame *out_frame = av_frame_alloc();
> + AVFrame *out_frame = NULL;
> +
> + if (!in_frame)
> + return DNN_ERROR;
> +
> + out_frame = av_frame_alloc();
> +
> + if (!out_frame) {
> + av_frame_free(&in_frame);
> + return DNN_ERROR;
> + }
> +
could you add some log message just like your first patch? thanks.
and, could you also fix the same issue in get_output_ov and get_output_tf if you'd like, thanks.
More information about the ffmpeg-devel
mailing list