Quel est le moyen le plus efficace pour copier un tampon fourni en externe en octets?

Question

Je m'interface avec une bibliothèque externe en utilisant des ctypes. Cette bibliothèque me renvoie un tampon binaire. L'interface ressemble à ceci:

int getBuff(unsigned char **buf, int *len); 

La bibliothèque exporte aussi deallocator pour que je puisse libérer le tampon quand je suis fait avec elle, mais cet aspect me pose pas de problème, donc je ne pense pas que nous avons besoin pour le couvrir. Dans mon code ctypes, je représente l'argument buf comme c_void_p. Je voudrais copier ce tampon dans un objet octets aussi efficacement que possible.

Au moment j'ai:

data = bytes(bytearray(ctypes.cast(buf, ctypes.POINTER(ctypes.c_ubyte*len.value))[0])) 

où buf est c_void_p et len est c_int. Si je comprends bien, cela fait deux copies. Une fois à l'objet bytearray, puis de nouveau à l'objet bytes.

Comment puis-je faire cela avec une seule copie? Mes efforts actuels se sont concentrés sur Python 2, mais en temps voulu, je devrai également prendre en charge Python 3 pour Python 3.

Answer 1

Apparemment, vous pouvez découper un pointeur ctypes. Pas c_void_p, c_char_p, ou c_wchar_p, mais POINTER types fonctionnent. Pour une POINTER(c_char), tranchage il vous donne bytes:

data = ctypes.POINTER(ctypes.c_char).from_buffer(buf)[:len.value] 

Merci à eryksun avoir soulevé cette question. En outre, il n'est pas clair pourquoi buf est un c_void_p au lieu d'être déjà POINTER(c_char). (Pour une POINTER(c_char), le code serait juste buf[:len.value].)

---

Pour obtenir bytes à partir d'un objet général qui prend en charge le protocole de la mémoire tampon, memoryview(...).tobytes() implique une moins copie de bytes(bytearray(...)):

data = memoryview(ctypes.cast(buf, ctypes.POINTER(ctypes.c_ubyte*len.value))[0]).tobytes() 

Ceci est compatible avec Python 2 et Python 3.

---

Gardez à l'esprit que le buf ici doit être un pointeur vers le tampon, pas un pointeur vers un pointeur vers le tampon. getBuff prend un pointeur vers un pointeur (donc probablement byref(buf)).