码迷,mamicode.com
首页 > Windows程序 > 详细

Windows下获取逻辑cpu数量和cpu核数量(用GetLogicalProcessorInformation,从XP3才开始有的API)

时间:2016-12-08 02:30:43      阅读:364      评论:0      收藏:0      [点我收藏+]

标签:mat   name   ase   data-   malloc   attribute   proc   mod   orm   

代码可在Windows NT下正常运行

具体API说明请参照如下文档:

GetLogicalProcessorInformation

点击打开链接

点击打开链接

点击打开链接

 

 

[html] view plain copy
 
 技术分享技术分享
  1. typedef BOOL (WINAPI *LPFN_GLPI)(PSYSTEM_LOGICAL_PROCESSOR_INFORMATION, PDWORD);  
  2.   
  3. DWORD CountSetBits(ULONG_PTR bitMask)  
  4. {  
  5.     DWORD LSHIFT = sizeof(ULONG_PTR)*8 - 1;  
  6.     DWORD bitSetCount = 0;  
  7.     ULONG_PTR bitTest = (ULONG_PTR)1 <LSHIFT;      
  8.     DWORD i;  
  9.       
  10.     for (i = 0; i <= LSHIFT; ++i)  
  11.     {  
  12.         bitSetCount += ((bitMask & bitTest)?1:0);  
  13.         bitTest/=2;  
  14.     }  
  15.   
  16.     return bitSetCount;  
  17. }  
  18. LPFN_GLPI glpi;  
  19. glpi = (LPFN_GLPI) GetProcAddress(GetModuleHandle(TEXT("kernel32")),"GetLogicalProcessorInformation");  
  20. if (NULL == glpi)   
  21. {  
  22.     printf("GetLogicalProcessorInformation is not supported.\n");  
  23. }  
  24. BOOL done = FALSE;  
  25. PSYSTEM_LOGICAL_PROCESSOR_INFORMATION buffer = NULL;  
  26. PSYSTEM_LOGICAL_PROCESSOR_INFORMATION ptr = NULL;  
  27. DWORD returnLength = 0;  
  28. while (!done)  
  29. {  
  30.     DWORD rc = glpi(buffer, &returnLength);  
  31.   
  32.     if (FALSE == rc)   
  33.     {  
  34.         if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)   
  35.         {  
  36.             if (buffer)   
  37.                 free(buffer);  
  38.   
  39.             buffer = (PSYSTEM_LOGICAL_PROCESSOR_INFORMATION)malloc(returnLength);  
  40.   
  41.             if (NULL == buffer)   
  42.             {  
  43.                 printf("Error: Allocation failure\n");  
  44.                 return (2);  
  45.             }  
  46.         }   
  47.         else   
  48.         {  
  49.             printf("Error %d\n", GetLastError());  
  50.             return (3);  
  51.         }  
  52.     }   
  53.     else  
  54.     {  
  55.         done = TRUE;  
  56.     }  
  57. }  
  58.   
  59. ptr = buffer;  
  60. DWORD byteOffset = 0;  
  61. DWORD logicalProcessorCount = 0;  
  62. DWORD processorCoreCount = 0;  
  63.   
  64. while (byteOffset + sizeof(SYSTEM_LOGICAL_PROCESSOR_INFORMATION) <= returnLength)   
  65. {  
  66.     switch (ptr->Relationship)   
  67.     {  
  68.         case RelationProcessorCore:  
  69.             processorCoreCount++;  
  70.             logicalProcessorCount += CountSetBits(ptr->ProcessorMask);  
  71.             break;  
  72.     }  
  73.     byteOffset += sizeof(SYSTEM_LOGICAL_PROCESSOR_INFORMATION);  
  74.     ptr++;  
  75. }  
  76. printf("logical:%d  core:%d\n", logicalProcessorCount, processorCoreCount);  

http://blog.csdn.net/tobacco5648/article/details/22201169

Windows下获取逻辑cpu数量和cpu核数量(用GetLogicalProcessorInformation,从XP3才开始有的API)

标签:mat   name   ase   data-   malloc   attribute   proc   mod   orm   

原文地址:http://www.cnblogs.com/findumars/p/6143334.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!