卓越飞翔博客卓越飞翔博客

卓越飞翔 - 您值得收藏的技术分享站
技术文章35491本站已运行396

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

使用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 中分配内存,因为它更快、更安全。

所有讨论此处一个>.

卓越飞翔博客
上一篇: 如何在Golang中创建从开始到结束的YYYY-MM-DD格式的日期范围?
下一篇: 返回列表
留言与评论(共有 0 条评论)
   
验证码:
隐藏边栏