Início / coding / Perguntas / 79564149
Accepted
Dominik Kaszewski
Asked: 2025-04-09 18:52:04 +0800 CST

Codificar/decodificar automaticamente argumentos ctype e valores de retorno

  • 772

Tenho uma biblioteca compartilhada expondo uma função:

const char* foo(const char* s);

Para chamá-lo do Python usando ctypes, posso fazer o seguinte:

import ctypes

foo = ctypes.cdll.LoadLibrary('foo.so')
foo.foo.argtypes = [ctypes.c_char_p]
foo.foo.restype = ctypes.c_char_p

print(foo.foo('Hello World!'.encode('utf8')).decode('utf8'))

Existe uma maneira de incorporar o texto encode/decode('utf8')dentro do próprio tipo? Quero fazer o seguinte:

foo.foo.argtypes = [c_utf8_p]
foo.foo.restype = c_utf8_p

print(foo.foo('Hello World!'))

Consegui lidar argtypescom o seguinte:

class c_utf8_p:
    @classmethod
    def from_param(cls, obj: str):
        return obj.encode('utf8')

Mas não consigo descobrir o tipo de retorno. restypeA documentação diz que atribuí-lo a qualquer coisa que não seja um ctypestipo está obsoleto, além de truncar o valor, into que destrói ponteiros de 64 bits. Gostaria de evitar o uso, errcheckpois, logicamente, não estou verificando nenhum erro, além de exigir a definição de dois atributos para cada função.

python

1 respostas

  1. Best Answer

    Não há uma maneira documentada de fazer o que você quer com .restype. Acho que você tem duas opções:

    1. Use .errcheck, pois permite acesso ao valor de retorno real e pode retornar uma versão modificada. Uma vantagem é que ele define apenas um atributo extra por função, em comparação com o nº 2...

    2. Função wrapper. Seria necessária para cada função que precisasse de pós-processamento.

    Exemplos:

    // test.c - cl /LD test.c
#include <stdlib.h>
#include <string.h>

__declspec(dllexport)
const char* foo(const char* s) {
    char* ret = malloc(strlen(s) * 2 + 1);
    strcpy(ret, s);
    strcat(ret, s);
    return ret;
}

__declspec(dllexport)
void free_foo(const char* s) {
    free((void*)s);
}

    Usando .errcheck:

    import ctypes as ct

class c_utf8_p:
    @classmethod
    def from_param(cls, obj):
        return obj.encode()

def decode_bytes(result, func, args):
    s = ct.string_at(result).decode()
    dll.free_foo(result)
    return s

dll = ct.CDLL('./test')
foo = dll.foo
dll.foo.argtypes = c_utf8_p,
foo.restype = ct.POINTER(ct.c_char)
foo.errcheck = decode_bytes
free_foo = dll.free_foo
free_foo.argtypes = ct.POINTER(ct.c_char),
free_foo.restype = None

s = foo('Hello World!')
print(type(s), s)

    Saída:

    <class 'str'> Hello World!Hello World!

    Usando um wrapper:

    import ctypes as ct

dll = ct.CDLL('./test')
_foo = dll.foo
_foo.argtypes = ct.c_char_p,
_foo.restype = ct.POINTER(ct.c_char)
dll.free_foo.argtypes = ct.POINTER(ct.c_char),
dll.free_foo.restype = None

def foo(s):
    p = _foo(s.encode())
    s = ct.string_at(p).decode()
    dll.free_foo(p)
    return s

s = foo('Hello World!')
print(type(s), s)

    Saída:

    <class 'str'> Hello World!Hello World!
    • 1

relate perguntas