第二讲：我的第一个驱动

Posted 朝闻道

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了第二讲：我的第一个驱动相关的知识，希望对你有一定的参考价值。

原文：http://blog.csdn.net/caperingrabbit/article/details/5285288

 

配置好了开发环境之后，下面就要通过具体的程序来了解驱动的开发了。下面我们以一个WDM驱动的框架来实现Windows驱动程序的HelloWorld。

作为一个驱动程序，首先应该写的是它的入口函数，这点跟MFC的WinMain或者C++中的Main函数一样，驱动的入口函数使用DriverEntry。在入口函数中主要实现的功能是一些分发例程的注册以及其他的需要初始化的事务。

DriverEntry的原型是NTSTATUS DriverEntry(IN PDRIVER_OBJECT pDriverObject,IN PUNICODE_STRING pRegistryPath)，IN、OUT这只是一个宏，用来表示参数的输入输出类型，并无实质意义，也不占空间。pDriverObject是我们所要创建的驱动对象，pRegistryPath是驱动程序在注册表中的地址。至于它们的类型，PDRIVER_OBJECT是驱动开发中一个重要的数据结构，这是驱动对象，具体的我们以后再讨论，而PUNICODE_STRING则是一个宽字符的指针，在内核开发中，一般使用的字符串都是宽字符，即Unicode string。  下面是一个范例：

[cpp] view plain copy
 
  1. extern"C" //一般用C++进行Windows驱动开发，在入口函数前面要加这个关键词，主要是为了在驱动程序中能够调用C的函数  
  2. NTSTATUS DriverEntry(IN PDRIVER_OBJECT pDriverObject,  
  3.     IN PUNICODE_STRING pRegistryPath  
  4.     )  
  5.     {  
  6.         KdPrint(("Hello!welcome to the driver entry!/n"));//KdPrint是一个宏，类似于MFC中的Trace  
  7.         //对于通用的例程，这里也使用一个通用的处理函数进行注册  
  8.         for(ULONG i = 0; i <= IRP_MJ_MAXIMUM_FUNCTION; i++)  
  9.         {  
  10.             pDriverObject->MajorFunction[i] = DispatchRoutine;  
  11.         }         
  12.           
  13.         //驱动的功能例程的注册,这是几个比较特殊的例程，我们分别注册，  
  14.         //而驱动开发的主要工作就是完成他们的这些例程的实现函数，为了简单方便讲解，这里使用的和上边一样  
  15.         pDriverObject->DriverExtension->AddDevice = MyDeviceAdd;  
  16.         pDriverObject->DriverUnload = DriverUnload;  
  17.         pDriverObject->MajorFunction[IRP_MJ_CREATE] = DispatchRoutine;  
  18.         pDriverObject->MajorFunction[IRP_MJ_CLOSE] = DispatchRoutine;  
  19.         pDriverObject->MajorFunction[IRP_MJ_READ] = DispatchRoutine;  
  20.         pDriverObject->MajorFunction[IRP_MJ_WRITE] = DispatchRoutine;  
  21.           
  22.         return STATUS_SUCCESS;  
  23. }  

 

 

完成了入口函数之后，接下来就要实现上面注册的分发例程的功能函数了。作为一个WDM驱动程序，首先需要创建设备对象，这个任务就交给AddDevice例程，通过上面的注册，实现部分放在MyDeviceAdd。

 

