Browse Source

libavfilter/dnn_backend_native: check mem allocation

check that frame allocations return non-null.
tags/n4.4
Chris Miceli Guo, Yejun 5 years ago
parent
commit
ad95e5e45d
1 changed files with 16 additions and 1 deletions
  1. +16
    -1
      libavfilter/dnn/dnn_backend_native.c

+ 16
- 1
libavfilter/dnn/dnn_backend_native.c View File

@@ -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;



Loading…
Cancel
Save