[FFmpeg-devel] [PATCH] libavfilter/dnn_backend_native: check mem allocation
Chris Miceli
chris at miceli.net.au
Wed Oct 14 03:19:50 EEST 2020
check that frame allocations return non-null.
---
libavfilter/dnn/dnn_backend_native.c | 17 ++++++++++++++++-
1 file changed, 16 insertions(+), 1 deletion(-)
diff --git a/libavfilter/dnn/dnn_backend_native.c b/libavfilter/dnn/dnn_backend_native.c
index d45e211f0c..4fc3ba2044 100644
--- a/libavfilter/dnn/dnn_backend_native.c
+++ b/libavfilter/dnn/dnn_backend_native.c
@@ -79,8 +79,23 @@ static DNNReturnType get_output_native(void *model, const char *input_name, int
{
DNNReturnType ret;
NativeModel *native_model = (NativeModel *)model;
+ NativeContext *ctx = &native_model->ctx;
AVFrame *in_frame = av_frame_alloc();
- AVFrame *out_frame = av_frame_alloc();
+ AVFrame *out_frame = NULL;
+
+ if (!in_frame) {
+ av_log(ctx, AV_LOG_ERROR, "Could not allocate memory for input frame\n");
+ return DNN_ERROR;
+ }
+
+ out_frame = av_frame_alloc();
+
+ if (!out_frame) {
+ av_log(ctx, AV_LOG_ERROR, "Could not allocate memory for output frame\n");
+ av_frame_free(&in_frame);
+ return DNN_ERROR;
+ }
+
in_frame->width = input_width;
in_frame->height = input_height;
--
2.28.0
More information about the ffmpeg-devel
mailing list