Программно проверить количество ядер, используемых приложением

Я собираюсь сделать второй ответ здесь, потому что последний уже достаточно длинный, и этот ответ будет идти в несколько ином направлении.

После дальнейших исследований я определил, что на самом деле есть способ точно определить, на каком ядре / может / работает каждый поток. Код, который я придумал, использует специфичные для Windows библиотеки, но, безусловно, есть функции, эквивалентные Linux.

Более конкретно его использование wbemuuid.lib, comdef.h а также Wbemidl.h,

Код:

#define _WIN32_DCOM

#include <iostream>
#include <comdef.h>
#include <Wbemidl.h>
#include <cstdarg>
#include <string>

#pragma comment(lib, "wbemuuid.lib")

using namespace std;

DWORD affinity(unsigned int ID)
{
HANDLE threadh = OpenThread(THREAD_SET_INFORMATION | THREAD_QUERY_INFORMATION, FALSE, ID);

DWORD mask = 1;
DWORD old = 0;

while (mask)
{
old = SetThreadAffinityMask(threadh, mask);
if (old)
{
SetThreadAffinityMask(threadh, old);
return old;
}
else
{
if (GetLastError() != ERROR_INVALID_PARAMETER) return 0;
}
mask <<= 1;
}

return 0;
}

HRESULT connect(IWbemLocator** pLoc, IWbemServices** pSvc)
{
HRESULT hres;

hres = CoInitializeEx(0, COINIT_MULTITHREADED);
if (FAILED(hres))
{
cout << "Failed to initialize COM library. Error code = 0x" << hex << hres << endl;
return hres;
}

hres = CoInitializeSecurity(NULL, -1, NULL, NULL, RPC_C_AUTHN_LEVEL_DEFAULT, RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE, NULL);
if (FAILED(hres))
{
cout << "Failed to initialize security. Error code = 0x" << hex << hres << endl;
CoUninitialize();
return hres;
}

hres = CoCreateInstance( CLSID_WbemLocator, 0, CLSCTX_INPROC_SERVER, IID_IWbemLocator, (LPVOID*)&(*pLoc));
if (FAILED(hres))
{
cout << "Failed to create IWbemLocator object." << " Error code = 0x" << hex << hres << endl;
CoUninitialize();
return hres;
}

hres = (*pLoc)->ConnectServer(_bstr_t(L"ROOT\\CIMV2"), NULL, NULL, 0, NULL, 0, 0, &(*pSvc));
if (FAILED(hres))
{
cout << "Could not connect. Error code = 0x" << hex << hres << endl;
(*pLoc)->Release();
CoUninitialize();
return hres;
}

hres = CoSetProxyBlanket((*pSvc), RPC_C_AUTHN_WINNT, RPC_C_AUTHZ_NONE, NULL, RPC_C_AUTHN_LEVEL_CALL, RPC_C_IMP_LEVEL_IMPERSONATE, NULL, EOAC_NONE);

if (FAILED(hres))
{
cout << "Could not set proxy blanket. Error code = 0x" << hex << hres << endl;
(*pSvc)->Release();
(*pLoc)->Release();
CoUninitialize();
return hres;
}

return hres;
}

HRESULT query(IWbemLocator** pLoc, IWbemServices** pSvc, IEnumWbemClassObject** pEnum, const char* qry)
{
HRESULT hres;

hres = (*pSvc)->ExecQuery(bstr_t("WQL"), bstr_t(qry), WBEM_FLAG_FORWARD_ONLY | WBEM_FLAG_RETURN_IMMEDIATELY, NULL, &(*pEnum));

if (FAILED(hres))
{
cout << "Query for operating system name failed." << " Error code = 0x" << hex << hres << endl;
(*pSvc)->Release();
(*pLoc)->Release();
CoUninitialize();
return 1;
}

return hres;
}

