Win XP/7/8 下 32/64-Bit 枚举多处理器 IDT

forwe

网上枚举多核IDT的代码或多或少都有些问题，以下代码在WinXP 32-Bit、Win7 32/64-Bit、Win8 32/64-Bit测试通过。

1. //
   
2. #define     HGMAKESIZE (a, b)               (( ULONG64)(((ULONG32 )(((ULONG64)( a)) & 0xFFFFFFFF)) | ((ULONG64)((ULONG32 )(((ULONG64)( b)) & 0xFFFFFFFF))) << 32))
   
3. //
   
4. // Define struct
   
5. //
   
6. typedef struct _HG_IDT_ENTRY
   
7. {
   
8.         unsigned short      LowOffset;
   
9.         unsigned short      selector;
   
10.         unsigned char       unused_lo;
    
11.         unsigned char       segment_type:4;                // 0x0E is an interrupt gate
    
12.         unsigned char       system_segment_flag:1;
    
13.         unsigned char       DPL:2;                               // descriptor privilege level
    
14.         unsigned char       P:1;                           // present
    
15. #if defined (_WIN64)
    
16.         UINT16                    MidOffset ;
    
17.         ULONG32                   HiOffset ;
    
18.         ULONG32                   Placeholder ;                   // not use
    
19. #else
    
20.         unsigned short      HiOffset;
    
21. #endif
    
22. } HG_IDT_ENTRY, * PHG_IDT_ENTRY ;
    
23. 
    
24. //
    
25. // sidt return idt in this format
    
26. //  Stores the interrupt descriptor table register (IDTR) in the destination operand. In legacy and
    
27. //  compatibility mode, the destination operand is 6 bytes; in 64-bit mode it is 10 bytes. In all modes,
    
28. //    operand-size prefixes are ignored.
    
29. //    In non-64-bit mode, the lower two bytes of the operand specify the 16-bit limit and the upper 4 bytes
    
30. //    specify the 32-bit base address.
    
31. //    In 64-bit mode, the lower two bytes of the operand specify the 16-bit limit and the upper 8 bytes
    
32. //    specify the 64-bit base address.
    
33. //
    
34. #pragma pack (1)   
    
35. typedef struct _HG_IDT_INFO
    
36. {
    
37.         unsigned short IDTLimit;
    
38. #if defined (_WIN64)
    
39.         ULONG32 LowIDTbase ;
    
40.         ULONG32 HiIDTbase ;
    
41. #else
    
42.         unsigned short LowIDTbase;
    
43.         unsigned short HiIDTbase;
    
44. #endif
    
45. }HG_IDT_INFO, * PHG_IDT_INFO;
    
46. #pragma pack ()
    
47. 
    
48. typedef ULONG (NTAPI * pfnKeQueryActiveProcessorCount)(PKAFFINITY ActiveProcessors);
    
49. 
    
50. KEVENT g_Event ;
    
51. ULONG_PTR g_CurrentCpuAffinity = 0;
    
52. 
    
53. //
    
54. // Private methods
    
55. //
    
56. VOID TraverseIdt (ULONG_PTR CpuOrdinal)
    
57. {
    
58.         HG_IDT_INFO idt ;
    
59.         SIZE_T i = 0;
    
60.         SIZE_T address = 0;
    
61.         PHG_IDT_ENTRY idt_entry = NULL;
    
62.         PHG_IDT_ENTRY idtTmp = NULL;
    
63. 
    
64. #if defined (_WIN64)
    
65.         __sidt(&idt );
    
66.         idt_entry = (PHG_IDT_ENTRY )HGMAKESIZE( idt.LowIDTbase , idt.HiIDTbase );
    
67. #else
    
68.         _asm{
    
69.              sidt idt
    
70.        }
    
71.         idt_entry = (PHG_IDT_ENTRY )MAKELONG( idt.LowIDTbase , idt.HiIDTbase );
    
72. #endif
    
73.       
    
74.         if (idt_entry )
    
75.        {
    
76.               for (i = 0; i <= 0xFF; i++)
    
77.              {
    
78.                     idtTmp = &idt_entry [i];
    
79. #if defined (_WIN64)
    
80.                     address = HGMAKESIZE (MAKELONG( idtTmp->LowOffset , idtTmp-> MidOffset), idtTmp->HiOffset );
    
81. #else
    
82.                     address = MAKELONG (idtTmp-> LowOffset, idtTmp->HiOffset );
    
83. #endif
    
84.                     IdtLog(("CPU Ordinal: %d, Index: %02X, Address: %p\n" , CpuOrdinal, i , address));
    
85.              }
    
86.        }
    
87. }
    
88. 
    
89. VOID TraverseIdtDpc (
    
90.         __in  struct _KDPC * Dpc,
    
91.         __in  ULONG DeferredContext,
    
92.         __in  PVOID SystemArgument1,
    
93.         __in  PVOID SystemArgument2)
    
94. {
    
95.         HG_IDT_INFO idt ;
    
96.         SIZE_T i = 0;
    
97.         SIZE_T address = 0;
    
98.         PHG_IDT_ENTRY idt_entry = NULL;
    
99.         PHG_IDT_ENTRY idtTmp = NULL;
    
100. 
     
101.         TraverseIdt(g_CurrentCpuAffinity );
     
102. 
     
103.         HgKeSetEvent(&g_Event , IO_NO_INCREMENT, FALSE);
     
104. }
     
105. 
     
106. VOID QueryCurrentIdt ()
     
107. {
     
108.         KAFFINITY CpuAffinity ;
     
109.         size_t nCpuCount = 0;
     
110.         size_t i = 0;
     
111.         KDPC Dpc ;
     
112. 
     
113.         CpuAffinity = HgKeQueryActiveProcessors ();
     
114. 
     
115.         for(i = 0; i < sizeof(KAFFINITY ); i ++){
     
116.               if ((CpuAffinity >> i) & 1){
     
117.                     nCpuCount ++;
     
118.              }
     
119.        }
     
120. 
     
121.         if (nCpuCount == 1){
     
122.               KIRQL OldIrql = KeRaiseIrqlToDpcLevel();
     
123.               TraverseIdt(0);
     
124.               KeLowerIrql(OldIrql );
     
125.        } else{
     
126.               for(i = 0; i < sizeof(KAFFINITY ); i ++){
     
127.                     if ((CpuAffinity >> i) & 1){
     
128.                           g_CurrentCpuAffinity = i ;
     
129.                           HgKeInitializeEvent(&g_Event , NotificationEvent, FALSE);
     
130.                           HgKeInitializeDpc(&Dpc , (PKDEFERRED_ROUTINE )TraverseIdtDpc, NULL);
     
131.                           HgKeSetTargetProcessorDpc(&Dpc , (CCHAR) i);
     
132.                           HgKeSetImportanceDpc(&Dpc , HighImportance);
     
133.                           HgKeInsertQueueDpc(&Dpc , NULL, NULL);
     
134. 
     
135.                           if (HgKeWaitForSingleObject (&g_Event, (KWAIT_REASON)0, 0, 0, 0) == STATUS_SUCCESS )
     
136.                          {
     
137.                                 continue;
     
138.                          }
     
139.                    }
     
140.              }
     
141.        }
     
142. }

复制代码