/*++ Copyright (C) 1999-2001 Microsoft Corporation Module Name: CLASSBROKER.CPP Abstract: History: --*/ #include "precomp.h" #include #include #include #include "adaputil.h" #include "classbroker.h" #include struct _CookingTypeRec { DWORD dwType; WCHAR wcsName[128]; } g_aCookingRecs[] = { 0x00000000, L"PERF_COUNTER_RAWCOUNT_HEX", 0x00000100, L"PERF_COUNTER_LARGE_RAWCOUNT_HEX", 0x00000B00, L"PERF_COUNTER_TEXT", 0x00010000, L"PERF_COUNTER_RAWCOUNT", 0x00010100, L"PERF_COUNTER_LARGE_RAWCOUNT", 0x00012000, L"PERF_DOUBLE_RAW", 0x00400400, L"PERF_COUNTER_DELTA", 0x00400500, L"PERF_COUNTER_LARGE_DELTA", 0x00410400, L"PERF_SAMPLE_COUNTER", 0x00450400, L"PERF_COUNTER_QUEUELEN_TYPE", 0x00450500, L"PERF_COUNTER_LARGE_QUEUELEN_TYPE", 0x00550500, L"PERF_COUNTER_100NS_QUEUELEN_TYPE", 0x00650500, L"PERF_COUNTER_OBJ_TIME_QUEUELEN_TYPE", 0x10410400, L"PERF_COUNTER_COUNTER", 0x10410500, L"PERF_COUNTER_BULK_COUNT", 0x20020400, L"PERF_RAW_FRACTION", 0x20410500, L"PERF_COUNTER_TIMER", 0x20470500, L"PERF_PRECISION_SYSTEM_TIMER", 0x20510500, L"PERF_100NSEC_TIMER", 0x20570500, L"PERF_PRECISION_100NS_TIMER", 0x20610500, L"PERF_OBJ_TIME_TIMER", 0x20670500, L"PERF_PRECISION_OBJECT_TIMER", 0x20C20400, L"PERF_SAMPLE_FRACTION", 0x21410500, L"PERF_COUNTER_TIMER_INV", 0x21510500, L"PERF_100NSEC_TIMER_INV", 0x22410500, L"PERF_COUNTER_MULTI_TIMER", 0x22510500, L"PERF_100NSEC_MULTI_TIMER", 0x23410500, L"PERF_COUNTER_MULTI_TIMER_INV", 0x23510500, L"PERF_100NSEC_MULTI_TIMER_INV", 0x30020400, L"PERF_AVERAGE_TIMER", 0x30240500, L"PERF_ELAPSED_TIME", 0x40000200, L"PERF_COUNTER_NODATA", 0x40020500, L"PERF_AVERAGE_BULK", 0x40030401, L"PERF_SAMPLE_BASE", 0x40030402, L"PERF_AVERAGE_BASE", 0x40030403, L"PERF_RAW_BASE", 0x40030500, L"PERF_PRECISION_TIMESTAMP", 0x40030503, L"PERF_LARGE_RAW_BASE", 0x42030500, L"PERF_COUNTER_MULTI_BASE", 0x80000000, L"PERF_COUNTER_HISTOGRAM_TYPE", }; HRESULT GetCounterTypeString( DWORD dwType, WCHAR** pwcsString ) { HRESULT hRes = WBEM_E_NOT_FOUND; DWORD dwLeft = 0, dwRight = sizeof( g_aCookingRecs ) / sizeof( _CookingTypeRec ), dwMid = ( dwLeft + dwRight ) / 2; while ( ( dwLeft <= dwRight ) && FAILED( hRes ) ) { if ( g_aCookingRecs[dwMid].dwType < dwType ) { dwLeft = dwMid + 1; } else if ( g_aCookingRecs[dwMid].dwType > dwType ) { dwRight = dwMid - 1; } else { *pwcsString = g_aCookingRecs[dwMid].wcsName; hRes = WBEM_NO_ERROR; break; } dwMid = ( dwLeft + dwRight ) / 2; } return hRes; } /////////////////////////////////////////////////////////////////////////////// // // CAdapPerfClassElem // /////////////////////////////////////////////////////////////////////////////// CClassBroker::CClassBroker( IWbemClassObject* pBaseClass, WString wstrClassName, CPerfNameDb* pDefaultNameDb ) : m_pPerfObj( NULL ), m_pBaseClass( pBaseClass ), m_wstrClassName( wstrClassName ), m_pDefaultNameDb( pDefaultNameDb ) { if ( NULL != m_pBaseClass ) m_pBaseClass->AddRef(); if ( NULL != m_pDefaultNameDb ) m_pDefaultNameDb->AddRef(); } CClassBroker::CClassBroker( PERF_OBJECT_TYPE* pPerfObj, BOOL bCostly, IWbemClassObject* pBaseClass, CPerfNameDb* pDefaultNameDb, WCHAR* pwcsServiceName ) : m_pPerfObj( pPerfObj ), m_bCostly( bCostly ), m_pBaseClass( pBaseClass ), m_pDefaultNameDb( pDefaultNameDb ), m_wstrServiceName( pwcsServiceName ) { if ( NULL != m_pBaseClass ) m_pBaseClass->AddRef(); if ( NULL != m_pDefaultNameDb ) m_pDefaultNameDb->AddRef(); } CClassBroker::~CClassBroker() { if ( NULL != m_pBaseClass ) m_pBaseClass->Release(); if ( NULL != m_pDefaultNameDb ) m_pDefaultNameDb->Release(); } HRESULT CClassBroker::Generate( DWORD dwType, IWbemClassObject** ppObj ) /////////////////////////////////////////////////////////////////////////////// // // Generates a class based on the object BLOB passed in via the constructor // // Parameters: // ppObj - A pointer to the new class object interface pointer // /////////////////////////////////////////////////////////////////////////////// { IWbemClassObject* pNewClass = NULL; // Create the new class // ==================== HRESULT hr = m_pBaseClass->SpawnDerivedClass( 0L, &pNewClass ); CReleaseMe rmNewClass( pNewClass ); // And initialize the data // ======================= if ( SUCCEEDED( hr ) ) { // Class name // ========== hr = SetClassName( dwType, pNewClass ); // Class Qualifiers // ================ if ( SUCCEEDED( hr ) ) { hr = SetClassQualifiers( pNewClass, dwType, ( ADAP_DEFAULT_OBJECT == m_pPerfObj->ObjectNameTitleIndex ) ); } // Standard Properties // =================== if ( SUCCEEDED( hr ) ) { hr = AddDefaultProperties( pNewClass ); } // Perf Counter Properties // ======================= if ( SUCCEEDED( hr ) ) { hr = EnumProperties( dwType, pNewClass ); } // Return the class object interface // ================================= if ( SUCCEEDED( hr ) ) { hr = pNewClass->QueryInterface( IID_IWbemClassObject, (void**) ppObj ); } } return hr; } HRESULT CClassBroker::SetClassName( DWORD dwType, IWbemClassObject* pClass ) /////////////////////////////////////////////////////////////////////////////// // // Sets the name of the new WMI class. The syntax is: // // Win32_Perf__ // // where the service name is the name of the namespace and the display name // is the name of the object located in the perf name database // // Parameters: // pClass - The object which requires the name // /////////////////////////////////////////////////////////////////////////////// { HRESULT hr = WBEM_NO_ERROR; WString wstrObjectName; WString wstrTempSvcName; if ( 0 == m_wstrClassName.Length() ) { try { switch( dwType ) { case WMI_ADAP_RAW_CLASS: m_wstrClassName = ADAP_PERF_RAW_BASE_CLASS L"_"; break; case WMI_ADAP_COOKED_CLASS: m_wstrClassName = ADAP_PERF_COOKED_BASE_CLASS L"_"; break; default: hr = WBEM_E_INVALID_PARAMETER_ID; break; } } catch(...) { return WBEM_E_OUT_OF_MEMORY; } // Process the performance class name // ================================== if ( SUCCEEDED( hr ) ) { // Get the performance class' display name // ======================================= hr = m_pDefaultNameDb->GetDisplayName( m_pPerfObj->ObjectNameTitleIndex, wstrObjectName ); // If no object name was returned, then log an error // ================================================= if ( FAILED( hr ) ) { CAdapUtility::NTLogEvent( EVENTLOG_WARNING_TYPE, WBEM_MC_ADAP_MISSING_OBJECT_INDEX, m_pPerfObj->ObjectNameTitleIndex, (LPCWSTR)m_wstrServiceName ); } // Replace reserved characters with proper names // ============================================= if ( SUCCEEDED( hr ) ) { hr = ReplaceReserved( wstrObjectName ); } // Remove whitespace and extraneous characters // =========================================== if ( SUCCEEDED( hr ) ) { hr = RemoveWhitespaceAndNonAlphaNum( wstrObjectName ); } } // Now do the same for the service name // ==================================== if ( SUCCEEDED( hr ) ) { // Get the service name // ==================== wstrTempSvcName = m_wstrServiceName; // Replace reserved characters with proper names // ============================================= if ( SUCCEEDED( hr ) ) { hr = ReplaceReserved( wstrTempSvcName ); } // Remove whitespace and extraneous characters // =========================================== if ( SUCCEEDED( hr ) ) { hr = RemoveWhitespaceAndNonAlphaNum( wstrTempSvcName ); } } // Now we can build the rest of the name and try setting it in the object // ====================================================================== if ( SUCCEEDED( hr ) ) { try { m_wstrClassName += wstrTempSvcName; m_wstrClassName += L"_"; m_wstrClassName += wstrObjectName; if ( m_bCostly ) { m_wstrClassName += "_Costly"; } } catch( ... ) { hr = WBEM_E_OUT_OF_MEMORY; } } } // Set the class name in the WMI object // ==================================== if ( SUCCEEDED( hr ) ) { _variant_t var = (LPWSTR) m_wstrClassName; hr = pClass->Put( L"__CLASS", 0L, &var, CIM_STRING ); } return hr; } HRESULT CClassBroker::RemoveWhitespaceAndNonAlphaNum( WString& wstr ) /////////////////////////////////////////////////////////////////////////////// // // Removes spaces, tabs, etc. and non-alphanumeric characters from the // input string // // Parameters: // wstr - The string to be processed // /////////////////////////////////////////////////////////////////////////////// { HRESULT hr = WBEM_S_NO_ERROR; WCHAR* pWstr = wstr.UnbindPtr(); CVectorDeleteMe vdmWstr( pWstr ); if ( NULL != pWstr ) { try { WCHAR* pNewWstr = new WCHAR[lstrlenW(pWstr) + 1]; if (NULL == pNewWstr) return WBEM_E_OUT_OF_MEMORY; int x = 0; int y = 0; // Dump all of the leading, trailing and internal whitespace // ========================================================= for ( ; NULL != pWstr[x]; x++ ) { if ( !iswspace( pWstr[x] ) && isunialphanum( pWstr[x] ) ) { pNewWstr[y] = pWstr[x]; y++; } } pNewWstr[y] = NULL; // This will cause the WString to acquire the new pointer // ====================================================== wstr.BindPtr( pNewWstr ); } catch(...) { hr = WBEM_E_OUT_OF_MEMORY; } } return hr; } HRESULT CClassBroker::ReplaceReserved( WString& wstr ) /////////////////////////////////////////////////////////////////////////////// // // This is a 2-pass filter. First we must determine the size of the new buffer by counting // the number of replacement candidates, and, after creating the new buffer, we copy the // data, replacing the restricted characters where required. // // Replaces: // "/" with "Per", // "%" with "Percent", // "#" with "Number", // "@" with "At", // "&" with "And" // // Parameters: // wstr - String to be processed // /////////////////////////////////////////////////////////////////////////////// { HRESULT hr = WBEM_S_NO_ERROR; int x = 0, y = 0; // Get the data buffer for processing // ================================== WCHAR* pWstr = wstr.UnbindPtr(); CVectorDeleteMe vdmWstr( pWstr ); if ( NULL != pWstr ) { // First pass: Count the number of reserved characters // =================================================== DWORD dwNumSlashes = 0, dwNumPercent = 0, dwNumAt = 0, dwNumNumber = 0, dwNumAmper = 0, dwNumReserved = 0; for ( ; NULL != pWstr[x]; x++ ) { switch ( pWstr[x] ) { case L'/': dwNumSlashes++; dwNumReserved++; break; case L'%': dwNumPercent++; dwNumReserved++; break; case L'@': dwNumAt++; dwNumReserved++; break; case L'#': dwNumNumber++; dwNumReserved++; break; case L'&': dwNumAmper++; dwNumReserved++; break; default: break; } } try { // Create the new buffer // ===================== DWORD dwBuffSize = lstrlenW(pWstr) + 1 + ( 3 * dwNumSlashes ) + ( 7 * dwNumPercent ) + ( 2 * dwNumAt ) + ( 6 * dwNumNumber ) + ( 3 * dwNumAmper ); WCHAR* pNewWstr = new WCHAR[dwBuffSize]; if (NULL == pNewWstr) return WBEM_E_OUT_OF_MEMORY; // Second pass: Replace reserved characters // ======================================== DWORD dwBuffSizeCurr = dwBuffSize; for ( x = 0; NULL != pWstr[x]; x++ ) { BOOL AllIsUpper = FALSE; DWORD Cnt; switch ( pWstr[x] ) { case L'/': // if all characters up to the end of string or to the next space are uppercase for (Cnt=1;pWstr[x+Cnt] && pWstr[x+Cnt]!=' ';Cnt++) { if (isupper(pWstr[x+Cnt])) { AllIsUpper = TRUE; } else { AllIsUpper = FALSE; break; } }; if (!AllIsUpper) { StringCchCopyW( &pNewWstr[y],dwBuffSizeCurr, L"Per" ); y+=3; dwBuffSizeCurr-=3; } else { x++; pNewWstr[y]=pWstr[x]; y++; dwBuffSizeCurr--; } break; case L'%': StringCchCopyW( &pNewWstr[y],dwBuffSizeCurr, L"Percent" ); y+=7; dwBuffSizeCurr-=7; break; case L'@': StringCchCopyW( &pNewWstr[y],dwBuffSizeCurr, L"At" ); y+=2; dwBuffSizeCurr-=2; break; case L'#': StringCchCopyW( &pNewWstr[y],dwBuffSizeCurr, L"Number" ); y+=6; dwBuffSizeCurr-=6; break; case L'&': StringCchCopyW( &pNewWstr[y],dwBuffSizeCurr, L"And" ); y+=3; dwBuffSizeCurr-=3; break; default: pNewWstr[y] = pWstr[x]; y++; dwBuffSizeCurr--; break; } } pNewWstr[y] = NULL; // This will cause the WString to acquire the new pointer // ====================================================== wstr.BindPtr( pNewWstr ); } catch(...) { hr = WBEM_E_OUT_OF_MEMORY; } } return hr; } //////////////////////////////////////////////////////////////////////////////////////////// // // CLocaleClassBroker // //////////////////////////////////////////////////////////////////////////////////////////// CLocaleClassBroker::CLocaleClassBroker( IWbemClassObject* pBaseClass, WString wstrClassName, CPerfNameDb* pDefaultNameDb, CPerfNameDb* pLocaleNameDb ) : CClassBroker( pBaseClass, wstrClassName, pDefaultNameDb ), m_pLocaleNameDb( pLocaleNameDb ) { if ( NULL != m_pLocaleNameDb ) m_pLocaleNameDb->AddRef(); } CLocaleClassBroker::CLocaleClassBroker( PERF_OBJECT_TYPE* pPerfObj, BOOL bCostly, IWbemClassObject* pBaseClass, CPerfNameDb* pDefaultNameDb, CPerfNameDb* pLocaleNameDb, LANGID LangId, WCHAR* pwcsServiceName ) : m_pLocaleNameDb( pLocaleNameDb ), m_LangId( LangId ), CClassBroker( pPerfObj, bCostly, pBaseClass, pDefaultNameDb, pwcsServiceName ) { if ( NULL != m_pLocaleNameDb ) m_pLocaleNameDb->AddRef(); } CLocaleClassBroker::~CLocaleClassBroker() { if ( NULL != m_pLocaleNameDb ) m_pLocaleNameDb->Release(); } HRESULT CLocaleClassBroker::SetClassQualifiers( IWbemClassObject* pClass, DWORD dwType, BOOL fIsDefault ) //////////////////////////////////////////////////////////////////////////////////////////// // // Sets the class' qualifiers per the localized object rules. Note that the operations // are performed directly on the IWbemClassObject // // The following qualifiers will be added: // - Amendment // - Locale(0x0409) // - DisplayName (Amended flavor) // - Genericperfctr (signals that this is a generic counter) // // Parameters: // pClass - The object to be massaged // fIsDefault - Indicator for the default object (not used in localized objects) // //////////////////////////////////////////////////////////////////////////////////////////// { HRESULT hr = WBEM_S_NO_ERROR; _variant_t var; try { IWbemQualifierSet* pQualSet = NULL; hr = pClass->GetQualifierSet( &pQualSet ); CReleaseMe rmQualSet( pQualSet ); // Amendment // ========= if ( SUCCEEDED( hr ) ) { var = (bool)true; hr = pQualSet->Put( L"Amendment", &var, 0L ); } // Locale // ====== if ( SUCCEEDED( hr ) ) { var.Clear(); V_VT(&var) = VT_I4; V_I4(&var) = m_LangId; hr = pQualSet->Put( L"locale", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // DisplayName // =========== if ( SUCCEEDED( hr ) ) { LPCWSTR pwcsDisplayName = NULL; var.Clear(); // Fetch the name from the Names' database // ======================================= hr = m_pLocaleNameDb->GetDisplayName( m_pPerfObj->ObjectNameTitleIndex, &pwcsDisplayName ); // If this is a localized Db, this is a benign error. We will just pull the value // from the default db (it must be there, we wouldn't have a class name if it didn't // ================================================================================= if ( FAILED( hr ) ) { hr = m_pDefaultNameDb->GetDisplayName( m_pPerfObj->ObjectNameTitleIndex, &pwcsDisplayName ); } if ( SUCCEEDED( hr ) ) { var = (LPWSTR) pwcsDisplayName ; hr = pQualSet->Put( L"DisplayName", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE | WBEM_FLAVOR_AMENDED ); } } // Genericperfctr // ============== if ( SUCCEEDED(hr) ) { var = (bool)true; hr = pQualSet->Put( L"genericperfctr", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Perfindex // ========= if ( SUCCEEDED( hr ) ) { var.Clear(); V_VT(&var) = VT_I4; V_I4(&var) = m_pPerfObj->ObjectNameTitleIndex; hr = pQualSet->Put( L"perfindex", (VARIANT*)&var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } } catch(...) { hr = WBEM_E_OUT_OF_MEMORY; } if ( FAILED( hr ) ) { // Something whacky happened: log an event // ======================================= CAdapUtility::NTLogEvent( EVENTLOG_WARNING_TYPE, WBEM_MC_ADAP_GENERAL_OBJECT_FAILURE, (LPCWSTR)m_wstrClassName, (LPCWSTR)m_wstrServiceName, CHex( hr ) ); } return hr; } HRESULT CLocaleClassBroker::AddDefaultProperties( IWbemClassObject* pObj ) //////////////////////////////////////////////////////////////////////////////////////////// // // Ignored for localized classes // //////////////////////////////////////////////////////////////////////////////////////////// { return WBEM_S_NO_ERROR; } HRESULT CLocaleClassBroker::EnumProperties( DWORD dwType, IWbemClassObject* pObj ) //////////////////////////////////////////////////////////////////////////////////////////// // // Ignored for localized classes // //////////////////////////////////////////////////////////////////////////////////////////// { return WBEM_S_NO_ERROR; } HRESULT CLocaleClassBroker::SetPropertyQualifiers( PERF_COUNTER_DEFINITION* pCtrDefinition, DWORD dwType, BOOL fIsDefault, LPCWSTR pwcsPropertyName, IWbemClassObject* pClass, BOOL bBase ) //////////////////////////////////////////////////////////////////////////////////////////// // // Adds localization qualifiers for the counter properties // // The following qualifiers will be added: // - DisplayName (Amended flavor) // // Properties: // pCtrDefinition - The portion of the performance blob related to the property // fIsDefault - Flag identifying default property // pwcsPropertyName - The name of the property // pClass - The WMI class containing the property // bBase - Base property identifier // //////////////////////////////////////////////////////////////////////////////////////////// { HRESULT hr = WBEM_S_NO_ERROR; _variant_t var; _variant_t varHelp; try { // DisplayName // =========== if ( SUCCEEDED( hr ) ) { LPCWSTR pwcsDisplayName = NULL; LPCWSTR pwcsHelpName = NULL; // Fetch the name from the Names' database // ======================================= if ( !bBase ) { hr = m_pLocaleNameDb->GetDisplayName( pCtrDefinition->CounterNameTitleIndex, &pwcsDisplayName ); // If this is a localized Db, this is a benign error. We will just pull the value // from the default db (it must be there, we wouldn't have a class name if it didn't // ================================================================================= if ( FAILED( hr ) ) { hr = m_pDefaultNameDb->GetDisplayName( pCtrDefinition->CounterNameTitleIndex, &pwcsDisplayName ); } if ( SUCCEEDED( hr ) ) { var = (LPWSTR) pwcsDisplayName ; } hr = m_pLocaleNameDb->GetHelpName( pCtrDefinition->CounterHelpTitleIndex, &pwcsHelpName ); // If this is a localized Db, this is a benign error. We will just pull the value // from the default db (it must be there, we wouldn't have a class name if it didn't // ================================================================================= if ( FAILED( hr ) ) { hr = m_pDefaultNameDb->GetHelpName( pCtrDefinition->CounterHelpTitleIndex, &pwcsHelpName ); } if ( SUCCEEDED( hr ) ) { varHelp = (LPWSTR) pwcsHelpName ; } } else { var = L""; varHelp = L""; } // Set the qualifier // ================= if ( SUCCEEDED( hr ) ) { IWbemQualifierSet* pQualSet = NULL; hr = pClass->GetPropertyQualifierSet( pwcsPropertyName, &pQualSet ); CReleaseMe rmQualSet( pQualSet ); if ( SUCCEEDED( hr ) ) { hr = pQualSet->Put( L"DisplayName", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE | WBEM_FLAVOR_AMENDED ); if (SUCCEEDED(hr)) { hr = pQualSet->Put( L"Description", &varHelp, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE | WBEM_FLAVOR_AMENDED ); } } } } } catch(...) { hr = WBEM_E_OUT_OF_MEMORY; } if ( FAILED( hr ) ) { // Something whacky happened: log an event // ======================================= CAdapUtility::NTLogEvent( EVENTLOG_WARNING_TYPE, WBEM_MC_ADAP_GENERAL_OBJECT_FAILURE, (LPCWSTR)m_wstrClassName, (LPCWSTR)m_wstrServiceName, CHex( hr ) ); } return hr; } HRESULT CLocaleClassBroker::AddProperty( PERF_COUNTER_DEFINITION* pCtrDefinition, DWORD dwType, BOOL fIsDefault, IWbemClassObject* pClass, WString &wstrLastCtrName, BOOL* pbLastCounterIsNotBase ) //////////////////////////////////////////////////////////////////////////////////////////// // // Ignored for localized classes // //////////////////////////////////////////////////////////////////////////////////////////// { return WBEM_S_NO_ERROR; } HRESULT CLocaleClassBroker::GenPerfClass( PERF_OBJECT_TYPE* pPerfObj, DWORD dwType, BOOL bCostly, IWbemClassObject* pBaseClass, CPerfNameDb* pDefaultNameDb, CPerfNameDb* pLocaleNameDb, LANGID LangId, WCHAR* pwcsServiceName, IWbemClassObject** ppObj) /////////////////////////////////////////////////////////////////////////////// // // A static member of the broker. It generates a WMI class based on the // object BLOB. // // Parameters: // pPerfObj - The object BLOB // bCostly - Costly object indicator // pBaseClass - The new object's base class // pDefaultNameDb - The default language names' database // pLocaleNameDb - The localized language names' database // LangId - The locale ID // pwcsServiceName - The name of the perflib service // ppObj - A pointer to the new class object interface pointer // /////////////////////////////////////////////////////////////////////////////// { HRESULT hr = WBEM_S_NO_ERROR; IWbemClassObject* pObject = NULL; CLocaleClassBroker Broker( pPerfObj, bCostly, pBaseClass, pDefaultNameDb, pLocaleNameDb, LangId, pwcsServiceName ); hr = Broker.Generate( dwType, &pObject ); if ( SUCCEEDED( hr ) ) *ppObj = pObject; return hr; } HRESULT CLocaleClassBroker::ConvertToLocale( IWbemClassObject* pDefaultClass, CLocaleDefn* pLocaleDefn, CLocaleDefn* pDefaultDefn, IWbemClassObject** ppObject) /////////////////////////////////////////////////////////////////////////////// // // A static member of the broker. It generates a new localized class based // on the default object // /////////////////////////////////////////////////////////////////////////////// { HRESULT hr = WBEM_S_NO_ERROR; _variant_t var; int nLocale = 0; int nPerfIndex = 0; int nHelpIndex = 0; WString wstrClassName; DWORD dwType = WMI_ADAP_RAW_CLASS; CPerfNameDb* pLocaleNameDb = NULL; CPerfNameDb* pDefaultNameDb = NULL; // Get references to the localized name's databases // ================================================ hr = pLocaleDefn->GetNameDb( &pLocaleNameDb ); CAdapReleaseMe armLocaleNameDb( pLocaleNameDb ); // Get references to the default name's databases // ============================================== if ( SUCCEEDED( hr ) ) { hr = pDefaultDefn->GetNameDb( &pDefaultNameDb ); } CAdapReleaseMe armDefaultNameDb( pDefaultNameDb ); // Get the locale ID // ================= if ( SUCCEEDED( hr ) ) { hr = pLocaleDefn->GetLID( nLocale ); } // Get the object's perf index // =========================== if ( SUCCEEDED( hr ) ) { IWbemQualifierSet* pQualSet = NULL; hr = pDefaultClass->GetQualifierSet( &pQualSet ); CReleaseMe rmQualSet( pQualSet ); if ( SUCCEEDED( hr ) ) { hr = pQualSet->Get( L"perfindex", 0L, &var, NULL ); if (SUCCEEDED(hr)) { nPerfIndex = V_I4(&var); } else { // see InitializeMembers nPerfIndex = 0; hr = WBEM_S_FALSE; } } if ( SUCCEEDED( hr ) ) { hr = pQualSet->Get( L"helpindex", 0L, &var, NULL ); if (SUCCEEDED(hr)) { nHelpIndex = V_I4(&var); } else { // see InitializeMembers nHelpIndex = 0; hr = WBEM_S_FALSE; } } if ( SUCCEEDED( hr ) ) { hr = pQualSet->Get( L"Cooked", 0L, &var, NULL ); if ( SUCCEEDED( hr ) && ( V_BOOL(&var) == VARIANT_TRUE ) ) { dwType = WMI_ADAP_COOKED_CLASS; } else { dwType = WMI_ADAP_RAW_CLASS; hr = WBEM_S_FALSE; } } } // Get the class name // ================== if ( SUCCEEDED( hr ) ) { var.Clear(); hr = pDefaultClass->Get( L"__CLASS", 0L, &var, NULL, NULL ); wstrClassName = V_BSTR(&var); } // Create locaized class // ===================== IWbemClassObject* pBaseClass = NULL; IWbemClassObject* pLocaleClass = NULL; if ( SUCCEEDED( hr ) ) { hr = pLocaleDefn->GetBaseClass( dwType, &pBaseClass ); CReleaseMe rmBaseClass( pBaseClass ); if ( SUCCEEDED( hr ) ) { hr = pBaseClass->SpawnDerivedClass( 0L, &pLocaleClass ); } } // Initialize the data // =================== // Set the name // ============ if ( SUCCEEDED( hr ) ) { var.Clear(); var = LPCWSTR(wstrClassName); hr = pLocaleClass->Put( L"__CLASS", 0L, &var, CIM_STRING ); } // Set Qualifiers // ============== if ( SUCCEEDED( hr ) ) { IWbemQualifierSet* pQualSet = NULL; hr = pLocaleClass->GetQualifierSet( &pQualSet ); CReleaseMe rmQualSet( pQualSet ); // Amendment // ========= if ( SUCCEEDED( hr ) ) { var.Clear(); var = bool(true); hr = pQualSet->Put( L"Amendment", &var, 0L ); } // Locale // ====== if ( SUCCEEDED( hr ) ) { var.Clear(); V_VT(&var) = VT_I4; V_I4(&var) = nLocale; hr = pQualSet->Put( L"locale", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // DisplayName // =========== if ( SUCCEEDED( hr ) ) { LPCWSTR pwcsDisplayName = NULL; hr = pLocaleNameDb->GetDisplayName( nPerfIndex, &pwcsDisplayName ); // If this is a localized Db, this is a benign error. We will just pull the value // from the default db (it must be there, we wouldn't have a class name if it didn't // ================================================================================= if ( FAILED( hr ) ) { hr = pDefaultNameDb->GetDisplayName( nPerfIndex, &pwcsDisplayName ); } if ( SUCCEEDED( hr ) ) { var.Clear(); var = (WCHAR *)( pwcsDisplayName ); hr = pQualSet->Put( L"DisplayName", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE | WBEM_FLAVOR_AMENDED ); } else { // the nPerfInedx was bad ERRORTRACE((LOG_WMIADAP,"class %S: DisplayName for counter %d not found\n",(WCHAR *)wstrClassName,nPerfIndex)); } } // Description // =========== if ( SUCCEEDED( hr ) ) { LPCWSTR pwcsHelpName = NULL; hr = pLocaleNameDb->GetHelpName( nHelpIndex, &pwcsHelpName ); // If this is a localized Db, this is a benign error. We will just pull the value // from the default db (it must be there, we wouldn't have a class name if it didn't // ================================================================================= if ( FAILED( hr ) ) { hr = pDefaultNameDb->GetHelpName( nHelpIndex, &pwcsHelpName ); } if ( SUCCEEDED( hr ) ) { var.Clear(); var = (WCHAR *)( pwcsHelpName ); hr = pQualSet->Put( L"Description", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE | WBEM_FLAVOR_AMENDED ); } else { // the nPerfInedx was bad ERRORTRACE((LOG_WMIADAP,"class %S: Description for counter %d not found\n",(WCHAR *)wstrClassName,nPerfIndex)); } } // Genericperfctr // ============== if ( SUCCEEDED(hr) ) { var.Clear(); var = bool(true); hr = pQualSet->Put( L"genericperfctr", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } } // Set Properties // ============== if ( SUCCEEDED( hr ) ) { BSTR bstrPropName; pDefaultClass->BeginEnumeration( WBEM_FLAG_LOCAL_ONLY ); while ( WBEM_S_NO_ERROR == pDefaultClass->Next( 0, &bstrPropName, NULL, NULL, NULL ) ) { CSysFreeMe sfm(bstrPropName); var.Clear(); CIMTYPE ct; int nCounterIndex = 0; int nHelpIndex2 = 0; WString wstrDefaultPropDisplayName; // Create the property based upon the default property // =================================================== V_VT(&var) = VT_NULL; V_I8(&var) = 0; hr = pDefaultClass->Get( bstrPropName, 0L, NULL, &ct, NULL ); if (FAILED(hr)) continue; hr = pLocaleClass->Put( bstrPropName, 0L, (VARIANT*)&var, ct ); if (FAILED(hr)) continue; // Grab the default property qualifier set // ======================================= IWbemQualifierSet* pQualSet = NULL; hr = pDefaultClass->GetPropertyQualifierSet( bstrPropName, &pQualSet ); CReleaseMe rmQualSet( pQualSet ); // Get the default perfindex to be (used to retrieve the display // name from the localized names' database) // ============================================================= if ( SUCCEEDED( hr ) ) { hr = pQualSet->Get( L"perfindex", 0L, &var, NULL ); nCounterIndex = V_UI4(&var); } // DisplayName // =========== if ( SUCCEEDED( hr ) ) { LPCWSTR pwcsDisplayName = NULL; hr = pLocaleNameDb->GetDisplayName( nCounterIndex, &pwcsDisplayName ); // If this is a localized Db, this is a benign error. We will just pull the value // from the default db (it must be there, we wouldn't have a class name if it didn't // ================================================================================= if ( FAILED( hr ) ) { hr = pDefaultNameDb->GetDisplayName( nCounterIndex, &pwcsDisplayName ); } if ( SUCCEEDED( hr ) ) { IWbemQualifierSet* pLocaleQualSet = NULL; hr = pLocaleClass->GetPropertyQualifierSet( bstrPropName, &pLocaleQualSet ); CReleaseMe rmLocaleQualSet( pLocaleQualSet ); var = (WCHAR *)( pwcsDisplayName ); hr = pLocaleQualSet->Put( L"DisplayName", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE | WBEM_FLAVOR_AMENDED ); } else { ERRORTRACE((LOG_WMIADAP,"class %S: Display for counter %d not found\n",(WCHAR *)wstrClassName,nCounterIndex)); } } // HelpIndex if ( SUCCEEDED( hr ) ) { var.Clear(); hr = pQualSet->Get( L"helpindex", 0L, &var, NULL ); nHelpIndex2 = V_UI4(&var); } // Description // =========== if ( SUCCEEDED( hr ) ) { LPCWSTR pwcsHelpName = NULL; hr = pLocaleNameDb->GetHelpName( nHelpIndex2, &pwcsHelpName ); // If this is a localized Db, this is a benign error. We will just pull the value // from the default db (it must be there, we wouldn't have a class name if it didn't // ================================================================================= if ( FAILED( hr ) ) { hr = pDefaultNameDb->GetHelpName( nHelpIndex2, &pwcsHelpName ); } if ( SUCCEEDED( hr ) ) { IWbemQualifierSet* pLocaleQualSet = NULL; hr = pLocaleClass->GetPropertyQualifierSet( bstrPropName, &pLocaleQualSet ); CReleaseMe rmLocaleQualSet( pLocaleQualSet ); var = (WCHAR *)( pwcsHelpName ); hr = pLocaleQualSet->Put( L"Description", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE | WBEM_FLAVOR_AMENDED ); } else { ERRORTRACE((LOG_WMIADAP,"class %S: Description for counter %d not found\n",(WCHAR *)wstrClassName,nCounterIndex)); } } } pDefaultClass->EndEnumeration(); } if ( SUCCEEDED( hr ) ) { *ppObject = pLocaleClass; if ( NULL != *ppObject ) (*ppObject)->AddRef(); } return hr; } //////////////////////////////////////////////////////////////////////////////////////////// // // CDefaultClassBroker // //////////////////////////////////////////////////////////////////////////////////////////// HRESULT CDefaultClassBroker::SetClassQualifiers( IWbemClassObject* pClass, DWORD dwType, BOOL fIsDefault ) //////////////////////////////////////////////////////////////////////////////////////////// // // Sets the class' qualifiers. Note that the operations are performed directly on the // IWbemClassObject. // // The following qualifiers will be added: // - Dynamic // - Provider("NT5_GenericPerfProvider_V1") // - Registrykey // - Locale(0x0409) // - Perfindex // - Helpindex // - Perfdetail // - Genericperfctr (signals that this is a generic counter) // - Singleton (if applicable) // - Costly (if applicable) // // Parameters: // pClass - The object to be massaged // fIsDefault - Indicator for the default object (not used in localized objects) // //////////////////////////////////////////////////////////////////////////////////////////// { HRESULT hr = WBEM_S_NO_ERROR; _variant_t var; try { IWbemQualifierSet* pQualSet = NULL; hr = pClass->GetQualifierSet( &pQualSet ); CReleaseMe rmQualSet( pQualSet ); switch ( dwType ) { case WMI_ADAP_RAW_CLASS: { // Default // ======= if ( SUCCEEDED( hr ) && fIsDefault ) { var = bool(true); hr = pQualSet->Put( L"perfdefault", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Dynamic // ======= if ( SUCCEEDED( hr ) ) { var = bool(true); hr = pQualSet->Put( L"dynamic", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Provider // ======== if ( SUCCEEDED( hr ) ) { var = L"Nt5_GenericPerfProvider_V1"; hr = pQualSet->Put( L"provider", &var, 0L ); } // Registrykey // =========== if ( SUCCEEDED( hr ) ) { var = (WCHAR *)(m_wstrServiceName); hr = pQualSet->Put( L"registrykey", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Locale // ====== if ( SUCCEEDED( hr ) ) { var.Clear(); V_VT(&var) = VT_I4; V_I4(&var) = 0x0409; hr = pQualSet->Put( L"locale", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Perfindex // ========= if ( SUCCEEDED( hr ) ) { V_VT(&var) = VT_I4; V_I4(&var) = m_pPerfObj->ObjectNameTitleIndex; hr = pQualSet->Put( L"perfindex", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Helpindex // ========= if ( SUCCEEDED( hr ) ) { V_VT(&var) = VT_I4; V_I4(&var) = m_pPerfObj->ObjectHelpTitleIndex; hr = pQualSet->Put( L"helpindex", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } #ifdef _PM_CHANGED_THEIR_MIND_ // Description // ========== if ( SUCCEEDED( hr ) ) { HRESULT hr2; LPCWSTR pwcsHelpName = NULL; hr2 = m_pDefaultNameDb->GetHelpName( m_pPerfObj->ObjectHelpTitleIndex, &pwcsHelpName ); var = (WCHAR *)pwcsHelpName; if (SUCCEEDED(hr2)) { hr = pQualSet->Put( L"Description", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } else { ERRORTRACE((LOG_WMIADAP,"class %S: Help for counter %d not found\n",(WCHAR *)m_wstrClassName,m_pPerfObj->ObjectHelpTitleIndex)); } var.Clear(); } #endif // Perfdetail // ========== if ( SUCCEEDED( hr ) ) { V_VT(&var) = VT_I4; V_I4(&var) = m_pPerfObj->DetailLevel; hr = pQualSet->Put( L"perfdetail", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Genericperfctr // ============== if ( SUCCEEDED(hr) ) { var = bool(true); hr = pQualSet->Put( L"genericperfctr", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // HiPerf // ============== if ( SUCCEEDED(hr) ) { var = bool(true); hr = pQualSet->Put( L"hiperf", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Singleton (set if the numinstances is PERF_NO_INSTANCES) // ======================================================== if ( SUCCEEDED(hr) && IsSingleton( ) ) { var = bool(true); // This will have default flavors hr = pQualSet->Put( L"singleton", &var, 0L ); } // Costly // ====== if ( SUCCEEDED(hr) && m_bCostly ) { var = bool(true); hr = pQualSet->Put( L"costly", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } }break; case WMI_ADAP_COOKED_CLASS: { var.Clear(); // Dynamic // ======= if ( SUCCEEDED( hr ) ) { var = bool(true); hr = pQualSet->Put( L"dynamic", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Provider // ======== if ( SUCCEEDED( hr ) ) { var = L"HiPerfCooker_v1"; hr = pQualSet->Put( L"provider", &var, 0L ); } // Locale // ====== if ( SUCCEEDED( hr ) ) { var.Clear(); V_VT(&var) = VT_I4; V_I4(&var) = 0x0409; hr = pQualSet->Put( L"locale", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Registrykey // =========== if ( SUCCEEDED( hr ) ) { var.Clear(); var = (WCHAR *)( m_wstrServiceName ); hr = pQualSet->Put( L"registrykey", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Cooked // ====== if ( SUCCEEDED( hr ) ) { var.Clear(); var = bool(true); hr = pQualSet->Put( L"Cooked", &var, 0 ); } // AutoCook // ======== if ( SUCCEEDED( hr ) ) { V_VT(&var) = VT_I4; V_I4(&var) = 1 ; hr = pQualSet->Put( L"AutoCook", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Genericperfctr // ============== if ( SUCCEEDED(hr) ) { var = bool(true); hr = pQualSet->Put( L"genericperfctr", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // HiPerf // ============== if ( SUCCEEDED(hr) ) { var = bool(true); hr = pQualSet->Put( L"hiperf", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // AutoCook_RawClass // ================= if ( SUCCEEDED( hr ) ) { _variant_t varClassName; hr = pClass->Get( L"__CLASS", 0, &varClassName, 0, 0 ); if ( SUCCEEDED( hr ) ) { WCHAR* wszRawClass = NULL; WCHAR* wszClassRoot = varClassName.bstrVal + wcslen ( ADAP_PERF_COOKED_BASE_CLASS ); size_t cchSizeTmp = wcslen( wszClassRoot ) + wcslen( ADAP_PERF_RAW_BASE_CLASS ) + 1; wszRawClass = new WCHAR[ cchSizeTmp ]; if (NULL == wszRawClass) hr = WBEM_E_OUT_OF_MEMORY; if (SUCCEEDED(hr)) { CDeleteMe dmRawClass( wszRawClass ); StringCchPrintfW( wszRawClass,cchSizeTmp, L"%s%s", ADAP_PERF_RAW_BASE_CLASS, wszClassRoot ); var = wszRawClass; hr = pQualSet->Put( L"AutoCook_RawClass", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); //WBEM_FLAVOR_FLAG_PROPAGATE_TO_DERIVED_CLASS } } } // Perfindex // ========= if ( SUCCEEDED( hr ) ) { var.Clear(); V_VT(&var) = VT_I4; V_I4(&var) = m_pPerfObj->ObjectNameTitleIndex; hr = pQualSet->Put( L"perfindex", (VARIANT*)&var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Helpindex // ========= if ( SUCCEEDED( hr ) ) { V_VT(&var) = VT_I4; V_I4(&var) = m_pPerfObj->ObjectHelpTitleIndex; hr = pQualSet->Put( L"helpindex", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } #ifdef _PM_CHANGED_THEIR_MIND_ // Description // ========== if ( SUCCEEDED( hr ) ) { HRESULT hr2; LPCWSTR pwcsHelpName = NULL; hr2 = m_pDefaultNameDb->GetHelpName( m_pPerfObj->ObjectHelpTitleIndex, &pwcsHelpName ); var = (WCHAR *)pwcsHelpName; if (SUCCEEDED(hr2)) { hr = pQualSet->Put( L"Description", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } else { ERRORTRACE((LOG_WMIADAP,"class %S: Help for counter %d not found\n",(WCHAR *)m_wstrClassName,m_pPerfObj->ObjectHelpTitleIndex)); } var.Clear(); } #endif // Singleton (set if the numinstances is PERF_NO_INSTANCES) // ======================================================== if ( SUCCEEDED(hr) && IsSingleton( ) ) { var.Clear(); var = bool(true); // This will have default flavors hr = pQualSet->Put( L"singleton", (VARIANT*)&var, 0L ); } }break; } } catch(...) { hr = WBEM_E_OUT_OF_MEMORY; } if ( FAILED( hr ) ) { // Something whacky happened: log an event // ======================================= CAdapUtility::NTLogEvent( EVENTLOG_WARNING_TYPE, WBEM_MC_ADAP_GENERAL_OBJECT_FAILURE, (LPCWSTR)m_wstrClassName, (LPCWSTR)m_wstrServiceName, CHex( hr ) ); } return hr; } HRESULT CDefaultClassBroker::AddDefaultProperties( IWbemClassObject* pClass ) /////////////////////////////////////////////////////////////////////////////// // // Adds appropriate default properties. // // The following qualifiers will be added: // - Name // // Parameters: // pClass - The object to be massaged // /////////////////////////////////////////////////////////////////////////////// { HRESULT hr = WBEM_S_NO_ERROR; // If we are not a singleton class, then we will // need a name property that is marked as a key // ============================================= if ( !IsSingleton() ) { _variant_t var; // Add the Name property // ===================== V_VT(&var) = VT_NULL; V_I8(&var) = 0; hr = pClass->Put( L"Name", 0L, &var, CIM_STRING ); // Add the property qualifiers // =========================== if ( SUCCEEDED( hr ) ) { IWbemQualifierSet* pQualSet = NULL; hr = pClass->GetPropertyQualifierSet( L"Name", &pQualSet ); CReleaseMe rmQualSet( pQualSet ); // Dynamic // ======= if ( SUCCEEDED( hr ) ) { var.Clear(); var = bool(true); hr = pQualSet->Put( L"key", (VARIANT*)&var, 0L ); } } } return hr; } HRESULT CDefaultClassBroker::EnumProperties( DWORD dwType, IWbemClassObject* pClass ) /////////////////////////////////////////////////////////////////////////////// // // Walks the counter definitions and generates corresponding properties // // Parameters: // pClass - The object to be massaged // /////////////////////////////////////////////////////////////////////////////// { HRESULT hr = WBEM_S_NO_ERROR; BOOL bLastCounterIsNotBase = FALSE; WString wstrLastCtrName; // Set to first counter definition // =============================== LPBYTE pbData = ((LPBYTE) m_pPerfObj) + m_pPerfObj->HeaderLength; // Cast to a counter definition // ============================ PERF_COUNTER_DEFINITION* pCounterDefinition = (PERF_COUNTER_DEFINITION*) pbData; // For each counter definition, add a corresponding property // ========================================================= for ( DWORD dwCtr = 0; SUCCEEDED( hr ) && dwCtr < m_pPerfObj->NumCounters; dwCtr++ ) { hr = AddProperty( pCounterDefinition, dwType, ( dwCtr == (DWORD) m_pPerfObj->DefaultCounter), pClass, wstrLastCtrName, &bLastCounterIsNotBase ); // Now go to the next counter definition // ===================================== pbData = ((LPBYTE) pCounterDefinition) + pCounterDefinition->ByteLength; pCounterDefinition = (PERF_COUNTER_DEFINITION*) pbData; } return hr; } HRESULT CDefaultClassBroker::AddProperty( PERF_COUNTER_DEFINITION* pCtrDefinition, DWORD dwType, BOOL fIsDefault, IWbemClassObject* pClass, WString &wstrLastCtrName, BOOL* pbLastCounterIsNotBase ) /////////////////////////////////////////////////////////////////////////////// // // Adds a property defined by the counter definition // // Properties: // pCtrDefinition - The counter BLOB // dwType - Raw or Formatted object? // fIsDefault - The default property flag // pClass - The class containing the property // wstrLastCtrName - The name of the last counter (required for base // properties) // pbLastCounterIsNotBase // - An indicator for the previous counter's baseness // /////////////////////////////////////////////////////////////////////////////// { if (NULL == pbLastCounterIsNotBase) return WBEM_E_INVALID_PARAMETER; HRESULT hr = WBEM_S_NO_ERROR; WString wstrPropertyName; DWORD dwCounterTypeMask = PERF_SIZE_VARIABLE_LEN; BOOL bBase = FALSE; if ( PERF_COUNTER_BASE == ( pCtrDefinition->CounterType & 0x00070000 ) ) { // It's a base property // ==================== if ( *pbLastCounterIsNotBase ) { try { // The property name is the same as the previous property, // but with "_Base" appended to the end // ======================================================= wstrPropertyName = wstrLastCtrName; if ( WMI_ADAP_RAW_CLASS == dwType ) { wstrPropertyName += "_Base"; } } catch(...) { hr = WBEM_E_OUT_OF_MEMORY; } bBase = TRUE; } else { // Cannot have 2 consequtive bases // =============================== CAdapUtility::NTLogEvent( EVENTLOG_WARNING_TYPE, WBEM_MC_ADAP_BAD_PERFLIB_INVALID_DATA, (LPCWSTR)m_wstrServiceName, CHex(0) ); hr = WBEM_E_FAILED; } } else { // It's not a base property so get the name from the names' database // ================================================================= hr = m_pDefaultNameDb->GetDisplayName( pCtrDefinition->CounterNameTitleIndex, wstrPropertyName ); if ( FAILED( hr ) ) { // Index does not exist in the Names' DB: log an event // =================================================== CAdapUtility::NTLogEvent( EVENTLOG_WARNING_TYPE, WBEM_MC_ADAP_MISSING_PROPERTY_INDEX, (LPCWSTR)m_wstrClassName, (LPCWSTR)m_wstrServiceName, pCtrDefinition->CounterNameTitleIndex ); } // Replace reserved characters with text // ===================================== if ( SUCCEEDED( hr ) ) { hr = ReplaceReserved( wstrPropertyName ); } // Remove restricted characters // ============================ if ( SUCCEEDED ( hr ) ) { hr = RemoveWhitespaceAndNonAlphaNum( wstrPropertyName ); } } if ( SUCCEEDED( hr ) ) { _variant_t varTest; DWORD dwBaseCtr = 1; // Ensure that the property does not exist // ======================================= if ( FAILED( pClass->Get( wstrPropertyName, 0L, &varTest, NULL, NULL ) ) ) { // Now check the perf counter type to see if it's a DWORD or LARGE. // If it's anything else, we will NOT support this object // ================================================================ DWORD dwCtrType = pCtrDefinition->CounterType & dwCounterTypeMask; if ( PERF_SIZE_DWORD == dwCtrType || PERF_SIZE_LARGE == dwCtrType ) { _variant_t var; CIMTYPE ct = ( PERF_SIZE_DWORD == dwCtrType ? CIM_UINT32 : CIM_UINT64 ); // Add the property // ================ V_VT(&var) = VT_NULL; V_I8(&var) = 0; hr = pClass->Put( wstrPropertyName, 0L, &var, ct ); // Set the property qualifiers // =========================== if ( SUCCEEDED( hr ) ) { hr = SetPropertyQualifiers( pCtrDefinition, dwType, fIsDefault, wstrPropertyName, pClass, bBase ); } } else if ( PERF_SIZE_ZERO == dwCtrType ) { // Ignore zero size properties // =========================== } else { // Illegal property type: log an event // =================================== CAdapUtility::NTLogEvent( EVENTLOG_WARNING_TYPE, WBEM_MC_ADAP_BAD_PERFLIB_BAD_PROPERTYTYPE, (LPCWSTR)m_wstrClassName, (LPCWSTR)m_wstrServiceName, (LPCWSTR)wstrPropertyName); hr = WBEM_E_FAILED; } } else if ( ( WMI_ADAP_COOKED_CLASS == dwType ) && ( bBase ) ) { hr = SetPropertyQualifiers( pCtrDefinition, dwType, fIsDefault, wstrPropertyName, pClass, bBase ); } else { // Raw Property already exists: log an event ( // ========================================= CAdapUtility::NTLogEvent( EVENTLOG_WARNING_TYPE, WBEM_MC_ADAP_DUPLICATE_PROPERTY, (LPCWSTR)m_wstrClassName, (LPCWSTR)m_wstrServiceName, (LPCWSTR) wstrPropertyName ); hr = WBEM_E_FAILED; } } if ( SUCCEEDED( hr ) ) { *pbLastCounterIsNotBase = !bBase; wstrLastCtrName = wstrPropertyName; } else { // Wierdness: log an event // ======================= CAdapUtility::NTLogEvent( EVENTLOG_WARNING_TYPE, WBEM_MC_ADAP_GENERAL_OBJECT_FAILURE, (LPCWSTR)m_wstrClassName, (LPCWSTR)m_wstrServiceName, CHex( hr ) ); } return hr; } HRESULT CDefaultClassBroker::SetPropertyQualifiers( PERF_COUNTER_DEFINITION* pCtrDefinition, DWORD dwType, BOOL fIsDefault, LPCWSTR pwcsPropertyName, IWbemClassObject* pClass, BOOL bBase ) /////////////////////////////////////////////////////////////////////////////// // // Sets the qualifier values of the properties defined by the counter // definition. // // The following qualifiers will be added: // - Perfdefault // - Display // - Countertype // - Perfindex // - Helpindex // - Defaultscale // - Perfdetail // // Properties: // pCtrDefinition - The portion of the performance blob related to the property // fIsDefault - Flag identifying default property // pwcsPropertyName - The name of the property // pClass - The WMI class containing the property // bBase - Base property identifier // /////////////////////////////////////////////////////////////////////////////// { HRESULT hr = WBEM_S_NO_ERROR; _variant_t var; try { IWbemQualifierSet* pQualSet = NULL; hr = pClass->GetPropertyQualifierSet( pwcsPropertyName, &pQualSet ); CReleaseMe rmQualSet( pQualSet ); switch ( dwType ) { case WMI_ADAP_RAW_CLASS: { // Perfdefault // =========== if ( SUCCEEDED( hr ) && fIsDefault ) { var = bool(true); hr = pQualSet->Put( L"perfdefault", (VARIANT*)&var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Display // ======= if ( SUCCEEDED( hr ) ) { LPCWSTR pwcsDisplayName = NULL; var.Clear(); // Fetch the name from the Names' database // ======================================= if ( !bBase ) { hr = m_pDefaultNameDb->GetDisplayName( pCtrDefinition->CounterNameTitleIndex, &pwcsDisplayName ); if ( SUCCEEDED( hr ) ) { var = pwcsDisplayName; } else { ERRORTRACE((LOG_WMIADAP,"class %S: DisplayName for counter %d not found\n",(WCHAR *)m_wstrClassName,pCtrDefinition->CounterNameTitleIndex)); } } else { var = L""; } // If this is a localized Db, this could be a benign error // ======================================================= if ( SUCCEEDED( hr ) ) { hr = pQualSet->Put( L"DisplayName", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } } #ifdef _PM_CHANGED_THEIR_MIND_ // Description // =========== if ( SUCCEEDED( hr ) ) { LPCWSTR pwcsHelpName = NULL; var.Clear(); if ( !bBase ) { hr = m_pDefaultNameDb->GetHelpName( pCtrDefinition->CounterHelpTitleIndex, &pwcsHelpName ); if ( SUCCEEDED( hr ) ) { var = pwcsHelpName; } else { ERRORTRACE((LOG_WMIADAP,"class %S: Help for counter %d not found\n",(WCHAR *)m_wstrClassName,pCtrDefinition->CounterNameTitleIndex)); } } else { var = L""; } if ( SUCCEEDED( hr ) ) { hr = pQualSet->Put( L"Description", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } } #endif // Countertype // =========== if ( SUCCEEDED( hr ) ) { var.Clear(); V_VT(&var) = VT_I4; V_I4(&var) = pCtrDefinition->CounterType ; hr = pQualSet->Put( L"countertype", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Perfindex // ========= if ( SUCCEEDED( hr ) ) { V_VT(&var) = VT_I4; V_I4(&var) = pCtrDefinition->CounterNameTitleIndex ; hr = pQualSet->Put( L"perfindex", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Helpindex // ========= if ( SUCCEEDED( hr ) ) { V_VT(&var) = VT_I4; V_I4(&var) = pCtrDefinition->CounterHelpTitleIndex ; hr = pQualSet->Put( L"helpindex", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Defaultscale // ============ if ( SUCCEEDED( hr ) ) { V_VT(&var) = VT_I4; V_I4(&var) = pCtrDefinition->DefaultScale ; hr = pQualSet->Put( L"defaultscale", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Perfdetail // ========== if ( SUCCEEDED( hr ) ) { V_VT(&var) = VT_I4; V_I4(&var) = pCtrDefinition->DetailLevel ; hr = pQualSet->Put( L"perfdetail", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } }break; case WMI_ADAP_COOKED_CLASS: { var.Clear(); #ifdef _PM_CHANGED_THEIR_MIND_ // Display // ======= if ( SUCCEEDED( hr ) ) { LPCWSTR pwcsDisplayName = NULL; var.Clear(); // Fetch the name from the Names' database // ======================================= if ( !bBase ) { hr = m_pDefaultNameDb->GetDisplayName( pCtrDefinition->CounterNameTitleIndex, &pwcsDisplayName ); if ( SUCCEEDED( hr ) ) { var = pwcsDisplayName; } else { ERRORTRACE((LOG_WMIADAP,"class %S: DisplayName for counter %d not found\n",(WCHAR *)m_wstrClassName,pCtrDefinition->CounterNameTitleIndex)); } } else { var = L""; } // If this is a localized Db, this could be a benign error // ======================================================= if ( SUCCEEDED( hr ) ) { hr = pQualSet->Put( L"DisplayName", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } } // Description // =========== if ( SUCCEEDED( hr ) ) { LPCWSTR pwcsHelpName = NULL; var.Clear(); if ( !bBase ) { hr = m_pDefaultNameDb->GetHelpName( pCtrDefinition->CounterHelpTitleIndex, &pwcsHelpName ); if ( SUCCEEDED( hr ) ) { var = pwcsHelpName; } } else { var = L""; } if ( SUCCEEDED( hr ) ) { hr = pQualSet->Put( L"Description", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } } #endif if ( !bBase ) { // CookingType // =========== if ( SUCCEEDED( hr ) ) { WCHAR* wszCookingType = NULL; hr = GetCounterTypeString( pCtrDefinition->CounterType, &wszCookingType ); if ( SUCCEEDED( hr ) ) { var = wszCookingType; hr = pQualSet->Put( L"CookingType", (VARIANT*)&var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } } // Counter // ======= if ( SUCCEEDED( hr ) ) { WString wstrPropertyName; var.Clear(); // Fetch the name from the Names' database // ======================================= hr = m_pDefaultNameDb->GetDisplayName( pCtrDefinition->CounterNameTitleIndex, wstrPropertyName ); // Replace reserved characters with proper names // ============================================= if ( SUCCEEDED( hr ) ) { hr = ReplaceReserved( wstrPropertyName ); } // Remove whitespace and extraneous characters // =========================================== if ( SUCCEEDED( hr ) ) { hr = RemoveWhitespaceAndNonAlphaNum( wstrPropertyName ); } if ( SUCCEEDED( hr ) ) { var = LPCWSTR(wstrPropertyName ); // If this is a localized Db, this could be a benign error // ======================================================= hr = pQualSet->Put( L"Counter", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } } // PerfTimeStamp & PerfTimeFreq // ============================ if ( SUCCEEDED( hr ) ) { _variant_t varStamp; _variant_t varFreq; if ( pCtrDefinition->CounterType & PERF_TIMER_100NS ) { varStamp = L"TimeStamp_Sys100NS"; varFreq = L"Frequency_Sys100NS"; } else if ( pCtrDefinition->CounterType & PERF_OBJECT_TIMER ) { varStamp = L"Timestamp_Object" ; varFreq = L"Frequency_Object" ; } else { varStamp = L"Timestamp_PerfTime"; varFreq = L"Frequency_PerfTime"; } hr = pQualSet->Put( L"PerfTimeStamp", &varStamp, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); if ( SUCCEEDED( hr ) ) { hr = pQualSet->Put( L"PerfTimeFreq", &varFreq, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } } // Perfindex // ========= if ( SUCCEEDED( hr ) ) { var.Clear(); V_VT(&var) = VT_I4; V_I4(&var) = pCtrDefinition->CounterNameTitleIndex; hr = pQualSet->Put( L"perfindex", (VARIANT*)&var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } // Helpindex // ========= if ( SUCCEEDED( hr ) ) { V_VT(&var) = VT_I4; V_I4(&var) = pCtrDefinition->CounterHelpTitleIndex ; hr = pQualSet->Put( L"helpindex", &var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } } else { // Base // ==== if ( SUCCEEDED( hr ) ) { WCHAR* wszCounterBase = NULL; _variant_t varCounter; hr = pQualSet->Get( L"Counter", 0L, &varCounter, NULL ); size_t cchSizeTmp = wcslen( varCounter.bstrVal ) + 5 + 1; wszCounterBase = new WCHAR[ cchSizeTmp ]; CDeleteMe dmCounterBase( wszCounterBase ); if ( NULL == wszCounterBase ) { hr = WBEM_E_OUT_OF_MEMORY; } else { StringCchPrintfW( wszCounterBase, cchSizeTmp , L"%s_Base", varCounter.bstrVal ); var = wszCounterBase; hr = pQualSet->Put( L"Base", (VARIANT*)&var, WBEM_FLAVOR_FLAG_PROPAGATE_TO_INSTANCE ); } } } }break; } } catch(...) { hr = WBEM_E_OUT_OF_MEMORY; } if ( FAILED( hr ) ) { // Weirdness: log an event // ======================= CAdapUtility::NTLogEvent( EVENTLOG_WARNING_TYPE, WBEM_MC_ADAP_GENERAL_OBJECT_FAILURE, (LPCWSTR)m_wstrClassName, (LPCWSTR)m_wstrServiceName, CHex( hr ) ); } return hr; } HRESULT CDefaultClassBroker::GenPerfClass( PERF_OBJECT_TYPE* pPerfObj, DWORD dwType, BOOL bCostly, IWbemClassObject* pBaseClass, CPerfNameDb* pDefaultNameDb, WCHAR* pwcsServiceName, IWbemClassObject** ppObj) /////////////////////////////////////////////////////////////////////////////// // // A static member of the broker. It generates a WMI class based on the // object BLOB. // // Parameters: // pPerfObj - The object BLOB // bCostly - Costly object indicator // pBaseClass - The new object's base class // pDefaultNameDb - The default language names' database // pwcsServiceName - The name of the perflib service // ppObj - A pointer to the new class object interface pointer // /////////////////////////////////////////////////////////////////////////////// { if (NULL == ppObj) return WBEM_E_INVALID_PARAMETER; HRESULT hr = WBEM_S_NO_ERROR; IWbemClassObject* pObject = NULL; CDefaultClassBroker Broker( pPerfObj, bCostly, pBaseClass, pDefaultNameDb, pwcsServiceName ); hr = Broker.Generate( dwType, &pObject ); if ( SUCCEEDED( hr ) ) *ppObj = pObject; return hr; }