• 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"));

    }

  • 相关阅读:
    ORBSLAM2的资源
    工程思想
    Linux中PATH、 LIBRARY_PATH、 LD_LIBRARY_PATH和ROS_PACKAGE_PATH
    CMkeList文件编写
    SLAM资料
    windows下隐藏文件夹
    Ubuntu16装Flash
    知识管理
    学习纲领
    lsof详解
  • 原文地址:https://www.cnblogs.com/alsofly/p/3623880.html
Copyright © 2020-2023  润新知