python - Python 扩展方法的段错误

Question

我一直在研究 Python C/C++ 扩展，到目前为止，我设法让所有的东西都能正常工作，除了一个方法，它添加了两个名为“int_obj”的结构，其中有一个整数，并返回一个名为“add_int_obj”的新结构。

到目前为止，该扩展程序只有两种方法，完美运行的“value”和有问题的“add”方法。

我正在使用以下 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。

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

Answer 1

感谢@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;
}