Python扩展方法的分段错误

时间:2017-10-16 22:58:29

标签: python c++ python-extensions

我一直致力于Python C / C ++扩展,到目前为止,我设法让一切工作,除了一个方法添加了两个名为“int_obj”的结构,里面有一个整数并返回一个名为“add_int_obj”的新结构。 / p>

到目前为止,扩展只有两种方法,“值”完美无瑕,“添加”是有问题的方法。

我正在使用以下test.py脚本测试它:

import intobj
val1 = intobj.IntObj(3)
val2 = intobj.IntObj(5)
print "Val1"
print val1
print "Val1 Value"
print val1.value()
print "Val2"
print val2
print "Val2 Value"
print val2.value()
val3 = intobj.add(val1, val2)
print "Val3"
print val3
print "Val3 Value"
print val3.value()

当我执行时,我得到以下输出:

$ python test.py 
Val1
<intobj.IntObj object at 0x7faf3fc8a0f0>
Val1 Value
3
Val2
<intobj.IntObj object at 0x7faf3fc8a108>
Val2 Value
5
IntObj_add start
Int Obj Value A: 3
Int Obj Value B: 5
Int Obj Value Result: 8
IntObj_add end
Val3
Segmentation fault (core dumped)

处理添加两个结构的扩展方法是:

static PyObject *IntObj_add(PyObject *self,PyObject *args)
{
    std::cout<<"IntObj_add start"<<std::endl;

    PyObject *a,*b;

    IntObj *result;

    if (!PyArg_ParseTuple(args, "OO", &a, &b))
            return NULL;

    std::cout<<"Int Obj Value A: "<<int_obj_value(((IntObj*)a)->content)<<std::endl;
    std::cout<<"Int Obj Value B: "<<int_obj_value(((IntObj*)b)->content)<<std::endl;

    result = new IntObj;

    Py_INCREF(result);

    result->content = add_int_obj(((IntObj*)a)->content,((IntObj*)b)->content);

    std::cout<<"Int Obj Value Result: "<<int_obj_value(result->content)<<std::endl;

    std::cout<<"IntObj_add end"<<std::endl;

    return (PyObject*)result;
}

由于当我尝试打印对象时,仅在方法之外发生分段错误,而不是当我尝试获取值时,我的猜测是垃圾收集器正在删除该方法试图返回的PyObject。

如何安全地返回结果对象?

1 个答案:

答案 0 :(得分:0)

感谢@ user2357112的评论和一些阅读刷新我的C / C ++,我记得我可以在使用它之前定义方法,这样我就可以通过以下方式解决我的问题:

我在文件开头定义了我的方法,如下所示:

static PyObject *IntObj_add(PyObject *, PyObject *);

然后在实现PyTypeObject之后,我实现了创建调用类型对象而不是new的对象的方法,如下所示:

static PyObject *IntObj_add(PyObject *self, PyObject *args)
{
    PyObject *a, *b;

    IntObj *result;

    if (!PyArg_ParseTuple(args, "OO", &a, &b))
        return NULL;

    result = (IntObj *)IntObjType.tp_alloc(&IntObjType, 0);

    result->content = add_int_obj(((IntObj *)a)->content, ((IntObj *)b)->content);

    return (PyObject *)result;
}