able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/admin/wmi/wbem/winmgmt/repdrvr/repdrvr.h

604 lines
24 KiB
C
Raw Permalink Normal View History

Add source files
2020-09-26 03:20:57 -05:00
//***************************************************************************
//
// (c) 1999-2001 by Microsoft Corp. All Rights Reserved.
//
// repdrvr.h
//
// cvadai 19-Mar-99 Created as prototype for Quasar.
//
//***************************************************************************
#ifndef _REPDRVR_H_
#define _REPDRVR_H_
#pragma warning( disable : 4251 ) // needs to have dll-interface to be used by clients of class
//#define InterlockedIncrement(l) (++(*(l)))
//#define InterlockedDecrement(l) (--(*(l)))
EXTERN_C const IID IID_IWbemQuery;
EXTERN_C const IID IID_IWmiDbHandle;
EXTERN_C const IID IID_IWmiDbController;
EXTERN_C const IID IID_IWmiDbSession;
EXTERN_C const IID IID_IWmiDbBatchSession;
EXTERN_C const IID IID_IWmiDbIterator;
EXTERN_C const IID IID_ISequentialStream2;
#define WMIDB_OBJECT_STATE_NORMAL 0
#define WMIDB_OBJECT_STATE_IN_USE 1
#define WMIDB_OBJECT_STATE_DELETED 2
#define WMIDB_OBJECT_STATE_PROTECTED 3
#define WMIDB_OBJECT_STATE_EXCLUSIVE 4
#define WMIDB_OBJECT_STATE_PERSISTENT 5
#define WMIDB_OBJECT_STATE_AUTODELETE 6
#define WMIDB_SECURITY_FLAG_CLASS 1
#define WMIDB_SECURITY_FLAG_INSTANCE 2
#include <wmiutils.h>
#include <winbase.h>
#include <sqlcache.h>
#include <repcache.h>
#include <wbemint.h>
#include <coresvc.h>
extern CRITICAL_SECTION g_csRepdrvr;
class _WMILockit
{
public:
_WMILockit(CRITICAL_SECTION *pCS);
~_WMILockit();
private:
CRITICAL_SECTION *m_cs;
};
class CWmiDbIterator;
struct MappedProperties
{
LPWSTR wPropName;
LPWSTR wTableName;
BOOL bReadOnly;
LPWSTR wScopeClass;
BOOL bIsKey;
BOOL bStoreAsBlob;
BOOL bStoreAsNumber;
BOOL bStoreAsMOFText;
LPWSTR * arrColumnNames;
LPWSTR * arrForeignKeys;
DWORD dwNumColumns;
DWORD dwNumForeignKeys;
// Special treatment for embedded objects
BOOL bDecompose;
LPWSTR wClassTable;
LPWSTR wClassNameCol;
LPWSTR wClassDataCol;
LPWSTR wClassForeignKey;
};
LPWSTR GetKeyString (LPWSTR lpString);
void ConvertDataToString(WCHAR * lpKey, BYTE* pData, DWORD dwType, BOOL bQuotes = FALSE);
LPWSTR StripUnresolvedName (LPWSTR lpPath);
BOOL IsDerivedFrom(IWbemClassObject *pObj, LPWSTR lpClassName, BOOL bDirectOnly=FALSE);
HRESULT ConvertBlobToObject (IWbemClassObject *pNewObj, BYTE *pBuffer, DWORD dwLen, _IWmiObject **ppNewObj);
//***************************************************************************
// CWmiDbHandle
//***************************************************************************
class CWmiDbHandle : public IWmiDbHandle
{
friend class CWmiDbSession;
friend class CWmiDbIterator;
friend class CLockCache;
friend class CObjectCache;
friend class CWmiCustomDbIterator;
friend class CWmiESEIterator;
public:
HRESULT STDMETHODCALLTYPE QueryInterface(
/* [in] */ REFIID riid,
/* [iid_is][out] */ void __RPC_FAR *__RPC_FAR *ppvObject);
ULONG STDMETHODCALLTYPE AddRef( );
ULONG STDMETHODCALLTYPE Release( );
HRESULT STDMETHODCALLTYPE GetHandleType(
/* [out] */ DWORD __RPC_FAR *pdwType) ;
CWmiDbHandle();
~CWmiDbHandle();
HRESULT QueryInterface_Internal(CSQLConnection *pConn, void __RPC_FAR *__RPC_FAR *ppvObject, LPWSTR lpKey = NULL);
private:
DWORD m_dwHandleType;
DWORD m_dwVersion;
SQL_ID m_dObjectId;
SQL_ID m_dClassId;
SQL_ID m_dScopeId;
ULONG m_uRefCount;
BOOL m_bSecDesc;
BOOL m_bDefault;
IWmiDbSession *m_pSession;
IWbemClassObject *m_pData;
};
class CESSHolder
{
public:
CESSHolder(long lType, LPWSTR lpNs, LPWSTR lpClass, _IWmiObject *pOld, _IWmiObject *pNew);
~CESSHolder() {};
HRESULT Deliver(_IWmiCoreServices *pCS, LPCWSTR lpRootNs);
private:
_bstr_t m_sNamespace;
_bstr_t m_sClass;
_IWmiObject *pOldObject;
_IWmiObject *pNewObject;
long m_lType;
};
class CESSManager
{
friend class CWmiDbController;
public:
typedef std::map <DWORD, CESSHolder *> ESSObjs;
CESSManager();
~CESSManager ();
void SetConnCache (CSQLConnCache *pConn) {m_Conns = pConn;};
void SetSchemaCache (CSchemaCache *pCache) {m_Schema = pCache;};
void SetObjectCache (CObjectCache *pObjects) {m_Objects = pObjects;};
void InitializeESS();
HRESULT AddInsertRecord(CSQLConnection *pConn, LPWSTR lpGUID, LPWSTR lpNamespace, LPWSTR lpClass, DWORD dwGenus,
IWbemClassObject *pOldObj, IWbemClassObject *pNewObj);
HRESULT AddDeleteRecord(CSQLConnection *pConn, LPWSTR lpGUID, LPWSTR lpNamespace, LPWSTR lpClass, DWORD dwGenus,
IWbemClassObject *pObj);
HRESULT CommitAll (LPCWSTR lpGUID, LPCWSTR lpRootNs);
HRESULT DeleteAll (LPCWSTR lpGUID);
private:
_IWmiCoreServices *m_EventSubSys;
CSQLConnCache *m_Conns;
CObjectCache *m_Objects;
CSchemaCache *m_Schema;
ESSObjs m_ESSObjs;
};
//***************************************************************************
// CWmiDbController
//***************************************************************************
class CWmiDbController : public IWmiDbController
{
friend class CWmiDbSession;
friend class CWmiDbIterator;
friend class CWmiCustomDbIterator;
friend class CWmiESEIterator;
friend class CWmiDbHandle;
typedef std::map<SQL_ID, bool> SQL_IDMap;
public:
HRESULT STDMETHODCALLTYPE QueryInterface(
/* [in] */ REFIID riid,
/* [iid_is][out] */ void __RPC_FAR *__RPC_FAR *ppvObject);
ULONG STDMETHODCALLTYPE AddRef( );
ULONG STDMETHODCALLTYPE Release( );
virtual HRESULT STDMETHODCALLTYPE Logon(
/* [in] */ WMIDB_LOGON_TEMPLATE __RPC_FAR *pLogonParms,
/* [in] */ DWORD dwFlags,
/* [in] */ DWORD dwRequestedHandleType,
/* [out] */ IWmiDbSession __RPC_FAR *__RPC_FAR *ppSession,
/* [out] */ IWmiDbHandle __RPC_FAR *__RPC_FAR *ppRootNamespace) ;
virtual HRESULT STDMETHODCALLTYPE GetLogonTemplate(
/* [in] */ LCID lLocale,
/* [in] */ DWORD dwFlags,
/* [out] */ WMIDB_LOGON_TEMPLATE __RPC_FAR *__RPC_FAR *ppTemplate) ;
virtual HRESULT STDMETHODCALLTYPE FreeLogonTemplate(
/* [out][in] */ WMIDB_LOGON_TEMPLATE __RPC_FAR *__RPC_FAR *pTemplate) ;
virtual HRESULT STDMETHODCALLTYPE Shutdown(
/* [in] */ DWORD dwFlags);
virtual HRESULT STDMETHODCALLTYPE SetCallTimeout(
/* [in] */ DWORD dwMaxTimeout);
virtual HRESULT STDMETHODCALLTYPE SetCacheValue(
/* [in] */ DWORD dwMaxBytes);
virtual HRESULT STDMETHODCALLTYPE FlushCache(
/* [in] */ DWORD dwFlags);
virtual HRESULT STDMETHODCALLTYPE GetStatistics(
/* [in] */ DWORD dwParameter,
/* [out] */ DWORD __RPC_FAR *pdwValue) ;
CWmiDbController();
~CWmiDbController();
private:
typedef std::vector<CWmiDbSession *> Sessions;
HRESULT GetUnusedSession(WMIDB_LOGON_TEMPLATE *pLogon,
DWORD dwFlags,
DWORD dwHandleType,
IWmiDbSession **ppSession);
HRESULT SetConnProps(WMIDB_LOGON_TEMPLATE *pLogon);
HRESULT ReleaseSession(IWmiDbSession *pSession);
void IncrementHitCount(bool bCacheUsed=false);
void AddHandle();
void SubtractHandle();
HINSTANCE GetResourceDll (LCID lLocale);
BOOL HasSecurityDescriptor(SQL_ID ObjId);
void AddSecurityDescriptor(SQL_ID ObjId);
void RemoveSecurityDescriptor(SQL_ID ObjId);
DWORD m_dwTimeOut;
DWORD m_dwCurrentStatus;
ULONG m_uRefCount;
DWORD m_dwTotalHits;
DWORD m_dwCacheHits;
DWORD m_dwTotalHandles;
CRITICAL_SECTION m_cs;
DBPROP *m_InitProperties;
DBPROPSET *m_rgInitPropSet;
IMalloc *m_pIMalloc;
CObjectCache ObjectCache;
CSchemaCache SchemaCache; // separate so it doesn't get flushed.
CLockCache LockCache; // separate so it doesn't get flushed.
CSQLConnCache ConnCache; // Cache of SQL connections.
CESSManager ESSMgr;
SQL_IDMap SecuredIDs; // List of objects with security descriptors.
Sessions m_Sessions;
BOOL m_bCacheInit;
BOOL m_bIsAdmin;
BOOL m_bESSEnabled;
};
class CWbemClassObjectProps
{
public:
LPWSTR lpClassName;
LPWSTR lpNamespace;
LPWSTR lpRelPath;
LPWSTR lpSuperClass;
LPWSTR lpDynasty;
LPWSTR lpKeyString;
DWORD dwGenus;
CWbemClassObjectProps(CWmiDbSession *pSession, CSQLConnection *pConn,
IWbemClassObject *pObj, CSchemaCache *pCache, SQL_ID dScopeID);
~CWbemClassObjectProps();
};
struct SessionLock
{
SQL_ID dObjectId;
DWORD dwHandleType;
};
//***************************************************************************
// CWmiDbSession
//***************************************************************************
class CWmiDbSession : public IWmiDbSession, IWmiDbBatchSession, IWbemTransaction, IWmiDbBackupRestore
{
friend class CWmiDbController;
friend class CWmiDbIterator;
friend class CWmiCustomDbIterator;
friend class CWmiESEIterator;
friend class CWmiDbHandle;
typedef std::vector <SQL_ID> SQLIDs;
typedef std::map <DWORD, DWORD> Properties;
typedef std::vector <SessionLock *> SessionLocks;
typedef std::map <DWORD, SQLIDs> SessionDynasties;
typedef std::map <DWORD, DWORD> RefCount;
public:
HRESULT STDMETHODCALLTYPE QueryInterface(
/* [in] */ REFIID riid,
/* [iid_is][out] */ void __RPC_FAR *__RPC_FAR *ppvObject);
ULONG STDMETHODCALLTYPE AddRef( );
ULONG STDMETHODCALLTYPE Release( );
virtual HRESULT STDMETHODCALLTYPE GetObject(
/* [in] */ IWmiDbHandle __RPC_FAR *pScope,
/* [in] */ IWbemPath __RPC_FAR *pPath,
/* [in] */ DWORD dwFlags,
/* [in] */ DWORD dwRequestedHandleType,
/* [out] */ IWmiDbHandle __RPC_FAR *__RPC_FAR *ppResult);
virtual HRESULT STDMETHODCALLTYPE GetObjectDirect(
/* [in] */ IWmiDbHandle __RPC_FAR *pScope,
/* [in] */ IWbemPath __RPC_FAR *pPath,
/* [in] */ DWORD dwFlags,
/* [in] */ REFIID riid,
/* [iid_is][out] */ LPVOID __RPC_FAR *pObj);
virtual HRESULT STDMETHODCALLTYPE PutObject(
/* [in] */ IWmiDbHandle __RPC_FAR *pScope,
/* [in] */ REFIID riid,
/* [iid_is][in] */ LPVOID pObjToPut,
/* [in] */ DWORD dwFlags,
/* [in] */ DWORD dwRequestedHandleType,
/* [out] */ IWmiDbHandle __RPC_FAR *__RPC_FAR *ppResult);
virtual HRESULT STDMETHODCALLTYPE DeleteObject(
/* [in] */ IWmiDbHandle __RPC_FAR *pScope,
/* [in] */ DWORD dwFlags,
/* [in] */ REFIID riid,
/* [iid_is][in] */ LPVOID pObj);
virtual HRESULT STDMETHODCALLTYPE ExecQuery(
/* [in] */ IWmiDbHandle __RPC_FAR *pScope,
/* [in] */ IWbemQuery __RPC_FAR *pQuery,
/* [in] */ DWORD dwFlags,
/* [in] */ DWORD dwHandleType,
/* [out] */ DWORD *dwMessageFlags,
/* [out] */ IWmiDbIterator __RPC_FAR *__RPC_FAR *pQueryResult) ;
virtual HRESULT STDMETHODCALLTYPE RenameObject(
/* [in] */ IWbemPath __RPC_FAR *pOldPath,
/* [in] */ IWbemPath __RPC_FAR *pNewPath,
/* [in] */ DWORD dwFlags,
/* [in] */ DWORD dwRequestedHandleType,
/* [out] */ IWmiDbHandle __RPC_FAR *__RPC_FAR *ppResult);
virtual HRESULT STDMETHODCALLTYPE Enumerate(
/* [in] */ IWmiDbHandle __RPC_FAR *pScope,
/* [in] */ DWORD dwFlags,
/* [in] */ DWORD dwRequestedHandleType,
/* [out] */ IWmiDbIterator __RPC_FAR *__RPC_FAR *ppQueryResult);
virtual HRESULT STDMETHODCALLTYPE AddObject(
/* [in] */ IWmiDbHandle __RPC_FAR *pScope,
/* [in] */ IWbemPath __RPC_FAR *pPath,
/* [in] */ DWORD dwFlags,
/* [in] */ DWORD dwRequestedHandleType,
/* [out] */ IWmiDbHandle __RPC_FAR *__RPC_FAR *ppResult);
virtual HRESULT STDMETHODCALLTYPE RemoveObject(
/* [in] */ IWmiDbHandle __RPC_FAR *pScope,
/* [in] */ IWbemPath __RPC_FAR *pPath,
/* [in] */ DWORD dwFlags);
virtual HRESULT STDMETHODCALLTYPE SetDecoration(
/* [in] */ LPWSTR lpMachineName,
/* [in] */ LPWSTR lpNamespacePath);
virtual HRESULT STDMETHODCALLTYPE SupportsQueries(
/* [in] */ DWORD *dwQuerySupportLevel
) {return WBEM_S_NO_ERROR;};
// batch session methods
virtual HRESULT STDMETHODCALLTYPE PutObjects(
/* [in] */ IWmiDbHandle __RPC_FAR *pScope,
/* [in] */ DWORD dwFlags,
/* [in] */ DWORD dwHandleType,
/* [out][in] */ WMIOBJECT_BATCH __RPC_FAR *pBatch);
virtual HRESULT STDMETHODCALLTYPE GetObjects(
/* [in] */ IWmiDbHandle __RPC_FAR *pScope,
/* [in] */ DWORD dwFlags,
/* [in] */ DWORD dwHandleType,
/* [in, out] */ WMIOBJECT_BATCH __RPC_FAR *pBatch);
virtual HRESULT STDMETHODCALLTYPE DeleteObjects(
/* [in] */ IWmiDbHandle __RPC_FAR *pScope,
/* [in] */ DWORD dwFlags,
/* [out][in] */ WMIOBJECT_BATCH __RPC_FAR *pBatch);
// IWbemTransaction methods.
virtual HRESULT STDMETHODCALLTYPE Begin(
/* [in] */ ULONG uTimeout,
/* [in] */ ULONG uFlags,
/* [in] */ GUID __RPC_FAR *pTransGUID);
virtual HRESULT STDMETHODCALLTYPE Rollback(
/* [in] */ ULONG uFlags);
virtual HRESULT STDMETHODCALLTYPE Commit(
/* [in] */ ULONG uFlags);
virtual HRESULT STDMETHODCALLTYPE QueryState(
/* [in] */ ULONG uFlags,
/* [out] */ ULONG __RPC_FAR *puState);
virtual HRESULT STDMETHODCALLTYPE Backup(
/* [in] */ LPCWSTR lpBackupPath,
/* [in] */ DWORD dwFlags);
virtual HRESULT STDMETHODCALLTYPE Restore(
/* [in] */ LPCWSTR lpRestorePath,
/* [in] */ LPCWSTR lpDestination,
/* [in] */ DWORD dwFlags);
HRESULT LoadSchemaCache ();
HRESULT LoadClassInfo (CSQLConnection *pConn, LPCWSTR lpDynasty = NULL, SQL_ID dScopeId = 0, BOOL bDeep = TRUE);
HRESULT LoadClassInfo (CSQLConnection *pConn, SQL_ID dClassId, BOOL bDeep = TRUE);
HRESULT GetObjectData (CSQLConnection *pConn, SQL_ID dObjectId, SQL_ID dClassId, SQL_ID dScopeId,
DWORD dwHandleType, DWORD &dwVersion, IWbemClassObject **pObj, BOOL bNoDefaults = FALSE,
LPWSTR lpKey = NULL, BOOL bGetSD = FALSE);
HRESULT GetObjectData2 (CSQLConnection *pConn, SQL_ID dObjectId, SQL_ID dClassId, SQL_ID dScopeId,
IWbemClassObject *pObj, BOOL bNoDefaults = FALSE, LPWSTR *lpKey = NULL);
HRESULT GetClassObject (CSQLConnection *pConn, SQL_ID dClassId, IWbemClassObject **ppObj);
HRESULT STDMETHODCALLTYPE GetObject_Internal(
/* [in] */ LPWSTR lpPath,
/* [in] */ DWORD dwFlags,
/* [in] */ DWORD dwRequestedHandleType,
/* [in] */ SQL_ID *dScopeId,
/* [out] */ IWmiDbHandle __RPC_FAR *__RPC_FAR *ppResult,
CSQLConnection *pConn = NULL);
HRESULT STDMETHODCALLTYPE PutObject(
CSQLConnection *pConn,
/* [in] */ IWmiDbHandle *pScope,
SQL_ID dScopeID,
LPWSTR lpScopePath,
/* [in] */ IUnknown __RPC_FAR *pObjToPut,
/* [in] */ DWORD dwFlags,
/* [in] */ DWORD dwRequestedHandleType,
_bstr_t &sPath,
/* [out] */ IWmiDbHandle __RPC_FAR *__RPC_FAR *ppResult,
BOOL bStoreAsClass = FALSE);
HRESULT CleanCache(SQL_ID dObjId, DWORD dwLockType = 0, void *pObj = NULL);
HRESULT ShutDown ();
HRESULT Delete(IWmiDbHandle *pHandle, CSQLConnection *pConn= NULL);
HRESULT Delete(SQL_ID dID, CSQLConnection *pConn= NULL);
HRESULT VerifyObjectSecurity (CSQLConnection *pConn, SQL_ID ObjectId, SQL_ID ClassId, SQL_ID ScopeId, SQL_ID ScopeClassId, DWORD AccessType);
HRESULT AccessCheck( PNTSECURITY_DESCRIPTOR pSD, DWORD dwAccessType, BOOL &bHasDACL );
CWmiDbController * GetController() { return (CWmiDbController *)m_pController;}
CSchemaCache * GetSchemaCache() { return (m_pController ? &((CWmiDbController *)m_pController)->SchemaCache : NULL);}
CObjectCache * GetObjectCache() { return (m_pController ? &((CWmiDbController *)m_pController)->ObjectCache : NULL);}
CSQLConnCache * GetSQLCache() { return (m_pController ? &((CWmiDbController *)m_pController)->ConnCache : NULL);}
CRITICAL_SECTION * GetCS() {return (m_pController ? &((CWmiDbController *)m_pController)->m_cs : NULL);}
void FixMethodParamIds(IWbemClassObject *pTemp);
CWmiDbSession(IWmiDbController *pController);
~CWmiDbSession();
private:
HRESULT CustomGetObject(IWmiDbHandle *pScope, IWbemPath *pPath, LPWSTR lpObjectKey,
DWORD dwFlags, DWORD dwRequestedHandleType, IWmiDbHandle **ppResult);
HRESULT CustomGetMapping(CSQLConnection *pConn, IWmiDbHandle *pScope, LPWSTR lpClassName, IWbemClassObject **ppMapping);
HRESULT CustomCreateMapping(CSQLConnection *pConn, LPWSTR lpClassName, IWbemClassObject *pClassObj, IWmiDbHandle *pScope);
HRESULT CustomPutInstance(CSQLConnection *pConn, IWmiDbHandle *pScope, SQL_ID dClassId,
DWORD dwFlags, IWbemClassObject **ppObjToPut, LPWSTR lpClassName = NULL);
HRESULT CustomFormatSQL(IWmiDbHandle *pScope, IWbemQuery *pQuery, _bstr_t &sSQL, SQL_ID *dClassId,
MappedProperties **ppMapping, DWORD *dwNumProps, BOOL *bCount=FALSE);
HRESULT CustomDelete(CSQLConnection *pConn, IWmiDbHandle *pScope, IWmiDbHandle *pHandle, LPWSTR lpClassName = NULL);
HRESULT SetEmbeddedProp (IWmiDbHandle *pScope, LPWSTR lpPropName, IWbemClassObject *pObj, VARIANT &vValue, CIMTYPE ct);
HRESULT GetEmbeddedClass (IWmiDbHandle *pScope, IWbemClassObject *pObj, LPWSTR lpEmbedProp, IWbemClassObject **ppClass);
HRESULT CustomSetProperties (IWmiDbHandle *pScope, IRowset *pRowset, IMalloc *pMalloc,
IWbemClassObject *pClassObj, MappedProperties *pProps,
DWORD dwNumProps, IWbemClassObject *pObj);
// Security helper functions.
HRESULT VerifyObjectSecurity (CSQLConnection *pConn, IWmiDbHandle *pHandle, DWORD AccessType);
HRESULT VerifyObjectSecurity(CSQLConnection *pConn, SQL_ID dScopeID, SQL_ID dScopeClassId, LPWSTR lpObjectPath,CWbemClassObjectProps *pProps,DWORD dwHandleType,
DWORD dwReqAccess, SQL_ID &dObjectId, SQL_ID &dClassId);
HRESULT VerifyClassSecurity (CSQLConnection *pConn, SQL_ID ClassId, DWORD AccessType);
HRESULT GetEffectiveObjectSecurity(CSQLConnection *pConn, SQL_ID ObjectId, SQL_ID ClassId, SQL_ID ScopeId, SQL_ID ScopeClassId, PNTSECURITY_DESCRIPTOR *ppSD, DWORD &dwSDLength);
HRESULT GetInheritedObjectSecurity(CSQLConnection *pConn, SQL_ID ObjectId, SQL_ID ClassId, SQL_ID ScopeId, SQL_ID ScopeClassId, PNTSECURITY_DESCRIPTOR *ppSD, DWORD &dwSDLength);
HRESULT GetInheritedClassSecurity(CSQLConnection *pConn, SQL_ID ClassId, SQL_ID ScopeId, SQL_ID ScopeClassId, PNTSECURITY_DESCRIPTOR *ppSD, DWORD &dwSDLength);
HRESULT GetInheritedInstanceSecurity(CSQLConnection *pConn, SQL_ID ObjectId, SQL_ID ClassId, SQL_ID ScopeId, SQL_ID ScopeClassId, PNTSECURITY_DESCRIPTOR *ppSD, DWORD &dwSDLength);
HRESULT GetInheritedContainerSecurity(CSQLConnection *pConn, SQL_ID ClassId, SQL_ID ScopeId, SQL_ID ScopeClassId, PNTSECURITY_DESCRIPTOR *ppSD, DWORD &dwSDLength);
HRESULT GetObjectSecurity (CSQLConnection *pConn, SQL_ID dObjectId, PNTSECURITY_DESCRIPTOR *pSD, DWORD dwSDLength, DWORD dwFlags, BOOL &bHasDacl);
HRESULT PutClass( CSQLConnection *pConn,SQL_ID dScopeID,LPCWSTR lpScopePath,CWbemClassObjectProps *pProps, IWbemClassObject *pObj,DWORD dwFlags,
SQL_ID &dObjectId,_bstr_t &sObjectPath, bool &bChg, BOOL bIgnoreDefaults = FALSE);
HRESULT PutInstance( CSQLConnection *pConn,IWmiDbHandle *pScope, SQL_ID dScopeID,LPCWSTR lpScopePath,CWbemClassObjectProps *pProps,
IWbemClassObject *pObj,DWORD dwFlags,SQL_ID &dObjectId,SQL_ID &dClassId, _bstr_t &sObjectPath, bool &bChg);
// These three functions will only generate sp_InsertClassData procs
HRESULT InsertPropertyDef(CSQLConnection *pConn,
IWbemClassObject *pObj, SQL_ID dScopeId, SQL_ID dObjectId,
LPWSTR lpPropName, VARIANT vDefault,
CIMTYPE cimtype, long lPropFlags,DWORD dRefId,
Properties &props);
HRESULT InsertQualifier(CSQLConnection *pConn,IWmiDbHandle *pScope, SQL_ID dObjectId,
LPWSTR lpQualifierName, VARIANT vValue,
long lQfrFlags,DWORD dwFlags, DWORD PropID,
Properties &props);
HRESULT InsertQualifiers (CSQLConnection *pConn,IWmiDbHandle *pScope, SQL_ID dObjectId, DWORD PropID, DWORD Flags, IWbemQualifierSet *pQS,
Properties &props);
HRESULT InsertArray(CSQLConnection *pConn,IWmiDbHandle *pScope, SQL_ID dObjectId, SQL_ID dClassId,
DWORD dwPropertyID, VARIANT &vDefault, long lFlavor=0, DWORD dwRefID = 0,
LPCWSTR lpObjectKey = NULL, LPCWSTR lpObjectPath = NULL, SQL_ID dScope = 0,
CIMTYPE ct = 0);
HRESULT InsertPropertyValues (CSQLConnection *pConn, IWmiDbHandle *pScope,
LPWSTR lpPath,SQL_ID &dObjectId,SQL_ID dClassId,SQL_ID dScopeId,
DWORD dwFlags,CWbemClassObjectProps *pProps, IWbemClassObject *pObj);
HRESULT FormatBatchInsQfrs (CSQLConnection *pConn,IWmiDbHandle *pScope,
SQL_ID dObjectId, SQL_ID dClassId,
DWORD dPropID, IWbemQualifierSet *pQS,
int &iPos, InsertQualifierValues **ppVals,
Properties &props, int &iNumProps);
HRESULT FormatBatchInsQfrValues(CSQLConnection *pConn,IWmiDbHandle *pScope,
SQL_ID dObjectId, DWORD dwQfrID,
VARIANT &vTemp, long lFlavor, InsertQualifierValues *pVals, Properties &props,
int &iPos, DWORD PropID);
HRESULT SetKeyhole (CSQLConnection *pConn, IWbemClassObject *pObj, DWORD dwKeyholePropID,
LPWSTR sKeyholeProp, LPCWSTR lpScopePath, _bstr_t &sPath);
HRESULT SetUnkeyedPath(CSQLConnection *pConn, IWbemClassObject *pObj, SQL_ID dClassId, _bstr_t &sObjPath);
HRESULT InsertQualifierValues (CSQLConnection *pConn, SQL_ID dObjectId,IWbemClassObject *pObj,Properties &props);
HRESULT NormalizeObjectPathGet (IWmiDbHandle __RPC_FAR *pScope, IWbemPath __RPC_FAR *pPath,
LPWSTR * lpNewPath, BOOL *bDefault = NULL, SQL_ID *dClassId = NULL, SQL_ID *dScopeId = NULL
, CSQLConnection *pConn = NULL);
HRESULT NormalizeObjectPath(IWmiDbHandle __RPC_FAR *pScope, LPCWSTR lpPath, LPWSTR * lpNewPath, BOOL bNamespace=FALSE,
CWbemClassObjectProps **ppProps=NULL, BOOL *bDefault = NULL, CSQLConnection *pConn = NULL, BOOL bNoTrunc = FALSE);
HRESULT NormalizeObjectPath(IWbemClassObject __RPC_FAR *pScope, IWmiDbHandle *pScope2, LPCWSTR lpPath, LPWSTR * lpNewPath, BOOL bNamespace=FALSE,
CWbemClassObjectProps **ppProps=NULL, BOOL *bDefault = NULL, BOOL bNoTrunc = FALSE);
HRESULT VerifyObjectLock (SQL_ID ObjectId, DWORD dwLockType, DWORD dwVersion);
BOOL IsDistributed() { return m_bIsDistributed;};
HRESULT IssueDeletionEvents (CSQLConnection *pConn, SQL_ID dObjectId,
SQL_ID dClassId, SQL_ID dScopeId, IWbemClassObject *pObj);
HRESULT AddTransLock(SQL_ID dObjectId, DWORD dwHandleType);
HRESULT CleanTransLocks();
BOOL LockExists (SQL_ID dObjId);
HRESULT UpdateHierarchy(CSQLConnection *pConn, SQL_ID dClassId, DWORD dwFlags, LPCWSTR lpScopePath,
CWbemClassObjectProps *pProps, _IWmiObject *pObj);
HRESULT UnlockDynasties(BOOL bDelete = FALSE);
HRESULT DeleteRows(IWmiDbHandle *pScope, IWmiDbIterator *pIterator, REFIID iid);
DWORD AddRef_Lock();
DWORD Release_Lock();
ULONG m_uRefCount;
IWmiDbController *m_pController;
IMalloc *m_pIMalloc;
_bstr_t m_sNamespacePath;
_bstr_t m_sMachineName;
BOOL m_bInUse;
DWORD m_dwLocale;
BOOL m_bIsDistributed;
_bstr_t m_sGUID; // GUID of the transaction.
SessionLocks m_TransLocks;
SessionDynasties m_Dynasties;
RefCount m_ThreadRefCount;
};
#endif
Reference in a new issue Copy permalink