// xxxxxx.cpp : コンソールアプリケーションのエントリポイントを定義します。
//
#include "stdafx.h"
#include "pdh.h"

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

//////////////////プロトタイプ宣言////////////////////////
//<Registry使用> Performance Object/Instance/Counter名・各説明の取得
void GetPerfDataFromRegistry();
//<Registry使用> RegistryよりObject/Counter又は説明データの取得
BOOL GetRawDataFromRegistry(LPTSTR **pppszStringArray,
                            DWORD  *pdwSize,
                            LPTSTR *ppszRawString,
                            BOOL   bHelp = FALSE);
//<Registry使用> Performance Instanceの親インスタンス名の取得
LPTSTR GetParentInstanceName(DWORD dwParentObjIdx,
                             DWORD dwParentInstIdx,
                             const PPERF_DATA_BLOCK pPerfData);
////////////////////////////////////////////////////////
//メイン関数
int _tmain(int argc, _TCHAR* argv[])
{
    GetPerfDataFromRegistry();
    return 0;
}
//<Registry使用> Performance Object/Instance/Counter名・各説明の取得
void GetPerfDataFromRegistry()
{
    //取得するPERF_DATA_BLOCKのサイズ 
    DWORD                     dwPerfDataSize  = 102400; 
    DWORD                     dwTempSize      = 102400;
    //パフォーマンス情報全体を保持する構造体
    PPERF_DATA_BLOCK          pPerfData       = NULL;
    //各Performance Object情報を保持する構造体
    PPERF_OBJECT_TYPE         pPerfObject     = NULL;
    //Objectに対応するInstance情報を保持する構造体
    PPERF_INSTANCE_DEFINITION pPerfInstance   = NULL;
    //Objectに対応するCounter情報を保持する構造体
    PPERF_COUNTER_DEFINITION  pPerfCounter    = NULL;
    //各PERF_INSTANCE_DEFINITIONの後に続く構造体
    PPERF_COUNTER_BLOCK       pCounterBlk     = NULL;
    //Instanceの親インスタンス名
    LPTSTR                    pParentInstName = NULL;

    //HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows NT\CurrentVersion\Perflib\011
    //キーの各エントリの文字列格納用
    LPTSTR *ppStringArray     = NULL; //Object/Counter/Instance名文字列の配列
    DWORD  dwStringArraySize  = 0;    //ppStringArrayの配列のサイズ
    LPTSTR pRawString         = NULL; //上記キーのCounterエントリの文字列
    LPTSTR *ppExplainArray    = NULL; //各Object/Counterの説明の文字列の配列
    DWORD  dwExplainArraySize = 0;    //ppExplainArryaの配列のサイズ
    LPTSTR pRawExplain        = NULL; //上記キーのHelpエントリの文字列

    //HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows NT\CurrentVersion\Perflib\011
    //キーのCounterエントリの文字列取得&インデックスに対応するの要素番号に
    //Object/Counter/Instance名格納
    if (!GetRawDataFromRegistry(&ppStringArray, &dwStringArraySize, &pRawString))
        return;

    //HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows NT\CurrentVersion\Perflib\011
    //キーのHelpエントリの文字列取得&インデックスに対応する要素番号に
    //Object/Counterの説明格納
    if (!GetRawDataFromRegistry(&ppExplainArray, 
                                &dwExplainArraySize, 
                                &pRawExplain, TRUE)
       )
        return;

    //pPerfDataにPERF_DATA_BLOCKのデータを取得
    while (::RegQueryValueEx(HKEY_PERFORMANCE_DATA, _T("Global"), NULL, NULL,
                                    (LPBYTE)pPerfData, &dwTempSize) == ERROR_MORE_DATA
          )
    {
        if (pPerfData)
            delete [] pPerfData;

        dwPerfDataSize += 1024;     dwTempSize = dwPerfDataSize;
        pPerfData = (PPERF_DATA_BLOCK) new BYTE[dwPerfDataSize];
    }

    //最初のPerformance Object取得
    pPerfObject = (PPERF_OBJECT_TYPE)((PBYTE)pPerfData + pPerfData->HeaderLength);

    //Performance Object列挙
    for (DWORD i=0; i<pPerfData->NumObjectTypes; i++)
    {
        _tprintf(_T("Object%d : %s -> %s\n"),
                    i,
                    ppStringArray[pPerfObject->ObjectNameTitleIndex],
                    ppExplainArray[pPerfObject->ObjectHelpTitleIndex]);

        //ObjectにInstanceが存在する場合
        if (pPerfObject->NumInstances > 0)
        {
            //最初のPerformance Instance 取得
            pPerfInstance = (PPERF_INSTANCE_DEFINITION)((PBYTE)pPerfObject
                                                        + pPerfObject->DefinitionLength);

            //インスタンス列挙
            for (LONG j=0; j<pPerfObject->NumInstances; j++)
            {
                //現在のインスタンス(pPerfInstance)の親Instance名取得(存在する場合)
                if (pPerfInstance->ParentObjectTitleIndex > 0)
                {
                    pParentInstName =
                           GetParentInstanceName(pPerfInstance->ParentObjectTitleIndex,
                                                 pPerfInstance->ParentObjectInstance,
                                                 pPerfData);

                    _tprintf(_T("\tInstance% 3d : %S/%S\n"),
                                j,
                                pParentInstName,
                                (TCHAR *)((PBYTE)pPerfInstance
                                                        + pPerfInstance->NameOffset));

                } else //親インスタンスがない時
                {
                    _tprintf(_T("\tInstance% 3d : %S\n"),
                                j,
                                (TCHAR *)((PBYTE)pPerfInstance
                                                        + pPerfInstance->NameOffset));
                }

                //次のインスタンス
                pCounterBlk = (PPERF_COUNTER_BLOCK)((PBYTE)pPerfInstance
                                                          + pPerfInstance->ByteLength);
                pPerfInstance = (PPERF_INSTANCE_DEFINITION)((PBYTE)pCounterBlk
                                                          + pCounterBlk->ByteLength);
            }
        }

        //最初のCounter取得
        pPerfCounter = (PPERF_COUNTER_DEFINITION)((PBYTE)pPerfObject
                                                         + pPerfObject->HeaderLength);

        //Performance Counter列挙//
        for (DWORD k=0; k<pPerfObject->NumCounters; k++)
        {
            _tprintf(_T("\tCounter% 4d : %s -> %s\n"),
                        k,
                        ppStringArray[pPerfCounter->CounterNameTitleIndex],
                        ppExplainArray[pPerfCounter->CounterHelpTitleIndex]);

            //次のCounte取得
            pPerfCounter = (PPERF_COUNTER_DEFINITION)((PBYTE)pPerfCounter
                                                        + pPerfCounter->ByteLength);
        }

        //次のPerfomance Object
        pPerfObject = (PPERF_OBJECT_TYPE)((PBYTE)pPerfObject
                                                    + pPerfObject->TotalByteLength);
        _tprintf(_T("\n"));
    }

    //各データの後始末
    delete [] ppStringArray;        delete [] pRawString;
    delete [] ppExplainArray;       delete [] pRawExplain;
    delete [] pPerfData;

    //キーのクローズ(WINPEF.Hを参照)
    ::RegCloseKey(HKEY_PERFORMANCE_DATA);
}