[cpp] view plain copy
 
  1. NTSTATUS MyDeviceAdd(IN PDRIVER_OBJECT DriverObject,  
  2.                            IN PDEVICE_OBJECT PhysicalDeviceObject)  
  3. {   
  4.     PAGED_CODE();//DDK提供的一个宏  
  5.     KdPrint(("Enter MyDrivers/n"));  
  6.   
  7.     NTSTATUS status;  
  8.     PDEVICE_OBJECT fdo;  
  9.     UNICODE_STRING devName;  
  10.     RtlInitUnicodeString(&devName,L"//Device//MyDevice");  
  11.     //创建设备对象  
  12.     status = IoCreateDevice(  
  13.         DriverObject,  
  14.         sizeof(DEVICE_EXTENSION),  
  15.         &(UNICODE_STRING)devName,  
  16.         FILE_DEVICE_UNKNOWN,  
  17.         0,  
  18.         FALSE,  
  19.         &fdo);  
  20.     if( !NT_SUCCESS(status))  
  21.         return status;  
  22.     //获得设备扩展的数据结构。这个数据结构一般是由用户自己定义，保存一些设备相关的信息  
  23.     PDEVICE_EXTENSION pdx = (PDEVICE_EXTENSION)fdo->DeviceExtension;  
  24.     //配置设备扩展的内容  
  25.     pdx->fdo = fdo;  
  26.     //设备的挂载  
  27.     pdx->NextStackDevice = IoAttachDeviceToDeviceStack(fdo, PhysicalDeviceObject);  
  28.     UNICODE_STRING symLinkName;  
  29.     RtlInitUnicodeString(&symLinkName,L"//DosDevices//MyDriver");  
  30.   
  31.     pdx->ustrDeviceName = devName;  
  32.     pdx->ustrSymLinkName = symLinkName;  
  33.     status = IoCreateSymbolicLink(&(UNICODE_STRING)symLinkName,&(UNICODE_STRING)devName);  
  34.   
  35.     if( !NT_SUCCESS(status))  
  36.     {  
  37.         IoDeleteSymbolicLink(&pdx->ustrSymLinkName);  
  38.         status = IoCreateSymbolicLink(&symLinkName,&devName);  
  39.         if( !NT_SUCCESS(status))  
  40.         {  
  41.             return status;  
  42.         }  
  43.     }  
  44.     //设置设备的IO状态  
  45.     fdo->Flags |= DO_BUFFERED_IO | DO_POWER_PAGABLE;  
  46.     fdo->Flags &= ~DO_DEVICE_INITIALIZING;  
  47.   
  48.     KdPrint(("LeaveMyDrivers/n"));  
  49.     return STATUS_SUCCESS;  
  50. }  

 

 

生成了设备对象之后，接下来就要处理发向设备的各种请求了。这里就用一个通用的处理例程来示范一下。其实这个处理函数只是将这些请求转交给真实的设备而已，没有实际的功能。

 

[cpp] view plain copy
 
  1. NTSTATUS HelloWDMDispatchRoutine(IN PDEVICE_OBJECT fdo,  
  2.                                  IN PIRP Irp)  
  3. {  
  4.     PAGED_CODE();  
  5.     KdPrint(("Enter HelloWDMDispatchRoutine/n"));  
  6.     Irp->iostatus.Status = STATUS_SUCCESS;  
  7.     Irp->IoStatus.Information = 0;   //   
  8.     IoCompleteRequest( Irp, IO_NO_INCREMENT );  
  9.     KdPrint(("Leave HelloWDMDispatchRoutine/n"));  
  10.     return STATUS_SUCCESS;  
  11. }  
  12. 执行完所有的请求之后，在最后卸载驱动。对于WDM驱动，标准的程序其实卸载例程没什么实质内容，一般都是有Pnp里边的RemoveDevice来处理的，这里只是一个演示，演示一下这个例程的功能和内容。  
  13. VOID DriverUnload(IN PDRIVER_OBJECT pDriverObject)   
  14. {  
  15.     PDEVICE_OBJECT  pNextObj;  
  16.     KdPrint(("Enter DriverUnload/n"));  
  17.     pNextObj = pDriverObject->DeviceObject;  
  18.     while (pNextObj != NULL)   
  19.     {  
  20.         PDEVICE_EXTENSION pDevExt = (PDEVICE_EXTENSION)  
  21.             pNextObj->DeviceExtension;  
  22.   
  23.         //删除符号链接  
  24.         UNICODE_STRING pLinkName = pDevExt->ustrSymLinkName;  
  25.         IoDeleteSymbolicLink(&pLinkName);  
  26.         pNextObj = pNextObj->NextDevice;  
  27.         IoDeleteDevice( pDevExt->pDevice );  
  28.     }  
  29. }  

 

 

