Ошибка сегментации в методе расширения Python
Я работал над расширением 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, который метод пытается вернуть.
Как я могу безопасно вернуть объект результата?
1 ответ
Благодаря комментарию @user2357112 и некоторому чтению, обновляющему мой C/C++, я вспомнил, что мог определить метод до его использования, и таким образом я мог решить свою проблему следующим образом:
Я определил свой метод в начале файла следующим образом:
static PyObject *IntObj_add(PyObject *, PyObject *);
И затем после того, как PyTypeObject реализован, я реализовал метод, создающий объект, вызывающий объект типа вместо нового, следующим образом:
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;
}