Windows 驅動與驅動之間的通訊

本文主要介紹 windows NT 驅動之間利用 CTL_CODE 如何通訊. 一般來講, 使用者態的應用程式通過調用 CreateFile 和 DeviceIOControl 就可以和核心態驅動之間進行通訊, 如果是核心態的一個驅動程式想和另外一個驅動程式進行類似操作又如何?呢, 接下來會做詳細的說明.

主要的思路是首先得到另外一個驅動的 handle, 再構建一個 IRP, 賦予必要的參數, 再把這個 IRP 發送給另外一個驅動, 等待另外一個驅動處理完畢後擷取返回的資料.

具體代碼如下:
    
PIRP SMBIrp;
PIO_STACK_LOCATION      irpStack;
 
RtlInitUnicodeString(&usDeviceToFilter, L"\Device\AdvSMBus");
status = IoGetDeviceObjectPointer(&usDeviceToFilter,
    FILE_WRITE_DATA,&SMBusObject,&g_LM87Data.SmbHc);
if( !NT_SUCCESS(status) )
{
    KdPrint(("LM87: Can't get SMBus host! "));
}
SMBIrp = IoAllocateIrp (g_LM87Data.SmbHc->StackSize, FALSE);
if(!SMBIrp)
{
    KdPrint(("LM87: Allocate irp failed! "));
}
irpStack = IoGetNextIrpStackLocation(SMBIrp);          
g_LM87Data.SMBusHostData.Address = LM87_SMBUS_ADDRESS;
g_LM87Data.SMBusHostData.Offset = LM87_INTER_TEMP_REGISTER;
g_LM87Data.SMBusHostData.Data = 0x0;           
irpStack->MajorFunction = IRP_MJ_DEVICE_CONTROL;
irpStack->Parameters.DeviceIoControl.IoControlCode      = IOCTL_SMBUS_READ_BYTE;
irpStack->Parameters.DeviceIoControl.InputBufferLength = sizeof(SMBUS_HOST);
irpStack->Parameters.DeviceIoControl.OutputBufferLength = sizeof(SMBUS_HOST);
SMBIrp->AssociatedIrp.SystemBuffer = &g_LM87Data.SMBusHostData;                  
IoSetCompletionRoutine (SMBIrp, LM87RequestComplete, &g_LM87Data.SyncEvent, TRUE, TRUE, TRUE);
IoCallDriver (g_LM87Data.SmbHc, SMBIrp);
KeWaitForSingleObject(&g_LM87Data.SyncEvent, Executive, KernelMode, FALSE, NULL);
KdPrint(("LM87: query result is [%X] ",g_LM87Data.SMBusHostData.Data));
result[1] = g_LM87Data.SMBusHostData.Data * 10; //compatible with DiagAnywhere
IoFreeIrp(SMBIrp);

在上面的這段代碼中, 我們先用 IoGetDeviceObjectPointer 得到另外一個驅動的 handle,
然後利用 IoAllocateIrp 產生一個 IRP, 在賦完必要的參數之後通過 IoCallDriver 發送給另外一個驅動, 這裡需要注意的是在 IoCallDriver 之前有 IoSetCompletionRoution 的調用, 當另外一個驅動處理完這個 IRP 之後進入它所設定的 Roution 之中, 通過一個 Event 我們就知道 IRP 已經處理完畢, 我們可以擷取想要的資料了.

思路：

第一步:IoGetDeviceObjectPointer 得到另外一個驅動的 handle

第二步:利用 IoAllocateIrp 產生一個 IRP

第三步:在賦完必要的參數之後通過 IoCallDriver 發送給另外一個驅動

 注意事項:

在 IoCallDriver 之前有 IoSetCompletionRoution 的調用, 當另外一個驅動處理完這個 IRP 之後進入它所設定的 Roution 之中, 通過一個 Event 我們就知道 IRP 已經處理完畢, 我們可以擷取想要的資料了.

NTSTATUS
LM87RequestComplete (
     IN PDEVICE_OBJECT       DeviceObject,
     IN PIRP                 Irp,
     IN PVOID                Context
     )
{
    PKEVENT         Event;
    Event = (PKEVENT) Context;
    KeSetEvent (Event, IO_NO_INCREMENT, FALSE);
    return STATUS_MORE_PROCESSING_REQUIRED;
}

這樣就實現了在核心態的一個驅動程式和另外一個驅動程式的通訊.

另外還有一種實現的方式是通過 WorKItem 和 IoBuildDeviceIoControlRequest 來實現, 這裡就不做詳細的介紹了.