sidt指令怎样获取多处理器系统IDT表地址

b571013930 2013-10-26 03:49:23

  

 

 ULONG count = 0; 

    UNICODE_STRING   ustr;    PKDPC pdpc; 

    PsGetVersion(0,0,&g_dwBuildNumber,0); 

    //__asm int 3 

   //动态获取地址，因为获取cpu数目的方法在vista sp1即以后版本有所改变， 

  //如果直接使用KeQueryActiveProcessorCount将导致驱动在xp下无法加载 

   if (g_dwBuildNumber == 2600 || g_dwBuildNumber == 6000) 

     { 

       RtlInitUnicodeString(&ustr,L"KeNumberProcessors"); 

        PVOID p = MmGetSystemRoutineAddress(&ustr); 

        if (p!=0) 

        { 

            count = *(ULONG *)p; 

        }     

     } 

     else 

    { 

          KAFFINITY procs; 

          RtlInitUnicodeString(&ustr,L"KeQueryActiveProcessorCount"); 

         fnKeQueryActiveProcessorCount MyQueryActiveProcessorCount = (fnKeQueryActiveProcessorCount)MmGetSystemRoutineAddress(&ustr); 

        if (MyQueryActiveProcessorCount != 0) 

        { 

              count = MyQueryActiveProcessorCount(&procs); 

          } 



      } 

    if (count==1) 

      { 

         ShowIDTinfo(0,0,0,0); 

     } 

     else 

      { 

          ULONG currrent_pro_num = KeGetCurrentProcessorNumber();          ShowIDTinfo(0,currrent_pro_num,0,0); 

         g_FinshedDPC = 1; 

         PKDPC temp_dpc; 

          temp_dpc = (PKDPC)ExAllocatePoolWithTag(NonPagedPool,sizeof(KDPC)*count,'rm'); 

         pdpc = temp_dpc; 

       if (temp_dpc == NULL) 

           return status; 

        for (ULONG i = 0;i<count;i++,*temp_dpc++) 

          { 

            if (i!=currrent_pro_num) 

           { 

                //传入一个cpu号就好 

               KeInitializeDpc(temp_dpc,(PKDEFERRED_ROUTINE)ShowIDTinfo,(PVOID)i); 

               KeSetTargetProcessorDpc(temp_dpc,i); 

                KeInsertQueueDpc(temp_dpc,NULL,NULL); 

            } 

           

        } 

        //等待所有的dpc历程完事儿 

       while(InterlockedCompareExchange(&g_FinshedDPC,count,count) != count) 

         { 

             __asm nop 

         } 

         ExFreePoolWithTag(pdpc,'rm'); 

     } 

 

     return status; 

} 



 VOID ShowIDTinfo( 

                 IN struct _KDPC   *Dpc, 

                 IN ULONG cpuNum, 

                  IN PVOID   SystemArgument1, 

                  IN PVOID   SystemArgument2) 

{ 

      IDTINFO idt_info; 



     __asm sidt   idt_info 

 

   PIDTENTRY pIDTen =( PIDTENTRY)MAKELONG(idt_info.LowIDTbase,idt_info.HiIDTbase); 

      PIDTENTRY pIDTtmp; 

      ULONG addr; 



     for (ULONG i=0;i<0xFF;i++) 

      { 

          pIDTtmp = &pIDTen[i]; 

         addr = MAKELONG(pIDTtmp->LowOffset,pIDTtmp->HiOffset); 

        DbgPrint("CPU:%d        Index:%d        Addr:0x%x    DPL:%d\n",cpuNum,i,addr,pIDTtmp->DPL);         

     } 



     InterlockedIncrement(&g_FinshedDPC); 

 

}