[common] cpuinfo: implement for Windows

This commit is contained in:
Quantum 2021-08-30 20:54:05 -04:00 committed by Geoffrey McRae
parent 34fb2f9076
commit 0faafbff47
2 changed files with 95 additions and 0 deletions

View File

@ -36,6 +36,7 @@ add_library(lg_common_platform_code STATIC
windebug.c windebug.c
ivshmem.c ivshmem.c
time.c time.c
cpuinfo.c
) )
target_link_libraries(lg_common_platform_code target_link_libraries(lg_common_platform_code

View File

@ -0,0 +1,94 @@
/**
* Looking Glass
* Copyright © 2017-2021 The Looking Glass Authors
* https://looking-glass.io
*
* This program is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License as published by the Free
* Software Foundation; either version 2 of the License, or (at your option)
* any later version.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
* more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc., 59
* Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
#include "common/cpuinfo.h"
#include "common/debug.h"
#include "common/windebug.h"
#include <windows.h>
static void getProcessorCount(int * procs)
{
if (!procs)
return;
SYSTEM_INFO si;
GetSystemInfo(&si);
*procs = si.dwNumberOfProcessors;
}
static bool getCPUModel(char * model, size_t modelSize)
{
if (!model)
return true;
LRESULT lr;
DWORD cb = modelSize;
if ((lr = RegGetValueA(HKEY_LOCAL_MACHINE, "HARDWARE\\DESCRIPTION\\SYSTEM\\CentralProcessor\\0",
"ProcessorNameString", RRF_RT_REG_SZ, NULL, model, &cb)))
{
DEBUG_WINERROR("Failed to query registry", lr);
return false;
}
return true;
}
static bool getCoreCount(int * cores)
{
if (!cores)
return true;
DWORD cb = 0;
GetLogicalProcessorInformationEx(RelationProcessorCore, NULL, &cb);
if (GetLastError() != ERROR_INSUFFICIENT_BUFFER)
{
DEBUG_WINERROR("Failed to call GetLogicalProcessorInformationEx", GetLastError());
return false;
}
BYTE buffer[cb];
if (!GetLogicalProcessorInformationEx(RelationProcessorCore,
(PSYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX) buffer, &cb))
{
DEBUG_WINERROR("Failed to call GetLogicalProcessorInformationEx", GetLastError());
return false;
}
*cores = 0;
DWORD offset = 0;
while (offset < cb)
{
PSYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX lpi =
(PSYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX) (buffer + offset);
if (lpi->Relationship == RelationProcessorCore)
++*cores;
offset += lpi->Size;
}
return true;
}
bool lgCPUInfo(char * model, size_t modelSize, int * procs, int * cores)
{
getProcessorCount(procs);
return getCPUModel(model, modelSize) && getCoreCount(cores);
}