用Golang和C编写的Python模块

时间:2016-03-20 19:09:11

标签: python c go cgo

我遵循此tutorial

在C:

中编写此代码
#define Py_LIMITED_API
#include <Python.h>

PyObject * startVM(PyObject *, PyObject *);

int PyArg_ParseTuple_S(PyObject * args, char* a) {  
    return PyArg_ParseTuple(args, "s", &a);
}

static PyMethodDef FooMethods[] = {  
    {"startVM", startVM, METH_VARARGS, "Starts."},
    {NULL, NULL, 0, NULL}
};

static struct PyModuleDef foomodule = {  
   PyModuleDef_HEAD_INIT, "foo", NULL, -1, FooMethods
};

PyMODINIT_FUNC PyInit_foo(void) {
    return PyModule_Create(&foomodule);
}

这个代码在GO中:

package main

import "fmt"


// #cgo pkg-config: python3
// #define Py_LIMITED_API
// #include <Python.h>
// int PyArg_ParseTuple_S(PyObject *,char *);
import "C"

//export startVM
func startVM(self, args *C.PyObject) {  
    var a *C.char
    if C.PyArg_ParseTuple_S(args, a) == 0 {
        //return nil
    }
    fmt.Println(a)
    //return C.PyBytes_FromString(&a)
}

func main() {}  

我可以在go中编译代码,但是当我使用此命令调用python模块时:python3 -c 'import foo; foo.startVM("hello")',它会打印nil并导致分段错误... 有人知道如何解决它吗? 提前谢谢。

1 个答案:

答案 0 :(得分:4)

无输出

此功能:

int PyArg_ParseTuple_S(PyObject * args, char* a) {
    return PyArg_ParseTuple(args, "s", &a);
}

只会设置a本地副本而不会将其返回到调用函数中,因为您按值传递字符串指针(通过复制)所以PyArg_ParseTuple只设置副本。

var a *C.char
C.PyArg_ParseTuple_S(args, a)
// Here `a` is not set, so it keeps its default value: nil.

您可以通过将指针传递给字符串而不是字符串本身来解决此问题:

// C
int PyArg_ParseTuple_S(PyObject * args, char** a) {
    return PyArg_ParseTuple(args, "s", a);
}

// Go
var a *C.char
if C.PyArg_ParseTuple_S(args, &a) == 0 {
    //return nil
}

正确打印

fmt.Println(a)将打印a所持的地址,而不是它指向的字符串。 Go有自己的字符串类型,不适用于C字符串。

如果要正确打印文本,必须使用C.GoString

进行转换
// C string to Go string
func C.GoString(*C.char) string

(来自https://golang.org/cmd/cgo/

例如:

str := C.GoString(a)
fmt.Println(str)

分段错误。

我不熟悉python模块开发,但我可以假设发生了错误,因为python方法应该返回有效的PyObject*NULL。但你的代码没有做到这一点。 startVM的返回值未设置且默认为nil,python接受此非零指针作为有效对象并取消引用它,这会导致分段错误。

指定startVM的返回类型可能会有所帮助:

//export startVM
func startVM(self, args *C.PyObject) *C.PyObject {  
    // ...some code...
    return nil
}