代码可在Windows NT下正常运行
具体API说明请参照如下文档:
GetLogicalProcessorInformation
[html] view plain copy
- typedef BOOL (WINAPI *LPFN_GLPI)(PSYSTEM_LOGICAL_PROCESSOR_INFORMATION, PDWORD);
- DWORD CountSetBits(ULONG_PTR bitMask)
- {
- DWORD LSHIFT = sizeof(ULONG_PTR)*8 - 1;
- DWORD bitSetCount = 0;
- ULONG_PTR bitTest = (ULONG_PTR)1 << LSHIFT;
- DWORD i;
- for (i = 0; i <= LSHIFT; ++i)
- {
- bitSetCount += ((bitMask & bitTest)?1:0);
- bitTest/=2;
- }
- return bitSetCount;
- }
- LPFN_GLPI glpi;
- glpi = (LPFN_GLPI) GetProcAddress(GetModuleHandle(TEXT("kernel32")),"GetLogicalProcessorInformation");
- if (NULL == glpi)
- {
- printf("GetLogicalProcessorInformation is not supported.\n");
- }
- BOOL done = FALSE;
- PSYSTEM_LOGICAL_PROCESSOR_INFORMATION buffer = NULL;
- PSYSTEM_LOGICAL_PROCESSOR_INFORMATION ptr = NULL;
- DWORD returnLength = 0;
- while (!done)
- {
- DWORD rc = glpi(buffer, &returnLength);
- if (FALSE == rc)
- {
- if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
- {
- if (buffer)
- free(buffer);
- buffer = (PSYSTEM_LOGICAL_PROCESSOR_INFORMATION)malloc(returnLength);
- if (NULL == buffer)
- {
- printf("Error: Allocation failure\n");
- return (2);
- }
- }
- else
- {
- printf("Error %d\n", GetLastError());
- return (3);
- }
- }
- else
- {
- done = TRUE;
- }
- }
- ptr = buffer;
- DWORD byteOffset = 0;
- DWORD logicalProcessorCount = 0;
- DWORD processorCoreCount = 0;
- while (byteOffset + sizeof(SYSTEM_LOGICAL_PROCESSOR_INFORMATION) <= returnLength)
- {
- switch (ptr->Relationship)
- {
- case RelationProcessorCore:
- processorCoreCount++;
- logicalProcessorCount += CountSetBits(ptr->ProcessorMask);
- break;
- }
- byteOffset += sizeof(SYSTEM_LOGICAL_PROCESSOR_INFORMATION);
- ptr++;
- }
- printf("logical:%d core:%d\n", logicalProcessorCount, processorCoreCount);
http://blog.csdn.net/tobacco5648/article/details/22201169
时间: 2024-10-13 06:56:03