ctypes를 사용하여 바이트 및 바이트 객체를 모두 외부 라이브러리로 효율적으로 전달할 수 있습니까?

Question

내가 외부 라이브러리에 다음 함수가 있다고 가정 :

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

내가 버퍼의 복사본을 만들지 않고, ctypes를 사용하여 내 파이썬 코드에서이 함수를 호출 할 수 있도록하고 싶습니다

. 버퍼가 상당히 클 수 있으므로 사본을 피하는 것이 명백한 성능 이점을 갖습니다. 그리고 내 코드의 소비자의 편의를 위해이 버퍼를 bytes 또는 bytearray으로 제공 할 수 있기를 바랍니다.

지금 으로 buf을 선언합니다. 내 argtypes 선언에.

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

잘 작동하며 bytes 개체를 전달할 수 있습니다. 나는 다음 bytearray 개체를 전달할 경우, 나는 다음과 같은 오류가 발생 :

ctypes.ArgumentError: argument 1: : wrong type

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] 

bytearray 위해 만들어지는 c_char 배열은 파이썬 버퍼 프로토콜을 통해 물체의 내부 버퍼를 얻을 필요가있다. 배열 크기는 중요하지 않으므로 bytearray의 가능한 모든 길이에 대해 배열 하위 클래스를 만들지 않아도됩니다. from_param 인수 목록에 캐시 된 길이 0 배열 유형을 사용하기 만하면됩니다.