Python CFFI调用外部文件时崩溃，堆损坏问题排查

问题背景 #

我用C语言实现自组织映射(SOM)，并通过Python CFFI完成绑定，包装器模块代码如下：

from ._SOM import ffi, lib

lib.init()


class SOM:
    def __init__(self, x, y, input_length, sigma, learning_rate):
        self.C_som = lib.createSOM(x, y, input_length, sigma, learning_rate)

    def random_weights_init(self, data):
        newData = []
        for d in data:
            temp = ffi.new(f"double[{len(data)}]")
            for index, elem in enumerate(d):
                temp[index] = elem
            newData.append(temp)
        lib.random_weights_init(self.C_som, newData, len(data))

    def train(self, data, iterations):
        newData = []
        for d in data:
            temp = ffi.new(f"double[{len(data)}]")
            for index, elem in enumerate(d):
                temp[index] = elem
            newData.append(temp)
        lib.train(self.C_som, newData, len(data), iterations)

    def winner(self, data):
        res = lib.winner(self.C_som, data)
        x = res[0]
        y = res[1]
        lib.free(res)
        return x, y


if __name__ == "__main__":
    S = SOM(10, 10, 3, 0.67, .05)
    S.random_weights_init([[.5, 6, 9], [7, 8, 7], [7, 7, 7]])
    S.train([[.5, 6, 9], [7, 8, 7], [7, 7, 7]], 10)
    r = S.winner([7, 8, 7])
    print(r)

直接运行该文件（将第1行改为from _SOM import ffi, lib）时程序运行正常，但在另一个文件中通过from .SOM.SOM_test import SOM as C_SOM导入该类并使用时，触发堆损坏错误：

Process finished with exit code -1073740940 (0xC0000374)

问题根源 #

经排查，问题出在C代码本身：结构体中存在一个double***类型的字段，但实际分配内存时使用的是float**、float*和float类型，类型不匹配导致内存操作异常，最终引发堆损坏。

解决措施 #

• 统一内存分配类型：将C代码中所有相关内存分配从float系列类型替换为double系列，确保与结构体字段的double***类型完全一致
• 校验指针操作：检查所有针对该结构体字段的读写逻辑，保证操作的是double类型数据，避免隐式类型转换错误
• 规范内存释放：确保释放内存时使用与分配类型匹配的释放逻辑，防止非法内存访问或内存泄漏

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