Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

686 lines
21 KiB

/*++
Copyright (C) 1996-2001 Microsoft Corporation
Module Name:
Abstract:
History:
--*/
//
// CookerUtils.cpp
//
#include "precomp.h"
#include <wbemint.h>
#include "cookerutils.h"
#include "strutils.h"
//////////////////////////////////////////////////////////////
//
//
// Helper Functions
//
//
//////////////////////////////////////////////////////////////
BOOL IsCookingClass( IWbemClassObject* pCookingClassObject )
{
HRESULT hResult = WBEM_NO_ERROR;
IWbemQualifierSet *pCookingClassQualifierSet = NULL;
hResult = pCookingClassObject->GetQualifierSet( &pCookingClassQualifierSet );
// Verify "AutoCook" qualifier
// ===========================
if ( SUCCEEDED( hResult ) )
{
CReleaseMe arQualifierSet( (IUnknown*)pCookingClassQualifierSet );
_variant_t vVal;
hResult = pCookingClassQualifierSet->Get( WMI_COOKER_AUTOCOOK_QUALIFIER, 0, &vVal, NULL );
if ( SUCCEEDED( hResult ) )
{
if ( vVal.vt != VT_I4 )
{
hResult = WBEM_E_INVALID_QUALIFIER;
}
else
{
if (WMI_COOKER_VERSION != V_I4(&vVal))
hResult = WBEM_E_INVALID_CLASS;
}
}
}
return SUCCEEDED( hResult );
}
// Table of the counter types and their Cooking type values
struct _CookingTypeRec
{
DWORD dwType;
WCHAR * wcsName;
DWORD dwReqProp;
}
g_aCookingRecs[] =
{
0x00000000, L"PERF_COUNTER_RAWCOUNT_HEX", REQ_1VALUE,
// BEGIN COOKER_ONLY
0x00000001, L"COOKER_AVERAGE", REQ_2VALUE,
0x00000002, L"COOKER_MIN", REQ_2VALUE,
0x00000003, L"COOKER_MAX", REQ_2VALUE,
0x00000004, L"COOKER_RANGE", REQ_2VALUE,
0x00000005, L"COOKER_VARIANCE", REQ_2VALUE,
// END COOKER_ONLY
0x00000100, L"PERF_COUNTER_LARGE_RAWCOUNT_HEX", REQ_1VALUE,
0x00000B00, L"PERF_COUNTER_TEXT", REQ_NONE,
0x00010000, L"PERF_COUNTER_RAWCOUNT", REQ_1VALUE,
0x00010100, L"PERF_COUNTER_LARGE_RAWCOUNT", REQ_1VALUE,
0x00012000, L"PERF_DOUBLE_RAW", REQ_1VALUE,
0x00400400, L"PERF_COUNTER_DELTA", REQ_2VALUE,
0x00400500, L"PERF_COUNTER_LARGE_DELTA", REQ_2VALUE,
0x00410400, L"PERF_SAMPLE_COUNTER", REQ_2VALUE|REQ_TIME|REQ_FREQ,
0x00450400, L"PERF_COUNTER_QUEUELEN_TYPE", REQ_2VALUE|REQ_TIME,
0x00450500, L"PERF_COUNTER_LARGE_QUEUELEN_TYPE", REQ_2VALUE|REQ_TIME,
0x00550500, L"PERF_COUNTER_100NS_QUEUELEN_TYPE", REQ_2VALUE|REQ_TIME,
0x00650500, L"PERF_COUNTER_OBJ_TIME_QUEUELEN_TYPE", REQ_2VALUE|REQ_TIME,
0x10410400, L"PERF_COUNTER_COUNTER", REQ_2VALUE|REQ_TIME|REQ_FREQ,
0x10410500, L"PERF_COUNTER_BULK_COUNT", REQ_2VALUE|REQ_TIME|REQ_FREQ,
0x20020400, L"PERF_RAW_FRACTION", REQ_1VALUE|REQ_BASE,
0x20410500, L"PERF_COUNTER_TIMER", REQ_2VALUE|REQ_TIME,
0x20470500, L"PERF_PRECISION_SYSTEM_TIMER", REQ_2VALUE|REQ_TIME,
0x20510500, L"PERF_100NSEC_TIMER", REQ_2VALUE|REQ_TIME,
0x20570500, L"PERF_PRECISION_100NS_TIMER", REQ_2VALUE|REQ_TIME,
0x20610500, L"PERF_OBJ_TIME_TIMER", REQ_2VALUE|REQ_TIME,
0x20670500, L"PERF_PRECISION_OBJECT_TIMER", REQ_2VALUE|REQ_TIME,
0x20C20400, L"PERF_SAMPLE_FRACTION", REQ_2VALUE|REQ_BASE,
0x21410500, L"PERF_COUNTER_TIMER_INV", REQ_2VALUE|REQ_TIME,
0x21510500, L"PERF_100NSEC_TIMER_INV", REQ_2VALUE|REQ_TIME,
0x22410500, L"PERF_COUNTER_MULTI_TIMER", REQ_2VALUE|REQ_TIME,
0x22510500, L"PERF_100NSEC_MULTI_TIMER", REQ_2VALUE|REQ_TIME,
0x23410500, L"PERF_COUNTER_MULTI_TIMER_INV", REQ_2VALUE|REQ_TIME,
0x23510500, L"PERF_100NSEC_MULTI_TIMER_INV", REQ_2VALUE|REQ_TIME,
0x30020400, L"PERF_AVERAGE_TIMER", REQ_2VALUE|REQ_BASE|REQ_FREQ,
0x30240500, L"PERF_ELAPSED_TIME", REQ_1VALUE|REQ_TIME|REQ_FREQ,
0x40000200, L"PERF_COUNTER_NODATA", REQ_NONE,
0x40020500, L"PERF_AVERAGE_BULK", REQ_2VALUE|REQ_BASE,
0x40030401, L"PERF_SAMPLE_BASE", REQ_NONE,
0x40030402, L"PERF_AVERAGE_BASE", REQ_NONE,
0x40030403, L"PERF_RAW_BASE", REQ_NONE,
0x40030500, L"PERF_PRECISION_TIMESTAMP", REQ_NONE,
0x40030503, L"PERF_LARGE_RAW_BASE", REQ_NONE,
0x42030500, L"PERF_COUNTER_MULTI_BASE", REQ_NONE,
0x80000000, L"PERF_COUNTER_HISTOGRAM_TYPE", REQ_NONE,
};
//
//
// performs a linear search in the above array
//
/////////////////////////////////////////////
HRESULT FindCounterType( WCHAR* pwcsString, DWORD* pdwType, DWORD *pdwReqProp)
{
HRESULT hRes = WBEM_S_NO_ERROR;
DWORD dwLeft = 0,
dwRight = sizeof( g_aCookingRecs ) / sizeof( _CookingTypeRec );
while ( ( dwLeft < dwRight ) && SUCCEEDED( hRes ) )
{
// Found it
if ( wbem_wcsicmp( g_aCookingRecs[dwLeft].wcsName, pwcsString ) == 0 )
{
*pdwType = g_aCookingRecs[dwLeft].dwType;
*pdwReqProp = g_aCookingRecs[dwLeft].dwReqProp;
break;
}
dwLeft++;
}
if ( dwLeft >= dwRight )
{
hRes = WBEM_E_NOT_FOUND;
}
return hRes;
}
//
//
// TRUE if a class has the singleton qualifier set to VARIANT_TRUE
//
////////////////////////////////////////////
BOOL IsSingleton(IWbemClassObject * pCls)
{
if (pCls)
{
HRESULT hr;
IWbemQualifierSet * pQS = NULL;
if (SUCCEEDED(hr = pCls->GetQualifierSet(&pQS)))
{
CReleaseMe rm1(pQS);
_variant_t VarBool;
hr = pQS->Get(L"singleton",0,&VarBool,NULL);
if (SUCCEEDED(hr) &&
VarBool.vt == VT_BOOL &&
V_BOOL(&VarBool) == VARIANT_TRUE)
{
return TRUE;
}
}
}
return FALSE;
}
//
// returns TRUE if the property qualifier set has the CookingType
// qualifier and if the CookingType is recognized
//
///////////////////////////////////////////////////////////////////
BOOL IsCookingProperty( BSTR strPropName, IWbemClassObject* pCookingClassObject, DWORD* pdwCounterType, DWORD * pdwReqProp )
{
HRESULT hResult = WBEM_NO_ERROR;
IWbemQualifierSet* pPropQualSet = NULL;
_variant_t vVal;
hResult = pCookingClassObject->GetPropertyQualifierSet( (LPCWSTR)strPropName, &pPropQualSet );
if ( SUCCEEDED( hResult ) )
{
CReleaseMe arPropQualSet( (IUnknown*)pPropQualSet );
hResult = pPropQualSet->Get( WMI_COOKER_COOKING_PROPERTY_ATTRIBUTE, 0, &vVal, NULL );
if ( SUCCEEDED ( hResult ) )
{
if ( vVal.vt == VT_BSTR )
{
hResult = FindCounterType( V_BSTR( &vVal ), pdwCounterType, pdwReqProp );
}
else
{
hResult = E_FAIL;
}
}
}
return SUCCEEDED( hResult );
}
//
//
// returns the portion of __RELPATH past the '=' sign
// return NULL in case of failure
//
///////////////////////////////////////////////////////////////////////////////
LPWSTR GetKey( IWbemObjectAccess* pClassObj )
{
WMISTATUS dwStatus = WBEM_NO_ERROR;
LPWSTR wszKeyName = NULL;
LPWSTR wszTempKeyPtr = NULL;
_variant_t var;
dwStatus = pClassObj->Get( L"__RELPATH", 0, &var, NULL, NULL );
if ( SUCCEEDED( dwStatus ) )
{
if ( var.vt != VT_BSTR )
{
dwStatus = WBEM_E_FAILED;
}
else
{
wszTempKeyPtr = wcsstr( var.bstrVal, L"=" );
wszTempKeyPtr++;
size_t length = wcslen( wszTempKeyPtr ) + 1 ;
wszKeyName = new WCHAR[ length];
if (wszKeyName)
StringCchCopyW( wszKeyName, length , wszTempKeyPtr );
else
{
// returns NULL
}
}
}
return wszKeyName;
}
WMISTATUS CopyBlob( IWbemClassObject* pSource, IWbemClassObject* pTarget )
{
HRESULT dwStatus = E_NOINTERFACE ;
HRESULT hr1,hr2;
_IWmiObject* pIntTarget = NULL;
_IWmiObject* pIntSource = NULL;
hr1 = pTarget->QueryInterface( IID__IWmiObject, (void**)&pIntTarget );
CReleaseMe arIntTarget( pIntTarget );
hr2 = pSource->QueryInterface( IID__IWmiObject, (void**)&pIntSource );
CReleaseMe arIntSource( pIntSource );
if ( SUCCEEDED(hr1) && SUCCEEDED(hr2) )
{
dwStatus = pIntTarget->CopyInstanceData( 0L, pIntSource );
}
return dwStatus;
}
//
//
// TRUE if the class/instance derives from Win32_PerfFormattedData
//
/////////////////////////////////////
BOOL IsHiPerfObj(IWbemObjectAccess* pObject)
{
_variant_t Var;
HRESULT hr = pObject->Get(L"__SUPERCLASS",0,&Var,NULL,NULL);
if (SUCCEEDED(hr))
{
if (Var.vt == VT_BSTR)
hr = (0 == wbem_wcsicmp(Var.bstrVal,L"Win32_PerfFormattedData"))?S_OK:E_FAIL;
else
hr = E_FAIL;
}
return SUCCEEDED(hr);
}
//
//
// TRUE if an object has the required qualifier to
// be regarded as an HiPerf CWebminstance
//
////////////////////////////////////////////////
BOOL IsHiPerf( IWbemObjectAccess* pObject )
{
WMISTATUS hr = WBEM_NO_ERROR;
IWbemQualifierSet* pQualSet = NULL;
hr = pObject->GetQualifierSet( &pQualSet );
CReleaseMe arQualSet( pQualSet );
// Verify the "Hiperf" class qualifier
if (SUCCEEDED(hr))
{
_variant_t var;
hr = pQualSet->Get( WMI_COOKER_HIPERF_QUALIFIER, 0, &var, NULL );
if ( SUCCEEDED(hr) && ( ( var.vt != VT_BOOL ) || ( var.boolVal != VARIANT_TRUE ) ) )
{
hr = WBEM_E_INVALID_QUALIFIER;
}
}
// Verify the "Cooked" class qualifier
if (SUCCEEDED(hr))
{
_variant_t var;
hr = pQualSet->Get( WMI_COOKER_COOKING_QUALIFIER, 0, &var, NULL );
if ( SUCCEEDED(hr) && ( ( var.vt != VT_BOOL ) || ( var.boolVal != VARIANT_TRUE ) ) )
{
hr = WBEM_E_INVALID_QUALIFIER;
}
}
// Verify the "AutoCook" class qualifier
if (SUCCEEDED(hr))
{
_variant_t var;
hr = pQualSet->Get( WMI_COOKER_AUTOCOOK_QUALIFIER, 0, &var, NULL );
if (SUCCEEDED(hr))
{
if ( var.vt != VT_I4 )
{
hr = WBEM_E_INVALID_QUALIFIER;
}
else
{
if (WMI_COOKER_VERSION != V_I4(&var))
hr = WBEM_E_INVALID_CLASS;
}
}
}
// Verify the "AutoCook_RawClass" class qualifier
if (SUCCEEDED(hr))
{
_variant_t var;
hr = pQualSet->Get( WMI_COOKER_RAWCLASS_QUALIFIER, 0, &var, NULL );
if ( SUCCEEDED(hr) && ( var.vt != VT_BSTR ) )
{
hr = WBEM_E_INVALID_QUALIFIER;
}
}
// verify the "AutoCook_RawDeafult" class qualifier if there
if (SUCCEEDED(hr))
{
_variant_t var;
hr = pQualSet->Get( WMI_COOKER_AUTOCOOK_RAWDEFAULT, 0, &var, NULL );
if ( SUCCEEDED(hr) )
{
if ((var.vt != VT_I4) ||
(V_I4(&var) != AUTOCOOK_RAWDEFAULT_CURRENT_ACCEPTED))
{
hr = WBEM_E_INVALID_QUALIFIER;
}
else // good qualifier, see the properties
{
HRESULT hr1 = pObject->Get(WMI_COOKER_REQ_TIMESTAMP_PERFTIME,0,NULL,NULL,NULL);
HRESULT hr2 = pObject->Get(WMI_COOKER_REQ_TIMESTAMP_SYS100NS,0,NULL,NULL,NULL);
HRESULT hr3 = pObject->Get(WMI_COOKER_REQ_TIMESTAMP_OBJECT ,0,NULL,NULL,NULL);
HRESULT hr4 = pObject->Get(WMI_COOKER_REQ_FREQUENCY_PERFTIME,0,NULL,NULL,NULL);
HRESULT hr5 = pObject->Get(WMI_COOKER_REQ_FREQUENCY_SYS100NS,0,NULL,NULL,NULL);
HRESULT hr6 = pObject->Get(WMI_COOKER_REQ_FREQUENCY_OBJECT ,0,NULL,NULL,NULL);
if (SUCCEEDED(hr1) &&
SUCCEEDED(hr2) &&
SUCCEEDED(hr3) &&
SUCCEEDED(hr4) &&
SUCCEEDED(hr5) &&
SUCCEEDED(hr6))
{
// ok
}
else
{
hr = WBEM_E_INVALID_CLASS;
}
}
}
else
{
// this qualifier is optional
hr = WBEM_NO_ERROR;
}
}
return SUCCEEDED(hr);
}
//
//
// TRUE if the given path points to a class has the required qualifier to
// be regarded as an HiPerf class
// it also verifies that the AutoCook_RawClass indeed exists and it is hiperf
//
////////////////////////////////////////////////
BOOL IsHiPerf( IWbemServices* pNamespace, LPCWSTR wszObject )
{
IWbemClassObject* pObject = NULL;
// Get the WMI object
BSTR strObject = SysAllocString(wszObject);
if (NULL == strObject) return FALSE;
CSysFreeMe fme(strObject);
HRESULT hr = pNamespace->GetObject( strObject, 0, NULL, &pObject, NULL );
CReleaseMe arObject( pObject );
if (SUCCEEDED(hr))
{
IWbemQualifierSet* pQualSet = NULL;
hr = pObject->GetQualifierSet( &pQualSet );
CReleaseMe arQualSet( pQualSet );
// Verify the "Hiperf" class qualifier
if ( SUCCEEDED(hr) )
{
_variant_t var;
hr = pQualSet->Get( WMI_COOKER_HIPERF_QUALIFIER, 0, &var, NULL );
if ( SUCCEEDED(hr) &&
( (var.vt != VT_BOOL) ||
(var.boolVal != VARIANT_TRUE)) )
{
hr = WBEM_E_INVALID_QUALIFIER;
}
}
// Verify the "Cooked" class qualifier
if ( SUCCEEDED(hr) )
{
_variant_t var;
hr = pQualSet->Get( WMI_COOKER_COOKING_QUALIFIER, 0, &var, NULL );
if ( SUCCEEDED(hr) &&
( (var.vt != VT_BOOL) ||
(var.boolVal != VARIANT_TRUE)) )
{
hr = WBEM_E_INVALID_QUALIFIER;
}
}
// Verify the "AutoCook" class qualifier
if ( SUCCEEDED(hr) )
{
_variant_t var;
hr = pQualSet->Get( WMI_COOKER_AUTOCOOK_QUALIFIER, 0, &var, NULL );
if ( SUCCEEDED(hr) )
{
if (var.vt != VT_I4 )
{
hr = WBEM_E_INVALID_QUALIFIER;
}
else
{
if (WMI_COOKER_VERSION != V_I4(&var))
hr = WBEM_E_INVALID_CLASS;
}
}
}
// Verify the "AutoCook_RawClass" class qualifier
// ==============================================
if ( SUCCEEDED(hr) )
{
_variant_t varRawClass;
hr = pQualSet->Get( WMI_COOKER_RAWCLASS_QUALIFIER, 0, &varRawClass, NULL );
if ( SUCCEEDED(hr) )
{
if ( varRawClass.vt == VT_BSTR )
{
IWbemClassObject * pRawClass = NULL;
hr = pNamespace->GetObject(V_BSTR(&varRawClass),0,NULL,&pRawClass,NULL);
CReleaseMe rm1(pRawClass);
if (SUCCEEDED(hr))
{
IWbemQualifierSet * pQS = NULL;
hr = pRawClass->GetQualifierSet(&pQS);
CReleaseMe rm2(pQS);
if (SUCCEEDED(hr))
{
// verify the hiperf qualifier on the RAW class
_variant_t varH;
hr = pQS->Get(WMI_COOKER_HIPERF_QUALIFIER, 0, &varH, NULL);
if ( SUCCEEDED(hr) &&
( (varH.vt != VT_BOOL) ||
(varH.boolVal != VARIANT_TRUE)) )
{
hr = WBEM_E_INVALID_QUALIFIER;
};
}
}
else
{
hr = WBEM_E_INVALID_CLASS;
}
}
else
{
hr = WBEM_E_INVALID_QUALIFIER;
}
}
}
// verify the "AutoCook_RawDeafult" class qualifier if there
// ==========================================================
if ( SUCCEEDED(hr) )
{
_variant_t var;
hr = pQualSet->Get( WMI_COOKER_AUTOCOOK_RAWDEFAULT, 0, &var, NULL );
if ( SUCCEEDED(hr) )
{
if ((var.vt != VT_I4) ||
(V_I4(&var) != AUTOCOOK_RAWDEFAULT_CURRENT_ACCEPTED))
{
hr = WBEM_E_INVALID_QUALIFIER;
}
else // good qualifier, see the properties
{
HRESULT hr1 = pObject->Get(WMI_COOKER_REQ_TIMESTAMP_PERFTIME,0,NULL,NULL,NULL);
HRESULT hr2 = pObject->Get(WMI_COOKER_REQ_TIMESTAMP_SYS100NS,0,NULL,NULL,NULL);
HRESULT hr3 = pObject->Get(WMI_COOKER_REQ_TIMESTAMP_OBJECT ,0,NULL,NULL,NULL);
HRESULT hr4 = pObject->Get(WMI_COOKER_REQ_FREQUENCY_PERFTIME,0,NULL,NULL,NULL);
HRESULT hr5 = pObject->Get(WMI_COOKER_REQ_FREQUENCY_SYS100NS,0,NULL,NULL,NULL);
HRESULT hr6 = pObject->Get(WMI_COOKER_REQ_FREQUENCY_OBJECT ,0,NULL,NULL,NULL);
if (SUCCEEDED(hr1) &&
SUCCEEDED(hr2) &&
SUCCEEDED(hr3) &&
SUCCEEDED(hr4) &&
SUCCEEDED(hr5) &&
SUCCEEDED(hr6))
{
// ok
}
else
{
hr = WBEM_E_INVALID_CLASS;
}
}
}
else
{
// this qualifier is optional
hr = WBEM_NO_ERROR;
}
}
return SUCCEEDED(hr);
}
return FALSE;
}
//
//
// returns pCookingInst->GetQualifierSet()->Get(L"AutoCook_RawClass")
//
//////////////////////////////////
WMISTATUS GetRawClassName( IWbemClassObject* pCookingInst, WCHAR** pwszRawClassName )
{
WMISTATUS dwStatus = WBEM_NO_ERROR;
IWbemQualifierSet* pQualSet = NULL; // The Instance qualifier set
_variant_t varRawInstName; // The raw instance name
dwStatus = pCookingInst->GetQualifierSet( &pQualSet );
CReleaseMe adQualSet( pQualSet );
if ( SUCCEEDED( dwStatus ) )
{
dwStatus = pQualSet->Get( WMI_COOKER_RAWCLASS_QUALIFIER, 0, &varRawInstName, NULL );
if ( SUCCEEDED( dwStatus ) )
{
if ( VT_BSTR == varRawInstName.vt )
{
size_t length = wcslen( varRawInstName.bstrVal ) + 1;
WCHAR* wszRawClassName = new WCHAR[length];
if ( NULL != wszRawClassName )
{
StringCchCopyW( wszRawClassName, length , varRawInstName.bstrVal);
*pwszRawClassName = wszRawClassName;
}
else
{
dwStatus = WBEM_E_OUT_OF_MEMORY;
}
}
else
{
dwStatus = WBEM_E_TYPE_MISMATCH;
}
}
}
return dwStatus;
}
//
//
// returns pAccess->Get("__CLASS")
//
//////////////////////////////////
WMISTATUS GetClassName( IWbemObjectAccess* pAccess, WCHAR** pwszClassName )
{
_variant_t vVal;
WMISTATUS dwStatus = pAccess->Get(L"__CLASS", 0, &vVal, NULL, NULL );
if ( VT_BSTR != V_VT(&vVal) )
{
dwStatus = E_FAIL;
}
if ( SUCCEEDED( dwStatus ) )
{
size_t length = wcslen( vVal.bstrVal ) + 1;
WCHAR* wszClassName = new WCHAR[ length ];
if ( wszClassName )
{
StringCchCopyW( wszClassName, length , vVal.bstrVal );
*pwszClassName = wszClassName;
}
else
{
dwStatus = WBEM_E_OUT_OF_MEMORY;
}
}
return dwStatus;
}