`

《Windows内核编程》---WDM驱动程序的基本结构和实例

 
阅读更多

WDM驱动的基本结构:

WDM驱动模型是建立在NT式驱动程序模型基础之上的。对于WDM驱动程序来说,一般都是基于分层的,即完成一个设备的操作,至少要由两个驱动设备共同完成。

1)物理设备对象和功能设备对象

物理设备对象(Physical Device ObjectPDO)和功能设备对象(Function Device ObjectFDO)的关系是“附加”与“被附加”的关系。

PC插入某个设备时,PDO会自动创建。确切的说,是由总线驱动创建的。PDO不能单独操作设备,需要配合FDO一起使用。系统会提示检测到新设备,要求安装驱动程序。需要安装的驱动程序就是WDM程序,此驱动程序负责创建FDO,并且附加到PDO上。

当一个FDO附加到PDO上时,PDO设备对象的子域AttachedDevice会记录FDO的位置。PDO被称作底层驱动或下层驱动,而FDO被称作高层驱动或上层驱动。

复杂一点的情况是,在FDOPDO之间还会存在过滤驱动。在FDO上面的过滤驱动被称作上层过滤驱动,在FDO下层的驱动被称作下层过滤驱动。每个设备对象中,有个StackSize子域,表明操作这个设备对象需要几层才能到达最下层的物理设备。

过滤驱动可以嵌套,即可以有很多个高层过滤驱动或多个底层过滤驱动。过滤驱动不是必须的,在WDM模型中PDOFDO是必须的。

NT设备是被动装入的,例如当有设备插入PC时,系统不会有提示,用户需要自己指定加载何种驱动;而WDM驱动则不然,当插入设备时,系统会自动创建出PDO,并提示请求用户安装FDO

2WDM驱动的入口程序

NT驱动一样也是DriverEntry,但是初始化作用被分散到其他例程中。如创建设备对象的工作被放在了AddService例程中,同时,在DriverEntry中,需要设置对IRP_MJ_PNP处理的派遣函数。

WDM驱动的DriverEntryNT式驱动的DriverEntry有以下几点不同:

(1)增加了对AddService函数的设置;AddService例程负责创建FDO,并附加到PDO上。

(2)创建设备对象的操作不在这个函数中了,而转到了AddService例程中。

(3)必须加入IRP_MJ_PNP的派遣回调函数。IRP_MJ_PNP主要是负责计算机中即插即用的处理。

3WDM驱动的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是为了定位设备的下一层设备。在附加操作完成后,需要设定符号链接,以便用户应用程序访问该设备。

4DriverUnload例程

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的处理函数中进行的。除了需要删除设备,取消符号链接外,在卸载函数中还需要将FDOPDO上的堆栈中移除掉,使用函数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_PNPIRP的处理。其中,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对其他PNPIRP做同样处理,即直接传递到底层驱动,并将底层驱动的结果返回:

/**

* 函数名称: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"));

}

分享到:
评论

相关推荐

Global site tag (gtag.js) - Google Analytics