这样，一个驱动的主要部分就完成了。接下来完成一些细碎的工作，编译一下，一个驱动就出现了。

头文件

 

[cpp] view plain copy
 
  1. #ifdef __cplusplus  
  2. extern "C"  
  3. {  
  4. #endif  
  5. #include <wdm.h>  
  6. #ifdef __cplusplus  
  7. }  
  8. #endif   
  9.   
  10. typedef struct _DEVICE_EXTENSION  
  11. {  
  12.     PDEVICE_OBJECT fdo;  
  13.     PDEVICE_OBJECT NextStackDevice;  
  14.     UNICODE_STRING ustrDeviceName;  // 设备名  
  15.     UNICODE_STRING ustrSymLinkName; // 符号链接名  
  16. } DEVICE_EXTENSION, *PDEVICE_EXTENSION;  
  17.   
  18. #define PAGEDCODE code_seg("PAGE")  
  19. #define LOCKEDCODE code_seg()  
  20. #define INITCODE code_seg("INIT")  
  21.   
  22. #define PAGEDDATA data_seg("PAGE")  
  23. #define LOCKEDDATA data_seg()  
  24. #define INITDATA data_seg("INIT")  
  25.   
  26. #define arraysize(p) (sizeof(p)/sizeof((p)[0]))  
  27.   
  28. NTSTATUS MyDeviceAdd(IN PDRIVER_OBJECT DriverObject,  
  29.                            IN PDEVICE_OBJECT PhysicalDeviceObject);  
  30. NTSTATUS DispatchRoutine(IN PDEVICE_OBJECT fdo,  
  31.                                  IN PIRP Irp);  
  32. void DriverUnload(IN PDRIVER_OBJECT DriverObject);  
  33.   
  34. extern "C"  
  35. NTSTATUS DriverEntry(IN PDRIVER_OBJECT DriverObject,  
  36.                      IN PUNICODE_STRING RegistryPath);  

 

 

新建一个文本文件，内容如下：

 

[cpp] view plain copy
 
  1. #  
  2. # DO NOT EDIT THIS FILE!!!  Edit ./sources. If you want to add a new source  
  3. # file to this component.  This file merely indirects to the real make file  
  4. # that is shared by all the driver components of the Windows NT DDK  
  5. #  
  6.   
  7. !INCLUDE $(NTMAKEENV)/makefile.def  

 

 

重命名文件，改为Makefile，后缀没有。然后重新建立一个文本文件，写入如下内容：

 

[cpp] view plain copy
 
  1. TARGETNAME=HellOWORLD  
  2. TARGETTYPE=DRIVER  
  3. DRIVERTYPE=WDM  
  4. TARGETPATH=OBJ  
  5.   
  6. INCLUDES=$(BASEDIR)/inc;/  
  7.          $(BASEDIR)/inc/ddk;/  
  8.   
  9. SOURCES=MyDrivers.cpp/  

 

 

保存，更改文件名为Sources，后缀没有，这样进入DDK的编译环境，已编译，一个后缀为.sys的驱动文件就出现了。

好了，一个简单的驱动就诞生了。

http://blog.csdn.net/chence19871/article/details/50697439

以上是关于第二讲：我的第一个驱动的主要内容，如果未能解决你的问题，请参考以下文章

病毒分析第二讲,分析病毒的主要功能

32位汇编第二讲,编写窗口程序,加载资源,响应消息,以及调用C库函数

C++ c++与C语言的区别第二讲

逆向课程第二讲,寻找main入口点

自然语言处理第二讲：单词计数

从零到壹学习密码学第二讲：hash 算法