|
|
/*++
Copyright (C) 1996-2001 Microsoft Corporation
Module Name:
MOFDATA.H
Abstract:
Defines MOF compiler classes related to complete MOF file representation and transfer of data into WinMgmt. Division of labor between these and the classes defined in MOFPROP.H/CPP is not clear-cut.
History:
11/27/96 a-levn Compiles.
--*/
#ifndef _MCAUX_H_
#define _MCAUX_H_
#include <windows.h>
#include <wbemidl.h>
#include <miniafx.h>
#include <mofprop.h>
//******************************************************************************
//******************************************************************************
//
// class CNamespaceCache
//
// Represents the cache of pointers to the various namespaces MOF compiler
// has connection to.
//
//******************************************************************************
//
// Constructor.
//
// Constructs the cache given the IWbemLocator pointer to WinMgmt. This class will
// use this pointer to connect to whatever namespace is required.
//
// PARAMETERS:
//
// ADDREF IWbemLocator* pLocator Locator pointer. This function AddRefs
// it. It is Released in destructor.
//
//******************************************************************************
//
// Destructor
//
// Releases the locator pointer we were given in the constructor.
// Releases all cached namespace pointers.
//
//******************************************************************************
//
// GetNamespace
//
// Retrieves a pointer to a given namespace. If in cache, the cached copy is
// returned. If not, a new connection is established and cached.
//
// PARAMETERS:
//
// COPY LPCWSTR wszName Full name of the namespace to connect to.
//
// RETURN VALUES:
//
// IWbemServices*: NULL if an error occurs. If not NULL, the caller must
// release this pointer when no longer necessary.
//
//******************************************************************************
class CNamespaceCache{private: IWbemLocator* m_pLocator;
struct CNamespaceRecord { LPWSTR m_wszName; IWbemServices* m_pNamespace;
CNamespaceRecord(COPY LPCWSTR wszName, ADDREF IWbemServices* pNamespace); ~CNamespaceRecord(); };
CPtrArray m_aRecords; // CNamespaceRecord*
public: CNamespaceCache(ADDREF IWbemLocator* pLocator); ~CNamespaceCache(); RELEASE_ME IWbemServices* GetNamespace(COPY LPCWSTR wszName, SCODE & scRet, WCHAR * pUserName, WCHAR * pPassword, WCHAR * pAuthority,IWbemContext * pCtx, GUID LocatorGUID, LONG fConnectFlags);};
//*****************************************************************************
//*****************************************************************************
//
// class CMofData
//
// Represents an entire MOF file, basically --- a collection of objects. See
// CMObject in mofprop.h for details of object representation.
//
// Capable of storing its data into WinMgmt.
//
//******************************************************************************
//
// AddObject
//
// Adds another object to the store.
//
// PARAMETERS:
//
// ACQUIRE CMObject* pObject The object to add. The class acquires this
// object and will delete it on destruction.
//
//******************************************************************************
//
// GetNumObjects
//
// RETURN VALUES:
//
// int: the number of objects in the store.
//
//******************************************************************************
//
// GetObject
//
// Returns the object stored at a given index
//
// PARAMETERS:
//
// int nIndex The index to retrieve the object at.
//
// RETURN VALUES:
//
// CMObject*: NULL if nIndex is out of range. Otherwise, an INTERNAL
// pointer which is NOT to be deleted by the caller.
//
//******************************************************************************
//
// Store
//
// Transfers all the data to WinMgmt.
//
// PARAMETERS:
//
// OLE_MODIFY IWbemLocator* pLocator WinMgmt locator pointer to store into.
// long lClassFlags WBEM_FLAG_CREATE_OR_UPATE,
// WBEM_FLAG_CREAYE_ONLY, or
// WBEM_FLAG_UPDATE_ONLY to apply to
// class operations.
// long lInstanceFlags Same as lClassFlags, but for
// instance operations.
// BOOL bRollBackable Not implemented. Must be TRUE.
//
class CMofParser;
class CMofData : private CMofAliasCollection{private: CPtrArray m_aObjects; // CMObject*
CPtrArray m_aQualDefaults; // CMoQualifier*
BYTE * m_pBmofData; BYTE * m_pBmofToFar; HRESULT RollBack(int nObjects);
INTERNAL LPCWSTR FindAliasee(READ_ONLY LPWSTR wszAlias); friend CMObject; PDBG m_pDbg;
BOOL GotLineNumber(int nIndex);
void PrintError(int nIndex, long lMsgNum, HRESULT hres, WBEM_COMPILE_STATUS_INFO *pInfo);public: void SetBmofBuff(BYTE * pData){m_pBmofData = pData;}; BYTE * GetBmofBuff(){return m_pBmofData;}; void SetBmofToFar(BYTE * pData){m_pBmofToFar = pData;}; BYTE * GetBmofToFar(){return m_pBmofToFar;}; BOOL CMofData::IsAliasInUse(READ_ONLY LPWSTR wszAlias); void AddObject(ACQUIRE CMObject* pObject) {m_aObjects.Add(pObject);} int GetNumObjects() {return m_aObjects.GetSize();} CPtrArray * GetObjArrayPtr(){return &m_aObjects;};
INTERNAL CMObject* GetObject(int nIndex) {return (CMObject*)m_aObjects[nIndex];}
void SetQualifierDefault(ACQUIRE CMoQualifier* pDefault); HRESULT SetDefaultFlavor(MODIFY CMoQualifier& Qual, bool bTopLevel, QUALSCOPE qs, PARSESTATE ps); int GetNumDefaultQuals(){return m_aQualDefaults.GetSize();}; CMoQualifier* GetDefaultQual(int nIndex){return (CMoQualifier*)m_aQualDefaults.GetAt(nIndex);};
CMofData(PDBG pDbg){m_pDbg = pDbg;}; ~CMofData() { int i; for(i = 0; i < m_aObjects.GetSize(); i++) delete (CMObject*)m_aObjects[i]; for(i = 0; i < m_aQualDefaults.GetSize(); i++) delete (CMoQualifier*)m_aQualDefaults[i]; }
HRESULT Store(CMofParser & Parser, OLE_MODIFY IWbemLocator* pLocator, IWbemServices *pOverride, BOOL bRollBackable, WCHAR * pUserName, WCHAR * pPassword, WCHAR *pAuthority, IWbemContext * pCtx, GUID LocatorGUID, WBEM_COMPILE_STATUS_INFO *pInfo, BOOL bClassOwnerUpdate, BOOL bInstanceOwnerUpdate, LONG fConnectFlags);
HRESULT Split(CMofParser & Parser, LPWSTR BMOFFileName, WBEM_COMPILE_STATUS_INFO *pInfo, BOOL bUnicode, BOOL bAutoRecovery,LPWSTR pwszAmendment); void RecursiveSetAmended(CMObject * pObj);
};
#endif
|
