|
|
/*++
Copyright (C) 2000-2001 Microsoft Corporation
--*/
#ifndef __A51PROV__H_
#define __A51PROV__H_
#include <windows.h>
#include <wbemidl.h>
#include <unk.h>
#include <wbemcomn.h>
#include <sync.h>
#include <reposit.h>
#include <wmiutils.h>
#include <objpath.h>
#include <filecach.h>
#include <hiecache.h>
#include <corex.h>
#include "a51fib.h"
#include "creposit.h"
class CAutoTransaction;class CDbIterator;class CRepEvent{public: DWORD m_dwType; WString m_wszArg1; WString m_wszNamespace; _IWmiObject* m_pObj1; _IWmiObject* m_pObj2;
CRepEvent(DWORD dwType, LPCWSTR wszNamespace, LPCWSTR wszArg1, _IWmiObject* pObj1, _IWmiObject* pObj2); virtual ~CRepEvent();};
class CEventCollector{protected: CUniquePointerArray<CRepEvent> m_apEvents; bool m_bNamespaceOnly; CCritSec m_csLock;
public: CEventCollector() : m_bNamespaceOnly(false){ } ~CEventCollector() { } bool AddEvent(CRepEvent* pEvent); void SetNamespaceOnly(bool bNamespaceOnly) {m_bNamespaceOnly = bNamespaceOnly;} bool IsNamespaceOnly() {return m_bNamespaceOnly;}
HRESULT SendEvents(_IWmiCoreServices* pCore);
void DeleteAllEvents();
void TransferEvents(CEventCollector &aEventsToTransfer);
int GetSize() { return m_apEvents.GetSize(); }};
class CSession : public CUnkBase<IWmiDbSessionEx, &IID_IWmiDbSessionEx>{friend CAutoTransaction;private: CEventCollector m_aTransactedEvents; bool m_bInWriteTransaction;
public: CSession(CLifeControl* pControl = NULL) : TUnkBase(pControl), m_bInWriteTransaction(false) {}
virtual ~CSession();
ULONG STDMETHODCALLTYPE Release();
HRESULT STDMETHODCALLTYPE GetObject( IWmiDbHandle *pScope, IWbemPath *pPath, DWORD dwFlags, DWORD dwRequestedHandleType, IWmiDbHandle **ppResult );
HRESULT STDMETHODCALLTYPE GetObjectDirect( IWmiDbHandle *pScope, IWbemPath *pPath, DWORD dwFlags, REFIID riid, LPVOID *pObj );
HRESULT STDMETHODCALLTYPE PutObject( IWmiDbHandle *pScope, REFIID riid, LPVOID pObj, DWORD dwFlags, DWORD dwRequestedHandleType, IWmiDbHandle **ppResult );
HRESULT STDMETHODCALLTYPE DeleteObject( IWmiDbHandle *pScope, DWORD dwFlags, REFIID riid, LPVOID pObj );
HRESULT STDMETHODCALLTYPE ExecQuery( IWmiDbHandle *pScope, IWbemQuery *pQuery, DWORD dwFlags, DWORD dwRequestedHandleType, DWORD *dwMessageFlags, IWmiDbIterator **ppQueryResult ); HRESULT STDMETHODCALLTYPE RenameObject( IWbemPath *pOldPath, IWbemPath *pNewPath, DWORD dwFlags, DWORD dwRequestedHandleType, IWmiDbHandle **ppResult );
HRESULT STDMETHODCALLTYPE Enumerate( IWmiDbHandle *pScope, DWORD dwFlags, DWORD dwRequestedHandleType, IWmiDbIterator **ppQueryResult );
HRESULT STDMETHODCALLTYPE AddObject( IWmiDbHandle *pScope, IWbemPath *pPath, DWORD dwFlags, DWORD dwRequestedHandleType, IWmiDbHandle **ppResult );
HRESULT STDMETHODCALLTYPE RemoveObject ( IWmiDbHandle *pScope, IWbemPath *pPath, DWORD dwFlags );
HRESULT STDMETHODCALLTYPE SetDecoration( LPWSTR lpMachineName, LPWSTR lpNamespacePath );
HRESULT STDMETHODCALLTYPE SupportsQueries( DWORD *dwQuerySupportLevel ) {return WBEM_E_FAILED;};
HRESULT STDMETHODCALLTYPE GetObjectByPath( IWmiDbHandle *pScope, LPCWSTR wszPath, DWORD dwFlags, REFIID riid, LPVOID *pObj );
HRESULT STDMETHODCALLTYPE DeleteObjectByPath( IWmiDbHandle *pScope, LPCWSTR wszObjectPath, DWORD dwFlags );
HRESULT STDMETHODCALLTYPE ExecQuerySink( IWmiDbHandle *pScope, IWbemQuery *pQuery, DWORD dwFlags, DWORD dwRequestedHandleType, IWbemObjectSink* pSink, DWORD *dwMessageFlags );
HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, void** ppv);
HRESULT STDMETHODCALLTYPE BeginWriteTransaction(DWORD dwFlags);
HRESULT STDMETHODCALLTYPE BeginReadTransaction(DWORD dwFlags);
HRESULT STDMETHODCALLTYPE CommitTransaction(DWORD dwFlags);
HRESULT STDMETHODCALLTYPE AbortTransaction(DWORD dwFlags);protected: HRESULT InternalBeginTransaction(bool bWriteOperation); HRESULT InternalAbortTransaction(bool bWriteOperation); HRESULT InternalCommitTransaction(bool bWriteOperation);};class CLocalizationUpgrade;
class CNamespaceHandle : public CUnkBase<IWmiDbHandle, &IID_IWmiDbHandle>{friend CLocalizationUpgrade;protected:
static long s_lActiveRepNs;
CRepository * m_pRepository; WString m_wsNamespace; WString m_wsScope; WString m_wsFullNamespace; WCHAR m_wszMachineName[MAX_COMPUTERNAME_LENGTH+1];
WCHAR m_wszClassRootDir[MAX_PATH]; long m_lClassRootDirLen;
WCHAR m_wszInstanceRootDir[MAX_PATH]; long m_lInstanceRootDirLen;
CHierarchyCache* m_pClassCache;
_IWmiObject* m_pNullClass; bool m_bCached;
bool m_bUseIteratorLock;
public: CNamespaceHandle(CLifeControl* pControl, CRepository * pRepository); ~CNamespaceHandle();
STDMETHOD(GetHandleType)(DWORD* pdwType) {*pdwType = 0; return S_OK;}
HRESULT Initialize(LPCWSTR wszNamespace, LPCWSTR wszScope = NULL); HRESULT Initialize2(LPCWSTR wszNamespace, LPCWSTR wszNamespaceHash);
HRESULT GetObject( IWbemPath *pPath, DWORD dwFlags, DWORD dwRequestedHandleType, IWmiDbHandle **ppResult );
HRESULT GetObjectDirect( IWbemPath *pPath, DWORD dwFlags, REFIID riid, LPVOID *pObj );
HRESULT PutObject( REFIID riid, LPVOID pObj, DWORD dwFlags, DWORD dwRequestedHandleType, IWmiDbHandle **ppResult, CEventCollector &aEvents );
HRESULT DeleteObject( DWORD dwFlags, REFIID riid, LPVOID pObj, CEventCollector &aEvents );
HRESULT ExecQuery( IWbemQuery *pQuery, DWORD dwFlags, DWORD dwRequestedHandleType, DWORD *dwMessageFlags, IWmiDbIterator **ppQueryResult );
HRESULT GetObjectByPath( LPWSTR wszPath, DWORD dwFlags, REFIID riid, LPVOID *pObj );
HRESULT ExecQuerySink( IWbemQuery *pQuery, DWORD dwFlags, DWORD dwRequestedHandleType, IWbemObjectSink* pSink, DWORD *dwMessageFlags );
HRESULT DeleteObjectByPath(DWORD dwFlags, LPWSTR wszPath, CEventCollector &aEvents); HRESULT GetErrorStatus(); void SetErrorStatus(HRESULT hres);
HRESULT CreateSystemClasses(CFlexArray &aSystemClasses);
void TellIteratorNotToLock() { m_bUseIteratorLock = false; }
protected: HRESULT GetObjectHandleByPath(LPWSTR wszBuffer, DWORD dwFlags, DWORD dwRequestedHandleType, IWmiDbHandle **ppResult); HRESULT PutInstance(_IWmiObject* pInst, DWORD dwFlags, CEventCollector &aEvents); HRESULT PutClass(_IWmiObject* pClass, DWORD dwFlags, CEventCollector &aEvents); HRESULT ConstructClassRelationshipsDir(LPCWSTR wszClassName, CFileName& wszDirPath); HRESULT WriteParentChildRelationship(LPCWSTR wszChildFileName, LPCWSTR wszParentName); HRESULT ExecClassQuery(QL_LEVEL_1_RPN_EXPRESSION* pQuery, IWbemObjectSink* pSink, DWORD dwFlags); HRESULT ExecInstanceQuery(QL_LEVEL_1_RPN_EXPRESSION* pQuery,
LPCWSTR wszClassName, bool bDeep, IWbemObjectSink* pSink); HRESULT GetClassDirect(LPCWSTR wszClassName, REFIID riid, void** ppObj, bool bClone, __int64* pnTime, bool* pbRead, bool *pbSystemClass); HRESULT DeleteInstance(LPCWSTR wszClassName, LPCWSTR wszKey, CEventCollector &aEvents); HRESULT DeleteInstanceByFile(LPCWSTR wszFilePath, _IWmiObject* pClass, bool bClassDeletion, CEventCollector &aEvents); HRESULT DeleteClass(LPCWSTR wszClassName, CEventCollector &aEvents,bool bDisableEvents); HRESULT DeleteClassInstances(LPCWSTR wszClassName, _IWmiObject* pClass, CEventCollector &aEvents); HRESULT FileToSystemClass(LPCWSTR wszFileName, _IWmiObject** ppClass, bool bClone, __int64* pnTime = NULL); HRESULT FileToClass(LPCWSTR wszFileName, _IWmiObject** ppClass, bool bClone, __int64* pnTime, bool *pbSystemClass); HRESULT FileToInstance(_IWmiObject* pClass, LPCWSTR wszFileName, BYTE *pRetrievedBlob, DWORD dwSize, _IWmiObject** ppInstance, bool bMustBeThere = false); HRESULT WriteInstanceReferences(_IWmiObject* pInst, LPCWSTR wszClassName, LPCWSTR wszFilePath); HRESULT WriteInstanceReference(LPCWSTR wszReferringFile, LPCWSTR wszReferringClass, LPCWSTR wszReferringProp, LPWSTR wszReference); HRESULT ExecClassRefQuery(LPCWSTR wszQuery, LPCWSTR wszClassName, IWbemObjectSink* pSink); HRESULT ExecReferencesQuery(LPCWSTR wszQuery, IWbemObjectSink* pSink); HRESULT ExecInstanceRefQuery(LPCWSTR wszQuery, LPCWSTR wszClassName, LPCWSTR wszKey, IWbemObjectSink* pSink); HRESULT GetReferrerFromFile(LPCWSTR wszReferenceFile, LPWSTR wszReferrerRelFile, LPWSTR* pwszReferrerNamespace, LPWSTR* pwszReferrerClass, LPWSTR* pwszReferrerProp); HRESULT DeleteInstanceReference(LPCWSTR wszOurFilePath, LPWSTR wszReference); HRESULT DeleteInstanceReferences(_IWmiObject* pInst, LPCWSTR wszFilePath);
HRESULT EnumerateClasses(IWbemObjectSink* pSink, LPCWSTR wszSuperClass, LPCWSTR wszAncestor, bool bClone, bool bDontIncludeAncestorInResultSet); HRESULT ListToEnum(CWStringArray& wsClasses, IWbemObjectSink* pSink, bool bClone);
bool Hash(LPCWSTR wszName, LPWSTR wszHash); HRESULT InstanceToFile(IWbemClassObject* pInst, LPCWSTR wszClassName, LPCWSTR wszFileName1, LPCWSTR wszFileName2, __int64 nClassTime); HRESULT ConstructInstanceDefName(CFileName& wszInstanceDefName, LPCWSTR wszKey); HRESULT ClassToFile(_IWmiObject* pSuperClass, _IWmiObject* pClass, LPCWSTR wszFileName, __int64 nFakeUpdateTime = 0); HRESULT ComputeKeyFromPath(LPWSTR wszPath, LPWSTR wszKey, size_t dwKeyLen, LPWSTR* pwszClassName, bool* pbIsClass, LPWSTR* pwszNamespace = NULL); HRESULT ParseKey(LPWSTR wszKeyStart, LPWSTR* pwcRealStart, LPWSTR* pwcNextKey);
HRESULT GetInstanceByKey(LPCWSTR wszClassName, LPCWSTR wszKey, REFIID riid, void** ppObj); HRESULT WriteClassRelationships(_IWmiObject* pClass, LPCWSTR wszFileName); HRESULT ConstructParentChildFileName(LPCWSTR wszChildFileName, LPCWSTR wszParentName, CFileName& wszParentChildFileName); HRESULT DeleteDerivedClasses(LPCWSTR wszClassName, CEventCollector &aEvents, bool bDisableEvents); HRESULT EraseParentChildRelationship(LPCWSTR wszChildFileName, LPCWSTR wszParentName); HRESULT EraseClassRelationships(LPCWSTR wszClassName, _IWmiObject* pClass, LPCWSTR wszFileName); HRESULT GetClassByHash(LPCWSTR wszHash, bool bClone, _IWmiObject** ppClass, __int64* pnTime, bool* pbRead, bool *pbSystemClass); HRESULT DeleteClassByHash(LPCWSTR wszHash, CEventCollector &aEvents,bool bDisableEvents); HRESULT DeleteClassInternal(LPCWSTR wszClassName, _IWmiObject* pClass, LPCWSTR wszFileName, CEventCollector &aEvents, bool bSystemClass, bool bDisableEvents); HRESULT GetChildHashes(LPCWSTR wszClassName, CWStringArray& wsChildHashes); HRESULT GetChildDefs(LPCWSTR wszClassName, bool bRecursive, IWbemObjectSink* pSink, bool bClone); HRESULT ConstructClassDefFileName(LPCWSTR wszClassName, CFileName& wszFileName); HRESULT ConstructClassDefFileNameFromHash(LPCWSTR wszHash, CFileName& wszFileName); HRESULT ConstructClassRelationshipsDirFromHash(LPCWSTR wszHash, CFileName& wszDirPath); HRESULT GetChildHashesByHash(LPCWSTR wszHash, CWStringArray& wsChildHashes); HRESULT GetChildDefsByHash(LPCWSTR wszHash, bool bRecursive, IWbemObjectSink* pSink, bool bClone); HRESULT FireEvent(CEventCollector &aEvents, DWORD dwType, LPCWSTR wszArg1, _IWmiObject* pObj1, _IWmiObject* pObj2 = NULL); HRESULT DeleteInstanceAsScope(_IWmiObject* pInst, CEventCollector &aEvents); HRESULT EnumerateChildNamespaces(LPCWSTR wsRootNamespace, CWStringArray &aNamespaces, CEventCollector &aEvents);
HRESULT DeleteInstanceSelf(LPCWSTR wszFilePath, _IWmiObject* pInst, bool bClassDeletion); HRESULT ConstructReferenceDir(LPWSTR wszTargetPath, CFileName& wszDir); HRESULT ConstructReferenceDirFromKey(LPCWSTR wszClassName, LPCWSTR wszKey, CFileName& wszReferenceDir); HRESULT ConstructReferenceFileName(LPWSTR wszReference, LPCWSTR wszReferringFile, CFileName& wszReferenceFile); HRESULT ConstructKeyRootDirFromClass(CFileName& wszDir, LPCWSTR wszClassName); HRESULT ConstructKeyRootDirFromKeyRoot(CFileName& wszDir, LPCWSTR wszKeyRootClass); HRESULT ConstructLinkDirFromClass(CFileName& wszDir, LPCWSTR wszClassName); HRESULT DeleteInstanceLink(_IWmiObject* pInst, LPCWSTR wszInstanceDefFilePath); HRESULT GetKeyRoot(LPCWSTR wszClass, TEMPFREE_ME LPWSTR* pwszKeyRootClass); HRESULT ConstructInstDefNameFromLinkName(CFileName& wszInstanceDefName, LPCWSTR wszInstanceLinkName); HRESULT ExecDeepInstanceQuery(QL_LEVEL_1_RPN_EXPRESSION* pQuery, LPCWSTR wszClassHash, IWbemObjectSink* pSink); HRESULT ExecShallowInstanceQuery(QL_LEVEL_1_RPN_EXPRESSION* pQuery, LPCWSTR wszClassHash, IWbemObjectSink* pSink); HRESULT ConstructLinkDirFromClassHash(CFileName& wszDir, LPCWSTR wszClassHash); HRESULT ConstructClassReferenceFileName(LPCWSTR wszReferredToClass, LPCWSTR wszReferringFile, LPCWSTR wszReferringProp, CFileName& wszFieName); HRESULT WriteClassReference(_IWmiObject* pReferringClass, LPCWSTR wszReferringFile, LPCWSTR wszReferringProp); HRESULT EraseClassReference(_IWmiObject* pReferringClass, LPCWSTR wszReferringFile, LPCWSTR wszReferringProp); //CFileCache* GetFileCache();
HRESULT CanClassBeUpdatedCompatible(DWORD dwFlags, LPCWSTR wszClassName, _IWmiObject *pOldClass, _IWmiObject *pNewClass); HRESULT DeleteInstanceBackReferences(LPCWSTR wszFilePath); HRESULT ConstructReferenceDirFromFilePath(LPCWSTR wszFilePath, CFileName& wszReferenceDir);
HRESULT ClassHasChildren(LPCWSTR wszClassName); HRESULT ClassHasInstances(LPCWSTR wszClassName); HRESULT ClassHasInstancesFromClassHash(LPCWSTR wszClassHash); HRESULT ClassHasInstancesInScopeFromClassHash( LPCWSTR wszInstanceRootDir, LPCWSTR wszClassHash);
HRESULT UpdateClassCompatible(_IWmiObject* pSuperClass, LPCWSTR wszClassName, _IWmiObject *pNewClass, _IWmiObject *pOldClass, __int64 nFakeUpdateTime = 0); HRESULT UpdateClassCompatibleHash(_IWmiObject* pSuperClass, LPCWSTR wszClassHash, _IWmiObject *pClass, _IWmiObject *pOldClass, __int64 nFakeUpdateTime = 0); HRESULT UpdateClassSafeForce(_IWmiObject* pSuperClass, DWORD dwFlags, LPCWSTR wcsClassName, _IWmiObject *pOldClass, _IWmiObject *pNewClass, CEventCollector &aEvents); HRESULT UpdateClassAggressively(_IWmiObject* pSuperClass, DWORD dwFlags, LPCWSTR wszClassName, _IWmiObject *pNewClass, _IWmiObject *pOldClass, CEventCollector &aEvents); HRESULT UpdateChildClassAggressively(DWORD dwFlags, LPCWSTR wszClassHash, _IWmiObject *pNewClass, CEventCollector &aEvents); };
class CDbIterator : public CUnkBase2<IWmiDbIterator, &IID_IWmiDbIterator, IWbemObjectSink, &IID_IWbemObjectSink>{protected: CCritSec m_cs; CRefedPointerQueue<IWbemClassObject> m_qObjects; long m_lCurrentIndex; void* m_pExecFiber; CFiberTask* m_pExecReq; HRESULT m_hresStatus; void* m_pMainFiber; DWORD m_dwNumRequested;
HRESULT m_hresCancellationStatus; bool m_bExecFiberRunning; bool m_bUseLock;
public: CDbIterator(CLifeControl* pControl, bool bUseLock); ~CDbIterator();
STDMETHOD(Cancel) (DWORD dwFlags, void* pFiber);
STDMETHOD(NextBatch)( DWORD dwNumRequested, DWORD dwTimeOutSeconds, DWORD dwFlags, DWORD dwRequestedHandleType, REFIID riid, void* pFiber, DWORD *pdwNumReturned, LPVOID *ppObjects );
void SetExecFiber(void* pFiber, CFiberTask* pReq); STDMETHOD(Indicate)(long lNumObjects, IWbemClassObject** apObjects); STDMETHOD(SetStatus)(long lFlags, HRESULT hresResult, BSTR, IWbemClassObject*);};
//
// implements swprintf(pDest,L"%s\\%s",Src1,pSrc2)
//
///////////////////////////////////////////////////////////////////
void inline Cat2Str(WCHAR * pDest, WCHAR * pSrc1, LPCWSTR pSrc2){ while(*pSrc1) *pDest++ = *pSrc1++; *pDest++ = L'\\'; while(*pSrc2) *pDest++ = *pSrc2++; *pDest = 0;}
#endif
|
