VS2019遍历DPC定时器源码

万胜

VS2019遍历DPC定时器源码

1. #include <ntddk.h>
   
2. #include <windef.h>
   
3. 
   
4. /*---------------------------------自定义函数声明------------------------------------*/
   
5. VOID DriverUnload(IN PDRIVER_OBJECT DriverObject); //前置说明 卸载例程
   
6. ULONG QueryTimerTableListHead();
   
7. ULONG GetDpcTimerInformation_XP();
   
8. 
   
9. 
   
10. /*---------------------------------驱动入口函数驱动卸载函数实现------------------------------------*/
    
11. VOID DriverUnload(IN PDRIVER_OBJECT DriverObject)
    
12. {
    
13.         DbgPrint("[MyDriver]Unloaded...\n");
    
14.         return;
    
15. }
    
16. 
    
17. NTSTATUS DriverEntry(IN PDRIVER_OBJECT DriverObject, IN PUNICODE_STRING RegistryPath)
    
18. {
    
19.         DriverObject->DriverUnload = DriverUnload;
    
20.         DbgPrint("[MyDriver]Loaded...\n");
    
21.         GetDpcTimerInformation_XP();
    
22.         return STATUS_SUCCESS;
    
23. }
    
24. 
    
25. 
    
26. /*---------------------------------自定义函数实现------------------------------------*/
    
27. ULONG GetDpcTimerInformation_XP()
    
28. {
    
29.         ULONG  NumberOfTimerTable = NULL;
    
30.         ULONG  i;
    
31.         ULONG  ulCount = 0;
    
32.         PLIST_ENTRY  pList = NULL;
    
33.         PLIST_ENTRY pNextList = NULL;
    
34.         PKTIMER  pTimer = NULL;
    
35. 
    
36.         NumberOfTimerTable = 0x100;
    
37. 
    
38.         pList = (PLIST_ENTRY)QueryTimerTableListHead();                    //取得链表头
    
39.         if (pList == NULL)
    
40.         {
    
41.                 KdPrint(("timer table failed\n"));
    
42.                 return 0;
    
43.         }
    
44.         for (i = 0; i < NumberOfTimerTable; i++, pList++)                      //NumberOfTimerTable 个list
    
45.         {
    
46.                 if (!MmIsAddressValid((PVOID)&pList))
    
47.                 {
    
48.             KdPrint(("pList Failed\r\n"));
    
49.                         return NULL;
    
50.                 }
    
51.                 if (!MmIsAddressValid((PVOID)pList->Blink) ||
    
52.                         !MmIsAddressValid((PVOID)pList->Flink))
    
53.                 {
    
54.                         KdPrint(("Blink Failed\r\n"));
    
55.                         continue;      
    
56.                 }
    
57.                 for (pNextList = pList->Blink; pNextList != pList; pNextList = pNextList->Blink)    //遍历blink链
    
58.                 {
    
59. 
    
60. 
    
61.                         if (!MmIsAddressValid(pNextList))
    
62.                         {
    
63.                                 break;               
    
64.                         }
    
65.                         pTimer = CONTAINING_RECORD(pNextList, KTIMER, TimerListEntry);            //得到结构首
    
66. 
    
67.                         //检查pTimer以及各成员
    
68.                         if (!MmIsAddressValid((PVOID)pTimer))
    
69.                         {
    
70.                                 break;               
    
71.                         }
    
72.                         if (!MmIsAddressValid((PVOID)pTimer->Dpc))
    
73.                         {
    
74.                                 break;            
    
75.                         }
    
76.                         if (MmIsAddressValid((PVOID)pTimer) &&
    
77.                                 MmIsAddressValid((PVOID)pTimer->Dpc) &&
    
78.                                 MmIsAddressValid((PVOID)pTimer->Dpc->DeferredRoutine) &&
    
79.                                 MmIsAddressValid((PVOID)&pTimer->Period))                 
    
80.                         {
    
81.                                 DbgPrint("定时器对象:%08x 函数入口:%08x\n", pTimer, pTimer->Dpc->DeferredRoutine);
    
82.                                 ulCount++;                                
    
83.                         }
    
84.                         if (!MmIsAddressValid(pNextList->Blink))
    
85.                         {
    
86.                                 break;           
    
87.                         }
    
88.                 }
    
89.         }
    
90.         return ulCount;
    
91. }
    
92. 
    
93. 
    
94. ULONG QueryTimerTableListHead()
    
95. {
    
96.         UNICODE_STRING UnicodeTimerHead;
    
97.         ULONG ulTimerTable;
    
98.         PUCHAR i;
    
99.         ULONG ulTimerTableListHead;
    
100.         
     
101.         RtlInitUnicodeString(&UnicodeTimerHead, (PWCHAR)L"KeUpdateSystemTime");
     
102.         ulTimerTable = (ULONG)MmGetSystemRoutineAddress(&UnicodeTimerHead);
     
103.         if (ulTimerTable == 0) return 0;
     
104. 
     
105.         for (i = ulTimerTable; i < ulTimerTable + 200; i++)
     
106.         {
     
107.                         if (*i == 0x8d)
     
108.                         {
     
109.                                 ulTimerTableListHead = *(PULONG)(i + 3);
     
110.                                 if (MmIsAddressValid(ulTimerTableListHead))
     
111.                                 {
     
112.                                         DbgPrint("ulTimerTableListHead%08x", ulTimerTableListHead);
     
113.                                         return ulTimerTableListHead;
     
114.                                 }
     
115.                         }               
     
116.         }
     
117.         return NULL;
     
118. }

复制代码