1 /*++
2 
3 Copyright (c) Microsoft Corporation
4 
5 Module Name:
6 
7     FxCallbackLock.hpp
8 
9 Abstract:
10 
11     This is the C++ header for the FxCallbackLock
12 
13     This represents an abstract class for implementing
14     the locks used when we callback into the device driver.
15 
16 Author:
17 
18 
19 
20 
21 Revision History:
22 
23 
24         Made it mode agnostic
25 
26 --*/
27 
28 #ifndef _FXCALLBACKLOCK_H_
29 #define _FXCALLBACKLOCK_H_
30 
31 extern "C" {
32 
33 #if defined(EVENT_TRACING)
34 #include "FxCallbackLock.hpp.tmh"
35 #endif
36 
37 }
38 
39 //
40 // Callback locks track current owner. This is used to determine if a callback
41 // event into the driver needs to be deferred to prevent recursive
42 // locking.
43 //
44 // A Callback lock supports recursive locking, but this is not exposed in
45 // the current Driver Frameworks <-> Device Driver interactions, and results
46 // in a verifier assert.
47 //
48 
49 class FxCallbackLock : public FxGlobalsStump {
50 
51 protected:
52     MxThread         m_OwnerThread;
53     ULONG            m_RecursionCount;
54 
55     // For Verifier
56     FxVerifierLock*  m_Verifier;
57 
58 public:
59 
60 
61 
62 
63 
64 
65 
66     KIRQL m_PreviousIrql;
67 
68 public:
69 
FxCallbackLock(PFX_DRIVER_GLOBALS FxDriverGlobals)70     FxCallbackLock(
71         PFX_DRIVER_GLOBALS FxDriverGlobals
72         ) :
73         FxGlobalsStump(FxDriverGlobals)
74     {
75         m_OwnerThread = NULL;
76         m_RecursionCount = 0;
77         m_Verifier = NULL;
78     }
79 
~FxCallbackLock()80     virtual ~FxCallbackLock()
81     {
82     }
83 
84     virtual
85     void
86     Initialize(
87         FxObject* ParentObject
88         ) = 0;
89 
90     virtual
91     void
92     Lock(
93         __out PKIRQL PreviousIrql
94         ) = 0;
95 
96     virtual
97     void
98     Unlock(
99         __in KIRQL PreviousIrql
100         ) = 0;
101 
102     // TRUE if the current thread is the owner
103     _Must_inspect_result_
104     virtual
105     BOOLEAN
106     IsOwner(
107         VOID
108         ) = 0;
109 
110     VOID
CheckOwnership(VOID)111     CheckOwnership(
112         VOID
113         )
114     {
115         PFX_DRIVER_GLOBALS pFxDriverGlobals;
116 
117         pFxDriverGlobals = GetDriverGlobals();
118 
119         //
120         // If verify locks is on, catch drivers
121         // returning with the lock released.
122         //
123         if (pFxDriverGlobals->FxVerifierLock) {
124             if (IsOwner() == FALSE) {
125                 DoTraceLevelMessage(
126                     pFxDriverGlobals, TRACE_LEVEL_ERROR, TRACINGDEVICE,
127                     "Callback: Driver released the callback lock 0x%p",
128                     this);
129                 FxVerifierDbgBreakPoint(pFxDriverGlobals);
130             }
131         }
132     }
133 };
134 
135 
136 
137 
138 
139 
140 
141 
142 
143 
144 
145 
146 
147 
148 
149 
150 
151 
152 
153 
154 
155 
156 
157 
158 
159 
160 
161 
162 
163 
164 
165 
166 
167 
168 
169 
170 
171 
172 
173 
174 
175 
176 
177 
178 
179 
180 
181 
182 
183 
184 
185 
186 
187 
188 
189 
190 
191 
192 
193 
194 
195 
196 
197 
198 
199 
200 
201 
202 
203 
204 
205 
206 
207 
208 
209 
210 
211 
212 
213 
214 
215 
216 
217 
218 
219 
220 
221 
222 
223 
224 
225 
226 
227 
228 
229 
230 
231 
232 
233 
234 
235 
236 
237 
238 
239 
240 
241 
242 
243 
244 
245 
246 
247 
248 
249 
250 
251 
252 
253 
254 
255 
256 
257 
258 
259 
260 
261 
262 
263 
264 
265 
266 
267 
268 
269 
270 
271 
272 
273 
274 
275 
276 
277 
278 
279 
280 
281 
282 
283 
284 
285 
286 
287 
288 
289 
290 
291 
292 
293 
294 
295 
296 
297 
298 
299 
300 
301 
302 
303 
304 
305 
306 
307 
308 
309 
310 
311 
312 
313 
314 
315 
316 
317 
318 
319 
320 
321 
322 
323 
324 
325 
326 
327 
328 
329 
330 
331 
332 
333 
334 
335 
336 
337 
338 
339 
340 
341 
342 
343 
344 
345 
346 
347 
348 
349 
350 
351 
352 
353 
354 
355 
356 
357 
358 
359 
360 
361 
362 
363 
364 
365 
366 
367 
368 
369 
370 
371 
372 
373 
374 
375 
376 
377 
378 
379 
380 
381 
382 
383 
384 
385 
386 
387 
388 
389 
390 
391 
392 
393 
394 
395 
396 
397 
398 
399 
400 
401 
402 
403 
404 
405 
406 
407 
408 
409 
410 
411 
412 
413 
414 
415 
416 
417 
418 
419 
420 
421 
422 
423 
424 
425 
426 
427 
428 
429 
430 
431 
432 
433 
434 
435 
436 
437 
438 
439 
440 
441 
442 
443 
444 
445 
446 
447 
448 
449 
450 
451 
452 
453 
454 
455 
456 
457 
458 
459 
460 
461 
462 
463 
464 
465 
466 
467 
468 
469 
470 
471 
472 
473 
474 
475 
476 
477 
478 
479 
480 
481 
482 
483 
484 
485 
486 
487 
488 
489 
490 
491 
492 
493 
494 
495 
496 
497 
498 
499 
500 
501 
502 
503 
504 
505 
506 
507 
508 
509 
510 
511 
512 
513 
514 
515 
516 
517 
518 
519 
520 
521 
522 
523 
524 
525 
526 
527 
528 
529 
530 
531 
532 
533 
534 
535 
536 
537 
538 
539 
540 
541 
542 
543 
544 
545 
546 
547 
548 
549 
550 
551 
552 
553 
554 
555 
556 
557 
558 
559 
560 
561 
562 
563 
564 
565 
566 
567 
568 
569 
570 
571 
572 
573 
574 
575 
576 
577 
578 
579 
580 
581 
582 
583 
584 
585 
586 
587 
588 
589 
590 
591 
592 
593 
594 
595 
596 
597 
598 
599 
600 
601 
602 
603 
604 
605 
606 
607 
608 
609 
610 
611 
612 
613 
614 
615 
616 
617 
618 
619 
620 
621 
622 
623 
624 
625 
626 
627 
628 
629 
630 
631 
632 
633 
634 
635 
636 
637 
638 
639 
640 
641 
642 
643 
644 
645 
646 
647 
648 #endif // _FXCALLBACKLOCK_H_
649