148 lines
6.3 KiB
C
148 lines
6.3 KiB
C
|
//***************************************************************************
|
||
|
|
|
||
|
|
//
|
||
|
|
|
||
|
|
// Copyright (c) 1997-2001 Microsoft Corporation, All Rights Reserved
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
#ifndef _WDMSHELL_HEADER
|
||
|
|
#define _WDMSHELL_HEADER
|
||
|
|
#include "wmicom.h"
|
||
|
|
//************************************************************************************************************
|
||
|
|
//============================================================================================================
|
||
|
|
//
|
||
|
|
// The Standard WDM Shell
|
||
|
|
//
|
||
|
|
//============================================================================================================
|
||
|
|
//************************************************************************************************************
|
||
|
|
class CWMIClassType
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
CWMIClassType() {}
|
||
|
|
~CWMIClassType(){}
|
||
|
|
BOOL IsHiPerfClass(WCHAR * wcsClass, IWbemServices * pServices);
|
||
|
|
|
||
|
|
};
|
||
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////////////////
|
||
|
|
class CWMIStandardShell
|
||
|
|
{
|
||
|
|
|
||
|
|
private:
|
||
|
|
|
||
|
|
CWMIProcessClass * m_pClass;
|
||
|
|
CProcessStandardDataBlock * m_pWDM;
|
||
|
|
BOOL m_fInit;
|
||
|
|
|
||
|
|
|
||
|
|
public:
|
||
|
|
CWMIStandardShell();
|
||
|
|
~CWMIStandardShell();
|
||
|
|
|
||
|
|
|
||
|
|
HRESULT Initialize(WCHAR * wcsClass, BOOL fInternalEvent, CHandleMap * pList,BOOL fUpdateNamespace, ULONG uDesiredAccess, IWbemServices __RPC_FAR * pServices,
|
||
|
|
IWbemObjectSink __RPC_FAR * pHandler, IWbemContext __RPC_FAR *pCtx);
|
||
|
|
|
||
|
|
inline BOOL HasMofChanged() { return m_pWDM->HasMofChanged(); }
|
||
|
|
|
||
|
|
//=============================================
|
||
|
|
// Process All and Single WMI Instances
|
||
|
|
//=============================================
|
||
|
|
HRESULT ProcessAllInstances();
|
||
|
|
|
||
|
|
HRESULT ProcessSingleInstance( WCHAR * wcsInstanceName);
|
||
|
|
//==========================================================
|
||
|
|
// The put instance group
|
||
|
|
//==========================================================
|
||
|
|
HRESULT FillInAndSubmitWMIDataBlob( IWbemClassObject * pIClass, int nTypeOfPut, CVARIANT & vList);
|
||
|
|
|
||
|
|
//=============================================
|
||
|
|
// Event functions
|
||
|
|
//=============================================
|
||
|
|
HRESULT ProcessEvent(WORD wBinaryMofType, PWNODE_HEADER WnodeHeader);
|
||
|
|
inline HRESULT RegisterWMIEvent( WCHAR * wcsGuid, ULONG_PTR uContext, CLSID & Guid, BOOL fRegistered)
|
||
|
|
{ return m_pWDM->RegisterWMIEvent(wcsGuid,uContext,Guid,fRegistered);}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
//=============================================
|
||
|
|
// method functions
|
||
|
|
//=============================================
|
||
|
|
HRESULT ExecuteMethod( WCHAR * wcsInstance,
|
||
|
|
WCHAR * MethodInstanceName,
|
||
|
|
IWbemClassObject * pParentClass,
|
||
|
|
IWbemClassObject * pInClassData,
|
||
|
|
IWbemClassObject * pInClass,
|
||
|
|
IWbemClassObject * pOutClass ) ;
|
||
|
|
//=============================================
|
||
|
|
// data processing functions
|
||
|
|
//=============================================
|
||
|
|
HRESULT GetGuid(WCHAR * pwcsGuid);
|
||
|
|
|
||
|
|
HRESULT SetGuidForEvent( WORD wType,WCHAR * wcsGuid );
|
||
|
|
HRESULT RegisterForWMIEvents( ULONG uContext, WCHAR * wcsGuid, BOOL fRegistered,CLSID & Guid );
|
||
|
|
inline BOOL CancelWMIEventRegistration( GUID gGuid , ULONG_PTR uContext ) { return m_pClass->WMI()->CancelWMIEventRegistration(gGuid,uContext);}
|
||
|
|
|
||
|
|
//=============================================
|
||
|
|
// The binary mof groupg
|
||
|
|
//=============================================
|
||
|
|
HRESULT ProcessBinaryGuidsViaEvent( PWNODE_HEADER WnodeHeader,WORD wType );
|
||
|
|
HRESULT QueryAndProcessAllBinaryGuidInstances(CNamespaceManagement & Namespace, BOOL & fMofHasChanged, KeyList * pArrDriversInRegistry);
|
||
|
|
//=============================================
|
||
|
|
// Misc
|
||
|
|
//=============================================
|
||
|
|
inline CLSID * GuidPtr() { return m_pClass->GuidPtr();}
|
||
|
|
inline HRESULT SetErrorMessage(HRESULT hr){ return m_pClass->WMI()->SetErrorMessage(hr,m_pClass->GetClassName(),m_pWDM->GetMessage());}
|
||
|
|
|
||
|
|
};
|
||
|
|
|
||
|
|
//************************************************************************************************************
|
||
|
|
//============================================================================================================
|
||
|
|
//
|
||
|
|
// The Hi Performance Shell
|
||
|
|
//
|
||
|
|
//============================================================================================================
|
||
|
|
//************************************************************************************************************
|
||
|
|
|
||
|
|
class CWMIHiPerfShell
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
|
||
|
|
CHiPerfHandleMap * m_pHiPerfMap;
|
||
|
|
CProcessHiPerfDataBlock * m_pWDM;
|
||
|
|
CWMIProcessClass * m_pClass;
|
||
|
|
BOOL m_fAutoCleanup;
|
||
|
|
BOOL m_fInit;
|
||
|
|
|
||
|
|
HRESULT QueryAllInstances(HANDLE WMIHandle,IWbemHiPerfEnum* pHiPerfEnum);
|
||
|
|
HRESULT QuerySingleInstance(HANDLE WMIHandle);
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
public:
|
||
|
|
|
||
|
|
CWMIHiPerfShell(BOOL fAuto);
|
||
|
|
~CWMIHiPerfShell();
|
||
|
|
|
||
|
|
HRESULT Initialize(BOOL fUpdate, ULONG uDesiredAccess, CHandleMap * pList,WCHAR * wcs, IWbemServices __RPC_FAR * pServices,
|
||
|
|
IWbemObjectSink __RPC_FAR * pHandler, IWbemContext __RPC_FAR *pCtx) ;
|
||
|
|
|
||
|
|
|
||
|
|
inline void SetHiPerfHandleMap(CHiPerfHandleMap * p) { m_pHiPerfMap = p; }
|
||
|
|
inline CCriticalSection * GetCriticalSection() { return m_pHiPerfMap->GetCriticalSection();}
|
||
|
|
inline CHiPerfHandleMap * HiPerfHandleMap() { return m_pHiPerfMap;}
|
||
|
|
|
||
|
|
HRESULT QueryAllHiPerfData();
|
||
|
|
HRESULT HiPerfQuerySingleInstance(WCHAR * wcsInstance);
|
||
|
|
HRESULT AddAccessObjectToRefresher(IWbemObjectAccess *pAccess, IWbemObjectAccess ** ppRefreshable, ULONG_PTR * plId);
|
||
|
|
HRESULT AddEnumeratorObjectToRefresher(IWbemHiPerfEnum* pHiPerfEnum, ULONG_PTR * plId);
|
||
|
|
HRESULT RemoveObjectFromHandleMap(ULONG_PTR lHiPerfId);
|
||
|
|
HRESULT RefreshCompleteList();
|
||
|
|
|
||
|
|
inline HRESULT SetErrorMessage(HRESULT hr){ return m_pClass->WMI()->SetErrorMessage(hr,m_pClass->GetClassName(),m_pWDM->GetMessage());}
|
||
|
|
|
||
|
|
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
#endif
|