Как определить длину слова операционной системы во время выполнения на C++?

Question

У меня есть программа на C++, которая скомпилирована для x86 (32 бит). Он вызывает драйвер режима ядра. Драйвер скомпилирован для того, чтобы размер слова операционной системы был запущен. Целевые операционные системы могут быть 32 или 64 бит (в моем случае окна).

Моя проблема заключается в определении размера указателя, возвращаемого драйвером, поскольку это необходимо вызову ОС.

Следующий вызов программы пользовательского режим работал, если система имела размер слова 32 бита:

HANDLE device = OpenDevice(); 
HANDLE packageReceivedEvent = CreateEvent(NULL, FALSE, FALSE, NULL); 
bool result = DeviceIoControl(
    device, 
    IOCTL_CODE, 
    &packageReceivedEvent, //for signaling 
    sizeof(HANDLE), //TODO does not work for 64 bit 
    nullptr, 
    0, 
    &recvBytes, 
    nullptr); 

Определения приходят от стандартного WinBase.h и winnt.h.

Я не могу использовать какие-либо решения времени компиляции, такие как sizeof(int), поскольку они будут рассматривать программу пользовательского режима, а не компиляцию драйвера, в зависимости от.

Возможно, было бы достаточно установить DeviceIoControlnInBufferSize просто наивысшим ожидаемым размером слова, но есть ли более приятное решение?

Answer 1

вам нужен код, как этот

union { 
    __int64 v; 
    HANDLE packageReceivedEvent; 
}; 
v = 0; 
BOOL fOk = FALSE; 
if (packageReceivedEvent = CreateEvent(NULL, FALSE, FALSE, NULL)) 
{ 
    BOOL Wow64Process; 
    if (IsWow64Process(NtCurrentProcess(), &Wow64Process)) 
    { 
     fOk = DeviceIoControl(
      device, 
      IOCTL_CODE, 
      &packageReceivedEvent, //for signaling 
      Wow64Process ? 8 : 4, 
      nullptr, 
      0, 
      &recvBytes, 
      nullptr); 
    } 
} 
ULONG err = fOk ? NOERROR : GetLastError(); 

, а не «длина слова» (это всегда 2 байта), а длина указателя, который составляет 4 байта для 32-битного кода и 8 байт для 64-битных кода. драйверы всегда должны быть «родными», поэтому только 64-битные драйверы могут запускаться в 64-битных окнах, когда приложения пользовательского режима могут быть 32 или 64 бит. для определения его во время выполнения - используйте IsWow64Process. некоторые драйверы допускают только 64-битные структуры компоновки, некоторые могут определять 32-битный процесс по вызову IoIs32bitProcess и ждать 32-битных структур компоновки из 32-битного процесса. очевидно, ваш драйвер принимает только 64-битные структуры компоновки. так что вам нужно передать 8 или 4 байта, как размер входного сигнала, зависит от окон 64 или 32

---

Вероятно, было бы достаточно, чтобы установить DeviceIOControl в nInBufferSize просто до самого высокого ожидаемого размера слова, но есть лучше решение?

это не решение:

• в 32 битной системе водитель может проверить, что InputBufferLength == sizeof(HANDLE) (и вернуть STATUS_INFO_LENGTH_MISMATCH если ложь) или что InputBufferLength >= sizeof(HANDLE)
• в 64 битной системе, если вы просто сказать, что InputBufferLength == 8, но какие данные будут в 32 битных входных буфера? в своем коде вы инициализации только низкое 32bits по packageReceivedEvent = CreateEvent(NULL,FALSE, FALSE, NULL); - так что вам нужно выделить 8 байты буфера первым и правильно INIT это