diff options
author | Jun Zhao <barryjzhao@tencent.com> | 2019-08-19 14:52:50 +0800 |
---|---|---|
committer | Jun Zhao <barryjzhao@tencent.com> | 2019-08-20 10:07:38 +0800 |
commit | 1b0a8e48f1d710800d2bd85795cfb3e93f28ab32 (patch) | |
tree | 244b5fa558ccf8ad6c5fc6c76e67144a2b70d50d /libavfilter | |
parent | f156f4ab2317f22bfef33c7eaead0d5d5f162903 (diff) | |
download | ffmpeg-streaming-1b0a8e48f1d710800d2bd85795cfb3e93f28ab32.zip ffmpeg-streaming-1b0a8e48f1d710800d2bd85795cfb3e93f28ab32.tar.gz |
lavfi/dnn/dnn_backend_native: fix memory leak in error path
fix memory leak in error path
Reviewed-by: Guo, Yejun <yejun.guo@intel.com>
Signed-off-by: Jun Zhao <barryjzhao@tencent.com>
Diffstat (limited to 'libavfilter')
-rw-r--r-- | libavfilter/dnn/dnn_backend_native.c | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/libavfilter/dnn/dnn_backend_native.c b/libavfilter/dnn/dnn_backend_native.c index 09c583b..d52abc6 100644 --- a/libavfilter/dnn/dnn_backend_native.c +++ b/libavfilter/dnn/dnn_backend_native.c @@ -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; } |