Вызов Go из Python - PullRequest
Новая
       34

Вызов Go из Python

1 голос
/ 13 июня 2019

Я попытался запустить простой скрипт go из python, и у меня возникла ошибка сегментации.Вот мой код:

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

Теперь, если я удаляю e и f из main.go, я получил 

None

и больше нет ошибки сегментации.

Почему число членов в структуре имеет значение здесь?

[EDIT] Оба работают в одном месте, я запускаю одну команду clear && go build -o main.so -buildmode=c-shared main.go && python3 main.py

1 Ответ

2 голосов
/ 14 июня 2019

Ваш код GO / C правильный.Проблема в скрипте Python.Вызов lib.Foo.restype = c_void_p ожидает указатель void, но библиотека возвращает структуру C.Вам нужно определить тип возвращаемого значения для структуры ctypes в python, тогда он будет работать так, как вы ожидаете.

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 {
    foo := C.foo{}
    foo.a = 1 // setting some values to avoid seeing zeros
    foo.b = 2
    return foo
}

main.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
...