将 ByteArray 从 Python 传递给 C 函数

Question

我想将一个 ByteArray 变量从我的 Python 程序传递给我用 C 编写的 DLL，以加速一些在 Python 中太慢的特定处理。我浏览了 Web，尝试了结合 byref、cast、memoryviews、addressof 的 Ctypes，但没有任何效果。有没有什么简单的方法可以实现这一点，而无需将我的 ByteArray 复制到其他可以通过的东西中？ 这是我想要做的：

/* My C DLL */
__declspec(dllexport) bool FastProc(char *P, int L)
{
    /* Do some complex processing on the char buffer */
    ;
    return true;
}

# My Python program
from ctypes import *
def main(argv):
    MyData = ByteArray([1,2,3,4,5,6])
    dll = CDLL('CHELPER.dll')
    dll.FastProc.argtypes = (c_char_p, c_int)
    dll.FastProc.restype = c_bool

    Result = dll.FastProc(MyData, len(MyData))
    print(Result)

但是在将第一个参数 (MyData) 传递给 C 函数时出现类型错误。

是否有任何不需要太多开销的解决方案会浪费我的 C 函数的好处？

奥利维尔

Answer 1

我假设ByteArray 应该是bytearray。我们可以使用create_string_buffer 创建一个可变字符缓冲区，它是ctypes 数组c_char。但是create_string_buffer 将不接受bytearray，我们需要传递一个bytes 对象来初始化它；幸运的是，在bytes 和bytearray 之间进行转换既快速又高效。

我没有你的 DLL，所以为了测试数组的行为是否正确，我将使用 libc.strfry 函数来打乱它的字符。

from ctypes import CDLL, create_string_buffer

libc = CDLL("libc.so.6") 

# Some test data, NUL-terminated so we can safely pass it to a str function.
mydata = bytearray([65, 66, 67, 68, 69, 70, 0])
print(mydata)

# Convert the Python bytearray to a C array of char
p = create_string_buffer(bytes(mydata), len(mydata))

#Shuffle the bytes before the NUL terminator byte, in-place.
libc.strfry(p)

# Convert the modified C array back to a Python bytearray
newdata = bytearray(p.raw)
print(newdata)

典型输出

bytearray(b'ABCDEF\x00')
bytearray(b'BFDACE\x00')