virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-01-13 21:28:01 +02:00
Code Issues Releases 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; DNNReturnType ret;
NativeModel *native_model = (NativeModel *)model; NativeModel *native_model = (NativeModel *)model;
NativeContext *ctx = &native_model->ctx;
AVFrame *in_frame = av_frame_alloc(); 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->width = input_width;
in_frame->height = input_height; in_frame->height = input_height;