Question

我正在尝试修复C ++ dll的Python包装器中的内存泄漏。问题是将字节缓冲区分配给在Python中创建的辅助对象：

struct ByteBuffer
{
    int length;
    uint8_t * dataBuf;       
};

我想将dataBuf作为Python数组提供，因此我提出（并且有效）的类型映射是：

%module(directors="1") mymodule

%typemap(in) uint8_t * (uint8_t *temp){
    int length = PySequence_Length($input);
    temp = new uint8_t[length];    // memory allocated here. How to free?
    for(int i=0; i<length; i++) {
        PyObject *o = PySequence_GetItem($input,i);
        if (PyNumber_Check(o)) {
            temp[i] = (uint8_t) PyLong_AsLong(o);
            //cout << (int)temp[i] << endl;
        } else {
            PyErr_SetString(PyExc_ValueError,"Sequence elements must be uint8_t");      
            return NULL;
        }
    }
    $1 = temp;
}

问题是typemap每次都为新的C数组分配内存，而且这个内存不会在dll中释放。换句话说，dll期望用户管理ByteBuffer的dataBuf的内存。例如，当在Python中按顺序创建10000个这样的对象然后删除它们时，内存使用量会稳定上升（泄漏）：

for i in range(10000):
    byteBuffer = mymodule.ByteBuffer()
    byteBuffer.length = 10000
    byteBuffer.dataBuf = [0]*10000
    # ... use byteBuffer
    del byteBuffer

有没有办法从python中删除已分配的dataBuf？感谢您的耐心等待！

编辑：我不发布整个工作代码以保持简短。如果需要，我会这样做。另外，我使用的是Python 3.5 x64和SWIG ver 3.0.7

Answer 1

这比我想象的要简单得多。我刚刚将它添加到.i文件

%typemap(freearg) uint8_t * {
    //cout << "Freeing uint8_t*!!! " << endl;
    if ($1) delete[]($1);
}

似乎工作。编辑：使用删除[]

自由切换

SWIG typemap中的内存释放

1 个答案: