ObReferenceObjectByName 使用

forwe

1. #include "ntddk.h"
   
2. 
   
3. #define INITCODE        code_seg("INIT")
   
4. #define PAGECODE        code_seg("PAGE")
   
5. 
   
6. 
   
7. NTKERNELAPI
   
8. NTSTATUS
   
9. ObReferenceObjectByName(
   
10.         IN PUNICODE_STRING ObjectName,
    
11.         IN ULONG Attributes,
    
12.         IN PACCESS_STATE PassedAccessState OPTIONAL,
    
13.         IN ACCESS_MASK DesiredAccess OPTIONAL,
    
14.         IN POBJECT_TYPE ObjectType,
    
15.         IN KPROCESSOR_MODE AccessMode,
    
16.         IN OUT PVOID ParseContext OPTIONAL,
    
17.         OUT PVOID *Object
    
18.         );
    
19. 
    
20. extern POBJECT_TYPE *IoDriverObjectType;
    
21. 
    
22. PDRIVER_OBJECT g_DriverObject;
    
23. PDRIVER_DISPATCH gfn_OrigReadCompleteRoutine;
    
24. 
    
25. #pragma PAGECODE
    
26. NTSTATUS FilterReadCompleteRoutine(PDEVICE_OBJECT pDeviceObj, PIRP pIrp)
    
27. {
    
28.         KdPrint(("IRP_MJ_DEVICE_CONTROL comming!"));
    
29.         return gfn_OrigReadCompleteRoutine(pDeviceObj, pIrp);
    
30. }
    
31. 
    
32. 
    
33. #pragma PAGECODE
    
34. void UnFilterDriverQuery()
    
35. {
    
36.         if (MmIsAddressValid(g_DriverObject))
    
37.         {
    
38.                 g_DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = gfn_OrigReadCompleteRoutine;
    
39.         }
    
40. }
    
41. 
    
42. 
    
43. #pragma PAGECODE
    
44. NTSTATUS FilterDriverQuery()
    
45. {
    
46.         UNICODE_STRING ObjectName;
    
47.         NTSTATUS Status;
    
48. 
    
49.         RtlInitUnicodeString(&ObjectName, L"\\Driver\\XueTr-火眼合作版本");
    
50. 
    
51.         Status = ObReferenceObjectByName(&ObjectName, OBJ_CASE_INSENSITIVE, NULL, 0,
    
52.                 *IoDriverObjectType, KernelMode, NULL, (PVOID *)&g_DriverObject);
    
53. 
    
54.         if (!NT_SUCCESS(Status))
    
55.         {
    
56.                 return Status;
    
57.         }
    
58. 
    
59.         gfn_OrigReadCompleteRoutine = g_DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL];
    
60.         g_DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = FilterReadCompleteRoutine;
    
61. 
    
62.         ObDereferenceObject(g_DriverObject);
    
63. 
    
64.         return STATUS_SUCCESS;
    
65. }
    
66. 
    
67. 
    
68. #pragma PAGECODE
    
69. void MyDriverUnload(PDRIVER_OBJECT pDriverObject)
    
70. {
    
71.         UNICODE_STRING symName;
    
72. 
    
73.         RtlInitUnicodeString(&symName, L"\\??\\FirstDeviceLinkName");
    
74. 
    
75.         UnFilterDriverQuery();
    
76. 
    
77.         if (pDriverObject->DeviceObject != NULL)
    
78.         {
    
79.                 IoDeleteSymbolicLink(&symName);
    
80.                 IoDeleteDevice(pDriverObject->DeviceObject);
    
81.                 KdPrint(("Delete device success!"));
    
82.         }
    
83. 
    
84. }
    
85. 
    
86. #pragma INITCODE
    
87. NTSTATUS CreateMyDevice(PDRIVER_OBJECT pDriverObject)
    
