mirror of https://github.com/tongzx/nt5src
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.
1697 lines
40 KiB
1697 lines
40 KiB
/*++
|
|
|
|
Copyright (C) 2000-2001 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
Abstract:
|
|
|
|
History:
|
|
|
|
--*/
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Cache.cpp
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "precomp.h"
|
|
#include <winperf.h>
|
|
#include <comdef.h>
|
|
#include <algorithm>
|
|
#include <wbemint.h>
|
|
#include <sync.h> // for CInCritSec
|
|
#include <autoptr.h>
|
|
|
|
#include "Cache.h"
|
|
#include "WMIObjCooker.h"
|
|
#include "CookerUtils.h"
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// CProperty
|
|
// =========
|
|
//
|
|
// The base property - used for raw properties and the base
|
|
// class for the CookedProperty.
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
CProperty::CProperty( LPWSTR wszName, long lHandle, CIMTYPE ct ) :
|
|
m_wszName( NULL ),
|
|
m_lPropHandle( lHandle ),
|
|
m_ct( ct )
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
// Parameters:
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
m_wszName = new WCHAR[ wcslen( wszName ) + 1 ];
|
|
if (m_wszName)
|
|
wcscpy( m_wszName, wszName);
|
|
}
|
|
|
|
CProperty::~CProperty(){
|
|
if (m_wszName) {
|
|
delete [] m_wszName;
|
|
}
|
|
}
|
|
|
|
LPWSTR CProperty::GetName()
|
|
{
|
|
return m_wszName?m_wszName:L"";
|
|
}
|
|
|
|
CIMTYPE CProperty::GetType()
|
|
{
|
|
return m_ct;
|
|
}
|
|
|
|
long CProperty::GetHandle()
|
|
{
|
|
return m_lPropHandle;
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// CCookingProperty
|
|
// ================
|
|
//
|
|
// The cooked property - used to model the data required to
|
|
// cook a property of a cooked class
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
CCookingProperty::CCookingProperty( LPWSTR wszName,
|
|
DWORD dwCounterType,
|
|
long lPropHandle,
|
|
CIMTYPE ct,
|
|
DWORD dwReqProp,
|
|
BOOL bUseWellKnownIfNeeded) :
|
|
CProperty( wszName, lPropHandle, ct ),
|
|
m_dwCounterType( dwCounterType ),
|
|
m_dwReqProp(dwReqProp),
|
|
m_nTimeFreq( 0 ),
|
|
m_lScale(0), // 10^0 = 1
|
|
m_pRawCounterProp( NULL ),
|
|
m_pTimeProp( NULL ),
|
|
m_pFrequencyProp( NULL ),
|
|
m_pBaseProp( NULL ),
|
|
m_nSampleWindow( 0 ),
|
|
m_nTimeWindow( 0 ),
|
|
m_bUseWellKnownIfNeeded(bUseWellKnownIfNeeded)
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Constructor
|
|
//
|
|
// Parameters:
|
|
// wszName - The property name
|
|
// dwCounterType - The property's counter type
|
|
// lPropHandle - The cooking property's WMI Access handle
|
|
// ct - The CIM type of the property
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{}
|
|
|
|
CCookingProperty::~CCookingProperty()
|
|
{
|
|
if (m_pRawCounterProp) {
|
|
delete m_pRawCounterProp;
|
|
}
|
|
|
|
if (m_pTimeProp){
|
|
delete m_pTimeProp;
|
|
}
|
|
|
|
if ( m_pFrequencyProp ){
|
|
delete m_pFrequencyProp;
|
|
}
|
|
if (m_pBaseProp){
|
|
delete m_pBaseProp;
|
|
}
|
|
}
|
|
|
|
WMISTATUS CCookingProperty::Initialize( IWbemQualifierSet* pCookingPropQualifierSet,
|
|
IWbemObjectAccess* pRawAccess,
|
|
IWbemQualifierSet* pCookingClassQSet)
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
// Parameters:
|
|
// pCookingClassAccess - The class definition for the cooking class
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
_variant_t vVal;
|
|
|
|
// Initialize the raw counter property ("Counter")
|
|
// ===============================================
|
|
dwStatus = pCookingPropQualifierSet->Get( WMI_COOKER_RAW_COUNTER, 0, &vVal, NULL );
|
|
|
|
if ( SUCCEEDED( dwStatus ) && ( vVal.vt != VT_BSTR ) )
|
|
{
|
|
dwStatus = E_FAIL;
|
|
}
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
// Get the raw data properties
|
|
// ===========================
|
|
|
|
CIMTYPE ct;
|
|
|
|
long lHandle = 0;
|
|
WCHAR* wszRawCounterName = vVal.bstrVal;
|
|
|
|
// Get the raw counter property
|
|
// ============================
|
|
|
|
dwStatus = pRawAccess->GetPropertyHandle( wszRawCounterName, &ct, &lHandle );
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
m_pRawCounterProp = new CProperty( wszRawCounterName, lHandle, ct );
|
|
|
|
if ( NULL == m_pRawCounterProp )
|
|
{
|
|
dwStatus = WBEM_E_OUT_OF_MEMORY;
|
|
}
|
|
}
|
|
|
|
// Get the raw base property
|
|
// =========================
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
_variant_t vProp;
|
|
dwStatus = pCookingPropQualifierSet->Get( WMI_COOKER_RAW_BASE, 0, &vProp, NULL );
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
if ( vProp.vt == VT_BSTR )
|
|
{
|
|
dwStatus = pRawAccess->GetPropertyHandle( vProp.bstrVal, &ct, &lHandle );
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
m_pBaseProp = new CProperty( vProp.bstrVal, lHandle, ct );
|
|
|
|
if ( NULL == m_pBaseProp )
|
|
{
|
|
dwStatus = WBEM_E_OUT_OF_MEMORY;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
dwStatus = WBEM_E_TYPE_MISMATCH;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// the property qualifier set failed, try the class one
|
|
_variant_t varProp; // does not throw, simple container
|
|
dwStatus = pCookingClassQSet->Get( WMI_COOKER_RAW_BASE, 0, &varProp, NULL );
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
if ( varProp.vt == VT_BSTR )
|
|
{
|
|
dwStatus = pRawAccess->GetPropertyHandle( varProp.bstrVal, &ct, &lHandle );
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
m_pBaseProp = new CProperty( varProp.bstrVal, lHandle, ct );
|
|
|
|
if ( NULL == m_pBaseProp )
|
|
{
|
|
dwStatus = WBEM_E_OUT_OF_MEMORY;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
dwStatus = WBEM_E_TYPE_MISMATCH;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
dwStatus = WBEM_NO_ERROR;
|
|
}
|
|
}
|
|
//
|
|
// no error so far, the BASE qulifier is REQUIRED, but none is found
|
|
//
|
|
if ( SUCCEEDED( dwStatus ) &&
|
|
IsReq(REQ_BASE) &&
|
|
(NULL == m_pBaseProp))
|
|
{
|
|
dwStatus = WBEM_E_INVALID_CLASS;
|
|
}
|
|
}
|
|
|
|
// Get the raw timestamp property record
|
|
// =====================================
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
_variant_t vProp2;
|
|
dwStatus = pCookingPropQualifierSet->Get( WMI_COOKER_RAW_TIME, 0, &vProp2, NULL );
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
if ( vProp2.vt == VT_BSTR )
|
|
{
|
|
dwStatus = pRawAccess->GetPropertyHandle( vProp2.bstrVal, &ct, &lHandle );
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
m_pTimeProp = new CProperty( vProp2.bstrVal, lHandle, ct );
|
|
|
|
if ( NULL == m_pTimeProp )
|
|
{
|
|
dwStatus = WBEM_E_OUT_OF_MEMORY;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
dwStatus = WBEM_E_TYPE_MISMATCH;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
|
|
// the property qualifier set failed, try the class one
|
|
|
|
//PERF_TIMER_TICK
|
|
//PERF_TIMER_100NS
|
|
//PERF_OBJECT_TIMER
|
|
|
|
_variant_t varProp; // does not throw, simple container
|
|
if (m_dwCounterType & PERF_OBJECT_TIMER)
|
|
{
|
|
dwStatus = pCookingClassQSet->Get( WMI_COOKER_RAW_TIME_OBJ, 0, &varProp, NULL );
|
|
if (FAILED(dwStatus) && m_bUseWellKnownIfNeeded)
|
|
{
|
|
dwStatus = WBEM_NO_ERROR;
|
|
varProp = WMI_COOKER_REQ_TIMESTAMP_PERFTIME;
|
|
}
|
|
}
|
|
else if (m_dwCounterType & PERF_TIMER_100NS)
|
|
{
|
|
dwStatus = pCookingClassQSet->Get( WMI_COOKER_RAW_TIME_100NS, 0, &varProp, NULL );
|
|
if (FAILED(dwStatus) && m_bUseWellKnownIfNeeded)
|
|
{
|
|
dwStatus = WBEM_NO_ERROR;
|
|
varProp = WMI_COOKER_REQ_TIMESTAMP_SYS100NS;
|
|
}
|
|
} else
|
|
{
|
|
dwStatus = pCookingClassQSet->Get( WMI_COOKER_RAW_TIME_SYS, 0, &varProp, NULL );
|
|
if (FAILED(dwStatus) && m_bUseWellKnownIfNeeded)
|
|
{
|
|
dwStatus = WBEM_NO_ERROR;
|
|
varProp = WMI_COOKER_REQ_TIMESTAMP_OBJECT;
|
|
}
|
|
}
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
if ( varProp.vt == VT_BSTR )
|
|
{
|
|
dwStatus = pRawAccess->GetPropertyHandle( varProp.bstrVal, &ct, &lHandle );
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
m_pTimeProp = new CProperty( varProp.bstrVal, lHandle, ct );
|
|
|
|
if ( NULL == m_pTimeProp )
|
|
{
|
|
dwStatus = WBEM_E_OUT_OF_MEMORY;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
dwStatus = WBEM_E_TYPE_MISMATCH;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
dwStatus = WBEM_NO_ERROR;
|
|
}
|
|
}
|
|
|
|
// get in cascade the frequency property
|
|
if (SUCCEEDED(dwStatus))
|
|
{
|
|
_variant_t VarFreqName; // simple container, does not throw
|
|
dwStatus = pCookingPropQualifierSet->Get( WMI_COOKER_RAW_FREQUENCY, 0, &VarFreqName, NULL );
|
|
|
|
if (SUCCEEDED(dwStatus))
|
|
{
|
|
if (VarFreqName.vt == VT_BSTR)
|
|
{
|
|
dwStatus = pRawAccess->GetPropertyHandle( VarFreqName.bstrVal, &ct, &lHandle );
|
|
|
|
if (SUCCEEDED(dwStatus))
|
|
{
|
|
m_pFrequencyProp = new CProperty( VarFreqName.bstrVal, lHandle, ct );
|
|
|
|
if ( NULL == m_pFrequencyProp )
|
|
{
|
|
dwStatus = WBEM_E_OUT_OF_MEMORY;
|
|
}
|
|
}
|
|
|
|
} else {
|
|
dwStatus = WBEM_E_TYPE_MISMATCH;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (m_dwCounterType & PERF_OBJECT_TIMER)
|
|
{
|
|
dwStatus = pCookingClassQSet->Get( WMI_COOKER_RAW_FREQ_OBJ, 0, &VarFreqName, NULL );
|
|
if (FAILED(dwStatus) && m_bUseWellKnownIfNeeded)
|
|
{
|
|
dwStatus = WBEM_NO_ERROR;
|
|
VarFreqName = WMI_COOKER_REQ_FREQUENCY_PERFTIME;
|
|
}
|
|
}
|
|
else if (m_dwCounterType & PERF_TIMER_100NS)
|
|
{
|
|
dwStatus = pCookingClassQSet->Get( WMI_COOKER_RAW_FREQ_100NS, 0, &VarFreqName, NULL );
|
|
if (FAILED(dwStatus) && m_bUseWellKnownIfNeeded)
|
|
{
|
|
dwStatus = WBEM_NO_ERROR;
|
|
VarFreqName = WMI_COOKER_REQ_FREQUENCY_SYS100NS;
|
|
}
|
|
} else
|
|
{
|
|
dwStatus = pCookingClassQSet->Get( WMI_COOKER_RAW_FREQ_SYS, 0, &VarFreqName, NULL );
|
|
if (FAILED(dwStatus) && m_bUseWellKnownIfNeeded)
|
|
{
|
|
dwStatus = WBEM_NO_ERROR;
|
|
VarFreqName = WMI_COOKER_REQ_FREQUENCY_OBJECT;
|
|
}
|
|
}
|
|
|
|
|
|
if (SUCCEEDED(dwStatus))
|
|
{
|
|
if (VarFreqName.vt == VT_BSTR)
|
|
{
|
|
dwStatus = pRawAccess->GetPropertyHandle( VarFreqName.bstrVal, &ct, &lHandle );
|
|
|
|
if (SUCCEEDED(dwStatus))
|
|
{
|
|
m_pFrequencyProp = new CProperty( VarFreqName.bstrVal, lHandle, ct );
|
|
|
|
if ( NULL == m_pFrequencyProp )
|
|
{
|
|
dwStatus = WBEM_E_OUT_OF_MEMORY;
|
|
}
|
|
}
|
|
|
|
} else {
|
|
dwStatus = WBEM_E_TYPE_MISMATCH;
|
|
}
|
|
} else {
|
|
dwStatus = WBEM_S_NO_ERROR;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
//
|
|
// Get the Scale factor from ONLY the property Qualifier
|
|
//
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
_variant_t VarScale; // does not throw, simple container
|
|
dwStatus = pCookingPropQualifierSet->Get( WMI_COOKER_SCALE_FACT, 0, &VarScale, NULL );
|
|
|
|
if ( SUCCEEDED( dwStatus ) && (V_VT(&VarScale) == VT_I4))
|
|
{
|
|
m_lScale = VarScale.intVal;
|
|
}
|
|
else
|
|
{
|
|
dwStatus = WBEM_S_NO_ERROR;
|
|
}
|
|
}
|
|
|
|
|
|
// Get the Sample and Time windows value
|
|
// =====================================
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
DWORD dwSampleStatus = WBEM_NO_ERROR,
|
|
dwTimeStatus = WBEM_NO_ERROR;
|
|
|
|
_variant_t vSampleProp; // does not throw, simple container
|
|
_variant_t vTimeProp; // does not throw, simple container
|
|
|
|
dwSampleStatus = pCookingPropQualifierSet->Get( WMI_COOKER_SAMPLE_WINDOW, 0, &vSampleProp, NULL );
|
|
dwTimeStatus = pCookingPropQualifierSet->Get( WMI_COOKER_TIME_WINDOW, 0, &vTimeProp, NULL );
|
|
|
|
|
|
if ( SUCCEEDED( dwSampleStatus ) && SUCCEEDED( dwTimeStatus ) )
|
|
{
|
|
dwStatus = WBEM_E_INVALID_PROPERTY;
|
|
}
|
|
else if ( SUCCEEDED( dwSampleStatus ) )
|
|
{
|
|
if ( vSampleProp.vt != VT_I4 )
|
|
{
|
|
dwStatus = E_FAIL;
|
|
}
|
|
else
|
|
{
|
|
m_nSampleWindow = vSampleProp.intVal;
|
|
}
|
|
}
|
|
else if ( SUCCEEDED( dwTimeStatus ) )
|
|
{
|
|
if ( vTimeProp.vt != VT_I4 )
|
|
dwStatus = E_FAIL;
|
|
else
|
|
m_nTimeWindow = vTimeProp.intVal;
|
|
}
|
|
else
|
|
{
|
|
m_nSampleWindow = WMI_DEFAULT_SAMPLE_WINDOW;
|
|
}
|
|
}
|
|
}
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
WMISTATUS CCookingProperty::Cook( DWORD dwNumSamples, __int64* aRawCounter, __int64* aBaseCounter, __int64* aTimeStamp, __int64* pnResult )
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
dwStatus = m_Cooker.CookRawValues( m_dwCounterType,
|
|
dwNumSamples,
|
|
aTimeStamp,
|
|
aRawCounter,
|
|
aBaseCounter,
|
|
m_nTimeFreq,
|
|
m_lScale,
|
|
pnResult );
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
CProperty* CCookingProperty::GetRawCounterProperty()
|
|
{
|
|
return m_pRawCounterProp;
|
|
}
|
|
|
|
CProperty* CCookingProperty::GetBaseProperty()
|
|
{
|
|
return m_pBaseProp;
|
|
}
|
|
|
|
CProperty* CCookingProperty::GetTimeProperty()
|
|
{
|
|
return m_pTimeProp;
|
|
}
|
|
|
|
HRESULT
|
|
CCookingProperty::SetFrequency(IWbemObjectAccess * pObjAcc){
|
|
|
|
if (m_nTimeFreq == 0){
|
|
// get the Frequency from the Raw Object
|
|
if (m_pFrequencyProp){
|
|
|
|
return GetPropValue(m_pFrequencyProp,pObjAcc,(__int64 *)&m_nTimeFreq);
|
|
|
|
} else if (!(m_dwReqProp & REQ_FREQ)) {
|
|
|
|
return WBEM_NO_ERROR;
|
|
} else {
|
|
|
|
LARGE_INTEGER li;
|
|
if (QueryPerformanceFrequency(&li)){
|
|
|
|
m_nTimeFreq = li.QuadPart;
|
|
return WBEM_NO_ERROR;
|
|
|
|
} else {
|
|
return WBEM_E_INVALID_PARAMETER;
|
|
}
|
|
|
|
}
|
|
} else {
|
|
return WBEM_NO_ERROR;
|
|
}
|
|
|
|
}
|
|
|
|
unsigned __int64 CCookingProperty::GetFrequency(void){
|
|
return m_nTimeFreq;
|
|
}
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// CPropertySampleCache
|
|
// ====================
|
|
//
|
|
// This class caches the sample data for a single property for a single
|
|
// instance
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
CPropertySampleCache::CPropertySampleCache():
|
|
m_aRawCounterVals(NULL),
|
|
m_aBaseCounterVals(NULL),
|
|
m_aTimeStampVals(NULL),
|
|
m_dwRefreshID(0)
|
|
{
|
|
}
|
|
|
|
CPropertySampleCache::~CPropertySampleCache(){
|
|
|
|
if (m_aRawCounterVals) {
|
|
delete [] m_aRawCounterVals;
|
|
}
|
|
if (m_aBaseCounterVals) {
|
|
delete [] m_aBaseCounterVals;
|
|
}
|
|
if (m_aTimeStampVals) {
|
|
delete [] m_aTimeStampVals;
|
|
}
|
|
}
|
|
|
|
|
|
WMISTATUS CPropertySampleCache::SetSampleInfo( DWORD dwNumActiveSamples, DWORD dwMinReqSamples )
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
m_dwNumSamples = 0;
|
|
m_dwTotSamples = dwNumActiveSamples;
|
|
|
|
m_aRawCounterVals = new __int64[dwNumActiveSamples];
|
|
if (!m_aRawCounterVals)
|
|
return WBEM_E_OUT_OF_MEMORY;
|
|
memset( m_aRawCounterVals, 0, sizeof(__int64) * dwNumActiveSamples );
|
|
|
|
m_aBaseCounterVals = new __int64[dwNumActiveSamples];
|
|
if (!m_aBaseCounterVals)
|
|
return WBEM_E_OUT_OF_MEMORY;
|
|
memset( m_aBaseCounterVals, 0, sizeof(__int64) * dwNumActiveSamples );
|
|
|
|
m_aTimeStampVals = new __int64[dwNumActiveSamples];
|
|
if (!m_aTimeStampVals)
|
|
return WBEM_E_OUT_OF_MEMORY;
|
|
memset( m_aBaseCounterVals, 0, sizeof(__int64) * dwNumActiveSamples );
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
WMISTATUS CPropertySampleCache::SetSampleData( DWORD dwRefreshID, __int64 nRawCounter, __int64 nRawBase, __int64 nTimeStamp )
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
if (dwRefreshID <= m_dwRefreshID){
|
|
return dwStatus;
|
|
} else {
|
|
m_dwRefreshID = dwRefreshID;
|
|
}
|
|
|
|
if ( m_dwNumSamples < m_dwTotSamples )
|
|
{
|
|
m_dwNumSamples++;
|
|
}
|
|
|
|
if ( m_dwTotSamples >= 2 ) {
|
|
for (LONG i = (LONG)(m_dwTotSamples-2); i>=0; i--){
|
|
m_aRawCounterVals[i+1] = m_aRawCounterVals[i];
|
|
m_aBaseCounterVals[i+1] = m_aBaseCounterVals[i];
|
|
m_aTimeStampVals[i+1] = m_aTimeStampVals[i];
|
|
}
|
|
}
|
|
|
|
m_aRawCounterVals[0] = nRawCounter;
|
|
m_aBaseCounterVals[0] = nRawBase;
|
|
m_aTimeStampVals[0] = nTimeStamp;
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
WMISTATUS CPropertySampleCache::GetData( DWORD* pdwNumSamples, __int64** paRawCounter, __int64** paBaseCounter, __int64** paTimeStamp )
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
*pdwNumSamples = m_dwNumSamples;
|
|
*paRawCounter = m_aRawCounterVals;
|
|
*paBaseCounter = m_aBaseCounterVals;
|
|
*paTimeStamp = m_aTimeStampVals;
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// CCookingInstance
|
|
// ================
|
|
//
|
|
// The cooking instance - used to model an instance of a cooked object. Each
|
|
// property maintains a cache of values that will be used to compute the
|
|
// final cooked value.
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
CCookingInstance::CCookingInstance( IWbemObjectAccess *pCookingInstance, DWORD dwNumProps ) :
|
|
m_wszKey( NULL ),
|
|
m_aPropertySamples( NULL ),
|
|
m_pCookingInstance( pCookingInstance ),
|
|
m_pRawInstance( NULL ),
|
|
m_dwNumProps( dwNumProps )
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
// Parameters:
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
if ( NULL != m_pCookingInstance )
|
|
{
|
|
m_pCookingInstance->AddRef();
|
|
m_wszKey = ::GetKey( m_pCookingInstance );
|
|
}
|
|
|
|
if (dwNumProps) {
|
|
m_aPropertySamples = new CPropertySampleCache[dwNumProps];
|
|
};
|
|
}
|
|
|
|
|
|
CCookingInstance::~CCookingInstance()
|
|
{
|
|
if ( NULL != m_wszKey )
|
|
delete [] m_wszKey;
|
|
|
|
if ( NULL != m_pCookingInstance ) {
|
|
m_pCookingInstance->Release();
|
|
}
|
|
|
|
if ( NULL != m_aPropertySamples )
|
|
delete [] m_aPropertySamples;
|
|
|
|
if ( NULL != m_pRawInstance ) {
|
|
m_pRawInstance->Release();
|
|
}
|
|
}
|
|
|
|
WMISTATUS CCookingInstance::InitProperty( DWORD dwProp, DWORD dwNumActiveSamples, DWORD dwMinReqSamples )
|
|
{
|
|
return m_aPropertySamples[dwProp].SetSampleInfo( dwNumActiveSamples, dwMinReqSamples );
|
|
}
|
|
|
|
WMISTATUS CCookingInstance::SetRawSourceInstance( IWbemObjectAccess* pRawSampleSource )
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
if ( NULL != m_pRawInstance )
|
|
{
|
|
m_pRawInstance->Release();
|
|
}
|
|
|
|
m_pRawInstance = pRawSampleSource;
|
|
|
|
if ( NULL != m_pRawInstance )
|
|
{
|
|
m_pRawInstance->AddRef();
|
|
}
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
WMISTATUS CCookingInstance::GetRawSourceInstance( IWbemObjectAccess** ppRawSampleSource )
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
*ppRawSampleSource = m_pRawInstance;
|
|
|
|
if ( NULL != m_pRawInstance )
|
|
{
|
|
m_pRawInstance->AddRef();
|
|
}
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
IWbemObjectAccess* CCookingInstance::GetInstance()
|
|
{
|
|
if ( NULL != m_pCookingInstance )
|
|
m_pCookingInstance->AddRef();
|
|
|
|
return m_pCookingInstance;
|
|
}
|
|
|
|
WMISTATUS CCookingInstance::AddSample( DWORD dwRefreshStamp, DWORD dwProp, __int64 nRawCounter, __int64 nRawBase, __int64 nTimeStamp )
|
|
{
|
|
return m_aPropertySamples[dwProp].SetSampleData( dwRefreshStamp, nRawCounter, nRawBase, nTimeStamp );
|
|
}
|
|
|
|
WMISTATUS CCookingInstance::Refresh( IWbemObjectAccess* pRawData, IWbemObjectAccess** ppCookedData )
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
WMISTATUS CCookingInstance::UpdateSamples()
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
WMISTATUS CCookingInstance::CookProperty( DWORD dwProp, CCookingProperty* pProperty )
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
// Parameters:
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
DWORD dwNumSamples = 0;
|
|
__int64* aRawCounter;
|
|
__int64* aBaseCounter;
|
|
__int64* aTimeStamp;
|
|
__int64 nResult = 0;
|
|
|
|
long lHandle = pProperty->GetHandle();
|
|
|
|
dwStatus = m_aPropertySamples[dwProp].GetData( &dwNumSamples, &aRawCounter, &aBaseCounter, &aTimeStamp );
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
#ifdef _VERBOSE
|
|
{
|
|
WCHAR pBuff[256];
|
|
unsigned __int64 Freq = pProperty->GetFrequency();
|
|
wsprintfW(pBuff,L"PropName %s sample %d\n"
|
|
L"Raw %I64d %I64d\n"
|
|
L"Base %I64d %I64d\n"
|
|
L"Time %I64d %I64d\n"
|
|
L"Freq %I64d\n",
|
|
pProperty->GetName(),dwNumSamples,
|
|
aRawCounter[0],aRawCounter[1],
|
|
aBaseCounter[0],aBaseCounter[1],
|
|
aTimeStamp[0],aTimeStamp[1],
|
|
Freq);
|
|
OutputDebugStringW(pBuff);
|
|
}
|
|
#endif
|
|
|
|
if (SUCCEEDED(dwStatus = pProperty->SetFrequency(m_pRawInstance))){
|
|
|
|
dwStatus = pProperty->Cook( dwNumSamples, aRawCounter, aBaseCounter, aTimeStamp, &nResult );
|
|
}
|
|
#ifdef _VERBOSE
|
|
{
|
|
char pBuff[128];
|
|
wsprintfA(pBuff,"Result %I64d dwStatus %08x\n",nResult,dwStatus);
|
|
OutputDebugStringA(pBuff);
|
|
}
|
|
#endif
|
|
};
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
switch ( pProperty->GetType() )
|
|
{
|
|
case CIM_UINT32:
|
|
dwStatus = m_pCookingInstance->WriteDWORD( lHandle, (DWORD) nResult );
|
|
break;
|
|
case CIM_UINT64:
|
|
dwStatus = m_pCookingInstance->WriteQWORD( lHandle, nResult );
|
|
break;
|
|
default:
|
|
dwStatus = WBEM_E_TYPE_MISMATCH;
|
|
}
|
|
};
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
// CEnumeratorCache
|
|
//
|
|
//
|
|
/////////////////////////////////////////////////////////////////////////
|
|
|
|
CEnumeratorCache::CEnumeratorCache() :
|
|
m_dwEnum( 0 )
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
// Parameters:
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
InitializeCriticalSection(&m_cs);
|
|
}
|
|
|
|
CEnumeratorCache::~CEnumeratorCache()
|
|
{
|
|
EnterCriticalSection(&m_cs);
|
|
DWORD i;
|
|
for (i=0;i<m_apEnumerators.size();i++){
|
|
CEnumeratorManager* pEnumMgr = m_apEnumerators[i];
|
|
if (pEnumMgr)
|
|
pEnumMgr->Release();
|
|
}
|
|
LeaveCriticalSection(&m_cs);
|
|
DeleteCriticalSection(&m_cs);
|
|
}
|
|
|
|
WMISTATUS CEnumeratorCache::AddEnum( LPCWSTR wszCookingClass,
|
|
IWbemClassObject* pCookedClass,
|
|
IWbemClassObject* pRawClass,
|
|
IWbemHiPerfEnum* pCookedEnum,
|
|
IWbemHiPerfEnum* pRawEnum,
|
|
long lIDRaw,
|
|
DWORD* pdwID )
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
// Parameters:
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
CEnumeratorManager* pEnumMgr = new CEnumeratorManager( wszCookingClass, pCookedClass, pRawClass, pCookedEnum, pRawEnum, lIDRaw );
|
|
|
|
//
|
|
CInCritSec ics(&m_cs);
|
|
//
|
|
|
|
if (pEnumMgr)
|
|
{
|
|
if (SUCCEEDED(pEnumMgr->GetInithResult()))
|
|
{
|
|
DWORD i;
|
|
for (i=0;i<m_apEnumerators.size();i++)
|
|
{
|
|
if(m_apEnumerators[i] == NULL)
|
|
{
|
|
m_apEnumerators[i] = pEnumMgr;
|
|
if (pdwID)
|
|
{
|
|
*pdwID = i;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
// we need to expand the array
|
|
if (i == m_apEnumerators.size())
|
|
{
|
|
try
|
|
{
|
|
m_apEnumerators.push_back(pEnumMgr);
|
|
if (pdwID)
|
|
{
|
|
*pdwID = m_apEnumerators.size()-1;
|
|
}
|
|
}
|
|
catch (...)
|
|
{
|
|
pEnumMgr->Release();
|
|
dwStatus = WBEM_E_OUT_OF_MEMORY;
|
|
}
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
dwStatus = pEnumMgr->GetInithResult();
|
|
}
|
|
|
|
} else {
|
|
dwStatus = WBEM_E_OUT_OF_MEMORY;
|
|
}
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
WMISTATUS CEnumeratorCache::RemoveEnum( DWORD dwID , long * pRawId )
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
// Parameters:
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
//
|
|
EnterCriticalSection(&m_cs);
|
|
//
|
|
|
|
if ( dwID < m_apEnumerators.size() ) {
|
|
|
|
CEnumeratorManager* pEnumMgr = m_apEnumerators[dwID];
|
|
m_apEnumerators[dwID] = NULL;
|
|
|
|
*pRawId = pEnumMgr->GetRawId();
|
|
|
|
pEnumMgr->Release();
|
|
|
|
} else {
|
|
dwStatus = E_FAIL;
|
|
}
|
|
|
|
//
|
|
LeaveCriticalSection(&m_cs);
|
|
//
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
WMISTATUS CEnumeratorCache::Refresh(DWORD dwRefreshId)
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
// Parameters:
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
|
|
CEnumeratorManager** ppEnumMang = new CEnumeratorManager*[m_apEnumerators.size()];
|
|
wmilib::auto_buffer<CEnumeratorManager*> rm_(ppEnumMang);
|
|
if (!ppEnumMang)
|
|
return WBEM_E_OUT_OF_MEMORY;
|
|
|
|
memset(ppEnumMang,0,sizeof(CEnumeratorManager*)*m_apEnumerators.size());
|
|
|
|
DWORD j=0;
|
|
DWORD i=0;
|
|
|
|
{
|
|
CInCritSec ics(&m_cs);
|
|
|
|
for (i=0;i<m_apEnumerators.size();i++){
|
|
|
|
CEnumeratorManager* pEnumMgr = m_apEnumerators[i];
|
|
if (pEnumMgr)
|
|
{
|
|
pEnumMgr->AddRef();
|
|
ppEnumMang[j] = pEnumMgr;
|
|
j++;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
|
|
for (i=0;i<j;i++)
|
|
{
|
|
dwStatus = ppEnumMang[i]->Refresh(dwRefreshId);
|
|
if (FAILED(dwStatus))
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
|
|
for (i=0;i<j;i++)
|
|
{
|
|
ppEnumMang[i]->Release();
|
|
}
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
// CInstanceRecord
|
|
//
|
|
//
|
|
/////////////////////////////////////////////////////////////////////////
|
|
|
|
CInstanceRecord::CInstanceRecord( LPWSTR wszName, long lID )
|
|
:m_lID( lID ),
|
|
m_wszName(NULL)
|
|
{
|
|
m_wszName = new WCHAR[ wcslen( wszName ) + 1];
|
|
if (m_wszName)
|
|
wcscpy( m_wszName, wszName );
|
|
}
|
|
|
|
CInstanceRecord::~CInstanceRecord()
|
|
{
|
|
if ( NULL != m_wszName )
|
|
{
|
|
delete [] m_wszName;
|
|
}
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
// CEnumeratorManager
|
|
//
|
|
//
|
|
/////////////////////////////////////////////////////////////////////////
|
|
|
|
|
|
|
|
CEnumeratorManager::CEnumeratorManager( LPCWSTR wszCookingClass,
|
|
IWbemClassObject* pCookedClass,
|
|
IWbemClassObject* pRawClass,
|
|
IWbemHiPerfEnum* pCookedEnum,
|
|
IWbemHiPerfEnum* pRawEnum,
|
|
long lRawID )
|
|
: m_pCookedClass( pCookedClass ),
|
|
m_pRawEnum(pRawEnum),
|
|
m_pCookedEnum( pCookedEnum ),
|
|
m_pCooker(NULL),
|
|
m_lRawID(lRawID),
|
|
m_dwSignature('mMnE'),
|
|
m_cRef(1), //-------------- initial refcount
|
|
m_dwVector(0)
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
// Parameters:
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
#ifdef _VERBOSE
|
|
{
|
|
char pBuff[256];
|
|
wsprintfA(pBuff,"CEnumeratorManager %08x\n",this);
|
|
OutputDebugStringA(pBuff);
|
|
}
|
|
#endif
|
|
m_wszCookingClassName = new WCHAR[ wcslen( wszCookingClass ) + 1 ];
|
|
|
|
if ( NULL != m_wszCookingClassName )
|
|
wcscpy( m_wszCookingClassName, wszCookingClass );
|
|
|
|
if ( NULL != m_pCookedClass )
|
|
m_pCookedClass->AddRef();
|
|
|
|
if ( NULL != m_pRawEnum )
|
|
m_pRawEnum->AddRef();
|
|
|
|
if ( NULL != m_pCookedEnum )
|
|
m_pCookedEnum->AddRef();
|
|
|
|
m_IsSingleton = IsSingleton(pRawClass);
|
|
|
|
InitializeCriticalSection(&m_cs);
|
|
|
|
m_InithRes = Initialize( pRawClass );
|
|
}
|
|
|
|
CEnumeratorManager::~CEnumeratorManager()
|
|
{
|
|
m_dwSignature = 'gmne';
|
|
|
|
if ( NULL != m_wszCookingClassName )
|
|
delete m_wszCookingClassName;
|
|
|
|
// one reference is hold by the CWMISimpleObjectCooker
|
|
if ( NULL != m_pCookedClass ) {
|
|
m_pCookedClass->Release();
|
|
}
|
|
|
|
if ( NULL != m_pRawEnum ) {
|
|
m_pRawEnum->Release();
|
|
}
|
|
|
|
if ( NULL != m_pCookedEnum ){
|
|
m_pCookedEnum->Release();
|
|
}
|
|
|
|
if (m_pCooker)
|
|
delete m_pCooker;
|
|
|
|
DeleteCriticalSection(&m_cs);
|
|
|
|
#ifdef _VERBOSE
|
|
{
|
|
char pBuff[256];
|
|
wsprintfA(pBuff,"~CEnumeratorManager %08x\n",this);
|
|
OutputDebugStringA(pBuff);
|
|
}
|
|
#endif
|
|
|
|
|
|
}
|
|
|
|
LONG CEnumeratorManager::AddRef()
|
|
{
|
|
return InterlockedIncrement(&m_cRef);
|
|
}
|
|
|
|
LONG CEnumeratorManager::Release()
|
|
{
|
|
LONG lRet = InterlockedDecrement(&m_cRef);
|
|
if (lRet == 0)
|
|
{
|
|
delete this;
|
|
}
|
|
|
|
return lRet;
|
|
}
|
|
|
|
|
|
WMISTATUS CEnumeratorManager::Initialize( IWbemClassObject* pRawClass )
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
// Parameters:
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
WMISTATUS dwStatus;
|
|
HRESULT hr1,hr2;
|
|
|
|
|
|
IWbemObjectAccess* pCookedAccess = NULL;
|
|
IWbemObjectAccess* pRawAccess = NULL;
|
|
|
|
hr1 = m_pCookedClass->QueryInterface( IID_IWbemObjectAccess, (void**)&pCookedAccess );
|
|
CAutoRelease arCookedAccess( pCookedAccess );
|
|
|
|
hr2 = pRawClass->QueryInterface( IID_IWbemObjectAccess, (void**)&pRawAccess );
|
|
CAutoRelease arRawAccess( pRawAccess );
|
|
|
|
if (SUCCEEDED(hr1) && SUCCEEDED(hr2))
|
|
{
|
|
m_pCooker = new CWMISimpleObjectCooker( m_wszCookingClassName,
|
|
pCookedAccess, // acquired by CWMISimpleObjectCooker
|
|
pRawAccess );
|
|
}
|
|
|
|
if (m_pCooker == NULL)
|
|
{
|
|
dwStatus = WBEM_E_OUT_OF_MEMORY;
|
|
}
|
|
else
|
|
{
|
|
dwStatus = m_pCooker->GetLastHR();
|
|
}
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
ULONG_PTR hash_string (WCHAR * pKey)
|
|
{
|
|
ULONG_PTR acc = 0;
|
|
ULONG_PTR i = 0;
|
|
WCHAR *this_char = pKey;
|
|
|
|
while (*this_char != NULL) {
|
|
acc ^= *(this_char++) << i;
|
|
i = (i + 1) % sizeof (void *);
|
|
}
|
|
|
|
return (acc<<1); // so we can save the LOWEST bit
|
|
}
|
|
|
|
|
|
WMISTATUS
|
|
CEnumeratorManager::GetRawEnumObjects(std::vector<IWbemObjectAccess*, wbem_allocator<IWbemObjectAccess*> > & refArray,
|
|
std::vector<ULONG_PTR, wbem_allocator<ULONG_PTR> > & refObjHashKeys)
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
// Parameters:
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
DWORD dwRet = 0,
|
|
dwNumRawObjects = 0;
|
|
|
|
IWbemObjectAccess** apObjAccess = NULL;
|
|
|
|
dwStatus = m_pRawEnum->GetObjects( 0L, 0, apObjAccess, &dwRet);
|
|
|
|
if ( WBEM_E_BUFFER_TOO_SMALL == dwStatus )
|
|
{
|
|
// Set the buffer size
|
|
// ===================
|
|
dwNumRawObjects = dwRet;
|
|
|
|
apObjAccess = new IWbemObjectAccess*[dwNumRawObjects];
|
|
memset( apObjAccess, 0, sizeof( apObjAccess ));
|
|
|
|
if ( NULL != apObjAccess )
|
|
{
|
|
dwStatus = m_pRawEnum->GetObjects( 0L, dwNumRawObjects, apObjAccess, &dwRet );
|
|
}
|
|
else
|
|
{
|
|
dwStatus = WBEM_E_OUT_OF_MEMORY;
|
|
}
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
try
|
|
{
|
|
refArray.reserve(dwNumRawObjects);
|
|
refObjHashKeys.reserve(dwNumRawObjects);
|
|
}
|
|
catch (...)
|
|
{
|
|
dwStatus = WBEM_E_OUT_OF_MEMORY;
|
|
dwNumRawObjects = 0;
|
|
}
|
|
|
|
for (DWORD i=0;i<dwNumRawObjects;i++)
|
|
{
|
|
HRESULT hr1;
|
|
_variant_t VarKey; // does not throw, just container
|
|
hr1 = apObjAccess[i]->Get(L"__RELPATH",0,&VarKey,NULL,NULL);
|
|
if (SUCCEEDED(hr1))
|
|
{
|
|
DWORD Hash = hash_string(VarKey.bstrVal);
|
|
refObjHashKeys.push_back(Hash);
|
|
refArray.push_back(apObjAccess[i]);
|
|
} else {
|
|
// if we cannot give out the ownership of a pointer, release
|
|
apObjAccess[i]->Release();
|
|
}
|
|
}
|
|
|
|
delete [] apObjAccess;
|
|
}
|
|
}
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
|
|
WMISTATUS
|
|
CEnumeratorManager::UpdateEnums(
|
|
std::vector<ULONG_PTR, wbem_allocator<ULONG_PTR> > & apObjKeyHash)
|
|
///////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
// Parameters:
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////
|
|
{
|
|
// cyclic logic:
|
|
// we have a 'circular array' of std::vector
|
|
// and the m_dwVector is the index
|
|
// circular increment of the index will decide
|
|
// who is the New and who is the Old
|
|
std::vector<ULONG_PTR, wbem_allocator<ULONG_PTR> > & Old = m_Delta[m_dwVector];
|
|
m_dwVector = (m_dwVector+1)%2;
|
|
m_Delta[m_dwVector].clear();
|
|
m_Delta[m_dwVector] = apObjKeyHash;
|
|
std::vector<ULONG_PTR, wbem_allocator<ULONG_PTR> > & New = m_Delta[m_dwVector];
|
|
|
|
DWORD j,k;
|
|
|
|
for (j=0;j<New.size();j++)
|
|
{
|
|
BOOL bFound = FALSE;
|
|
for (k=0;k<Old.size();k++)
|
|
{
|
|
if (Old[k] == New[j])
|
|
{
|
|
Old[k] |= 1;
|
|
bFound = TRUE;
|
|
break;
|
|
}
|
|
}
|
|
if (!bFound)
|
|
{
|
|
New[j] |= 1; // ad the very NEW bit
|
|
}
|
|
}
|
|
|
|
return WBEM_S_NO_ERROR;
|
|
|
|
}
|
|
|
|
WMISTATUS CEnumeratorManager::Refresh( DWORD dwRefreshStamp )
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
std::vector<IWbemObjectAccess*, wbem_allocator<IWbemObjectAccess*> > apObjAccess;
|
|
std::vector<ULONG_PTR, wbem_allocator<ULONG_PTR> > apObjHashKeys;
|
|
|
|
dwStatus = GetRawEnumObjects( apObjAccess, apObjHashKeys );
|
|
|
|
// calculate the Delta of the caches
|
|
if (SUCCEEDED(dwStatus))
|
|
{
|
|
dwStatus = UpdateEnums(apObjHashKeys);
|
|
}
|
|
|
|
std::vector<ULONG_PTR, wbem_allocator<ULONG_PTR> > & New = m_Delta[m_dwVector];
|
|
std::vector<ULONG_PTR, wbem_allocator<ULONG_PTR> > & Old = m_Delta[(m_dwVector-1)%2];
|
|
|
|
{
|
|
CInCritSec ics(&m_cs);
|
|
|
|
// Merge into the cache
|
|
if ( SUCCEEDED(dwStatus) )
|
|
{
|
|
//
|
|
// Elements in the New array with the bit set are really new
|
|
//
|
|
DWORD j;
|
|
for (j=0; j< New.size(); j++)
|
|
{
|
|
if (New[j] & 1) // test the very new BIT
|
|
{
|
|
EnumCookId thisEnumCookId;
|
|
dwStatus = InsertCookingRecord( apObjAccess[j], &thisEnumCookId, dwRefreshStamp );
|
|
if (SUCCEEDED(dwStatus))
|
|
{
|
|
try {
|
|
m_mapID[New[j]] = thisEnumCookId;
|
|
} catch (...) {
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
//remove the bit
|
|
New[j] &= (~1);
|
|
}
|
|
}
|
|
|
|
for (j=0; j<Old.size(); j++)
|
|
{
|
|
if (Old[j] & 1)
|
|
{
|
|
Old[j] &= (~1); // remove the ALREADY_THERE bit
|
|
}
|
|
else
|
|
{
|
|
EnumCookId thisEnumCookId;
|
|
thisEnumCookId = m_mapID[Old[j]];
|
|
m_mapID.erase(Old[j]);
|
|
RemoveCookingRecord(&thisEnumCookId);
|
|
}
|
|
}
|
|
m_pCooker->Recalc(dwRefreshStamp);
|
|
}
|
|
|
|
}
|
|
|
|
// in any case ....
|
|
for (DWORD i=0;i<apObjAccess.size();i++)
|
|
{
|
|
apObjAccess[i]->Release();
|
|
};
|
|
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
WMISTATUS
|
|
CEnumeratorManager::InsertCookingRecord(
|
|
IWbemObjectAccess* pRawObject,
|
|
EnumCookId * pEnumCookId,
|
|
DWORD dwRefreshStamp)
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
// Parameters:
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
if (!pRawObject || !pEnumCookId)
|
|
{
|
|
return WBEM_E_INVALID_PARAMETER;
|
|
}
|
|
|
|
IWbemObjectAccess* pCookedObject = NULL;
|
|
|
|
long lID = 0;
|
|
|
|
dwStatus = CreateCookingObject( pRawObject, &pCookedObject );
|
|
CAutoRelease rm1(pCookedObject);
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
dwStatus = m_pCooker->SetCookedInstance( pCookedObject, &lID );
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
dwStatus = m_pCooker->BeginCooking( lID, pRawObject,dwRefreshStamp);
|
|
}
|
|
}
|
|
|
|
if ( SUCCEEDED( dwStatus ) )
|
|
{
|
|
DWORD dwTarget;
|
|
long EnumId = lID;
|
|
|
|
dwStatus = m_pCookedEnum->AddObjects( 0L, 1, &EnumId, &pCookedObject );
|
|
if (SUCCEEDED(dwStatus))
|
|
{
|
|
pEnumCookId->CookId = lID;
|
|
pEnumCookId->EnumId = EnumId;
|
|
m_dwUsage++;
|
|
}
|
|
else
|
|
{
|
|
pEnumCookId->CookId = 0;
|
|
pEnumCookId->EnumId = 0;
|
|
}
|
|
}
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
|
|
|
|
|
|
WMISTATUS CEnumeratorManager::CreateCookingObject(
|
|
IWbemObjectAccess* pRawObject,
|
|
IWbemObjectAccess** ppCookedObject )
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Create a new instance of the cooked object and set the key(s) based on the
|
|
// raw object's key(s) value.
|
|
//
|
|
// Parameters:
|
|
//
|
|
// pRawObject - The new object's corresponding raw object
|
|
// ppCookedObject - The new cooked object
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
|
|
HRESULT hr = WBEM_E_FAILED;
|
|
IWbemClassObject * pCookedInst = NULL;
|
|
|
|
hr = m_pCookedClass->SpawnInstance(0,&pCookedInst);
|
|
CAutoRelease rm1(pCookedInst);
|
|
|
|
if (SUCCEEDED(hr) &&
|
|
!m_IsSingleton){
|
|
|
|
// get the 'list' of all the key property
|
|
// if you haven't got it in the past
|
|
if (m_pKeyProps.size() == 0)
|
|
{
|
|
hr = pRawObject->BeginEnumeration(WBEM_FLAG_KEYS_ONLY);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
BSTR bstrName;
|
|
|
|
while(WBEM_S_NO_ERROR == pRawObject->Next(0,&bstrName,NULL,NULL,NULL))
|
|
{
|
|
try
|
|
{
|
|
m_pKeyProps.push_back(bstrName);
|
|
}
|
|
catch (...)
|
|
{
|
|
hr = WBEM_E_OUT_OF_MEMORY;
|
|
};
|
|
SysFreeString(bstrName);
|
|
};
|
|
|
|
pRawObject->EndEnumeration();
|
|
}
|
|
}
|
|
|
|
// copy all the key properties from the Raw to the cooked instance
|
|
if (m_pKeyProps.size() > 0 && SUCCEEDED(hr))
|
|
{
|
|
for(int i=0;i<m_pKeyProps.size();i++)
|
|
{
|
|
// does not thorow, just a container
|
|
_variant_t VarVal;
|
|
CIMTYPE ct;
|
|
hr = pRawObject->Get(m_pKeyProps[i],0,&VarVal,&ct,NULL);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = pCookedInst->Put(m_pKeyProps[i],0,&VarVal,0);
|
|
|
|
if (FAILED(hr))
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
VarVal.Clear();
|
|
}
|
|
} else {
|
|
|
|
hr = WBEM_E_INVALID_CLASS;
|
|
|
|
}
|
|
};
|
|
|
|
if (SUCCEEDED(hr)){
|
|
hr = pCookedInst->QueryInterface( IID_IWbemObjectAccess, (void**)ppCookedObject );
|
|
}
|
|
|
|
return hr;
|
|
}
|
|
|
|
WMISTATUS CEnumeratorManager::RemoveCookingRecord( EnumCookId * pEnumCookID )
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
// Parameters:
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
if (!pEnumCookID)
|
|
{
|
|
return WBEM_E_INVALID_PARAMETER;
|
|
}
|
|
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
dwStatus = m_pCookedEnum->RemoveObjects( 0L, 1, &pEnumCookID->EnumId );
|
|
|
|
m_pCooker->StopCooking(pEnumCookID->CookId);
|
|
|
|
m_pCooker->Remove(pEnumCookID->CookId);
|
|
|
|
|
|
--m_dwUsage;
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
// Predicate Function for the std::sort
|
|
//
|
|
///////////////////////////////////////////////////////////////////////
|
|
|
|
bool Pr(IWbemObjectAccess* pFirst,IWbemObjectAccess* pSec){
|
|
|
|
|
|
WCHAR * pStr1 = GetKey(pFirst);
|
|
WCHAR * pStr2 = GetKey(pSec);
|
|
|
|
int res = -1; // to force true when GetKey fails
|
|
|
|
if (pStr1 && pStr2)
|
|
{
|
|
res = _wcsicmp(pStr1,pStr2);
|
|
}
|
|
|
|
if (pStr1){
|
|
delete [] pStr1;
|
|
}
|
|
if (pStr2){
|
|
delete [] pStr2;
|
|
}
|
|
|
|
return (res<0);
|
|
}
|
|
|
|
WMISTATUS CEnumeratorManager::SortRawArray(std::vector<IWbemObjectAccess*, wbem_allocator<IWbemObjectAccess*> > & refArray )
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
//
|
|
//
|
|
// Parameters:
|
|
//
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
{
|
|
WMISTATUS dwStatus = WBEM_NO_ERROR;
|
|
|
|
std::sort(refArray.begin(),refArray.end(),Pr);
|
|
|
|
return dwStatus;
|
|
}
|
|
|