在 C 中使用 Go 切片

Question

我正在尝试使用返回切片的函数构建一个 go 共享库。
如何使用 C 代码中的切片？

package main

import "C"

type T struct {
    A C.int
    B *C.char
}

//export Test
func Test() []T {
    arr := make([]T, 0)
    arr = append(arr, T{C.int(1), C.CString("a")})
    arr = append(arr, T{C.int(2), C.CString("abc")})
    return arr
}

func main() {}

go build -o lib.so -buildmode=c-shared main.go

我现在有一个lib.so 和一个lib.h

打印数组值的 C 代码是什么？

#include <stdio.h>
#include "lib.h"

typedef struct {
  int   A;
  char* B;
} T;

int main() {
  GoSlice a = Test();
  for (int i = 0; i < 2; i++){
    printf("%s\n", ((T *)a.data)[i].B);
  }
}

gcc -o main main.c ./lib.so

Answer 1

首先，C 不知道 Go 切片类型，也不知道如何操作它，因此您应该像在 C 中一样返回一个数组指针。

你也不能在 Go 中分配切片然后返回一个指向已分配内存的指针，而没有某种方法可以防止它被 GC'ed。在此示例中，仅在 C 中分配数组可能更简单。

p := C.malloc(2 * C.size_t(unsafe.Sizeof(T{})))

// convert the C pointer to a slice of `[]T` for convenient indexing
arr := unsafe.Slice((*T)(p), 2)
arr[0].A = C.int(1)
arr[0].B = C.CString("a")
arr[1].A = C.int(2)
arr[1].B = C.CString("abc")

您当然还需要处理数组大小，您可以像处理字符串一样将其组合成一个结构体，或者接受第二个指针参数来设置返回大小。

不要忘记，由于您在 C 中分配内存，因此您也有责任释放它。这适用于malloc 调用以及C.CString 数组。

Answer 2

在 C 中使用 Go 切片

---

这是一个 C 打印 Go byte 切片的示例：

package main

/*
#include <stdio.h>

void printbuf(size_t len, unsigned char *buf) {
    printf("%lu [", len);
    if (!buf) {
        len = 0;
    }
    size_t maxwidth = 16;
    size_t width = len <= maxwidth ? len : maxwidth;
    for (size_t i = 0; i < width; i++) {
        if (i > 0) {
            printf(" ");
        }
        printf("%02X", buf[i]);
    }
    if (width < len) {
        printf(" ...");
    }
    printf("]\n");
}
*/
import "C"

func cbuf(buf []byte) (size C.size_t, ptr *C.uchar) {
    var bufptr *byte
    if cap(buf) > 0 {
        bufptr = &(buf[:1][0])
    }
    return C.size_t(len(buf)), (*C.uchar)(bufptr)
}

func main() {
    buf := make([]byte, 24, 32)
    for i := range buf {
        buf[i] = byte(i)
    }
    bufsize, bufptr := cbuf(buf)
    C.printbuf(bufsize, bufptr)
}

输出：

24 [00 01 02 03 04 05 06 07 08 09 0A 0B 0C 0D 0E 0F ...]

---

func Test() []T 对 C 有错误的返回值。返回一个长度和一个数组指针，例如 cbuf。