4. 内核模式与用户模式通信：IOCTL控制码与缓冲区方式

在Windows驱动开发中，用户模式应用程序与内核模式驱动程序之间的数据交换是最核心的交互场景。由于两种模式运行在不同的地址空间（用户态虚拟地址空间与内核态虚拟地址空间），直接指针传递会引发访问违规（Access Violation）。Windows通过IOCTL（I/O Control Code）控制码与缓冲区管理方式提供了标准化的通信机制。

4.1 IOCTL控制码的结构与定义

IOCTL控制码是一个32位无符号整数，由四个字段组成，用于唯一标识一个控制操作。驱动程序通过IRP_MJ_DEVICE_CONTROL主功能代码接收IOCTL请求。

定义IOCTL控制码的标准宏：

#define IOCTL_SAMPLE_READ \
    CTL_CODE(FILE_DEVICE_UNKNOWN,          // 设备类型
             0x801,                         // 功能码（自定义范围）
             METHOD_BUFFERED,               // 缓冲区方式
             FILE_READ_DATA)                // 访问权限

4.2 缓冲区方式：METHOD_BUFFERED

METHOD_BUFFERED是最常用、最安全的通信方式。I/O管理器在系统非分页池中分配一个中间缓冲区，用于在用户模式和内核模式之间复制数据。

驱动中处理METHOD_BUFFERED的典型代码：

NTSTATUS DispatchDeviceControl(PDEVICE_OBJECT DeviceObject, PIRP Irp) {
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation(Irp);
    ULONG ioctlCode = irpStack->Parameters.DeviceIoControl.IoControlCode;
    PVOID systemBuffer = Irp->AssociatedIrp.SystemBuffer;
    ULONG inputLen = irpStack->Parameters.DeviceIoControl.InputBufferLength;
    ULONG outputLen = irpStack->Parameters.DeviceIoControl.OutputBufferLength;

    switch (ioctlCode) {
    case IOCTL_SAMPLE_READ: {
        // 从系统缓冲区读取输入
        PULONG inputData = (PULONG)systemBuffer;
        // 处理逻辑...
        // 将结果写回系统缓冲区
        *((PULONG)systemBuffer) = result;
        Irp->IoStatus.Information = sizeof(ULONG);
        break;
    }
    default:
        Irp->IoStatus.Status = STATUS_INVALID_DEVICE_REQUEST;
        Irp->IoStatus.Information = 0;
        break;
    }
    IoCompleteRequest(Irp, IO_NO_INCREMENT);
    return Irp->IoStatus.Status;
}

4.3 直接方式：METHOD_IN_DIRECT / METHOD_OUT_DIRECT

直接方式适用于大数据量传输（如视频帧、音频流、磁盘读写）。I/O管理器将用户模式缓冲区锁定在物理内存中，并映射到内核地址空间，避免数据复制开销。

驱动中访问MDL的典型代码：

NTSTATUS DispatchDirectRead(PDEVICE_OBJECT DeviceObject, PIRP Irp) {
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation(Irp);
    PMDL mdl = Irp->MdlAddress;  // 输出缓冲区的MDL
    PVOID systemBuffer = Irp->AssociatedIrp.SystemBuffer;  // 输入参数

    // 从系统缓冲区读取输入参数
    ULONG offset = *((PULONG)systemBuffer);

    // 锁定MDL并获取内核映射地址
    PVOID mappedBuffer = MmGetSystemAddressForMdlSafe(mdl, NormalPagePriority);
    if (!mappedBuffer) {
        Irp->IoStatus.Status = STATUS_INSUFFICIENT_RESOURCES;
        IoCompleteRequest(Irp, IO_NO_INCREMENT);
        return STATUS_INSUFFICIENT_RESOURCES;
    }

    // 直接向用户缓冲区写入数据（通过映射地址）
    RtlCopyMemory(mappedBuffer, deviceData + offset, transferSize);
    Irp->IoStatus.Information = transferSize;

    IoCompleteRequest(Irp, IO_NO_INCREMENT);
    return STATUS_SUCCESS;
}

4.4 其他方式：METHOD_NEITHER

METHOD_NEITHER方式不提供任何缓冲区管理。I/O管理器直接将用户模式虚拟地址传递给驱动程序，驱动必须自行处理地址验证和访问。

驱动中处理METHOD_NEITHER的代码：

NTSTATUS DispatchNeither(PDEVICE_OBJECT DeviceObject, PIRP Irp) {
    PIO_STACK_LOCATION irpStack = IoGetCurrentIrpStackLocation(Irp);
    PVOID inputBuffer = irpStack->Parameters.DeviceIoControl.Type3InputBuffer;
    PVOID outputBuffer = Irp->UserBuffer;

    __try {
        // 验证输入缓冲区可读
        ProbeForRead(inputBuffer, inputLen, sizeof(UCHAR));
        // 验证输出缓冲区可写
        ProbeForWrite(outputBuffer, outputLen, sizeof(UCHAR));

        // 在异常保护下访问用户缓冲区
        RtlCopyMemory(outputBuffer, inputBuffer, min(inputLen, outputLen));
        Irp->IoStatus.Information = min(inputLen, outputLen);
    }
    __except(EXCEPTION_EXECUTE_HANDLER) {
        Irp->IoStatus.Status = GetExceptionCode();
        Irp->IoStatus.Information = 0;
    }

    IoCompleteRequest(Irp, IO_NO_INCREMENT);
    return Irp->IoStatus.Status;
}

4.5 四种方式对比总结

4.6 用户模式调用示例

用户模式应用程序通过DeviceIoControl函数发起通信：

HANDLE hDevice = CreateFile(L"\\\\.\\SampleDevice",
                            GENERIC_READ | GENERIC_WRITE,
                            0, NULL, OPEN_EXISTING, 0, NULL);
if (hDevice == INVALID_HANDLE_VALUE) {
    printf("打开设备失败，错误码: %d\n", GetLastError());
    return 1;
}

ULONG inputData = 0x1234;
ULONG outputData = 0;
DWORD bytesReturned = 0;

BOOL result = DeviceIoControl(
    hDevice,
    IOCTL_SAMPLE_READ,          // 控制码
    &inputData, sizeof(inputData),  // 输入缓冲区
    &outputData, sizeof(outputData), // 输出缓冲区
    &bytesReturned,             // 返回字节数
    NULL);                      // 不使用重叠I/O

if (result) {
    printf("IOCTL成功，返回数据: 0x%X, 字节数: %lu\n",
           outputData, bytesReturned);
} else {
    printf("IOCTL失败，错误码: %d\n", GetLastError());
}

CloseHandle(hDevice);