mirror of https://git.ffmpeg.org/ffmpeg.git
lavfi/dnn_backend_openvino.c: fix mem leak for input_blob and output_blob upon error
This commit is contained in:
parent
3ce2ee7f54
commit
bd3ca0859e
|
@ -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;
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue