python中使用ctypes调用so传参设置遇到的问题及解决方法
时间：2021-01-31 08:12:13|栏目：Python代码|点击： 次
问题
近日在做一组声纹聚类时，使用了另一团队同学开发的声纹距离算法。该算法对外提供的是一组so包，需要使用方自己去使用。在python中调用纯so包一般使用ctypes类库，用起来看起来简单但也有不少细节容易犯错。本次使用过程中，就遇到传参的问题。
目标so库中对外export的函数是大致如下的三个函数:
void* create_handler(); int extract_feature(void* hander); bool destroy(void* handler);
这三个函数使用起来倒也简单，顺序使用就可以了。但发现写成如下形式的python代码后，执行会直接segment fault。
import sys
import ctypes
so = ctypes.CDLL("./lib/libbase.so")
p = so.create_handler()
feature = so.extract_feature(p)
so.destroy(p)
解决
这段代码中p是int类型，由void*自动转来，在ctyeps中这种转型本身是没问题的。segment fault发生在extract_feature函数调用中，问题应当出在参数上，回传的handler已经不是原来的pointer了，导致访问指针出错。
查阅ctypes的文档后，发现ctypes可以声明so库中函数的参数，返回类型。试了试，显示声明后问题得到了解决，证明我们的猜想是对的，确实指针发生了变化。修改后代码如下:
import sys
import ctypes
so = ctypes.CDLL("./lib/libbase.so")
so.create_handler.restype=ctypes.c_void_p
so.extract_feature.argtypes=[ctypes.c_void_p]
so.destroy.argtypes=[ctypes.c_void_p]
p = so.create_handler()
feature = so.extract_feature(p)
so.destroy(p)
结论:
ctypes中传递指针类型参数需要显示声明c函数的参数，返回类型。
总结
