使用runtime.Pinner将二维数组从Go传递到C

问题内容

我试图将二维数组从 Go 传递到某个 C 函数 void foo(in **float, out *double)。因为我想要这个 C 函数的包装器，所以我希望 Go 函数具有像 func FooWrapper([][]float32) []float64 这样的定义。最简单但效率不高的实现是通过下面列出的 C 来分配所有内存：

func FooWrapper(values [][]float32) []float64 {
    totalObj := len(values)
    totalParams := len(values[0])
    results := make([]float64, totalObj)

    ptrArrLength := uintptr(totalObj) * cPointerSize
    paramArrLength := uintptr(totalParams) * cFloatSize

    ptr := C.malloc(C.size_t(ptrArrLength + paramArrLength*uintptr(totalObj)))
    defer C.free(ptr)

    ptrSlice := unsafe.Slice((**C.float)(ptr), totalObj)
    for i, obj := range values {
        paramArrPtr := (*C.float)(unsafe.Add(ptr, ptrArrLength+uintptr(i)*paramArrLength))
        ptrSlice[i] = paramArrPtr

        paramSlice := unsafe.Slice(paramArrPtr, totalParams)
        for j, param := range obj {
            paramSlice[j] = (C.float)(param)
        }
    }

    C.foo((**C.float)(ptr), (*C.double)(&results[0]))

    return results
}

这样的实施安全吗？我可以传递result数据的指针吗？据我所知，这个指针将被固定，因为它传递给了 C 函数。

但是我想分配更少的内存，只是重用Go内存，我已经了解了 runtime.Pinner ，它使指针固定直到 runtime.Pinner.Unpin() 调用。我尝试使用 pinner 编写另一个实现：

func FooWrapper(values [][]float32) []float64 {
    length := len(values)
    results := make([]float64, length)

    pinner := runtime.Pinner{}
    defer pinner.Unpin()

    arr := (**C.float)(C.malloc(C.size_t(uintptr(length) * cPointerSize)))
    defer C.free(unsafe.Pointer(arr))
    slice := unsafe.Slice(arr, length)
    for i, v := range values {
        pinner.Pin(&v[0])
        slice[i] = (*C.float)(&v[0])
    }

    C.foo(arr, (*C.double)(&results[0]))

    return results
}

但是，不幸的是，这段代码不起作用

runtime: pointer 0xc016ecbfc0 to unused region of span span.base()=0xc016eca000 span.limit=0xc016ecbfa0 span.state=1
fatal error: found bad pointer in Go heap (incorrect use of unsafe or cgo?)

我使用runtime.Pinner是否错误（据我所知，我可以固定切片数据）？或者这段代码还有另一个错误。除了将所有数据分配和复制到 C 内存之外，是否有一些将 3d（4d 等）数组传递给 C 函数的实现？

解决方法

我找到了问题的答案。使用 malloc 是危险的，因为当我尝试将 Go 指针写入 C 内存时，GC 可以将未初始化的内存识别为“坏”内存，因此如果我将 malloc 更改为 calloc 就不会出现问题。当然这不是最好的解决方案，最好在 Go 中分配内存，因为它更快、更安全。

所有讨论此处一个>.

以上就是使用runtime.Pinner将二维数组从Go传递到C的详细内容，更多请关注编程网其它相关文章！