Python C++扩展函数是否存在内存泄漏或段错误？

你的这个C扩展函数确实存在内存泄漏问题，同时还有可能引发引用计数错误，具体问题和修复方式如下：

问题分析 #

1. 循环中的内存泄漏
   每次调用PyObject_GetAttrString都会返回一个新引用（对象的引用计数会+1），但你的代码在循环里直接把新获取的对象赋值给global_var，完全覆盖了之前的对象指针。这导致前一个对象的引用计数从未被递减，这些对象无法被Python的垃圾回收机制回收，最终造成内存泄漏。

2. 空列表场景的引用错误
   如果传入的constant_names是空列表，函数会直接返回PyImport_AddModule的结果。但PyImport_AddModule返回的是借用引用，不是新创建的引用。调用者如果对返回值执行Py_DECREF，会导致模块对象的引用计数错误，进而引发程序崩溃或内存损坏。

3. 唯一正确的部分：PyImport_ImportModule返回的temp_module你已经做了Py_DECREF，这部分没有问题。

修复后的代码 #

PyObject* get_global_constant(const char* module_name, std::vector<std::string> constant_names) {
    /* Gets a global variable from a Python module */
    PyObject *module = PyImport_ImportModule(module_name);
    if (!module) {
        return nullptr; // 模块导入失败，直接返回空，交由Python处理异常
    }

    PyObject *current = module;
    for (const std::string& constant : constant_names) {
        PyObject *next = PyObject_GetAttrString(current, constant.c_str());
        if (!next) {
            Py_DECREF(module);
            return nullptr; // 获取属性失败，释放已持有的引用后返回
        }
        // 释放前一个对象的引用（初始模块引用最后统一释放）
        if (current != module) {
            Py_DECREF(current);
        }
        current = next;
    }

    Py_DECREF(module); // 释放初始的模块引用
    return current; // 返回最终获取到的对象，调用者需负责后续的Py_DECREF
}

修复要点 #

• 用PyImport_ImportModule获取模块的新引用，替代PyImport_AddModule，彻底避免借用引用带来的风险。
• 循环中保留前一个对象的指针，获取新对象后及时释放前一个的引用，避免泄漏。
• 添加错误检查，确保在导入模块或获取属性失败时，释放已持有的所有引用，防止泄漏。
• 确保返回的是合法的新引用，调用者可以正确处理其引用计数。

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