如何避免unique_ptr从c++释放到C时的内存泄漏

Question

我有一个C程序，它在一个大型c++库周围调用c++包装器。c++库的一个功能是提供一些预测，这些预测将在C中使用。我想知道我应该在哪里free或delete内存。下面是一些示例代码:

cpp_wrapper.cpp

#include "cpp_wrapper.h"

uint8_t * run_model(struct_with_model model_struct, const double * input_data) {
// the model_struct is new'd in C++ in an earlier method
// I have a method to convert the input data to std::vector for use in C++ code
std::vector<double> data = get_data(input_data); 
model_struct->model->LoadData(data);
model_struct->model->run();
std::vector<uint8_t> predictions = model_struct->model->GetPredictions();
std::unique_ptr<uint8_t[]> classValues (new uint8_t[predictions.size()]);
memcpy(classValues.get(), predictions.data(), sizeof(uint8_t) * predictions.size());
model_struct->model->ClearData(); //clears data from model for future runs, if necessary
return classValues.release()
}
void model_delete(model_struct) {
// method to delete the model_struct when necessary
delete model_struct;
}

我有一个cpp_wrapper.h头文件，声明这些函数并在必要时调用extern C。C端，c_code.c:

#include "cpp_wrapper.h"

/*
There's a bunch of code here for ingesting data, initializing the model_struct, etc
*/
uint8_t * predictions = run_model(model_struct, input_data);
// Do stuff with predictions, as necessary
model_delete(model_struct);
free(predictions); // HERE is the question

Inew是c++代码中的初始变量classValues，而Irelease是std::unique_ptr作为c++run_model函数的返回值。我的理解是，当我release内存到C时，C代码负责free内存。是这样吗?我应该有一个c++方法来删除predictions变量(从C中)，就像我有一个c++方法来删除model_struct变量一样?我对如何最好地管理这里的内存感到困惑。我被建议不要在c++中使用malloc(或calloc)，但由于我将值传递回C，也许这是一个更好的选择…?我不确定。

Answer 1

最简单的解决方案是将c++调用分成两个调用。第一个决定所需缓冲区的大小。然后可以在C部分分配内存。在第二次调用中，数据被复制到这个缓冲区中。这样，内存管理完全保留在C部分。

或者，c++接口必须提供一个函数来清除内存。当要释放内存时，C代码调用这个函数。通常的规则是:

• malloc->free
• new->delete
• new[]->delete[]

如果我正确地看到这一点，你正在实现第二种方法，但不正确地使用new[]->delete!