Python - ctypes и изменяемые буферы

Question

Я пытаюсь работать с ctypes, и я не могу получить вызов FormatMessage() для правильной работы.

Вот код, который у меня есть до сих пор; Я думаю, что единственная проблема заключается в передаче изменяемого буфера; Я получаю ArgumentError от ctypes о lpBuffer

import ctypes 
from ctypes.wintypes import DWORD 

def main(): 
    fm = ctypes.windll.kernel32.FormatMessageA 
    fm.argtypes = [DWORD,DWORD,DWORD,DWORD,ctypes.wintypes.LPWSTR(),DWORD] 

    dwFlags = DWORD(0x1000) # FORMAT_MESSAGE_ALLOCATE_BUFFER |FORMAT_MESSAGE_FROM_SYSTEM 
    lpSource = DWORD(0) 
    dwMessageId = DWORD(0x05) 
    dwLanguageId = DWORD(0) 
    #buf = ctypes.wintypes.LPWSTR() 
    #lpBuffer = ctypes.byref(buf) 
    lpBuffer = ctypes.create_string_buffer(512) 
    nSize = DWORD(512) 

    res = fm(dwFlags,lpSource,dwMessageId,dwLanguageId,lpBuffer,nSize) 
    print res 

Я получаю сообщение об ошибке на аргумент lpBuffer говоря, что это неправильный тип, но я пробовал так много вариантов прохождения в буфере, как я мог думать из. Я пробовал сделать это аналогично здесь: https://gist.github.com/CBWhiz/6135237 и установить FORMAT_MESSAGE_ALLOCATE_BUFFER, а затем передать в файле LPWSTR() byref, я также попытался изменить argtype, указатель и кастинг на различные файлы LPWSTR(), c_char_p и т. Д., Но неважно что я делаю, он продолжает жаловаться.

Каков правильный синтаксис для правильной работы функции? Я знаю, что ctypes может быть finnicky, но я ничего не нашел в документации, чтобы решить проблему (я знаю, что документация использует прототип(), но я хотел бы сделать это таким образом в настоящее время)

Благодаря

Answer 1

Вот argtypes определение FormatMessageW (примечание «W» для Unicode):

import ctypes 
from ctypes import wintypes 

fm = ctypes.windll.kernel32.FormatMessageW 
fm.argtypes = [ 
    wintypes.DWORD, # dwFlags 
    wintypes.LPCVOID, # lpSource 
    wintypes.DWORD, # dwMessageId 
    wintypes.DWORD, # dwLanguageId 
    wintypes.LPWSTR, # lpBuffer 
    wintypes.DWORD, # nSize 
    wintypes.LPVOID, # Arguments (va_list *) 
] 

FORMAT_MESSAGE_ALLOCATE_BUFFER = 0x100 
FORMAT_MESSAGE_FROM_SYSTEM = 0x1000 

Если FormatMessage выделяет буфер, вы должны вместо этого передать ссылку на lpBuffer. Просто cast ссылка, чтобы обойти TypeError. Кроме того, не забудьте позвонить kernel32.LocalFree чтобы освободить буфер:

def main(): 
    dwFlags = FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ALLOCATE_BUFFER 
    lpSource = None 
    dwMessageId = 5 
    dwLanguageId = 0  
    lpBuffer = wintypes.LPWSTR() 
    nSize = 0 # minimum size 
    Arguments = None 

    if not fm(dwFlags, lpSource, dwMessageId, dwLanguageId, 
       ctypes.cast(ctypes.byref(lpBuffer), wintypes.LPWSTR), 
       nSize, Arguments): 
     raise ctypes.WinError() 

    msg = lpBuffer.value.rstrip() 
    ctypes.windll.kernel32.LocalFree(lpBuffer) 

    return msg