#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); // закрываем запрос
}