//<Registry使用> RegistryよりObject/Counter又は説明データの取得
//第一引数: 第三引数で取得したMULTI_SZ形式のObject/Counter/Instanceのインデックスと
//          名前文字列へのポインタを配列に格納して受け取る
//          (*pppszStringArrayに結果を受け取る)
//          (例)
//          2 ObjectA 4 CounterB 6 CounterD・・・ (第三引数で受け取るMULTI_SZ形式)
//              ↓
//          (*pppszStringArray)[2]=ObjectA, (*pppszStringArray)[4]=CounterB,
//          としてこの引数に出力
//第二引数: 第一引数の文字列の配列の要素数(*pppszStringArrayの要素数)を出力
//第三引数: HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows NT\CurrentVersion\Perflib\011
//          HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows NT\CurrentVersion\Perflib\009
//          キーのCounterまたはHelpエントリにある文字列を出力
//          (Object/Counter/Instance名又はその説明と対応するIndexのMULTI_SZ形式の文字列)
//第四引数: HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows NT\CurrentVersion\Perflib\011
//          のCounterエントリ(Object/Counter/Instance名)を取得する時はFALSE,
//          Helpエントリ(Object/Counter/Instanceに対応した説明)を取得する時はTRUE
//戻値 : 成功TRUE
//仕様 : 第一、第三引数は呼出し側で必要メモリを確保する必要はないが、使い終わったら
//       delete [] *pppszStringArray, delete [] *ppszRawStringで解放する事
BOOL GetRawDataFromRegistry(LPTSTR **pppszStringArray,
                            DWORD  *pdwSize,
                            LPTSTR *ppszRawString,
                            BOOL   bHelp /*=FALSE*/)
{
    HKEY   hPerflibKey;           //レジストリキーハンドル取得用
    DWORD  dwLastIdx      = 0;    //Counter名に対応したインデックスの最後の値
    DWORD  dwIdxSize      = sizeof DWORD; //dwLastIdxのサイズ
    LPTSTR *ppStringArray = NULL; //第一引数に対応する各Counter名などを収める文字列の配列
    LPTSTR pRawString     = NULL; //第三引数に対応するRegistryから取得したMULTI_SZ文字列
    LPTSTR pCurrentString = NULL; //pRawStringから切り出したObject/Counter名の保持用
    DWORD  dwMaxLen       = 0;    //レジストリキーのすべてのエントリの中で最大の文字数取得

    //第一〜第三引数のうちつでもNULLの場合は終了
    if (!pppszStringArray || !ppszRawString || !pdwSize)
        return FALSE;

    //SOFTWARE\Microsoft\Windows NT\CurrentVersion\Perflibキーのハンドル取得
    if (::RegOpenKeyEx(HKEY_LOCAL_MACHINE,
                       _T("SOFTWARE\Microsoft\Windows NT\CurrentVersion\Perflib"),
                       0,
                       KEY_READ,
                       &hPerflibKey) != ERROR_SUCCESS)
    {
        return FALSE;
    }

    //Counter名またはその説明の最後のインデックス番号取得
    if (::RegQueryValueEx(hPerflibKey,
                          bHelp ? _T("Last Help") : _T("Last Counter"),
                          NULL, NULL, (LPBYTE)&dwLastIdx, &dwIdxSize) != ERROR_SUCCESS)
    {
        ::RegCloseKey(hPerflibKey);     return FALSE;
    }
    ::RegCloseKey(hPerflibKey);

    //第一引数に出力する(最後のインデックス番号+1)要素数の文字列へポインタの配列作成
    ppStringArray = new LPTSTR[dwLastIdx+1];
    for (UINT i=0; i<dwLastIdx+1; i++)
        ppStringArray[i] = NULL;

    //第三引数に出力する文字列が格納されているレジストリキーハンドル取得
    if (::RegOpenKeyEx(HKEY_LOCAL_MACHINE,
                       _T("SOFTWARE\Microsoft\Windows NT\CurrentVersion\Perflib\011"),
                       0,
                       KEY_READ,
                       &hPerflibKey) != ERROR_SUCCESS)
    {
        delete [] ppStringArray;     return FALSE;
    }

    //開いたレジストリキーの各エントリの中で最大の文字数を取得
    if (::RegQueryInfoKey(hPerflibKey, NULL, NULL, NULL, NULL, NULL, NULL, NULL,
                          NULL, &dwMaxLen, NULL, NULL) != ERROR_SUCCESS)
    {
        delete [] ppStringArray;     ::RegCloseKey(hPerflibKey);
        return FALSE;
    }

    //第三引数に出力するMULTI_SZ文字列用のメモリ確保
    pRawString = new TCHAR[++dwMaxLen];

    //Counter又は説明の文字列を取得
    if (::RegQueryValueEx(hPerflibKey,
                          bHelp ? _T("Help") : _T("Counter"),
                          NULL, NULL,
                          (LPBYTE)pRawString, &dwMaxLen) != ERROR_SUCCESS)
    {
        delete [] pRawString;
        delete [] ppStringArray;
        ::RegCloseKey(hPerflibKey);
        return FALSE;
    }
    ::RegCloseKey(hPerflibKey);

    //取得したpRawStringを切り出してインデックスに対応したppStringArray配列の要素番号に
    //Counter/Instance名などを格納
    for (pCurrentString = pRawString;
         *pCurrentString != 0;
         pCurrentString += (lstrlen(pCurrentString) + 1)
        )
    {
        DWORD dwIdx = 0;
        dwIdx = _tstol(pCurrentString);
        pCurrentString += (lstrlen(pCurrentString) + 1);
        ppStringArray[dwIdx] = pCurrentString;
    }

    //引数に結果を渡す
    *pppszStringArray = ppStringArray;
    *ppszRawString    = pRawString;
    *pdwSize          = dwLastIdx + 1;

    return TRUE;
}

