郁金香驱动学习>第三课：添加设备对象

ymxkybqw

1. 
   
2. 
   
3. 
   
4. #include <ntddk.h>
   
5. 
   
6. //定义个INITCODE宏，这个宏表示 初始化的时候载入内存，然后可以从内存中卸载掉
   
7. #define INITCODE code_seg("INIT") //;注意这里是不能加分号的
   
8. #define PAGECODE code_seg("PAGE") //表示内存不够用时，可以置换到虚拟内存(硬盘)
   
9. 
   
10. //驱动设备
    
11. #pragma INITCODE //表示这个函数运行后，就从内存中释放掉
    
12. NTSTATUS CreateMyDevice (IN PDRIVER_OBJECT pDriverObject)
    
13. {
    
14.         NTSTATUS status; // NTSTATUS 等价于 LONG
    
15.         PDEVICE_OBJECT pDevObj;/*用来返回创建设备*/
    
16. 
    
17.         //UNICODE_STRING 类型为UNICODE的字符结构，里面包含3个成员，第一个Length 表示buffer字节的长度，不包含NULL，第二个MaximumLength：buffer总的字节大小，第三个Buffer：指向宽字符的指针
    
18.         //创建设备名称
    
19.         UNICODE_STRING devName;  
    
20.         UNICODE_STRING symLinkName;
    
21. 
    
22.         //会动态分配一块指向“\\Device\\yjxDDK_Device”的内存指针，赋值给devName.Buffer;
    
23.         RtlInitUnicodeString(&devName,L"\\Device\\yjxDDK_Device");/*对devName初始化字串为 "\\Device\\yjxDDK_Device"*/
    
24. 
    
25.         //IoCreateDevice 创建设备对象，并更新pDriverObject->DeviceObject 为新创建的设备对象
    
26.         //介绍每个参数
    
27.         // pDriverObject  该参数用于在驱动程序和新设备对象之间建立连接，
    
28.         // 0              第二个参数是设备扩展结构的大小
    
29.         // &devName       命名该设备对象的UNICODE_STRING串的地址
    
30.         // &FILE_DEVICE_UNKNOWN       设备类型
    
31.         // 0              为设备对象提供Characteristics标志
    
32.         // TRUE           指出设备是否是排斥的
    
33.         // &pDevObj       设备对象的名字，有了名字应用程序就可以访问该设备的驱动对象了
    
34.         status = IoCreateDevice( pDriverObject,\
    
35.                 0,\
    
36.                 &devName,\
    
37.                 FILE_DEVICE_UNKNOWN,\
    
38.                 0, TRUE,\
    
39.                 &pDevObj);
    
40. 
    
41.         //NT_SUCCESS宏定义：表示判断 status>=0
    
42.         if (!NT_SUCCESS(status)) //成功返回0  0>=0 吗 true 取反  false 不执行
    
43.         {
    
44.                 if(status == STATUS_INSUFFICIENT_RESOURCES)
    
45.                 {
    
46.                         KdPrint(("STATUS_INSUFFICIENT_RESOURCES 资源不足"));
    
47.                 }
    
48.                 if(status == STATUS_OBJECT_NAME_EXISTS)
    
49.                 {
    
50.                         KdPrint(("STATUS_OBJECT_NAME_EXISTS 指定对象不存在"));
    
51.                 }
    
52.                 if(status == STATUS_OBJECT_NAME_COLLISION)
    
53.                 {
    
54.                         KdPrint(("STATUS_INSUFFICIENT_RESOURCES 对象名有冲突"));
    
55.                 }
    
56.                 KdPrint(("创建设备错误，返回值是=%d ",status));
    
57.                 return status;
    
58.         }
    
59.                
    
60.         KdPrint(("创建设备成功，返回值是=%d ",status));
    
61. 
    
62.         //所创建设备被设定为直接缓冲I/O,这是设备读写三种方式中的一种
    
63.         pDevObj->Flags |= DO_BUFFERED_IO;
    
64.         //创建符号链接
    
65. 
    
66.         RtlInitUnicodeString(&symLinkName,L"\\??\\yjx888");
    
67. 
    
68.         //创建一个设备链接。驱动程序虽然有了设备名称，但是这种设备名只能在内核状态可见，而对于应用程序是不可见的，因此，驱动需要要暴露一个符号链接,来方便应用程序的访问，该链接指向真正的设备名称
    
69.         status = IoCreateSymbolicLink( &symLinkName,&devName );
    
70.         if (!NT_SUCCESS(status))
    
71.         {
    
72.                 KdPrint(("创建符号连接错误，返回值是=%d ",status));
    
73.                 //删除设备对象
    
74.                 IoDeleteDevice( pDevObj );
    
75.                 return status;
    
76.         }
    
77.         KdPrint(("创建符号连接成功，返回值是=%d ",status));
    
78.         return STATUS_SUCCESS;
    
79. }
    
80. 
    
81. 
    
82. 
    
83. //卸载函数的前置说明
    
84. VOID DDK_Unload(IN PDRIVER_OBJECT pDriverObject);  
    
85. 
    
86. //设置函数代码存放的代码段，这是是放在init段中
    
87. #pragma INITCODE
    
88. //NTSTATUS 等价与 LONG  
    
89. NTSTATUS DriverEntry(PDRIVER_OBJECT pDriverObject, PUNICODE_STRING B)
    
90. {
    
91.         KdPrint(("驱动成功被加载..."));
    
92. 
    
93.         //创建设备对象
    
94.         CreateMyDevice(pDriverObject);
    
95. 
    
96.         //Driverunload 回调函数的一个指针,这里的回调函数是卸载驱动的函数
    
97.         pDriverObject->DriverUnload = DDK_Unload;
    
98.         return 1;
    
99. }
    
100. VOID DDK_Unload(IN PDRIVER_OBJECT pDriverObject)
     
101. {
     
102.        
     
103.         //KdPrint使用方法类似printf，注意KdPrint((" ",  ))；使用的是双括号。
     
104.         //这个比DbgPrint 调用要稍好。因为在free 版不被编译。
     
105.         KdPrint(("驱动成功被卸载..."));
     
106. 
     
107.         DbgPrint("DbgPrint功能与KdPrint相同");
     
108. }
     

复制代码