RepoMirrors/ffmpeg
1
0
Fork 0
mirror of https://git.ffmpeg.org/ffmpeg.git synced 2025-01-29 18:53:02 +00:00
Code Issues Projects Releases Wiki Activity

libavfilter/dnn/dnn_backend{openvino, tf}: check memory alloc non-NULL

Browse Source 
These previously would not check that the return value was non-null
meaning it was susceptible to a sigsegv. This checks those values.
This commit is contained in:
Chris Miceli 2020-10-14 11:59:44 +11:00 committed by Guo, Yejun
parent ad95e5e45d
commit 6bdfea8d4b
2 changed files with 28 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
libavfilter/dnn/dnn_backend_openvino.c
View File

@ -141,8 +141,20 @@ static DNNReturnType get_output_ov(void *model, const char *input_name, int inpu
{
DNNReturnType ret;
OVModel *ov_model = (OVModel *)model;
OVContext *ctx = &ov_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, "Failed to allocate memory for input frame\n");
return DNN_ERROR;
}
out_frame = av_frame_alloc();
if (!out_frame) {
av_log(ctx, AV_LOG_ERROR, "Failed to allocate memory for output frame\n");
av_frame_free(&in_frame);
return DNN_ERROR;
}
in_frame->width = input_width;
in_frame->height = input_height;

16
libavfilter/dnn/dnn_backend_tf.c
View File

@ -159,8 +159,22 @@ static DNNReturnType get_output_tf(void *model, const char *input_name, int inpu
{
DNNReturnType ret;
TFModel *tf_model = (TFModel *)model;
TFContext *ctx = &tf_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, "Failed to allocate memory for input frame\n");
return DNN_ERROR;
}
out_frame = av_frame_alloc();
if (!out_frame) {
av_log(ctx, AV_LOG_ERROR, "Failed to allocate memory for output frame\n");
av_frame_free(&in_frame);
return DNN_ERROR;
}
in_frame->width = input_width;
in_frame->height = input_height;