//<Registry使用> Performance Instanceの親インスタンス名の取得
//第一引数: 親Performance Objectのインデックス番号
//          (PERF_OBJECT_TYPE構造体のObjectNameTitleIndexメンバ)
//第二引数: PERF_OBJECT_TYPEに続くPERF_INTANCE_DEFINITION構造体のインデックス(順番)
//第三引数: HKEY_PERFORMANCE_DATAキーで取得したPERF_DATA_BLOCK構造体へのポインタ
//戻値    : RegQueryValueで取得したPERF_DATA_BLOCKに続く該当PERF_INSTANCE_DEFINITION
//          のNameOffsetメンバにより取得するUnicodeのInstance名
//          (WINPERF.HのPERF_INSTANCE_DEFINITION参照)
LPTSTR GetParentInstanceName(DWORD dwParentObjIdx,
                             DWORD dwParentInstIdx,
                             const PPERF_DATA_BLOCK pPerfData)
{
    //現在のPerformance Objectへのポインタ
    PPERF_OBJECT_TYPE pPerfObject           = NULL;
    //現在のPerformance Instanceへのポインタ
    PPERF_INSTANCE_DEFINITION pPerfInstance = NULL;
    //PPER_COUNTER_BLOCKへのポインタ
    PPERF_COUNTER_BLOCK pPerfCounterBlk     = NULL;
    DWORD i                                 = 0;

    //最初のPerformance Object
    pPerfObject = (PPERF_OBJECT_TYPE)((PBYTE)pPerfData + pPerfData->HeaderLength);

    //第一引数で指定した親Objectのインデックスと一致するPERF_OBJECT_TYPEを検索
    for (i = 0;
        (i < pPerfData->NumObjectTypes)
                          && (pPerfObject->ObjectNameTitleIndex != dwParentObjIdx);
         i++)
    {
        //次のPerformance Object
        pPerfObject = (PPERF_OBJECT_TYPE)((PBYTE)pPerfObject
                                                   + pPerfObject->TotalByteLength);
    }

    //該当Performance Objectがない時
    if (i >= pPerfData->NumObjectTypes)
        return NULL;

    //該当Performance Objectの最初のInstance 取得
    pPerfInstance = (PPERF_INSTANCE_DEFINITION)((PBYTE)pPerfObject
                                                  + pPerfObject->DefinitionLength);

    //第二引数で指定した親InstanceのIndexと一致するPERF_INSTANCE_DEFINITIONを検索
    for (i = 0;
        (static_cast<LONG>(i) < pPerfObject->NumInstances)
                                                    && (i != dwParentInstIdx);
         i++)
    {
        //次のInstance
        pPerfCounterBlk = (PPERF_COUNTER_BLOCK)((PBYTE)pPerfInstance
                                                       + pPerfInstance->ByteLength);
        pPerfInstance = (PPERF_INSTANCE_DEFINITION)((PBYTE)pPerfCounterBlk
                                                       + pPerfCounterBlk->ByteLength);
    }

    //該当Performantce Instanceがない時
    if (static_cast<LONG>(i) >= pPerfObject->NumInstances)
        return NULL;

    //親Instance名を返却
    return (LPTSTR)((PBYTE)pPerfInstance + pPerfInstance->NameOffset);
}