从 Python 调用 Go

Question

我尝试在 python 的 go 脚本中运行一个简单的脚本，但遇到了分段错误。这是我的代码：

main.go

package main

import (
    /*
typedef struct foo{
int a;
int b;
int c;
int d;
int e;
int f;
} foo;
*/
    "C"
)

func main() {}

//export Foo
func Foo(t []int) C.foo {
    return C.foo{}
}

main.py

# loading shared object
lib = cdll.LoadLibrary("main.so")

# go type
class GoSlice(Structure):
    _fields_ = [("data", POINTER(c_void_p)), ("len", c_longlong), ("cap", c_longlong)]

lib.Foo.argtypes = [GoSlice]
lib.Foo.restype = c_void_p

t = GoSlice((c_void_p * 5)(1, 2, 3, 4, 5), 5, 5)
f = lib.Foo(t)
print(f)

使用该代码，我得到了

140362617287784
[1]    23067 segmentation fault  python3 main.py

现在如果我从 main.go 中删除 e 和 f 我得到了

None

不再有分段错误。

为什么结构中的成员数量在这里很重要？

[编辑] 两者都在同一个地方运行，我运行一个命令clear && go build -o main.so -buildmode=c-shared main.go && python3 main.py

Answer 1

您的 GO/C 代码是正确的。问题出在python脚本中。 lib.Foo.restype = c_void_p 调用需要一个 void 指针，但库返回一个 C 结构。您需要在 python 中定义一个 ctypes 结构的返回类型，然后它将按您的预期工作。

ma​​in.go：

package main

import (
    /*
    typedef struct foo{
    int a;
    int b;
    int c;
    int d;
    int e;
    int f;
    } foo;
    */
    "C"
)

func main() {
}

//export Foo
func Foo(t []int) C.foo {
    foo := C.foo{}
    foo.a = 1 // setting some values to avoid seeing zeros
    foo.b = 2
    return foo
}

ma​​in.py：

from ctypes import *

# loading shared object
lib = cdll.LoadLibrary("main.so")


# go type
class GoSlice(Structure):
    _fields_ = [("data", POINTER(c_void_p)), ("len", c_longlong), ("cap", c_longlong)]


class Foo(Structure):
    _fields_ = [('a', c_int),
                ('b', c_int),
                ('c', c_int),
                ('d', c_int),
                ('e', c_int),
                ('f', c_int)]


lib.Foo.argtypes = [GoSlice]
lib.Foo.restype = Foo

t = GoSlice((c_void_p * 5)(1, 2, 3, 4, 5), 5, 5)
f = lib.Foo(t)
print(f)
print(f.a)
print(f.b)

然后运行go build -o main.so -buildmode=c-shared main.go && python main.py，它会打印：

go build -o main.so -buildmode=c-shared main.go && python3 main.py 
<__main__.Foo object at 0x102608830>
1
2