使用runtime.Pinner将Go二维数组传递给C函数的问题及优化咨询

1. C内存分配方案的安全性 #

你的C内存分配方案是否安全，取决于内存管理是否严谨：

• 安全的前提：
  • 所有在C侧分配的内存（包括输入的**float指针数组、输出的*double数组），必须在Go侧使用完成后调用C.free释放，避免内存泄漏。
  • 确保C函数不会保留输入/输出指针的长期引用——如果C函数在返回后还会访问这些指针，Go侧释放内存后会导致野指针错误。
  • 严格匹配数组长度，避免C函数越界读写内存。
• 风险点：如果遗漏释放C内存、计算长度错误导致越界，或者C函数非法持有指针，都会引发内存泄漏、崩溃等不安全问题。

2. 能否直接传递Go侧result的指针？ #

不能直接传递Go切片的底层数组指针给C函数的out *double参数，除非用runtime.Pinner固定内存。原因是Go的内存由GC自动管理，在C函数执行期间，GC可能移动或释放Go内存，导致C持有的指针失效，引发崩溃或数据损坏。

如果要复用Go内存，必须先通过runtime.Pinner固定输出切片的底层数组，再将指针传递给C函数。

3. runtime.Pinner的常见错误用法 #

你的代码报错大概率是以下原因之一：

• 未完整固定多层内存：对于[][]float32这类二维数组，需要固定两部分内存：
  1. 存储子切片指针的顶层数组（即[][]float32的底层指针数组）；
  2. 每个子切片的底层数据数组。
     如果只固定了子切片数据，顶层指针数组可能被GC移动，导致C侧的**float指向无效地址。
• 错误固定了切片结构体而非底层数组：p.Pin(slice)固定的是切片的元数据结构体（包含指针、长度、容量），而非底层数据数组。正确做法是固定底层数组的首地址：p.Pin(&slice[0])（切片非空时）。
• 提前调用Unpin()：如果在C函数执行完成前就调用p.Unpin()，内存会被GC解除固定，可能被移动或释放，导致C访问失效。
• 输出切片长度不足：如果C函数需要写入的元素数量超过Go切片的长度，会越界覆盖内存，引发崩溃。

正确使用示例：

import (
    "runtime"
    "unsafe"
)

func FooWrapper(input [][]float32) []float64 {
    // 假设已知输出长度，需根据实际逻辑调整
    outputLen := calculateOutputLen(input)
    output := make([]float64, outputLen)

    p := &runtime.Pinner{}
    defer p.Unpin()

    // 固定输出切片的底层数组
    if outputLen > 0 {
        p.Pin(&output[0])
    }

    // 构建C侧输入的**float指针数组，并固定所有子切片数据
    cInput := C.malloc(C.size_t(len(input)) * C.size_t(unsafe.Sizeof(uintptr(0))))
    defer C.free(unsafe.Pointer(cInput))
    inputPtrs := (*[1 << 30]*C.float)(cInput) // 用大数组占位避免越界

    for i, subSlice := range input {
        if len(subSlice) == 0 {
            inputPtrs[i] = nil
            continue
        }
        // 固定子切片的底层数组
        p.Pin(&subSlice[0])
        inputPtrs[i] = (*C.float)(unsafe.Pointer(&subSlice[0]))
    }

    // 调用C函数
    C.foo((**C.float)(cInput), (*C.double)(unsafe.Pointer(&output[0])))

    return output
}

// 需实现计算输出长度的逻辑
func calculateOutputLen(input [][]float32) int {
    // 示例逻辑，根据实际需求修改
    total := 0
    for _, sub := range input {
        total += len(sub)
    }
    return total
}

4. 高维数组（三维及以上）的传递方案（非拷贝） #

除了内存拷贝，有两种可行方案：

方案一：用runtime.Pinner多层固定内存 #

以三维数组[][][]float32为例：

1. 在C侧分配***float指针数组，每个元素指向一个二维数组的**float指针数组；
2. 每个二维数组的**float指针数组，元素指向对应一维子切片的底层数组；
3. 用runtime.Pinner固定所有层级的指针数组和最底层的数据数组；
4. 将顶层的***float指针传递给C函数。

这种方案严格匹配C函数的指针层级，但代码复杂度随维度增加而上升，需注意所有内存的固定和释放。

方案二：扁平化数组+维度信息传递 #

将高维数组扁平为一维数组，同时将各维度的长度传递给C函数，C函数内部通过维度计算索引访问元素：

• Go侧：将[x][y][z]float32扁平为[]float32，长度为x*y*z；
• C函数：修改为接受float*（扁平化数组指针）+ 各维度长度参数（如int dimX, int dimY, int dimZ）；
• C侧访问：通过index = i*y*z + j*z + k计算元素索引，直接操作扁平化数组。

这种方案无需处理多层指针，只需固定一维数组的内存，代码更简洁，效率更高，适合允许修改C函数接口的场景。

内容的提问来源于stack exchange，提问作者Denis319199