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

[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;
+
+    if (!out_frame) {
+        av_frame_free(&in_frame);
+        return DNN_ERROR;
+    }
+
     in_frame->width = input_width;
     in_frame->height = input_height;
 
-- 
2.28.0