HRESULT parse(IWbemLocator** pLoc, IWbemServices** pSvc, IEnumWbemClassObject** pEnum, IWbemClassObject** pCls, size_t n_args, ...)
{
HRESULT hres;

ULONG uReturn = 0;

while (pEnum)
{
hres = (*pEnum)->Next(WBEM_INFINITE, 1, &(*pCls), &uReturn);

if (0 == uReturn)
{
break;
}

VARIANT vtProp;

va_list vl;
va_start(vl, n_args);
for (size_t i = 0; i < n_args; i++)
{
const char* name = va_arg(vl, const char*);

int wchars_num =  MultiByteToWideChar(CP_UTF8, 0, name, -1, NULL, 0);
wchar_t* wname = new wchar_t[wchars_num];
MultiByteToWideChar(CP_UTF8 , 0, name, -1, wname, wchars_num);

hres = (*pCls)->Get(wname, 0, &vtProp, 0, 0);
wcout << wname << " : " << std::to_wstring((size_t)vtProp.bstrVal) << " : " << affinity((DWORD)vtProp.bstrVal) << endl;

delete[] wname;
}
va_end(vl);

VariantClear(&vtProp);
}

return hres;
}

int main(int argc, char **argv)
{
string qry = "SELECT * FROM Win32_PerfFormattedData_PerfProc_Thread WHERE IDProcess = 7424";

HRESULT hres;

IWbemLocator* pLoc = NULL;
IWbemServices* pSvc = NULL;

IEnumWbemClassObject* pEnum = NULL;

IWbemClassObject* pCls = NULL;

hres = connect(&pLoc, &pSvc);
if (FAILED(hres)) return 1;

hres = query(&pLoc, &pSvc, &pEnum, qry.c_str());
if (FAILED(hres)) return 1;

hres = parse(&pLoc, &pSvc, &pEnum, &pCls, 1, "IDThread");
if (FAILED(hres)) return 1;

pSvc->Release();
pLoc->Release();
pEnum->Release();
pCls->Release();
CoUninitialize();

return 0;
}

Выходы, когда Prime95 остановлен:

IDThread : 9072 : 15
IDThread : 7052 : 15

Выводится, когда Prime95 работает с 4 рабочими потоками:

IDThread : 9072 : 15
IDThread : 7052 : 15
IDThread : 5600 : 1
IDThread : 5888 : 2
IDThread : 2888 : 4
IDThread : 9348 : 8

PercentProcessorTime : 0
PercentProcessorTime : 0
PercentProcessorTime : 70
PercentProcessorTime : 83
PercentProcessorTime : 80
PercentProcessorTime : 75

Выводится, когда Prime95 работает с 2 рабочими потоками:

IDThread : 9072 : 15
IDThread : 7052 : 15
IDThread : 2352 : 15
IDThread : 8396 : 15

Объяснение:

Чтобы объяснить код немного:

• 7424 это PID Prime95.
• SELECT * FROM Win32_PerfFormattedData_PerfProc_Thread WHERE IDProcess = 7424 это запрос, который я использую, чтобы перечислить все темы, связанные с конкретным PID. Вы можете найти список всей информации, которую вы можете получить от Win32_PerfFormattedData_PerfProc_Thread Вот. Все, что вам нужно сделать, это переключить аргумент parse() от ThreadID чтобы сказать PercentProcessorTime например, и он собирается вывести процент использования процессора.
• Код очень уродливый и, возможно, небезопасный, это также сильно измененная версия Пример: получение данных WMI с локального компьютера из MSDN.

Affinity:

Функция affinity() устанавливает привязку потока к новому, чтобы получить старый, а затем устанавливает его обратно к старому. Теперь я не уверен, как получить фактический номер ядра из сходства, что я знаю, что если его, например, 1 он работает на ядре номер 1, если его 2 он работает с ядром № 2, если его 7 он работает на ядрах 4 и 3 или что-то в этом роде. Я еще не до конца разобрался.

Портирование на Linux:

В Linux все немного проще, например, получить ядро ​​можно с помощью чего-то вроде sched_getcpu/sched_getaffinity. Немного погуглив, я уверен, что вы можете найти метод, чтобы вывести список всех потоков, связанных с процессом.