是否可以使用ctypes有效地将字节和字节数组对象传递给外部库？

Question

假设我在外部库中具有以下功能：

void foo(const unsigned char *buf, const int len);

我希望能够使用ctypes从我的Python代码中调用此函数，而无需复制缓冲区。 缓冲区可能很大，因此避免复制会带来明显的性能优势。 为了方便我的代码使用者，我希望能够以bytes或bytearray提供此缓冲区。

目前，我在argtypes声明ctypes.POINTER(ctypes.c_char) buf声明为ctypes.POINTER(ctypes.c_char) 。

lib.foo.argtypes = [ctypes.POINTER(ctypes.c_char), ctypes.c_int]
buf = bytes(...)
lib.foo(buf, len(buf))

这工作正常，我可以传递一个bytes对象。 但是，如果传递bytearray数组对象，则会遇到以下错误：

ctypes.ArgumentError：参数1：：错误的类型

有没有办法让我可以传递bytes bytearray ，最好与bytes互换？

Answer 1

您可以创建一个覆盖from_param的指针类型的子类，以适应bytearray 。 例如：

class Pchar(ctypes.POINTER(ctypes.c_char)):
    _type_ = ctypes.c_char
    @classmethod
    def from_param(cls, param, array_t=ctypes.c_char * 0):
        if isinstance(param, bytearray):
            param = array_t.from_buffer(param)
        return super(Pchar, cls).from_param(param)

lib.foo.argtypes = [Pchar, ctypes.c_int]

所述c_char一个专门针对所创建的阵列bytearray只需要得到在对象的经由Python的缓冲协议内部缓冲器。 数组大小无关紧要，因此我们可以避免为bytearray每个可能长度创建一个数组子类。 只需使用from_param参数列表中缓存的长度为0的数组类型。