Windows内核 WDM驱动程序的基本结构和实例
WDM驱动的基本结构：
WDM驱动模型是建立在NT式驱动程序模型基础之上的。对于WDM驱动程序来说，一般都是基于分层的，即完成一个设备的操作，至少要由两个驱动设备共同完成。
1）物理设备对象和功能设备对象
物理设备对象（Physical Device Object，PDO）和功能设备对象（Function Device Object，FDO）的关系是“附加”与“被附加”的关系。
当PC插入某个设备时，PDO会自动创建。确切的说，是由总线驱动创建的。PDO不能单独操作设备，需要配合FDO一起使用。系统会提示检测到新设备，要求安装驱动程序。需要安装的驱动程序就是WDM程序，此驱动程序负责创建FDO，并且附加到PDO上。
当一个FDO附加到PDO上时，PDO设备对象的子域AttachedDevice会记录FDO的位置。PDO被称作底层驱动或下层驱动，而FDO被称作高层驱动或上层驱动。
复杂一点的情况是，在FDO和PDO之间还会存在过滤驱动。在FDO上面的过滤驱动被称作上层过滤驱动，在FDO下层的驱动被称作下层过滤驱动。每个设备对象中，有个StackSize子域，表明操作这个设备对象需要几层才能到达最下层的物理设备。
过滤驱动可以嵌套，即可以有很多个高层过滤驱动或多个底层过滤驱动。过滤驱动不是必须的，在WDM模型中PDO和FDO是必须的。
NT设备是被动装入的，例如当有设备插入PC时，系统不会有提示，用户需要自己指定加载何种驱动；而WDM驱动则不然，当插入设备时，系统会自动创建出PDO，并提示请求用户安装FDO。
2）WDM驱动的入口程序
和NT驱动一样也是DriverEntry，但是初始化作用被分散到其他例程中。如创建设备对象的工作被放在了AddService例程中，同时，在DriverEntry中，需要设置对IRP_MJ_PNP处理的派遣函数。
WDM驱动的DriverEntry和NT式驱动的DriverEntry有以下几点不同：
(1)增加了对AddService函数的设置；AddService例程负责创建FDO，并附加到PDO上。
(2)创建设备对象的操作不在这个函数中了，而转到了AddService例程中。
(3)必须加入IRP_MJ_PNP的派遣回调函数。IRP_MJ_PNP主要是负责计算机中即插即用的处理。
3）WDM驱动的AddService例程
AddService例程是WDM独有的。在DriverEntry中需设置AddService例程的函数地址，设置的方式是驱动对象中有个DriverExtension子域，DriverExtension中有个AddService子域，将该子域指向AddService例程的函数地址即可。
和DriverEntry不同，AddService例程的名字可以任意命名：
DRIVER_ADD_DEVICE AddDevice;
NTSTATUS AddDevice(
__in struct _DRIVER_OBJECT *DriverObject, //驱动对象(I/O管理器创建的)
__in struct _DEVICE_OBJECT *PhysicalDeviceObject //设备对象(底层总线驱动创建的PDO设备对象)
//传入该参数的目的是将FDO附加在PDO上
)
{ ... }
AddService函数的功能可分为：
(1)通过IoCreateDevice等函数，创建设备对象，该设备对象就是FDO。
(2)创建FDO后，需要将FDO的地址保存下来，保存的位置是在设备扩展中。
(3)驱动程序将创建的FDO附加到PDO上，附加这个动作是利用函数IoAttachDeviceToDeviceStack实现的：
PDEVICE_OBJECT IoAttachDeviceToDeviceStack(
__in PDEVICE_OBJECT SourceDevice, //要附加到其他设备上的设备，
//将FDO附加在PDO之上时，此处即填FDO的地址
__in PDEVICE_OBJECT TargetDevice //被附加的设备，将FDO直接附加在PDO上时
//（不考虑过滤驱动），此处即时PDO的地址
);
返回值：附加后，返回附加设备的下层设备。如果中间没有过滤驱动，返回值就是PDO；如果中间有过滤驱动，返回的是过滤驱动。
当FDO附加到PDO上时，PDO会通过AttachedDevice子域知道它上层的设备是FDO（或过滤驱动）；而FDO无法知道下层是什么设备，因此，需要通过设备扩展设置FDO下层的设备，例如：
//设备扩展结构体
typedef struct _DEVICE_EXTENSION
{
PDEVICE_OBJECT fdo; //功能设备对象FDO
PDEVICE_OBJECT NextStackDevice; //FDO的下层驱动设备
UNICODE_STRING ustrDeviceName; //设备名
UNICODE_STRING ustrSymLinkName; //符号链接
}DEVICE_EXTENSION, *PDEVICE_EXTENSION;
一般根据自己的需要定制自己的设备扩展。子域fdo是为了保存FDO的地址以备后用；子域NextStackDevice是为了定位设备的下一层设备。在附加操作完成后，需要设定符号链接，以便用户应用程序访问该设备。
4）DriverUnload例程
在NT式驱动中，DriverUnload例程主要负责做删除设备和取消符号链接。而在WDM驱动中，这部分操作被IRP_MN_REMOVE_DEVICE的处理函数所负责。因此，如果在DriverEntry中有申请内存的操作，可以在DriverUnload例程中回收这些内存，DriverUnload例程变得相对简单了。
5）对IRP_MN_REMOVE_DEVICE IRP的处理
驱动程序内部是由IRP驱动的。创建IRP的原因很多，IRP_MN_REMOVE_DEVICE这个IRP是当设备被卸载时，由即插即用管理器创建，并发送到驱动程序中的。IRP一般由两个号码指定该IRP的具体意义：一是主IRP号（Major IRP）；一是辅IRP号（Minor IRP）。
每个IRP都由对应的派遣函数所处理，派遣函数是在DriverEntry中指定的。
在WDM驱动程序中，对设备的卸载一般是在对IRP_MN_REMOVE_DEVICE的处理函数中进行的。除了需要删除设备，取消符号链接外，在卸载函数中还需要将FDO从PDO上的堆栈中移除掉，使用函数IoDetachDevice：
VOID IoDetachDevice(
__inout PDEVICE_OBJECT TargetDevice //下层堆栈上的设备对象
);
调用此函数后，可将FDO从设备链上删除，但PDO还是存在的，PDO的删除是由操作系统负责的。
WDM驱动程序的编写实例如下，先定义头文件HelloWDM.h：
#pragma once
#ifdef __cplusplus
extern "C"
{
#endif
#include <wdm.h>
#ifdef __cplusplus
}
#endif
//设备扩展结构体
typedef struct _DEVICE_EXTENSION
{
PDEVICE_OBJECT fdo;
PDEVICE_OBJECT NextStackDevice;
UNICODE_STRING ustrDeviceName;
UNICODE_STRING ustrSymLinkName;
}DEVICE_EXTENSION, *PDEVICE_EXTENSION;
//定义分页内存、非分页内存和INIT段内存标志
#define PAGEDCODE code_seg("PAGE")
#define LOCKEDCODE code_seg()
#define INITCODE code_seg("INIT")
#define PAGEDDATA data_seg("PAGE")
#define LOCKEDCODE code_seg()
#define INITDATA data_seg("INIT")
#define ARRAYSIZE(p) (sizeof(p)/sizeof((p)[0]))
NTSTATUS HelloWDMAddDevice(IN PDRIVER_OBJECT DriverObject,
IN PDEVICE_OBJECT PhysicalDeviceObject);
NTSTATUS HelloWDMPnp(IN PDEVICE_OBJECT fdo, IN PIRP Irp);
NTSTATUS HelloWDMDispatchRoutine(IN PDEVICE_OBJECT fdo, IN PIRP Irp);
NTSTATUS DefaultPnpHandler(PDEVICE_EXTENSION pdx, PRIP Irp);
NTSTATUS HandleRemoveDevice(PDEVICE_EXTENSION pdx, PRIP Irp);
void HelloWDMUnload(IN PDRIVER_OBJECT DriverObject);
extern "C"
NTSTATUS DriverEntry(IN PDRIVER_OBJECT pDriverObject,
IN PUNICODE_STRING pRegistryPath);
和NT式驱动程序一样，WDM的入口函数地址也是DriverEntry，且在C++编译时需要使用extern “C”修饰：
/**
* 函数名称：DriverEntry
* 功能描述：初始化驱动程序，定位和申请硬件资源，创建内核对象
* 参数列表：
* pDriverObject：从I/O管理器中传进来的驱动对象
* pRegistryPath：驱动程序在注册表中的路径
* 返回值：返回初始化驱动状态
*/
#pragma INITCODE //此函数放在INIT段中，当驱动加载结束后，此函数就可以从内存中卸载了
extern "C" NTSTATUS DriverEntry(IN PDRIVER_OBJECT pDriverObject,
IN PUNICODE_STRING pRegistryPath)
{
KdPrint(("Enter DriverEntry/n"));
pDriverObject->DriverExtension->AddDevice = HelloWDMAddDevice;
pDriverObject->MajorFunction[IRP_MJ_PNP] = HelloWDMPnp;
pDriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = HelloWDMDispatchRoutine;
pDriverObject->MajorFunction[IRP_MJ_CREATE] = HelloWDMDispatchRoutine;
pDriverObject->MajorFunction[IRP_MJ_READ] = HelloWDMDispatchRoutine;
pDriverObject->MajorFunction[IRP_MJ_WRITE] = HelloWDMDispatchRoutine;
pDriverObject->DriverUnload = HelloWDMUnload;
KdPrint(("Leave DriverEntry/n"));
return STATUS_SUCCESS;
}
在WDM驱动程序中，创建设备对象的任务不再由DriverEntry承担，而需要驱动程序向系统注册一个称为AddDevice的例程。此例程由PNP管理器负责调用，其函数主要职责是创建设备。
/**
* 函数名称：HelloWDMAddDevice
* 功能描述：添加新设备
* 参数列表：
* DriverObject：从I/O管理器中传进来的驱动对象
* PhysicalDeviceObject：从I/O管理器中传进来的物理设备对象
* 返回值：返回新添加设备的状态
*/
#pragma PAGEDCODE
NTSTATUS HelloWDMAddDevice(IN PDRIVER_OBJECT DriverObject,
IN PDEVICE_OBJECT PhysicalDeviceObject)
{
PAGED_CODE(); //DDK提供的宏，只在check版中有效，
//用于确保该例程运行在低于APC_LEVEL的中断优先级别上
KdPrint(("Enter HelloWDMAddDevice/n"));
NTSTATUS status;
PDEVICE_OBJECT fdo;
UNICODE_STRING devName;
RtlInitUnicodeString(&devName, L"//Device//ASCEWDMDevice");
status = IoCreateDevice(DriverObject,sizeof(DEVICE_EXTENSION),
&(UNICODE_STRING)devName,
FILE_DEVICE_UNKNOWN,
0, FALSE, &fdo);
if(!NT_SUCCESS(status))
{
return status;
}
PDEVICE_EXTENSION pdx = (PDEVICE_EXTENSION)fdo->DeviceExtension;
pdx->fdo = fdo;
//将fdo(功能设备对象)挂接到设备堆栈上，将返回值(下层堆栈的位置)
//记录在设备扩展结构中
pdx->NextStackDevice = IoAttachDeviceToDeviceStack(
fdo, PhysicalDeviceObject);
UNICODE_STRING symLinkName;
RtlInitUnicodeString(&symLinkName, L"//DosDevices//HelloWDM");
pdx->ustrDeviceName = devName;
pdx->ustrSymLinkName = symLinkName;
status = IoCreateSymbolicLink(&(UNICODE_STRING)symLinkName,
&(UNICODE_STRING)devName);
if(!NT_SUCCESS(status))
{
IoDeleteSymbolicLink(&pdx->ustrSymLinkName);
status = IoCreateSymbolicLink(&symLinkName, &devName);
if(!NT_SUCCESS(status))
{
return status;
}
}
fdo->Flags |= DO_BUFFERED_IO | DO_POWER_PAGABLE;
fdo->Flags &= ~DO_DEVICE_INITIALIZING;
KdPrint(("Leaving HelloWDMAddDevice/n"));
}
WDM式驱动程序主要区别在于对IRP_MJ_PNP的IRP的处理。其中，IRP_MJ_PNP会细分为若干个子类。下面代码除了对IRP_MN_REMOVE_DEVICE做特殊处理外，其他IRP则做相同处理：
/**
* 函数名称：HelloWDMPnp
* 功能描述：对即插即用IRP进行处理
* 参数列表：
* fdo：功能设备对象
* Irp：从I/O请求包
* 返回值：返回状态
*/
#pragma PAGEDCODE
NTSTATUS HelloWDMPnp(IN PDEVICE_OBJECT fdo, IN PIRP Irp)
{
PAGED_CODE();
KdPrint(("Enter HelloWDMPnp/n"));
NTSTATUS status = STATUS_SUCCESS;
PDEVICE_EXTENSION pdx = (PDEVICE_EXTENSION)fdo->DeviceExtension;
//得到当前IRP的堆栈
PIO_STACK_LOCATION stack = IoGetCurrentIrpStackLocation(Irp);
static NTSTATUS (*fcntab[])(PDEVICE_EXTENSION pdx, PRIP Irp) =
{
DefaultPnpHandler, //IRP_MN_START_DEVICE
DefaultPnpHandler, //IRP_MN_QUERY_REMOVE_DEVICE
HandleRemoveDevice,//IRP_MN_REMOVE_DEVICE
DefaultPnpHandler, //IRP_MN_CANCEL_REMOVE_DEVICE
DefaultPnpHandler, //IRP_MN_STOP_DEVICE
DefaultPnpHandler, //IRP_MN_QUERY_STOP_DEVICE
DefaultPnpHandler, //IRP_MN_CANCEL_STOP_DEVICE
DefaultPnpHandler, //IRP_MN_QUERY_DEVICE_RELATIONS
DefaultPnpHandler, //IRP_MN_QUERY_INTERFACE
DefaultPnpHandler, //IRP_MN_QUERY_CAPABILITIES
DefaultPnpHandler, //IRP_MN_QUERY_RESOURCES
DefaultPnpHandler, //IRP_MN_QUERY_RESOURCE_REQUIREMENTS
DefaultPnpHandler, //IRP_MN_QUERY_DEVICE_TEXT
DefaultPnpHandler, //IRP_MN_FILTER_RESOURCE_REQUIREMENTS
DefaultPnpHandler, //
DefaultPnpHandler, //IRP_MN_READ_CONFIG
DefaultPnpHandler, //IPR_MN_WRITEZ_CONFIG
DefaultPnpHandler, //IRP_MN_EJECT
DefaultPnpHandler, //IRP_MN_SET_LOCK
DefaultPnpHandler, //IRP_MN_QUERY_ID
DefaultPnpHandler, //IRP_MN_QUERY_PNP_DEVICE_STATE
DefaultPnpHandler, //IRP_MN_QUERY_BUS_INFORMATION
DefaultPnpHandler, //IRP_MN_DEVICE_USAGE_NOTIFICATION
DefaultPnpHandler, //IRP_MN_SURPRISE_REMOVAL
};
ULONG fcn = stack->MinorFunction;
if(fcn >= ARRAYSIZE(fcntab))
{
//未知的设备类型
status = DefaultPnpHandler(pdx, Irp); //对于未知的IRP类型，我们让
//DefaultPnpHandler函数处理
return status;
}
#if DBG
static char* fcnname[] =
{
"IRP_MN_START_DEVICE",
"IRP_MN_QUERY_REMOVE_DEVICE",
"IRP_MN_REMOVE_DEVICE",
"IRP_MN_CANCEL_REMOVE_DEVICE",
"IRP_MN_STOP_DEVICE",
"IRP_MN_QUERY_STOP_DEVICE",
"IRP_MN_CANCEL_STOP_DEVICE",
"IRP_MN_QUERY_DEVICE_RELATIONS",
"IRP_MN_QUERY_INTERFACE",
"IRP_MN_QUERY_CAPABILITIES",
"IRP_MN_QUERY_RESOURCES",
"IRP_MN_QUERY_RESOURCE_REQUIREMENTS",
"IRP_MN_QUERY_DEVICE_TEXT",
"IRP_MN_FILTER_RESOURCE_REQUIREMENTS",
"",
"IRP_MN_READ_CONFIG",
"IRP_MN_WRITE_CONFIG",
"IRP_MN_EJECT",
"IRP_MN_SET_LOCK",
"IRP_MN_QUERY_ID",
"IRP_MN_QUERY_PNP_DEVICE_STATE",
"IRP_MN_QUERY_BUS_INFORMATION",
"IRP_MN_DEVICE_USAGE_NOTIFICATION",
"IRP_MN_SURPRISE_REMOVAL"
};
Kdprint(("PNP request (%s)/n", fcnname[fcn]));
#endif //DBG
status = (*fcntab[fcn])(pdx, Irp);
KdPrint(("Leave HelloWDMPnp/n"));
return status;
}
除了IRP_MN_STOP_DEVICE之外，HelloWDM对其他PNP的IRP做同样处理，即直接传递到底层驱动，并将底层驱动的结果返回：
/**
* 函数名称：DefaultPnpHandler
* 功能描述：对PNP IRP进行默认处理
* 参数列表：
* pdx：设备对象的扩展
* Irp：从I/O请求包
* 返回值：返回状态
*/
#pragma PAGEDCODE
NTSTATUS DefaultPnpHandler(PDEVICE_EXTENSION pdx, PRIP Irp)
{
PAGED_CODE();//确保该例程处于APC_LEVEL之下
KdPrint(("Enter DefaultPnpHandler/n"));
IoSkipCurrentIrpStackLocation(Irp); //略过当前堆栈
KdPrint(("Leave DefaultPnpHandler/n"));
return IoCallDriver(pdx->NextStackDevice, Irp); //用下层堆栈的驱动设备对象处理此IRP
}
对IRP_MN_REMOVE_DEVICE的处理类似于在NT式驱动中的卸载例程，而在WDM驱动中，卸载例程几乎不用做处理：
/**
* 函数名称： HandleRemoveDevice
* 功能描述：对IRP_MN_REMOVE_DEVICE进行处理
* 参数列表
* fdo：功能设备对象
* Irp：从I/O请求包
* 返回值：返回状态
*/
#pragma PAGEDCODE
NTSTATUS HandleRemoveDevice(PDEVICE_EXTENSION pdx, PIRP Irp)
{
PAGED_CODE();
KdPrint(("Enter HandleRemoveDevice/n"));
Irp->IoStatus.Status = STATUS_SUCCESS;
NTSTATUS status = DefaultPnpHandler(pdx, Irp);
IoDeleteSymbolicLink(&(UNICODE_STRING)pdx->ustrSymLinkName);
//调用IoDetachDevice()把fdo从设备栈中脱开
if(pdx->NextStackDevice)
IoDetachDevice(pdx->NextStackDevice);
//删除fdo
IoDeleteDevice(pdx->fdo);
KdPrint(("Leave HandleRemoveDevice/n"));
return status;
}
/**
* 函数名称：HelloWDMDispatchRoutine
* 功能描述：对默认IRP进行处理
* 参数列表：
* fdo：功能设备对象
* Irp：从I/O请求包
* 返回值：返回状态
*/
#pragma PAGEDCODE
NTSTATUS HelloWDMDispatchRoutine(IN PDEVICE_OBJECT fdo, IN PIRP Irp)
{
PAGED_CODE();
KdPrint(("Enter HelloWDMDispatchRoutine/n"));
Irp->IoStatus.Status = STATUS_SUCCESS;
Irp->IoStatus.Information = 0;
IoCompleteRequest(Irp, IO_NO_INCREMENT);
KdPrint(("Leave HelloWDMDispatchRoutine/n"));
return STATUS_SUCCESS;
}
由于WDM式的驱动程序将主要的卸载任务放在了对IRP_MN_REMOVE_DEVICE的处理函数中，在标准的卸载例程中几乎没有什么需要做的：
/**
* 函数名称：HelloWDMUnload
* 功能描述：负责驱动程序的卸载操作
* 参数列表：
* DriverObject：驱动对象
* 返回值：返回状态
*/
#pragma PAGEDCODE
void HelloWDMUnload(IN PDRIVER_OBJECT DriverObject)
{
PAGED_CODE();
KdPrint(("Enter HelloWDMUnload/n"));
KdPrint(("Leave HelloWDMUnload/n"));
}
