增加CFUNCTYPE的参考计数器

时间:2014-12-01 16:21:47

标签: python c ctypes

我有一个用C语言编写的库。库中的一个例程接受一个函数指针,将其保存到内存供以后使用,然后返回。我正在尝试传递Python回调函数。

我看到的问题是稍后调用此函数有时会导致分段错误。这是我正在玩的一些玩具代码。 Python代码:

import ctypes
import sys

class problem(ctypes.Structure):
    _fields_ = [("val",ctypes.c_int),("f",ctypes.POINTER(ctypes.c_void_p))]

def change_struct_data(s):
    s.contents.val = 99

class UsefulInfo:
    def __init__(self,library_name):
        self.lib = ctypes.pydll.LoadLibrary(library_name)
        self.lib.create_data.restype = ctypes.POINTER(problem)
        self.data = self.lib.create_data(12)

    def setcallback(self,f):
        cback = ctypes.CFUNCTYPE(None,ctypes.POINTER(problem))
        funcy = cback(f)
        self.lib.pass_func(self.data,funcy)

    def makecall(self):
        self.data.contents.val = 0
        self.lib.use_callback(self.data)
        print 'val is now',self.data.contents.val


test = UsefulInfo('./libfoo.so')
test.setcallback(change_struct_data)
test.makecall()

C代码:

//gcc test.c -I/usr/include/python2.7/ -L/usr/lib/python2.7/ -lpython2.7 -lm -fPIC -c -g
//gcc -shared -Wl,-soname,libfoo.so -o libfoo.so  test.o
#include <Python.h>
#include <stdio.h>
#include <stdlib.h>

typedef struct problem
{
    int val;
        void (*f)(struct problem*);
} problem;


int C_inc_ref(PyObject* obj)
{
    Py_INCREF(obj);
    return 0;
}

void pass_func(problem* prob,void (*funcy)(problem*))
{
    prob->f = funcy;
}

void use_callback(problem* prob)
{
    prob->f(prob);
}

problem* create_data(int n)
{
    problem* prob = (problem*) malloc(sizeof(problem));
    prob->val = n;
    prob->f = NULL;
    return prob;
}

经过深思熟虑,我怀疑在例程setcallback中转换为CFUNCTYPE的函数正在被垃圾收集。我用print语句稍微修改了例程:

def setcallback(self,f):
    cback = ctypes.CFUNCTYPE(None,ctypes.POINTER(problem))
    funcy = cback(f)
    self.lib.pass_func(self.data,funcy)
    print 'ref count is',sys.getrefcount(funcy)

毫不奇怪,我获得了funcy的2个引用。我尝试将它设置为一个全局变量,以防止垃圾收集,这似乎工作(它确实增加了引用计数),但肯定不是很漂亮。

我认为更好的解决方案是手动增加引用计数。但我根本无法使其正常工作。我再次将例程更改为:

def setcallback(self,f):
    cback = ctypes.CFUNCTYPE(None,ctypes.POINTER(problem))
    funcy = cback(f)
    self.lib.pass_func(self.data,funcy)
    #self.lib.C_inc_ref(funcy)
    ctypes.pythonapi.Py_IncRef(funcy)
    print 'ref count is',sys.getrefcount(funcy)

使用对Py_IncRef的调用或对C代码的调用来增加引用计数器。但是,无论使用哪种方法递增我使用的计数器,我要么得到一个分段错误,要么计数器没有递增(保持在2)。

有没有理由不能递增回调函数的引用计数器?也许我犯了一个错误?任何帮助表示赞赏。

1 个答案:

答案 0 :(得分:2)

cback是一个返回包装函数的函数。还有什么工作方式?装修:

# global scope, the callback type
cback = ctypes.CFUNCTYPE(None,ctypes.POINTER(problem))

# decorate the callback
@cback
def change_struct_data(s):
    s.contents.val = 99

现在change_struct_data实际上是一个实例化的CFUNCTYPE对象,它不会超出范围。你不能再直接从Python中调用它,因为它已被装饰为从C调用,但这不应该是一个问题。