Ist es möglich, sowohl Bytes als auch Bytearray-Objekte mit ctypes effizient an eine externe Bibliothek zu übergeben?

Question

Angenommen, ich die folgende Funktion in einer externen Bibliothek habe:

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

ich diese Funktion nutzen können anrufen möchte von meinem Python-Code, ctypes verwenden, ohne eine Kopie des Puffers zu machen. Der Puffer könnte ziemlich groß sein, so dass das Vermeiden einer Kopie offensichtliche Leistungsvorteile hat. Und für den Komfort des Verbrauchers meines Codes würde ich gerne in der Lage sein, diesen Puffer entweder als bytes oder bytearray zu liefern.

Im Moment erkläre ich buf als ctypes.POINTER(ctypes.c_char) in meiner argtypes Deklaration.

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

Dies funktioniert gut, und ich kann ein bytes Objekt übergeben. Wenn ich jedoch ein bytearray Objekt übergeben dann begegne ich den folgenden Fehler:

ctypes.ArgumentError: argument 1: : wrong type

Gibt es für mich einen Weg zu ermöglichen bytearray vorzugsweise mit bytes austauschbar übergeben, werden?

Answer 1

Sie können eine Unterklasse des Zeigertyps erstellen, die from_param überschreibt, um eine bytearray anzupassen. Zum Beispiel:

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] 

Die c_char Array, das für die bytearray erstellt werden nur an dem internen Puffer des Objekts über Python-Puffer-Protokoll erhalten benötigt. Die Array-Größe spielt keine Rolle, daher können wir die Erstellung einer Array-Unterklasse für jede mögliche Länge von bytearray vermeiden. Verwenden Sie einfach einen Array-Typ Länge 0, der in der Argumentliste from_param zwischengespeichert wird.