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.
 
 
 
 
 
 

528 lines
19 KiB

/*++
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