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 ustrSymbolicName;
16 }DEVICE_EXTENSION, *PDEVICE_EXTENSION;
17
18 #define PAGEDCODE code_seg("PAGE")
19 #define LOCKEDCODE code_seg()
20 #define INITCODE code_seg("INIT")
21 #define PAGEDDATA data_seg("PAGE")
22 #define LOCKEDDATA data_seg();
23 #define INITDATA data_seg("INIT")
24
25 #define arraysize(p) (sizeof(p) / sizeof((p)[0]))
26 #pragma PAGEDCODE
27
28
29 //////////////////////////////////////////////////////////////////////////
30 // 添加设备的处理例程
31 #pragma PAGEDCODE
32 NTSTATUS HelloWDMAddDevice(IN PDRIVER_OBJECT pDriverObject, IN PDEVICE_OBJECT PhysicalDeviceObject)
33 {
34 PAGED_CODE(); // PAGED_CODE()宏确保本函数运行在低于APC_LEVEL的中断优先级级别上
35 KdPrint(("Enter HelloWDMAddDevice\n"));
36
37 NTSTATUS status;
38 PDEVICE_OBJECT fdo;
39 UNICODE_STRING devName;
40
41 RtlInitUnicodeString(&devName, L"\\Device\\MyWDMDevice");
42 IoCreateDevice(pDriverObject, sizeof(DEVICE_EXTENSION), &devName, FILE_DEVICE_UNKNOWN,
43 0, FALSE, &fdo);
44
45 PDEVICE_EXTENSION pdx = (PDEVICE_EXTENSION)fdo->DeviceExtension;
46 pdx->fdo = fdo;
47 pdx->ustrDeviceName = devName;
48 // 将fdo附加到pdo上
49 pdx->NextStackDevice = IoAttachDeviceToDeviceStack(fdo, PhysicalDeviceObject);
50
51 UNICODE_STRING symName;
52 RtlInitUnicodeString(&symName, L"\\DosDevice\\helloWDM");
53 status = IoCreateSymbolicLink(&symName, &devName);
54
55 if (!NT_SUCCESS(status))
56 {
57 IoDeleteSymbolicLink(&symName);
58 status = IoCreateSymbolicLink(&symName, &devName);
59 if (!NT_SUCCESS(status))
60 {
61 return status;
62 }
63 }
64
65 fdo->Flags |= DO_BUFFERED_IO | DO_POWER_PAGABLE;
66 fdo->Flags &= ~DO_DEVICE_INITIALIZING;
67
68 KdPrint(("Leaving AddDevice!"));
69 return STATUS_SUCCESS;
70 }
71 //////////////////////////////////////////////////////////////////////////
72 // 对PNP IRP进行默认处理(叫下层驱动对此pnp进行处理)
73 #pragma PAGEDCODE
74 NTSTATUS DefaultPnpHandler(PDEVICE_EXTENSION pdx, PIRP Irp)
75 {
76 PAGED_CODE();
77 KdPrint(("Enter DefaultPnpHandler..."));
78 IoSkipCurrentIrpStackLocation(Irp);
79 KdPrint(("Leaving DefaultPnpHandler"));
80 return IoCallDriver(pdx->NextStackDevice, Irp);
81 }
82
83
84 //////////////////////////////////////////////////////////////////////////
85 // IRP_MN_PNP_REMOVE_DEVICE这个irp的处理例程
86 #pragma PAGEDCODE
87 NTSTATUS HandleRemoveDevice(PDEVICE_EXTENSION pdx, PIRP Irp)
88 {
89 PAGED_CODE();
90 KdPrint(("Entering HandleRemoveDevice"));
91
92 Irp->IoStatus.Status = STATUS_SUCCESS;
93 NTSTATUS status = DefaultPnpHandler(pdx, Irp);
94 IoDeleteSymbolicLink(&pdx->ustrSymbolicName);
95
96 // 调用IoDetachDevice将fdo从设备栈中脱离
97 if (pdx->NextStackDevice)
98 {
99 IoDetachDevice(pdx->NextStackDevice);
100 }
101
102 // 删除FDO
103 IoDeleteDevice(pdx->fdo);
104 KdPrint(("Leave HandleRemoveDevice"));
105 return status;
106 }
107
108 #pragma PAGEDCODE
109 NTSTATUS HelloWDMDispatchRoutine(IN PDEVICE_OBJECT fdo,
110 IN PIRP Irp)
111 {
112 PAGED_CODE();
113 KdPrint(("Entering HelloWDMDispatchRoutine..."));
114 Irp->IoStatus.Status = STATUS_SUCCESS;
115 Irp->IoStatus.Information = 0;
116 IoCompleteRequest(Irp, IO_NO_INCREMENT);
117 KdPrint(("leaving HelloWDMDispatchRoutine..."));
118 return STATUS_SUCCESS;
119 }
120
121 #pragma PAGEDCODE
122 void HelloWDMUnload(IN PDRIVER_OBJECT pDriverObject)
123 {
124 PAGED_CODE();
125 KdPrint(("Enter HelloWDMUnload"));
126 KdPrint(("Leaving HelloWDMUnload"));
127 }
128
129 //////////////////////////////////////////////////////////////////////////
130 // pnp处理例程
131 #pragma PAGEDCODE
132 NTSTATUS HelloWDMPnp(IN PDEVICE_OBJECT fdo, IN PIRP Irp)
133 {
134 PAGED_CODE();
135
136 KdPrint(("Entering HelloWDMPnp.."));
137 NTSTATUS status = STATUS_SUCCESS;
138 PDEVICE_EXTENSION pdx = (PDEVICE_EXTENSION)fdo->DeviceExtension;
139 PIO_STACK_LOCATION stack = IoGetCurrentIrpStackLocation(Irp);
140
141 static NTSTATUS (*fcntab[])(PDEVICE_EXTENSION pdx, PIRP Irp) =
142 {
143 DefaultPnpHandler, // IRP_MN_START_DEVICE
144 DefaultPnpHandler, // IRP_MN_QUERY_REMOVE_DEVICE
145 HandleRemoveDevice, // IRP_MN_REMOVE_DEVICE !!注意，在这里添加了相应的处理程序
146 DefaultPnpHandler, // IRP_MN_CANCEL_REMOVE_DEVICE
147 DefaultPnpHandler, // IRP_MN_STOP_DEVICE
148 DefaultPnpHandler, // IRP_MN_QUERY_STOP_DEVICE
149 DefaultPnpHandler, // IRP_MN_CANCEL_STOP_DEVICE
150 DefaultPnpHandler, // IRP_MN_QUERY_DEVICE_RELATIONS
151 DefaultPnpHandler, // IRP_MN_QUERY_INTERFACE
152 DefaultPnpHandler, // IRP_MN_QUERY_CAPABILITIES
153 DefaultPnpHandler, // IRP_MN_QUERY_RESOURCES
154 DefaultPnpHandler, // IRP_MN_QUERY_RESOURCE_REQUIREMENTS
155 DefaultPnpHandler, // IRP_MN_QUERY_DEVICE_TEXT
156 DefaultPnpHandler, // IRP_MN_FILTER_RESOURCE_REQUIREMENTS
157 DefaultPnpHandler,
158 DefaultPnpHandler, // IRP_MN_READ_CONFIG
159 DefaultPnpHandler, // IRP_MN_WRITE_CONFIG
160 DefaultPnpHandler, // IRP_MN_EJECT
161 DefaultPnpHandler, // IRP_MN_SET_LOCK
162 DefaultPnpHandler, // IRP_MN_QUERY_ID
163 DefaultPnpHandler, // IRP_MN_QUERY_PNP_DEVICE_STAT
164 DefaultPnpHandler, // IRP_MN_QUERY_BUS_INFORMATION
165 DefaultPnpHandler, // IRP_MN_DEVICE_USAGE_NOTIFICATION
166 DefaultPnpHandler, // IRP_MN_SURPRISE_REMOVAL
167 };
168
169 ULONG fcn = stack->MinorFunction;
170 if (fcn >= arraysize(fcntab))
171 {
172 // 未知的IRP类型
173 status = DefaultPnpHandler(pdx, Irp);
174 return status;
175 }
176
177 #if DBG
178 static char * fcnname[] =
179 {
180 "IRP_MN_START_DEVICE",
181 "IRP_MN_QUERY_REMOVE_DEVICE",
182 "IRP_MN_REMOVE_DEVICE",
183 "IRP_MN_CANCEL_REMOVE_DEVICE",
184 "IRP_MN_STOP_DEVICE",
185 "IRP_MN_QUERY_STOP_DEVICE",
186 "IRP_MN_CANCEL_STOP_DEVICE",
187 "IRP_MN_QUERY_DEVICE_RELATIONS",
188 "IRP_MN_QUERY_INTERFACE",
189 "IRP_MN_QUERY_CAPABILITIES",
190 "IRP_MN_QUERY_RESOURCES",
191 "IRP_MN_QUERY_RESOURCE_REQUIREMENTS",
192 "IRP_MN_QUERY_DEVICE_TEXT",
193 "IRP_MN_FILTER_RESOURCE_REQUIREMENTS",
194 "",
195 "IRP_MN_READ_CONFIG",
196 "IRP_MN_WRITE_CONFIG",
197 "IRP_MN_EJECT",
198 "IRP_MN_SET_LOCK",
199 "IRP_MN_QUERY_ID",
200 "IRP_MN_QUERY_PNP_DEVICE_STAT",
201 "IRP_MN_QUERY_BUS_INFORMATION",
202 "IRP_MN_DEVICE_USAGE_NOTIFICATION",
203 "IRP_MN_SURPRISE_REMOVAL"
204 };
205
206 KdPrint(("PNP Request(%s)\n", fcnname[fcn]));
207 #endif //DBG
208 status = (*fcntab[fcn])(pdx, Irp);
209 KdPrint(("Leave helloWDM Pnp\n"));
210
211 return status;
212 }
213
214 //////////////////////////////////////////////////////////////////////////
215 // 驱动程序入口函数
216 extern "C" NTSTATUS DriverEntry(PDRIVER_OBJECT DriverObject, PUNICODE_STRING RegistryPath)
217 {
218 KdPrint(("Enter DriverEntry -- gmh"));
219
220 DriverObject->DriverExtension->AddDevice = HelloWDMAddDevice;
221 DriverObject->MajorFunction[IRP_MJ_PNP] = HelloWDMPnp;
222 DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] =
223 DriverObject->MajorFunction[IRP_MJ_CREATE] =
224 DriverObject->MajorFunction[IRP_MJ_READ] =
225 DriverObject->MajorFunction[IRP_MJ_WRITE] = HelloWDMDispatchRoutine;
226 DriverObject->DriverUnload = HelloWDMUnload;
227
228 KdPrint(("Leaving DriverEntry -- gmh!"));
229 return STATUS_SUCCESS;
230 }
