ffmpeg/FFmpeg
1
0
Fork 0
mirror of https://git.ffmpeg.org/ffmpeg.git synced 2024-09-21 22:06:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

libavfilter/dnn_backend_native: check mem allocation

Browse Source 
check that frame allocations return non-null.
This commit is contained in:
Chris Miceli 2020-10-14 11:19:50 +11:00 committed by Guo, Yejun
parent 00772ef4f7
commit ad95e5e45d
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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