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;
+
+ if (!out_frame) {
+ av_frame_free(&in_frame);
+ return DNN_ERROR;
+ }
+
in_frame->width = input_width;
in_frame->height = input_height;
--
2.28.0
_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
To unsubscribe, visit link above, or email
ffmpeg-devel-requ...@ffmpeg.org with subject "unsubscribe".
