diff options
| author | Guo, Yejun <yejun.guo@intel.com> | 2021-03-13 13:52:21 +0800 |
|---|---|---|
| committer | Guo, Yejun <yejun.guo@intel.com> | 2021-03-18 09:30:09 +0800 |
| commit | bd3ca0859e993a41aae9105e32cd7f2a3c57d468 (patch) | |
| tree | dfa8f705c256f50e2b62330a118dd4d2c839da6e /libavfilter | |
| parent | 3ce2ee7f54ccbb20c88c1e8e0cc5796e06580ad0 (diff) | |
| download | ffmpeg-bd3ca0859e993a41aae9105e32cd7f2a3c57d468.tar.gz | |
lavfi/dnn_backend_openvino.c: fix mem leak for input_blob and output_blob upon error
Diffstat (limited to 'libavfilter')
| -rw-r--r-- | libavfilter/dnn/dnn_backend_openvino.c | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/libavfilter/dnn/dnn_backend_openvino.c b/libavfilter/dnn/dnn_backend_openvino.c index d86fb124d5..df1631e86e 100644 --- a/libavfilter/dnn/dnn_backend_openvino.c +++ b/libavfilter/dnn/dnn_backend_openvino.c @@ -141,12 +141,14 @@ static DNNReturnType fill_model_input_ov(OVModel *ov_model, RequestItem *request status |= ie_blob_get_dims(input_blob, &dims); status |= ie_blob_get_precision(input_blob, &precision); if (status != OK) { + ie_blob_free(&input_blob); av_log(ctx, AV_LOG_ERROR, "Failed to get input blob dims/precision\n"); return DNN_ERROR; } status = ie_blob_get_buffer(input_blob, &blob_buffer); if (status != OK) { + ie_blob_free(&input_blob); av_log(ctx, AV_LOG_ERROR, "Failed to get input blob buffer\n"); return DNN_ERROR; } @@ -211,6 +213,7 @@ static void infer_completion_callback(void *args) status = ie_blob_get_buffer(output_blob, &blob_buffer); if (status != OK) { + ie_blob_free(&output_blob); av_log(ctx, AV_LOG_ERROR, "Failed to access output memory\n"); return; } @@ -218,6 +221,7 @@ static void infer_completion_callback(void *args) status |= ie_blob_get_dims(output_blob, &dims); status |= ie_blob_get_precision(output_blob, &precision); if (status != OK) { + ie_blob_free(&output_blob); av_log(ctx, AV_LOG_ERROR, "Failed to get dims or precision of output\n"); return; } |