Как получить загруженность процессора


#include <stdio.h>
#include <pdh.h>
#include <conio.h>

/*****************************************************************************/
/********                                                             ********/
/********           не забудьте включить в проект "pdh.lib"           ********/
/********                                                             ********/
/********    Project -> Settings -> Link -> Object/Library Modules    ********/
/********                                                             ********/
/*****************************************************************************/


/*****************************************************************************/
/**********                                                         **********/
/**********            PDH - performance data helper                **********/
/**********                                                         **********/
/*****************************************************************************/


/************************** Описание наших функций ***************************/

#define PDH_ERROR -1

// Функция инициализирует PDH (performance data helper)
PDH_STATUS pdh_Init (HQUERY &hQuery, HCOUNTER &hCounter);



// Функция возвращает в случае удачи загруженность проца, в противном случае PHG_ERROR (-1)
int GetCPUUsage (HQUERY hQuery, HCOUNTER hCounter, PDH_STATUS &pdhStatus);


// Функция анинициализирует PDH
void pdh_Uninit (HQUERY hQuery, HCOUNTER hCounter);


// Функция показывает сообщение об ошибке
void ShowErrorMessage(PDH_STATUS pdhStatus);


// хмм... в названии функции все сказано!
void begin();

/****************************************************************************/


void main(void)
{
    
begin();
    
    
printf(TEXT("Press any key to continue")); // нажмите "any key" на клавиатуре :P
    
while (!_getch()); // ждем, пока юзверь не "клацнет" "any key"
}

void begin()
// хмм... в названии функции все сказано!
{
    
PDH_STATUS  pdhStatus      = 0;

    
HQUERY hQuery;
    
HQUERY hCounter;
    
int iUsage = 0;
    
pdhStatus = pdh_Init(hQuery, hCounter); // инициализируем PDH

// если не произошел облом с иницализации PDH, то
    
if ( pdhStatus == ERROR_SUCCESS )
    {
// пока юзверь не зажмет клавишу Esc, то
        
while (GetKeyState(VK_ESCAPE) >= 0)
        {
// получаем загруженность проца
            
iUsage = GetCPUUsage(hQuery, hCounter, pdhStatus);

// если не произошла ошибка (т.е. не вернулось значение PHG_ERROR)
            
if (iUsage != PDH_ERROR)
                
printf("%3i%%\r\n", iUsage); // выводим значение
            
else // а если произошла ошибка, то

// показываем сообщение, что произошла ошибка такая-то
            
ShowErrorMessage(pdhStatus);

            
Sleep(500); // ждем 0.5 секунд(ы)
        
}

// если юзверь все-таки клацнул Esc, то флаг ему в руки :)
        
pdh_Uninit(hQuery, hCounter); // заканчиваем работу с PDH
    
}
    
else // если произошла ошибка при инициализации PDH
        
ShowErrorMessage(pdhStatus); // показываем Еррорное сообщение
}

void ShowErrorMessage(PDH_STATUS pdhStatus)
// Сама функция показа сообщения об ошибке
{
    
LPSTR szMessage = NULL;

// грузим строку с ошибкой в szMessage. Память выделит функция
    
FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER |
                    
FORMAT_MESSAGE_FROM_HMODULE,
                    
GetModuleHandle("PDH.DLL"), pdhStatus,
                    
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
                    
szMessage, 0, NULL);

    
printf("Error №%i:  %s\r\n", pdhStatus, szMessage); // показываем ошибку

    
LocalFree(szMessage); // очищаем буфер
}

PDH_STATUS pdh_Init(HQUERY &hQuery, HCOUNTER &hCounter)
// инициализируем PDH
{
    
PDH_STATUS  pdhResult   = 0;
    
TCHAR       szCounterPath[1024];
    
DWORD       dwPathSize  = 1024;
    
PDH_COUNTER_PATH_ELEMENTS pe;
    
DWORD       dwType      = 0;
    
    
pdhResult = PdhOpenQuery( NULL, 0, &hQuery ); // запрос на открытие PDH
    
    
if ( pdhResult != ERROR_SUCCESS ) // если была ошибка, то
        
return pdhResult; // выходим, возвращая номер ошибки

    
pe.szMachineName     = 0l;
    
pe.szObjectName      = "Processor";
    
pe.szInstanceName    = "_Total";
    
pe.szParentInstance  = NULL;
    
pe.dwInstanceIndex   = 0;
    
pe.szCounterName     = "% Processor Time";

// создаем путь к счетчику PDH
    
pdhResult = PdhMakeCounterPath(&pe, szCounterPath, &dwPathSize, 0);

    
if (pdhResult != ERROR_SUCCESS) // если была ошибка, то
        
return pdhResult; // выходим, возвращая номер ошибки

// открываем счетчик PDH
    
pdhResult = PdhAddCounter(hQuery, szCounterPath, 0, &hCounter);

    
if (pdhResult != ERROR_SUCCESS) // если была ошибка, то
        
return pdhResult; // выходим, возвращая номер ошибки

// если дошли до этого места, то ошибки на 100% не было!
    
return pdhResult;
}

int GetCPUUsage(HQUERY hQuery, HCOUNTER hCounter, PDH_STATUS &pdhStatus)
// получаем загруженность проца
{
    
DWORD       dwType      = 0;

// даем запрос на сборы инфы
    
pdhStatus = PdhCollectQueryData(hQuery);

    
if (pdhStatus != ERROR_SUCCESS) // если была ошибка, то
        
return PDH_ERROR; // выходим, возвращая ошибку

    
_PDH_FMT_COUNTERVALUE pfci;
    
ZeroMemory(&pfci, sizeof(pfci)); // очищаем структуру pfci

// Получаем инфу о загруженности проца!
    
pdhStatus = PdhGetFormattedCounterValue( hCounter, PDH_FMT_DOUBLE, &dwType, &pfci);
    
    
if (pdhStatus != ERROR_SUCCESS) // если была ошибка, то
        
return PDH_ERROR; // выходим, возвращая ошибку

    
return (int)pfci.doubleValue; // возвращаем загруженность проца!
}

void pdh_Uninit(HQUERY hQuery, HCOUNTER hCounter)
// анинициализация
{
    
PdhRemoveCounter(hCounter); // удаляем счетчик
    
PdhCloseQuery(hQuery); // закрываем запрос
}
 
« Предыдущая статья   Следующая статья »