c++ - 带 DWORD 的 DeviceIoControl

Question

我的操作系统是 Win7 64 位。我正在尝试通过 DeviceIoControl 将变量的值而不是它的地址(进程 ID，DWORD)传递给我的驱动程序。试了好几次，都是蓝屏或者错误码998。

在用户模式方面，我确保 hDevice 有效，CTL_CODE 如下所示:

#define SENDPID CTL_CODE(FILE_DEVICE_UNKNOWN, 0x808, METHOD_BUFFERED, FILE_READ_DATA | FILE_WRITE_DATA)

然后我这样做:

DWORD pID = FindProcessName("someprocess.exe");
if (!pID) return false;
if (!DeviceIoControl(hDevice, SENDPID, &pID, sizeof(pID), NULL, 0, &BytesIO, 0))
{
    cout << GetLastError() << endl;
    return false;
}

在内核模式下(DriverEntry):

NTSTATUS DriverEntry(PDRIVER_OBJECT Object, PUNICODE_STRING RegistryPath)
{
UNICODE_STRING dNUS = { 0 };
RtlInitUnicodeString(&dNUS, L"\\Device\\testdriver");

UNICODE_STRING dSLU = { 0 };
RtlInitUnicodeString(&dSLU, L"\\DosDevices\\testdriver");

IoCreateDevice(Object, 0, &dNUS, FILE_DEVICE_UNKNOWN, FILE_DEVICE_SECURE_OPEN, FALSE, &deviceObj);
IoCreateSymbolicLink(&dSLU, &dNUS);

Object->MajorFunction[IRP_MJ_CREATE] = CCreate;
Object->MajorFunction[IRP_MJ_CLOSE] = CClose;
Object->MajorFunction[IRP_MJ_DEVICE_CONTROL] = IOCTL;
Object->DriverUnload = Unload;

return(STATUS_SUCCESS);
}

和我的 IOCTL 函数:

NTSTATUS IOCTL(PDEVICE_OBJECT Object, PIRP IRP)
{
PIO_STACK_LOCATION StackLocation = IoGetCurrentIrpStackLocation(IRP);
size_t size = 0;
DWORD pID = 0;

if (StackLocation->Parameters.DeviceIoControl.IoControlCode == SENDPID)
{
    pID = IRP->AssociatedIrp.SystemBuffer;
    size = sizeof(pID);
}

IRP->IoStatus.Status = STATUS_SUCCESS;
IRP->IoStatus.Information = size;
IofCompleteRequest(IRP, IO_NO_INCREMENT);
return(STATUS_SUCCESS);
}

如果有人能告诉我要更改什么，请告诉我:)

Answer 1

在应用端，为什么 FindProcessName() 返回一个 DWORD* 指针而不是一个 DWORD 值？或者，它是否真的返回一个 DWORD 值，而您只是错误地将其类型转换为 DWORD* 指针？

通过将 pID 定义为 DWORD* 指针，sizeof(pID) 是传递给 DeviceIoControl() 的错误值。它传递的是指针本身的大小(32 位为 4，64 位为 8)，而不是所指向的 DWORD 值的大小(sizeof(DWORD) 总是4).

如果 pID 是指向内存中 DWORD 的真实指针，则需要使用 sizeof(*pID) 而不是 大小(pID)。但是，如果 FindProcessName() 返回 0 怎么办？然后你最终会传递一个 NULL 指针到 DeviceIoControl()，即使你告诉 DeviceIoControl() 你传递了 4 个字节。

您应该更改 FindProcessName() 以返回一个 DWORD 值(如果它还没有)而不是 DWORD* 指针，然后使用 & 运算符将该 DWORD 的地址传递给 DeviceIoControl()，例如:

DWORD pID = FindProcessName("DayZ.exe");
if (!DeviceIoControl(hDevice, SENDPID, &pID, sizeof(pID), NULL, 0, &BytesIO, 0))

然后在内核端，只需对 SystemBuffer 进行类型转换即可访问该值(当然，在验证 SystemBuffer 至少包含 4 个字节之后):

NTSTATUS IOCTL(PDEVICE_OBJECT Object, PIRP IRP)
{
    PIO_STACK_LOCATION StackLocation = IoGetCurrentIrpStackLocation(IRP);
    size_t size = 0;
    DWORD pID = 0;

    if ((StackLocation->Parameters.DeviceIoControl.IoControlCode == SENDPID) &&
        (StackLocation->Parameters.DeviceIoControl.InputBufferLength >= sizeof(pID)))
    {
        pID = *(DWORD*)(IRP->AssociatedIrp.SystemBuffer);
        size = sizeof(pID);
    }

    IRP->IoStatus.Status = STATUS_SUCCESS;
    IRP->IoStatus.Information = size;
    IofCompleteRequest(IRP, IO_NO_INCREMENT);

    return STATUS_SUCCESS;
}