Question

我一直在学习用C语言扩展Python，我终于以某种方式使用了它，但是我得到了这个非常奇怪的错误。我用一个函数制作了一个测试模块来测试一个数字是否为素数。这是代码：

cmodule.c

#include <stdbool.h>
#include <math.h>
#include <Python.h>

bool isprime(int n) {
    if (n == 0 || n == 1)
        return false;

    for (int i = 2; i < (int)sqrt(n) + 1; i++) {
            if (n % i == 0)
                    return false;
    }
    return true;
}

static PyObject * isprime_wrapper(PyObject * self, PyObject * args) {
    int n;
    if (!PyArg_ParseTuple(args, "i", &n))
        return NULL;

    bool retval = isprime(n);
    if (retval)
        return Py_True;
    else
        return Py_False;    
}

static PyMethodDef methods[] = {
    {"isprime", isprime_wrapper, METH_VARARGS, "Tests if a number is prime."},
    {NULL, NULL, 0, NULL}
};

static struct PyModuleDef cmodule = {
    PyModuleDef_HEAD_INIT,
    "cmodule",
    "Test module.",
    -1,
    methods
};

PyMODINIT_FUNC PyInit_cmodule(void) {
    return PyModule_Create(&cmodule);
}

setup.py

from distutils.core import setup, Extension

cmodule = Extension("cmodule", sources = ["cmodule.c"])

setup(name = "CModuleTest",
    ext_modules = [cmodule])

乍一看它工作正常。我运行python3.6 setup.py build并且它构建没有错误，我可以使用.so文件进入目录，启动Python会话，并调用cmodule.isprime（）。然而，奇怪的部分是当我在循环中使用代码时，我得到这个错误：

>>> import cmodule
>>> for i in range(1, 1000001):
...     n = cmodule.isprime(i)
... 
*** Error in `python3.6': free(): invalid pointer: 0x0000561cc1faf620 ***

如果有人觉得有用，我可以添加完整的回溯。我在执行此操作时打印出i的所有值，并且在尝试运行isprime时似乎总是会出现此错误（154）。但是，当我在终端中打开Python并尝试运行isprime（154）或任何其他数字时，它可以正常工作。我能够复制它的唯一一次是在循环中，我不知道它可能来自哪里。我看过其他有类似问题的人，似乎是因为他们在C代码中使用free（）不正确，但我从来没有从我的代码中调用free（）。有谁知道可能导致这种情况的原因是什么？谢谢你的帮助。

Answer 1

CPython对象被引用计数，并且返回Py_True和Py_False可能是问题。改为使用Py_RETURN_TRUE等。它们适当地增加引用计数并返回它们各自的单例对象。

Python C扩展 - “python3.6”中的错误：free（）：无效指针：0x

1 个答案: