一尘不染

将Go [] byte转换为C * char

go

我有一个byte.Buffer,我使用binary.Write()函数包装了数据。然后,我需要将此字节数组发送到C函数。使用Go
1.6时,我未能成功解决这一问题。

buf := new(bytes.Buffer) //create my buffer
....
binary.Write(buf, binary.LittleEndian, data) //write my data to buffer here
addr := (*C.uchar)(unsafe.Pointer(&buf.Bytes()[0])) //convert buffers byte array to a C array
rc := C.the_function(addr, C.int(buf.Len())) //Fails here

它在调用C函数的行上失败:

panic: runtime error: cgo argument has Go pointer to Go pointer

C函数:

int the_function(const void *data, int nbytes);

我能够使以下内容正常工作,但将字节数组转换为字符串感觉不对。有一个更好的方法吗?这种方法是否会对数据产生副作用?

addr := unsafe.Pointer(C.CString(string(buf.Bytes()[0]))

同样,这需要在Go 1.6(引入了更严格的cgo指针规则)下工作。

谢谢。


阅读 929

收藏
2020-07-02

共1个答案

一尘不染

如果要使用第一种方法,则需要在函数调用参数之外创建切片,并避免临时分配的切片标头或参数中的外部结构,因此cgo检查不会将其视为存储在Go中的指针。

b := buf.Bytes()
rc := C.the_function(unsafe.Pointer(&b[0]), C.int(buf.Len()))

C.CString方法将更安全,因为数据被复制到C缓冲区中,因此没有指向Go内存的指针,并且没有机会bytes.Buffer修改后面的切片或超出范围。您将要转换整个字符串,而不仅仅是第一个字节。这种方法确实需要分配和复制两次,但是,如果数据量很小,则与cgo调用本身的开销相比,这可能不是问题。

str := buf.String()
p := unsafe.Pointer(C.CString(str))
defer C.free(p)
rc = C.the_function(p, C.int(len(str)))

如果该解决方案中不接受数据的2个副本,则存在第三个选项,您可以自己分配C缓冲区,然后将单个副本复制到该缓冲区中:

p := C.malloc(C.size_t(len(b)))
defer C.free(p)

// copy the data into the buffer, by converting it to a Go array
cBuf := (*[1 << 30]byte)(p)
copy(cBuf[:], b)
rc = C.the_function(p, C.int(buf.Len()))

但是,使用这两个选项,不要忘记释放malloc的指针。

2020-07-02