开发驱动基本框架源代码

海军上将

开发驱动基本框架源代码，发一篇驱动的基本框架吧！

1. 
   
2. // DriverDemo.cpp文件
   
3. extern "C"
   
4. {
   
5.     #include <ntddk.h>
   
6. }
   
7. 
   
8. //自定义函数的声明
   
9. NTSTATUS DispatchCreateClose(PDEVICE_OBJECT pDevObj,PIRP pIrp);
   
10. void DriverUnload(PDRIVER_OBJECT pDriverObj);
    
11. 
    
12. //驱动内部名称和连接名称
    
13. #define DEVICE_NAME L"\\Device\\devDriverDemo "
    
14. #define LINK_NAME L“\\??\\sldevDriverDemo ”
    
15. 
    
16. 驱动程序加载时调用DriverEntry例程
    
17. NTSTATUS DriverEntry(PDRIVER_OBJECT pDriverObj, PUNICODE_STRING pRegistryString)
    
18. {
    
19.     NTSTATUS status=STATUS_SUCCESS;
    
20.     DbgPrint("DriverDemo:DriverEntry...\n");
    
21.     //初始化各派遣例程
    
22.     pDriverObj->MajorFunction[IRP_MJ_CREATE]=DispatchCreateClose;
    
23.     pDriverObj->MajorFunction[IRP_MJ_CLOSE]=DispatchCreateClose;
    
24.     pDriverObj->DriverUnload=DriverUnload;
    
25.     //设备名称
    
26.     UNICODE_STRING ustrDeviceName;
    
27.     RtlInitUnicodeString(&ustrDeviceName,DEVICE_NAME);
    
28.     //创建设备对象
    
29.     PDEVICE_OBJECT pDevObj;
    
30.     status = IoCreateDevice(pDriverObj,0,&ustrDeviceName,FILE_DEVICE_UNKNOWN,0,false,&pDevObj);
    
31.     if(!NT_SUCCESS(status))
    
32.     {
    
33.         DbgPrint("IoCreateDevice创建设备失败! \n");
    
34.         return status;
    
35.     }
    
36.      //符号连接名称
    
37.     UNICODE_STRING ustrLinkName;
    
38.     RtlInitUnicodeString(&ustrLinkName,LINK_NAME);
    
39.     //创建关联
    
40.     status=IoCreateSymbolicLink(&ustrLinkName,&ustrDeviceName);
    
41.     if(!NT_SUCCESS(status))
    
42.     {
    
43.         IoDeleteDevice(pDevObj);
    
44.         DbgPrint("IoCreateSymbolicLink创建关联失败! \n");
    
45.         return status;
    
46.     }
    
47.     return STATUS_SUCCESS;
    
48. }
    
49. 
    
50. //自定义驱动卸载函数
    
51. void DriverUnload(PDRIVER_OBJECT pDriverObj)
    
52. {
    
53.      DbgPrint("DriverDemo:DriverUnload...\n");
    
54.      //删除符号连接名称
    
55.      UNICODE_STRING strLink;
    
56.      RtlInitUnicodeString(&strLink,LINK_NAME);
    
57.      IoDeleteSymbolicLink(&strLink);
    
58.      IoDeleteDevice(pDriverObj->DeviceObject);
    
59. }
    
60. 
    
61. //处理IRP_MJ_CREATE
    
62. NTSTATUS DispatchCreateClose(PDEVICE_OBJECT pDevObj,PIRP pIrp)
    
63. {
    
64.     DbgPrint("DriverDemo:DriverUnload...\n");
    
65.     pIrp->IoStatus.Status=STATUS_SUCCESS;
    
66.     IoCompleteRequest(pIrp,IO_NO_INCREMENT);
    
67.     return STATUS_SUCCESS;
    
68. }
    

复制代码