ctypes与cgo交互时的字符串所有权问题（Python调用Go生成的DLL场景）

背景 #

我最近在Python里调用自己用Go生成的DLL，为了调试时做个合理性验证（sanity check），特意加了个return_string()函数——它接收一个C风格的字符串指针（char*），转成Go字符串后再返回结果。主要就是用这个工具排查两种语言之间的编码问题。

具体问题 #

我已经在Python里把这个函数封装好了，大部分功能都正常，不过现在碰到了字符串所有权的棘手问题：当Go返回字符串给Python后，我不知道这块内存该由谁来负责释放，处理不好很容易出现内存泄漏或者野指针；而且偶尔返回的字符串还会乱码，怀疑和内存管理、编码转换的细节有关。

核心问题拆解 #

在Go和Python通过ctypes交互时，字符串的内存管理是最大的坑：

• Python传字符串给Go时，内存是Python管理的，Go不能随意修改或释放；
• Go返回字符串给Python时，要是用C.CString()生成的char*，这块内存是在C堆上的，Go的GC管不到，必须手动处理。

解决方案 #

1. 正确处理Go接收Python字符串的逻辑

在Go的cgo代码里，直接用C.GoString()把char*转成Go字符串就行——它会拷贝一份内容到Go的内存空间，原Python的内存完全不受影响，不用担心内存冲突：

//export return_string
func return_string(cStr *C.char) *C.char {
    // 把C字符串转成Go字符串，自动拷贝内容
    goStr := C.GoString(cStr)
    // 这里可以对goStr做任意处理，比如修改、编码校验等
    // 再把处理后的Go字符串转成C风格字符串返回
    return C.CString(goStr)
}

2. 妥善处理Go返回字符串的内存释放

这里有两种可靠的做法：

方案A：Python手动释放内存

在Python里调用完return_string()后，一定要用ctypes.free()释放返回的指针，避免内存泄漏：

import ctypes

# 加载Go生成的DLL
my_go_dll = ctypes.CDLL("./my_dll.dll")
# 明确函数的参数和返回值类型，避免类型错误
my_go_dll.return_string.argtypes = [ctypes.c_char_p]
my_go_dll.return_string.restype = ctypes.c_char_p

# 准备输入字符串，必须转成UTF-8字节串
input_str = "测试交互字符串".encode("utf-8")
# 调用DLL函数
result_ptr = my_go_dll.return_string(input_str)
# 读取指针指向的内容并转成Python字符串
result_str = ctypes.string_at(result_ptr).decode("utf-8")
# 关键：手动释放C堆上的内存
ctypes.free(result_ptr)

print("返回结果：", result_str)

方案B：Go提供专门的释放函数

如果担心Python那边忘记释放，可以在Go里加一个释放C字符串的函数，让Python调用：

//export free_cstring
func free_cstring(cStr *C.char) {
    // 用C的free函数释放内存
    C.free(unsafe.Pointer(cStr))
}

然后在Python里调用完返回函数后，调用这个释放函数：

# 先定义释放函数的参数类型
my_go_dll.free_cstring.argtypes = [ctypes.c_char_p]
# 调用释放函数
my_go_dll.free_cstring(result_ptr)

编码避坑提示 #

• Python传字符串时，必须用encode("utf-8")转成字节串，和Go默认的UTF-8编码对齐，避免乱码；
• Go返回的字符串，Python要用ctypes.string_at()读取原始字节，再用decode("utf-8")转成Python字符串，不要直接强转指针。

内容来源于stack exchange