Commit 1b0a8e48 authored by Jun Zhao's avatar Jun Zhao

lavfi/dnn/dnn_backend_native: fix memory leak in error path

fix memory leak in error path
Reviewed-by: 's avatarGuo, Yejun <yejun.guo@intel.com>
Signed-off-by: 's avatarJun Zhao <barryjzhao@tencent.com>
parent f156f4ab
......@@ -185,6 +185,7 @@ DNNModel *ff_dnn_load_model_native(const char *model_filename)
if (dnn_size > file_size || conv_params->input_num <= 0 ||
conv_params->output_num <= 0 || conv_params->kernel_size <= 0){
avio_closep(&model_file_context);
av_freep(&conv_params);
ff_dnn_free_model_native(&model);
return NULL;
}
......@@ -192,6 +193,9 @@ DNNModel *ff_dnn_load_model_native(const char *model_filename)
conv_params->biases = av_malloc(conv_params->output_num * sizeof(float));
if (!conv_params->kernel || !conv_params->biases){
avio_closep(&model_file_context);
av_freep(&conv_params->kernel);
av_freep(&conv_params->biases);
av_freep(&conv_params);
ff_dnn_free_model_native(&model);
return NULL;
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment