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.
 
 
 
 
 
 

648 lines
22 KiB

/*++
Copyright (C) 1996-2001 Microsoft Corporation
Module Name:
Abstract:
History:
--*/
// RefreshCooker.cpp
#include "precomp.h"
#include <wbemint.h>
#include <comdef.h>
#include <autoptr.h>
#include "Refresher.h"
#include "CookerUtils.h"
////////////////////////////////////////////////////////////////////////////
//
// CRefresher
// ==========
//
// The refresher class implements both the IWbemRefresher and the
// IWMIRefreshableCooker interfaces. It contains an instance cache and
// an enumerator cache as well as maintaining an internal refresher to
// track the raw data.
//
////////////////////////////////////////////////////////////////////////////
CRefresher::CRefresher() :
m_pRefresher( NULL ),
m_pConfig( NULL ),
m_lRef( 0 ),
m_bOK( FALSE ),
m_dwRefreshId(0)
{
#ifdef _VERBOSE
DbgPrintfA(0,"------------ CRefresher %08x \n",this);
#endif
WMISTATUS dwStatus = WBEM_NO_ERROR;
// Initialize the internal refresher
// =================================
dwStatus = CoCreateInstance( CLSID_WbemRefresher,
NULL,
CLSCTX_INPROC_SERVER,
IID_IWbemRefresher,
(void**) &m_pRefresher );
// Get the refresher configuration interface
// =========================================
if ( SUCCEEDED( dwStatus ) )
{
dwStatus = m_pRefresher->QueryInterface( IID_IWbemConfigureRefresher, (void**)&m_pConfig );
}
m_bOK = SUCCEEDED( dwStatus );
}
CRefresher::~CRefresher()
{
if (m_pRefresher ) m_pRefresher->Release();
if ( m_pConfig ) m_pConfig->Release();
}
///////////////////////////////////////////////////////////////////////////////
//
// Private methods
//
///////////////////////////////////////////////////////////////////////////////
WMISTATUS CRefresher::SearchCookingClassCache(
WCHAR* wszCookingClass,
/*out*/ CWMISimpleObjectCooker* & pObjectCooker_out )
///////////////////////////////////////////////////////////////////////////////
//
// SearchCookingClassCache enumerates the cache looking for a class name
// that matches the wszCookingClass parameter
//
// Parameters:
// wszCookingClass - The name of the WMI cooking class
// ppObjectCooker - The instance of the object cooker
//
///////////////////////////////////////////////////////////////////////////////
{
WMISTATUS dwStatus = WBEM_E_NOT_FOUND;
CWMISimpleObjectCooker* pObjectCooker = NULL;
// Enumerate through the cache looking for the record
m_CookingClassCache.BeginEnum();
while ( S_OK == m_CookingClassCache.Next( &pObjectCooker ) )
{
// Compare the names
// =================
if ( 0 == wbem_wcsicmp( pObjectCooker->GetCookingClassName(), wszCookingClass ) )
{
pObjectCooker_out = pObjectCooker;
dwStatus = WBEM_NO_ERROR;
break;
}
}
m_CookingClassCache.EndEnum();
return dwStatus;
}
WMISTATUS CRefresher::CreateObjectCooker(
WCHAR* wszCookingClassName,
IWbemObjectAccess* pCookingAccess,
IWbemObjectAccess* pRawAccess,
CWMISimpleObjectCooker** ppObjectCooker,
IWbemServices * pNamespace)
///////////////////////////////////////////////////////////////////////////////
//
// CreateObjectCooker will create and initialize a new object cooker and add
// it to the cache
//
// Parameters:
// pNamespace - The namespace pointer where the objects are located
// pCookingAccess - The WMI cooking object in need of a cooker
// wszCookingClassName
// - The name of the cooking class
// ppObjectCooker - The parameter to pass back the new object cooker
//
///////////////////////////////////////////////////////////////////////////////
{
if (NULL == ppObjectCooker) return WBEM_E_INVALID_PARAMETER;
CWMISimpleObjectCooker* pObjectCooker = NULL;
WCHAR* wszRawClassName;
long lID;
pObjectCooker = new CWMISimpleObjectCooker( wszCookingClassName, pCookingAccess, pRawAccess, pNamespace );
if ( NULL == pObjectCooker ) return WBEM_E_OUT_OF_MEMORY;
WMISTATUS dwStatus = pObjectCooker->GetLastHR();
// Add the object cooker to the cache
if ( SUCCEEDED( dwStatus ) )
{
dwStatus = m_CookingClassCache.Add( pObjectCooker, wszCookingClassName, &lID );
}
if (FAILED(dwStatus))
{
delete pObjectCooker;
pObjectCooker = NULL;
}
*ppObjectCooker = pObjectCooker;
return dwStatus;
}
WMISTATUS CRefresher::AddRawInstance(
IWbemServices* pService,
IWbemContext * pCtx,
IWbemObjectAccess* pCookingInst,
IWbemObjectAccess** ppRawInst )
///////////////////////////////////////////////////////////////////////////////
//
// AddRawInstance is called to add the corresponding raw instance of a
// cooked object to the internal refresher. We first extract the key value
// from the cooked object and create the raw instance path using the raw
// class name
//
// Parameters:
// pService - The namespace pointer where the objects are located
// pCookingInst - The WMI cooking instance
// ppRawInst - The WMI raw instance that was added to the internal
// refresher
//
///////////////////////////////////////////////////////////////////////////////
{
WMISTATUS dwStatus = WBEM_NO_ERROR;
IWbemClassObject* pObj = NULL; // The alternate representation of pCookingInst
_variant_t varRelPath; // The RELPATH value
WCHAR* wszRawClassName = NULL; // The name of the raw class
// Get the fully specified instance path for the cooking object
// ============================================================
pCookingInst->QueryInterface( IID_IWbemClassObject, (void**)&pObj );
CReleaseMe arObj( pObj );
dwStatus = pObj->Get( L"__RELPATH", 0, &varRelPath, NULL, NULL );
if ( SUCCEEDED( dwStatus ) )
{
// Verify the property type
// ========================
if ( varRelPath.vt != VT_BSTR )
{
dwStatus = WBEM_E_TYPE_MISMATCH;
}
if ( SUCCEEDED( dwStatus ) )
{
IWbemClassObject* pRawInst = NULL;
WCHAR* wszKeys = NULL;
WCHAR* wszRawInst = NULL;
// Extract the key name
// ====================
wszKeys = wcsstr( varRelPath.bstrVal, L"=" ) + 1;
// Get the raw class name
// ======================
dwStatus = GetRawClassName( pCookingInst, &wszRawClassName );
if (SUCCEEDED(dwStatus))
{
wmilib::auto_buffer<WCHAR> adRawClassName( wszRawClassName );
// Append the key to the raw class name
// ====================================
size_t length = wcslen( wszRawClassName ) + wcslen( wszKeys ) + 10;
wszRawInst = new WCHAR[ length ];
if (!wszRawInst)
return WBEM_E_OUT_OF_MEMORY;
wmilib::auto_buffer<WCHAR> adRawInst( wszRawInst );
StringCchPrintfW( wszRawInst, length , L"%s=%s", wszRawClassName, wszKeys );
// Add a raw instance to the internal refresher
// ============================================
dwStatus = m_pConfig->AddObjectByPath( pService, wszRawInst, 0, pCtx, &pRawInst, NULL );
CReleaseMe arRawInst( pRawInst );
if (SUCCEEDED(dwStatus)) {
// Return the IWbemObjectAccess interface of the raw instance
// ==========================================================
dwStatus = pRawInst->QueryInterface( IID_IWbemObjectAccess, (void**)ppRawInst );
}
}
}
}
return dwStatus;
}
WMISTATUS CRefresher::AddRawEnum(
IWbemServices* pNamespace,
IWbemContext * pCtx,
WCHAR * wszRawClassName,
IWbemHiPerfEnum** ppRawEnum,
long* plID )
///////////////////////////////////////////////////////////////////////////////
//
// AddRawEnum is called to add the corresponding raw enumerator to the
// internal refrehser. In order to add the raw enumerator to the refresher,
// we must determine the raw class name, therefore, we must create a
// cooking class in order to get the AutoCook_RawClass qualifier.
//
// Parameters:
// pNamespace - The namespace pointer where the objects are located
// wszRawClassName - The name of the cooking class
// ppRawEnum - The raw WMI enumerator that was added to the
// internal refresher
// plID - The refresher ID of the raw enumerator
//
///////////////////////////////////////////////////////////////////////////////
{
// Add the Raw enumerator to the internal refresher
WMISTATUS dwStatus = m_pConfig->AddEnum( pNamespace, wszRawClassName, 0, pCtx, ppRawEnum, plID );
#ifdef _VERBOSE
DbgPrintfA(0,"wszRawClassName %S pEnum %08x hr %08x\n",wszRawClassName,*ppRawEnum,dwStatus);
#endif
return dwStatus;
}
///////////////////////////////////////////////////////////////////////////////
//
// COM methods
//
///////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CRefresher::QueryInterface(REFIID riid, void** ppv)
///////////////////////////////////////////////////////////////////////////////
//
// Standard QueryInterface
//
// Parameters:
// riid - the ID of the requested interface
// ppv - a pointer to the interface pointer
//
///////////////////////////////////////////////////////////////////////////////
//ok
{
if (NULL == ppv) return E_POINTER;
if(riid == IID_IUnknown)
*ppv = (LPVOID)(IUnknown*)(IWMIRefreshableCooker*)this;
else if(riid == IID_IWMIRefreshableCooker)
*ppv = (LPVOID)(IWMIRefreshableCooker*)this;
else if(riid == IID_IWbemRefresher)
*ppv = (LPVOID)(IWbemRefresher*)this;
else
{
*ppv = NULL;
return E_NOINTERFACE;
}
((IUnknown*)*ppv)->AddRef();
return S_OK;
}
STDMETHODIMP_(ULONG) CRefresher::AddRef()
{
return InterlockedIncrement(&m_lRef);
}
STDMETHODIMP_(ULONG) CRefresher::Release()
{
long lRef = InterlockedDecrement(&m_lRef);
if(lRef == 0) delete this;
return lRef;
}
STDMETHODIMP CRefresher::AddInstance(
/*[in] */ IWbemServices* pNamespace, // The object's namespace
/*[in] */ IWbemContext * pCtx, // The Context
/*[in] */ IWbemObjectAccess* pCookingInstance, // Cooking class definition
/*[in] */ IWbemObjectAccess* pRefreshableRawInstance, // Raw instance
/*[out] */ IWbemObjectAccess** ppRefreshableInstance, // Cooking instance
/*[out] */ long* plId )
///////////////////////////////////////////////////////////////////////////////
//
// AddInstance is called to add a WMI cooking instance to the refresher. The
// refreshable instance is a clone of the WMI instance that is passed in
// by pCookingInstance. Once the instance is cloned, the corresponding raw
// instance is added to the internal refresher, and then the cloned
// cooked instance and the refreshable raw instance are added to the object
// cooker. If a cooker does not already exist in the cooker cache, one
// is created.
//
// Parameters:
// pNamespace - The namespace where the objects are located
// pCtx - IWbemContext implementation
// pCookingInstance - The instance to be cooked
// pRefreshableRawInstance
// - U N U S E D P A R A M
// ppRefreshableInstance
// - The refreshable cooking instance passed back to
// the client
// plId - The ID of the instance
//
///////////////////////////////////////////////////////////////////////////////
{
HRESULT hResult = S_OK;
CWMISimpleObjectCooker* pObjectCooker = NULL;
IWbemObjectAccess* pInternalRawInst = NULL; // The raw instance for the short term local refresher solution
// For now, we expect that the pRefreshableRawInstance parameter will be NULL
// since we are using an internal refresher to manage the raw instances
// ==========================================================================
if ( NULL == pNamespace || NULL == pCookingInstance || NULL != pRefreshableRawInstance )
{
return WBEM_E_INVALID_PARAMETER;
}
IWbemClassObject* pNewClassObj = NULL;
IWbemClassObject* pClassObj = pCookingInstance;
pClassObj->AddRef();
CReleaseMe arClassObj( pClassObj );
hResult = pClassObj->Clone( &pNewClassObj );
if (SUCCEEDED(hResult))
{
CReleaseMe arNewClassObj( pNewClassObj );
hResult = pNewClassObj->QueryInterface( IID_IWbemObjectAccess, (void**)ppRefreshableInstance );
// Add the instance to the object cooker
if ( SUCCEEDED( hResult ) )
{
// Get the raw instance (add it to the internal refresher)
hResult = AddRawInstance( pNamespace, pCtx, *ppRefreshableInstance, &pInternalRawInst );
CReleaseMe arInternalRawInst( pInternalRawInst );
// Retrieve the class cooker
if ( SUCCEEDED( hResult ) )
{
WCHAR* wszClassName = NULL;
// Get the cooked class' name
hResult = GetClassName( pCookingInstance, &wszClassName );
wmilib::auto_buffer<WCHAR> adaClassName( wszClassName );
if ( SUCCEEDED( hResult ) )
{
// Search for an existing cooking cache object
hResult = SearchCookingClassCache( wszClassName, pObjectCooker );
// If it does not exist, create a new one
if ( FAILED ( hResult ) )
{
hResult = CreateObjectCooker( wszClassName, pCookingInstance, pInternalRawInst, &pObjectCooker, pNamespace );
}
}
}
}
// Add the cooking instance
if ( SUCCEEDED( hResult ) )
{
hResult = pObjectCooker->SetCookedInstance( *ppRefreshableInstance, plId );
if ( SUCCEEDED( hResult ) )
{
// Add the raw instance to the cooker
hResult = pObjectCooker->BeginCooking( *plId, pInternalRawInst, m_dwRefreshId );
}
}
}
return hResult;
}
STDMETHODIMP CRefresher::AddEnum(
/*[in] */ IWbemServices* pNamespace,
/*[in] */ IWbemContext * pContext,
/*[in, string] */ LPCWSTR wszCookingClass,
/*[in] */ IWbemHiPerfEnum* pRefreshableEnum,
/*[out] */ long* plId )
///////////////////////////////////////////////////////////////////////////////
//
// AddEnum is called whenever a new cooked enumerator is added to the
// refresher. WMI passes an IWbemHiPerfEnum object to the provider which
// will be used for the cooked enumerator. The corresponding raw enumerator
// is obtained when the added to the internal refresher. Both of these
// enumerators as well as a cooking class template is added to the
// enumerator cache.
//
// Parameters:
// pNamespace - The namespace where the objects are located
// wszCookingClass - The name of the enumerators' cooking class
// pRefreshableEnum
// - The enumerator to be used for the cooked classes
// plId - The ID of the enumerator
//
///////////////////////////////////////////////////////////////////////////////
{
HRESULT hResult = WBEM_NO_ERROR;
IWbemHiPerfEnum* pRawEnum = NULL;
long lRawID = 0;
// Verify our 'in' parameters
// ==========================
if ( NULL == pNamespace || NULL == wszCookingClass || NULL == pRefreshableEnum )
{
hResult = WBEM_E_INVALID_PARAMETER;
}
if ( SUCCEEDED( hResult ) )
{
// Get the cooking object
// ======================
IWbemClassObject* pCookedObject = NULL;
IWbemClassObject* pRawObject = NULL;
BSTR strCookedClassName = SysAllocString( wszCookingClass );
if (NULL == strCookedClassName) return WBEM_E_OUT_OF_MEMORY;
CSysFreeMe afCookedClassName( strCookedClassName );
hResult = pNamespace->GetObject( strCookedClassName, 0, NULL, &pCookedObject, NULL );
CReleaseMe arCookedObject( pCookedObject );
if ( SUCCEEDED( hResult ) )
{
WCHAR* wszRawClassName = NULL;
hResult = GetRawClassName( pCookedObject, &wszRawClassName );
wmilib::auto_buffer<WCHAR> adRawClassName( wszRawClassName );
if ( SUCCEEDED( hResult ))
{
BSTR strRawClassName = SysAllocString(wszRawClassName);
if (NULL == strRawClassName) return WBEM_E_OUT_OF_MEMORY;
CSysFreeMe sfm(strRawClassName);
hResult = pNamespace->GetObject( strRawClassName, 0, NULL, &pRawObject, NULL );
if ( SUCCEEDED( hResult ) )
{
CReleaseMe arRawObject( pRawObject );
// Add the raw enumerator to our internal refresher
// ================================================
hResult = AddRawEnum( pNamespace, pContext, wszRawClassName, &pRawEnum, &lRawID );
CReleaseMe arRawEnum( pRawEnum );
if ( SUCCEEDED( hResult ) )
{
// Add the cooked enumerator to the enumerator cache
// =================================================
hResult = m_EnumCache.AddEnum(
wszCookingClass,
pCookedObject, // this is acquired by CWMISimpleObjectCooker and CEnumeratorManager
pRawObject,
pRefreshableEnum,
pRawEnum,
lRawID,
(DWORD*)plId );
// set the three bits
*plId |= WMI_COOKED_ENUM_MASK;
}
}
}
}
}
return hResult;
}
STDMETHODIMP CRefresher:: Remove(
/*[in] */ long lId )
///////////////////////////////////////////////////////////////////////////////
//
// Remove is used to remove an object from the refresher. Depending on the
// object, the corresponding removal is performed.
//
// Parameters:
// lID - The ID of the object to be removed
//
///////////////////////////////////////////////////////////////////////////////
{
HRESULT hResult = S_OK;
// Is it an instance ID?
// =====================
if ( lId == ( lId & ~WMI_COOKED_ENUM_MASK ) )
{
CWMISimpleObjectCooker* pCooker = NULL;
hResult = m_CookingClassCache.BeginEnum();
while ( S_OK == m_CookingClassCache.Next( &pCooker ) )
{
pCooker->Remove( lId );
}
hResult = m_CookingClassCache.EndEnum();
}
else
{
long RawId;
hResult = m_EnumCache.RemoveEnum( (lId & ~WMI_COOKED_ENUM_MASK) , &RawId );
if (SUCCEEDED(hResult)){
m_pConfig->Remove(RawId,0);
}
}
return hResult;
}
STDMETHODIMP CRefresher::Refresh()
///////////////////////////////////////////////////////////////////////////////
//
// Refresh is called when the refreshers' objects are to be updated. The
// instances are updated by explicitly enumerating through the instance
// cache. The enumerators' refresh is performed with the enumerator
// cache.
//
// Parameters: (none)
//
///////////////////////////////////////////////////////////////////////////////
{
HRESULT hResult = S_OK;
CWMISimpleObjectCooker* pCooker = NULL;
// Refresh the internal refresher
// ==============================
m_dwRefreshId++;
hResult = m_pRefresher->Refresh( 0L );
if ( SUCCEEDED( hResult ) )
{
// INSTANCES: Update the instance values for every class
// =====================================================
hResult = m_CookingClassCache.BeginEnum();
while ( S_OK == m_CookingClassCache.Next( &pCooker ) )
{
// And update all of the instances
// ===============================
pCooker->Recalc(m_dwRefreshId);
}
hResult = m_CookingClassCache.EndEnum();
// ENUMERATORS: Merge and update the values for items in the enumerator
// ====================================================================
if ( SUCCEEDED( hResult ) )
{
hResult = m_EnumCache.Refresh(m_dwRefreshId);
}
}
return hResult;
}
STDMETHODIMP CRefresher::Refresh( long lFlags )
///////////////////////////////////////////////////////////////////////////////
//
// This is the IWbemRefresher::Refresh implementation and is simply a call
// through.
//
///////////////////////////////////////////////////////////////////////////////
{
HRESULT hResult = WBEM_NO_ERROR;
hResult = Refresh();
return hResult;
}