一个WDM驱动开发框架

nowayings

下面是我起初学习WDM的时候的一个驱动开发框架。分享之。。。。。

1. #include "DriverWDM.h"
   
2. 
   
3. #pragma INITCODE
   
4. extern "C" NTSTATUS DriverEntry(IN PDRIVER_OBJECT pDriverObject,IN PUNICODE_STRING RegistryPath)
   
5. {
   
6.         NTSTATUS status;
   
7.         KdPrint(("enter DriverEntry\n"));
   
8. 
   
9.         pDriverObject->DriverExtension->AddDevice = HelloWDMAddDevice;
   
10.         pDriverObject->MajorFunction[IRP_MJ_PNP] = HelloWDMPnp;
    
11.         pDriverObject->MajorFunction[IRP_MJ_CREATE] =
    
12.         pDriverObject->MajorFunction[IRP_MJ_READ] =
    
13.         pDriverObject->MajorFunction[IRP_MJ_WRITE] = HelloDDKDispatchRoutine;
    
14. }
    
15. 
    
16. NTSTATUS HelloWDMAddDevice(IN PDRIVER_OBJECT pDriverObject,IN PDEVICE_OBJECT pPhysicalDeviceObject)
    
17. {
    
18.         PAGED_CODE();
    
19.         KdPrint(("Enter HelloWDMAddDevice\n"));
    
20.         UNICODE_STRING devName;                //设备名字
    
21.         NTSTATUS status        ;
    
22.         PDEVICE_OBJECT fdo;        //功能设备对象
    
23.         PDEVICE_EXTENSION pdx;        //设备扩展
    
24. 
    
25.         RtlInitUnicodeString(&devName,L"\\Device\\MyWDMDevice");
    
26.         status = IoCreateDevice(pDriverObject,sizeof(DEVICE_EXTENSION),&(UNICODE_STRING)devName,FILE_DEVICE_UNKNOWN,0,false,&fdo);
    
27.         //判断是否成功创建设备
    
28.         if(!NT_SUCCESS(status))
    
29.         {
    
30.                 return status;
    
31.         }
    
32. 
    
33.         //得到设备扩展,然后填充它
    
34.         pdx = (PDEVICE_EXTENSION)fdo->DeviceExtension;
    
35.         pdx->fdo = fdo;
    
36. 
    
37.         UNICODE_STRING symLinkName;
    
38.         RtlInitUnicodeString(&symLinkName,L"\\DosDevices\\HelloWDM");
    
39.         status = IoCreateSymbolicLink(&symLinkName,&devName);
    
40. 
    
41.         //如果创建符号连接失败。则进行相应的删除操作
    
42.         if (!NT_SUCCESS(status))
    
43.         {
    
44.                 //再一次创建符号连接
    
45.                 IoDeleteSymbolicLink(&symLinkName);
    
46.                 status = IoCreateSymbolicLink(&symLinkName,&devName);
    
47.                 if(!NTSTATUS(status))
    
48.                 {
    
49.                         return status;
    
50.                 }
    
51.         }
    
52. 
    
53.         //在设备扩展中记录设备的符号名字和设备名字
    
54.         pdx->ustrSymLinkName = symLinkName;
    
55.         pdx->ustrDeviceName = devName;
    
56. 
    
57.         //设置设备标志
    
58.         fdo->Flags |= DO_BUFFERED_IO | DO_POWER_PAGABLE;
    
59.         fdo->Flags &= ~DO_DEVICE_INITIALIZING;
    
60. 
    
61.         KdPrint(("leave HelloWDMDevice\n"));
    
62.         return STATUS_SUCCESS;
    
63. 
    
64. }

复制代码