88. {
    
89.         NTSTATUS Status;
    
90.         PDEVICE_OBJECT pDevObj;
    
91.         UNICODE_STRING DevName;
    
92.         UNICODE_STRING symName;
    
93. 
    
94.         RtlInitUnicodeString(&DevName, L"\\Device\\FirstDevice");
    
95. 
    
96.         Status = IoCreateDevice(pDriverObject, 0, &DevName, FILE_DEVICE_UNKNOWN,
    
97.                 0, TRUE, &pDevObj);
    
98. 
    
99.         if (!NT_SUCCESS(Status))
    
100.         {
     
101.                 return Status;
     
102.         }
     
103.        
     
104.         pDevObj->Flags |= DO_BUFFERED_IO;
     
105. 
     
106.         RtlInitUnicodeString(&symName, L"\\??\\FirstDeviceLinkName");
     
107.         Status = IoCreateSymbolicLink(&symName, &DevName);
     
108. 
     
109.         if (!NT_SUCCESS(Status))
     
110.         {
     
111.                 IoDeleteDevice(pDevObj);
     
112.                 return Status;
     
113.         }
     
114.         return STATUS_SUCCESS;
     
115. }
     
116. 
     
117. #pragma PAGECODE
     
118. NTSTATUS CreateCompleteRoutine(PDEVICE_OBJECT pDeviceObj, PIRP pIrp)
     
119. {
     
120.         NTSTATUS Status;
     
121.         Status = STATUS_SUCCESS;
     
122. 
     
123.         KdPrint(("Create"));
     
124. 
     
125.         pIrp->IoStatus.Status = Status;
     
126.         pIrp->IoStatus.Information = 0;
     
127.         IoCompleteRequest(pIrp, IO_NO_INCREMENT);
     
128.         return Status;
     
129. }
     
130. 
     
131. #pragma PAGECODE
     
132. NTSTATUS CloseCompleteRoutine(PDEVICE_OBJECT pDeviceObj, PIRP pIrp)
     
133. {
     
134.         NTSTATUS Status;
     
135.         Status = STATUS_SUCCESS;
     
136. 
     
137.         KdPrint(("Close"));
     
138. 
     
139.         pIrp->IoStatus.Status = Status;
     
140.         pIrp->IoStatus.Information = 0;
     
141.         IoCompleteRequest(pIrp, IO_NO_INCREMENT);
     
142.         return Status;
     
143. }
     
144. 
     
145. #pragma PAGECODE
     
146. NTSTATUS ReadCompleteRoutine(PDEVICE_OBJECT pDeviceObj, PIRP pIrp)
     
147. {
     
148.         NTSTATUS Status;
     
149.         Status = STATUS_SUCCESS;
     
150. 
     
151.         KdPrint(("Read"));
     
152. 
     
153.         pIrp->IoStatus.Status = Status;
     
154.         pIrp->IoStatus.Information = 0;
     
155.         IoCompleteRequest(pIrp, IO_NO_INCREMENT);
     
156.         return Status;
     
157. }
     
158. 
     
159. #pragma PAGECODE
     
160. NTSTATUS WriteCompleteRoutine(PDEVICE_OBJECT pDeviceObj, PIRP pIrp)
     
161. {
     
162.         NTSTATUS Status;
     
163.         Status = STATUS_SUCCESS;
     
164. 
     
165.         KdPrint(("Write"));
     
166. 
     
167.         pIrp->IoStatus.Status = Status;
     
168.         pIrp->IoStatus.Information = 0;
     
169.         IoCompleteRequest(pIrp, IO_NO_INCREMENT);
     
170.         return Status;
     
171. }
     
172. 
     
173. #pragma INITCODE
     
174. NTSTATUS DriverEntry(PDRIVER_OBJECT pDriverObject, PUNICODE_STRING pRegistryPath)
     
175. {
     
176.         NTSTATUS Status;
     
177.         Status = CreateMyDevice(pDriverObject);
     
178. 
     
179.         if (!NT_SUCCESS(Status))
     
180.         {
     
181.                 KdPrint(("Create device failed!"));
     
182.         }
     
183.         else
     
184.         {
     
185.                 KdPrint(("Create device success!"));
     
186.                 KdPrint(("%wZ", pRegistryPath));
     
187.         }
     
188. 
     
189.         pDriverObject->MajorFunction[IRP_MJ_CREATE] = CreateCompleteRoutine;
     
190.         pDriverObject->MajorFunction[IRP_MJ_CLOSE] = CloseCompleteRoutine;
     
191.         pDriverObject->MajorFunction[IRP_MJ_READ] = ReadCompleteRoutine;
     
192.         pDriverObject->MajorFunction[IRP_MJ_WRITE] = WriteCompleteRoutine;
     
193. 
     
194.         pDriverObject->DriverUnload = MyDriverUnload;
     
195. 
     
196.         FilterDriverQuery();
     
197. 
     
198.         return STATUS_SUCCESS;
     
199. }

复制代码