C++向C传递unique_ptr释放的内存时如何避免内存泄漏？

核心结论 #

你当前在C代码中用free(predictions)释放C++里new[]分配的内存是未定义行为，绝对不能这么做。必须严格保证内存分配与释放的配对规则：new[]对应delete[]，malloc对应free，跨机制混用会引发堆损坏、内存泄漏甚至程序崩溃。

为什么不能混用new[]和free #

尽管uint8_t是POD类型（无构造/析构逻辑），但new[]和malloc属于两套独立的内存分配体系：

• new[]调用C++标准库的内存分配器，可能附带额外元数据（比如数组长度，用于delete[]时计算释放范围）。
• free仅能正确识别malloc/calloc/realloc分配的内存结构，无法处理new[]的内存布局，混用必然导致问题。

两种正确的实现方案 #

方案1：在C++包装器中提供专门的释放函数（推荐）

和model_delete的设计思路一致，新增一个C可调用的函数来释放预测结果，确保用delete[]配对new[]。

修改cpp_wrapper.cpp，添加释放函数：

extern "C" void predictions_delete(uint8_t* predictions) {
    delete[] predictions;
}

在cpp_wrapper.h中声明该函数：

extern "C" void predictions_delete(uint8_t* predictions);

然后在C代码中替换free(predictions)：

// ... 使用predictions之后
predictions_delete(predictions);

该方案的优势：

• 严格遵循C++内存管理规则，扩展性强：若未来返回类型需要构造/析构（非POD对象），无需修改即可兼容。
• 彻底避免内存分配器混用的风险，稳定性更高。

方案2：C++中用malloc分配内存，C端用free释放

如果希望C端保持malloc/free的使用习惯，可以在C++包装器中改用malloc分配内存，此时C端用free释放完全合法。

修改run_model中的内存分配代码：

// 替换new[]为malloc
uint8_t* classValues = static_cast<uint8_t*>(malloc(sizeof(uint8_t) * predictions.size()));
if (classValues == nullptr) {
    // 处理内存分配失败，比如返回nullptr
    return nullptr;
}
memcpy(classValues, predictions.data(), sizeof(uint8_t) * predictions.size());
return classValues;

此时C端的free(predictions)是安全的。

该方案的注意点：

• 必须处理malloc返回nullptr的情况，避免后续memcpy触发崩溃。
• 仅适用于POD类型，若未来返回非POD对象，无法处理构造/析构逻辑。

最终建议 #

优先选择方案1，它更符合C++设计规范，能规避潜在内存问题，同时具备更好的扩展性。

内容的提问来源于stack exchange，提问作者CopyOfA