|
|
///////////////////////////////////////////////////////////////////////////////////
//
// Microsoft WMIOLE DB Provider
// (C) Copyright 1999 Microsoft Corporation. All Rights Reserved.
//
// Purpose: WmiOleDBMap.h: interface for the CWmiOleDBMap class.
//
//
///////////////////////////////////////////////////////////////////////////////////
#ifndef _WMIOLEDBMAP_HEADER
#define _WMIOLEDBMAP_HEADER
#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000
#include "headers.h"
const WCHAR strIndex[] = L"Index";
class CQuery;
#define ALL_QUALIFIERS 0x00000003
#define NO_QUALIFIERS 0x00000000
#define CLASS_QUALIFIERS 0x00000001
#define PROPERTY_QUALIFIERS 0x00000002
#define QUALIFIER_ L"Qualifier_"
#define DEFAULT_QUALIFIER_COUNT 10
#define WMI_CLASS_QUALIFIER 1
#define WMI_PROPERTY_QUALIFIER 2
#define WMI_PROPERTY 3
const CIM_OBJECTARRAY= CIM_FLAG_ARRAY | CIM_OBJECT ;
class cRowColumnInfoMemMgr;class CRowDataMemMgr;///////////////////////////////////////////////////////////////////////////////////////////////////////////
class CWmiOleDBMap {
private:
CWbemClassParameters * m_pWbemClassParms; // Common class information, such as name, context
CWbemClassDefinitionWrapper * m_pWbemClassDefinition; // Just class definitions
CWbemInstanceList * m_paWbemClassInstances; // Just instance info
CWbemCommandManager * m_pWbemCommandManager; // To manage commands
CWbemCollectionManager * m_pWbemCollectionManager;
CWbemClassWrapper * m_pWbemCurInst; ULONG m_cRef; BOOL m_bMethodRowset; CPropertyMemoryMgr m_PropMemMgr;
BOOL ParseQualifiedName(WCHAR * Root, WCHAR *& Parent, WCHAR *& Child ); HRESULT ValidProperty(const DBCOLUMNDESC * prgColDesc); HRESULT MapDBPROPToStdPropertyQualifier( DBPROP pProp, CVARIANT & Qualifier, CVARIANT & Value, LONG & lFlavor ); HRESULT MapDBPROPToStdClassQualifier( DBPROP pProp, CVARIANT & Qualifier, CVARIANT & Value, LONG & uFlavor );
HRESULT SetWMIProperty(const DBCOLUMNDESC* prgColDesc); HRESULT SetWMIClassQualifier(const DBCOLUMNDESC * prgColDesc,BOOL bDefault = TRUE); HRESULT SetWMIPropertyQualifier(const DBCOLUMNDESC * prgColDesc,BOOL bDefault = TRUE);
HRESULT GetPropertyColumnInfo( cRowColumnInfoMemMgr * pColumn, DBCOLUMNINFO ** pCol,CBSTR & pProperty,LONG &lFlavor);// HRESULT GetNextPropertyQualifier( CWbemClassWrapper *pInst,BSTR &strPropName,CVARIANT &vValue, LONG &lType);
HRESULT GetProperties(CRowDataMemMgr * pRow,CWbemClassWrapper * pClass,cRowColumnInfoMemMgr *pColInfoMgr);
HRESULT GetEmbededObjects(CWbemClassInstanceWrapper * pClass,BSTR strProperty,CVARIANT &vValue); HRESULT CommitRowDataForQualifier(CRowDataMemMgr * pRow,BSTR strQualifier,CVARIANT &vValue, ULONG lType, ULONG lFlavor); void GenerateURL(CWbemClassInstanceWrapper * pClass,BSTR strProperty,ULONG nIndex,CBSTR &strIn); void GetDefaultValue(const DBCOLUMNDESC * prgColDesc, VARIANT & varDefault); HRESULT SetColumns( const DBORDINAL cColumnDescs, const DBCOLUMNDESC rgColumnDescs[]); HRESULT GetPropertyColumnInfoForMixedRowsets( cRowColumnInfoMemMgr * pColumn, DBCOLUMNINFO ** pCol);
void SetColumnTypeURL(DBCOLUMNINFO * pCol) { pCol->dwFlags |= DBCOLUMNFLAGS_ISROWURL; }
public:
// CWmiOleDBMap(DWORD dwFlags, IDispatch *pDisp,CWbemConnectionWrapper * pWrap);
/* CWmiOleDBMap(int nSchemaType, DWORD dwFlags, WCHAR * pClassName, WCHAR * pSpecificTable, CWbemConnectionWrapper * Connect);
CWmiOleDBMap(DWORD dwFlags, WCHAR * pClassName, CWbemConnectionWrapper * Connect,BOOL fSchema = FALSE); CWmiOleDBMap(DWORD dwFlags, CQuery* p, CWbemConnectionWrapper * Connect); CWmiOleDBMap(DWORD dwFlags, WCHAR * pObjectPath, CWbemConnectionWrapper * Connect,INSTANCELISTTYPE instListType);*/ CWmiOleDBMap();
HRESULT FInit(int nSchemaType, DWORD dwFlags, WCHAR * pClassName, WCHAR * pSpecificTable, CWbemConnectionWrapper * Connect); HRESULT FInit(DWORD dwFlags, WCHAR * pClassName, CWbemConnectionWrapper * Connect); HRESULT FInit(DWORD dwFlags, CQuery* p, CWbemConnectionWrapper * Connect); HRESULT FInit(DWORD dwFlags, WCHAR * pObjectPath, CWbemConnectionWrapper * Connect,INSTANCELISTTYPE instListType);
~CWmiOleDBMap();
//===================================================================================
//
//===================================================================================
HRESULT CreateTable( DBORDINAL cColumnDescs, const DBCOLUMNDESC rgColumnDescs[], ULONG cPropertySets, DBPROPSET rgPropertySets[]); HRESULT AddColumn(const DBCOLUMNDESC* prgColDesc, DBID** ppColumnID); HRESULT DropColumn(const DBID* pColumnID);
HRESULT SetCommonDBCOLUMNINFO(DBCOLUMNINFO ** pCol,DBORDINAL uCurrentIndex); void SetColumnReadOnly(DBCOLUMNINFO * pCol, BOOL bReadOnly); HRESULT GetQualifiedNameColumnInfo( cRowColumnInfoMemMgr * pParentCol,DBCOLUMNINFO ** pCol, WCHAR * pName);
HRESULT DropTable();
HRESULT GetColumnCount( DBCOUNTITEM & cTotalColumns,DBCOUNTITEM & cParentColumns,DBCOUNTITEM &cNestedCols); HRESULT GetColumnInfoForParentColumns(cRowColumnInfoMemMgr * pParentCol);
HRESULT ResetInstances(); HRESULT ResetInstancesToNewPosition(DBROWOFFSET); HRESULT GetNextInstance(CWbemClassWrapper *&ppInst, CBSTR &strKey ,BOOL bFetchBack); HRESULT GetDataForInstance(CRowDataMemMgr * pRow,CWbemClassWrapper *pInst,cRowColumnInfoMemMgr *pColInfoMgr); HRESULT GetDataForSchemaInstance(CRowDataMemMgr * pRow,CWbemClassWrapper *pInst,cRowColumnInfoMemMgr *pColInfoMgr);
HRESULT GetPropertyQualifier(BSTR strPropName, BSTR strQualifierName ,VARIANT &vValue); BOOL IsPropQualiferIncluded(); HRESULT GetDataForPropertyQualifier(CRowDataMemMgr * pRow,CWbemClassWrapper *pInst,BSTR strPropName, BSTR strQualifier, cRowColumnInfoMemMgr *pColInfoMgr); BOOL IsSystemProperty(BSTR strProperty); HRESULT GetNextPropertyQualifier(CWbemClassWrapper *pInst,BSTR strPropName,BSTR &strQualifier,BOOL bFetchBack = FALSE); HRESULT ResetPropQualiferToNewPosition(CWbemClassWrapper *pInst,DBROWOFFSET lRowOffset,BSTR strPropertyName);
HRESULT GetNextClassQualifier(CWbemClassWrapper *pInst,BSTR &strQualifier,BOOL bFetchBack = FALSE); HRESULT GetDataForClassQualifier(CRowDataMemMgr * pRow,CWbemClassWrapper *pInst, BSTR strQualifier, cRowColumnInfoMemMgr *pColInfoMgr); HRESULT ResetClassQualiferToNewPosition(CWbemClassWrapper *pInst,DBROWOFFSET lRowOffset);
HRESULT DeleteInstance(CWbemClassWrapper * pClass ); HRESULT UpdateInstance(CWbemClassWrapper *pInst , BOOL bNewInst); HRESULT AddNewInstance(CWbemClassWrapper ** ppClass ); HRESULT RefreshInstance(CWbemClassWrapper * pInstance ); void ReleaseAllQualifiers(CWbemClassWrapper *pInst) {pInst->ReleaseAllPropertyQualifiers(); } void ReleaseQualifier(CWbemClassWrapper *pInst,BSTR strQualifier); WCHAR * GetClassName(); void SetNavigationFlags(DWORD dwFlags); void SetQueryFlags(DWORD dwFlags); HRESULT DeleteQualifier(CWbemClassWrapper *pInst, BSTR strQualifierName, BOOL bClassQualifier = TRUE , BSTR strPropertyName = NULL);
HRESULT SetProperty(CWbemClassWrapper *pInst,BSTR bstrColName,VARIANT *pvarData) { return pInst->SetProperty(bstrColName,pvarData); } HRESULT SetQualifier(CWbemClassWrapper *pInst,BSTR bstrColName,BSTR bstrQualifier ,VARIANT *pvarData,LONG lFlavor); DWORD GetFlags() { return m_pWbemClassParms->GetFlags(); }
HRESULT GetProperty(CWbemClassWrapper *pInst,BSTR pProperty, BYTE *& pData,DBTYPE &dwType ,DBLENGTH & dwSize, DWORD &dwFlags );
// This method is to be called before fetching any instance
HRESULT SetClass(WCHAR *pClassName) {return m_pWbemClassDefinition->SetClass(pClassName); }
CWbemClassWrapper *GetInstance(BSTR strPath); void GetInstanceKey(CWbemClassWrapper *pInst, CBSTR &strPath){ ((CWbemClassInstanceWrapper *)pInst)->GetKey(strPath);} CWbemClassWrapper *GetEmbededInstance(BSTR strPath,BSTR strProperty,int nIndex);
HRESULT GetKeyPropertyNames( SAFEARRAY **ppsaNames) { return m_pWbemClassDefinition->GetKeyPropertyNames(ppsaNames); } int ParseQualifiedNameToGetColumnType(WCHAR * wcsName ); HRESULT CWmiOleDBMap::GetInstanceCount(ULONG_PTR &lInstanceCount) { return m_paWbemClassInstances->GetNumberOfInstanceInEnumerator(&lInstanceCount);// return m_pWbemClassDefinition->GetInstanceCount(lInstanceCount);
}
HRESULT AddPropertyQualifier(CWbemClassWrapper *pInst,BSTR pProperty, BSTR Qualifier, VARIANT * vValue, LONG Flavor) { return pInst->SetPropertyQualifier(pProperty,Qualifier, vValue, Flavor); } HRESULT SetColumnProperties(const DBCOLUMNDESC * prgColDesc);
HRESULT AddIndex(const DBID* pColumnID); HRESULT DropIndex(const DBID* pColumnID); HRESULT UnlinkObjectFromContainer(BSTR strContainerObj,BSTR pstrObject); HRESULT LinkObjectFromContainer(BSTR strContainerObj,BSTR pstrObject); HRESULT CloneAndAddNewObjectInScope(BSTR strObj, BSTR strScope,WCHAR *& pstrNewPath);
void SetSytemPropertiesFlag(BOOL bSystemProperties) { m_pWbemClassParms->SetSytemPropertiesFlag(bSystemProperties);}
DWORD GetInstanceStatus(CWbemClassWrapper *pInst) { return ((CWbemClassInstanceWrapper *)pInst)->GetStatus(); } void SetInstanceStatus(CWbemClassWrapper *pInst, DWORD dwStatus) { ((CWbemClassInstanceWrapper *)pInst)->SetStatus(dwStatus); } INSTANCELISTTYPE GetObjListType(); STDMETHODIMP_(ULONG) AddRef(void); STDMETHODIMP_(ULONG) Release(void);
HRESULT SetSearchPreferences(ULONG cProps , DBPROP rgProp[]);
HRESULT GetRelativePath(CWbemClassInstanceWrapper *pInst,WCHAR *& strRelPath) { return pInst->GetRelativePath(strRelPath); }
FETCHDIRECTION GetCurFetchDirection() { return m_paWbemClassInstances->GetCurFetchDirection(); } void SetCurFetchDirection(FETCHDIRECTION FetchDir) { m_paWbemClassInstances->SetCurFetchDirection(FetchDir); }};
#endif
|
