560 lines
17 KiB
C++
560 lines
17 KiB
C++
/*++
|
|
|
|
Copyright (C) 1996-2001 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
PathParse.H
|
|
|
|
Abstract:
|
|
|
|
Implements the default object path parser.
|
|
|
|
History:
|
|
|
|
a-davj 5-feb-00 Created.
|
|
|
|
--*/
|
|
|
|
#ifndef _PATHPARSE_H_
|
|
#define _PATHPARSE_H_
|
|
|
|
#include "genlex.h"
|
|
//#include "opathlex.h"
|
|
#include <wmiutils.h>
|
|
#include <umi.h>
|
|
#include "wbemcli.h"
|
|
#include "flexarry.h"
|
|
typedef LPVOID * PPVOID;
|
|
|
|
class CSafeInCritSec
|
|
{
|
|
protected:
|
|
CRITICAL_SECTION* m_pcs;
|
|
BOOL m_bOK;
|
|
public:
|
|
CSafeInCritSec(CRITICAL_SECTION* pcs) : m_pcs(pcs)
|
|
{
|
|
m_bOK = TRUE;
|
|
__try
|
|
{
|
|
EnterCriticalSection(m_pcs);
|
|
}
|
|
__except(1)
|
|
{
|
|
m_bOK = FALSE;
|
|
};
|
|
|
|
}
|
|
inline ~CSafeInCritSec()
|
|
{
|
|
if(m_bOK)
|
|
LeaveCriticalSection(m_pcs);
|
|
}
|
|
BOOL IsOK(){return m_bOK;};
|
|
|
|
};
|
|
|
|
class CRefCntCS
|
|
{
|
|
private:
|
|
long m_lRef;
|
|
DWORD m_guard1; // test
|
|
CRITICAL_SECTION m_cs;
|
|
DWORD m_guard2; // test
|
|
HRESULT m_Status;
|
|
|
|
public:
|
|
CRefCntCS();
|
|
~CRefCntCS()
|
|
{
|
|
if(m_Status == S_OK)
|
|
DeleteCriticalSection(&m_cs);
|
|
m_guard1 = 0;
|
|
}
|
|
HRESULT GetStatus(){return m_Status;};
|
|
long AddRef(void){long lRef = InterlockedIncrement(&m_lRef);return lRef;};
|
|
long Release(void);
|
|
CRITICAL_SECTION * GetCS(){return &m_cs;};
|
|
};
|
|
|
|
|
|
//***************************************************************************
|
|
//
|
|
// STRUCT NAME:
|
|
//
|
|
// CKeyRef
|
|
//
|
|
// DESCRIPTION:
|
|
//
|
|
// Holds information for a single key. Includes name, data, and data type.
|
|
//
|
|
//***************************************************************************
|
|
|
|
struct CKeyRef
|
|
{
|
|
LPWSTR m_pName;
|
|
DWORD m_dwType;
|
|
DWORD m_dwSize;
|
|
void * m_pData;
|
|
|
|
CKeyRef();
|
|
CKeyRef(LPCWSTR wszKeyName, DWORD dwType, DWORD dwSize, void * pData);
|
|
HRESULT SetData(DWORD dwType, DWORD dwSize, void * pData);
|
|
~CKeyRef();
|
|
|
|
// note that the caller is to free the returned string.
|
|
LPWSTR GetValue(BOOL bQuotes=TRUE);
|
|
|
|
DWORD GetValueSize();
|
|
DWORD GetTotalSize();
|
|
};
|
|
|
|
class CUmiParsedComponent : public IUmiURLKeyList
|
|
{
|
|
public:
|
|
CUmiParsedComponent(IWbemPathKeyList * pParent) : m_pParent(pParent){};
|
|
|
|
//IUnknown members
|
|
|
|
STDMETHODIMP QueryInterface(REFIID riid, PPVOID ppv)
|
|
{
|
|
return m_pParent->QueryInterface(riid, ppv);
|
|
};
|
|
|
|
STDMETHODIMP_(ULONG) AddRef(void){return m_pParent->AddRef();};
|
|
STDMETHODIMP_(ULONG) Release(void){return m_pParent->Release();};
|
|
|
|
HRESULT STDMETHODCALLTYPE GetCount(
|
|
/* [out] */ ULONG __RPC_FAR *puKeyCount){return m_pParent->GetCount(puKeyCount);};
|
|
|
|
HRESULT STDMETHODCALLTYPE SetKey(
|
|
/* [string][in] */ LPCWSTR pszName,
|
|
/* [string][in] */ LPCWSTR pszValue){return m_pParent->SetKey(pszName, 0, CIM_STRING,
|
|
(void *)pszValue);};
|
|
|
|
HRESULT STDMETHODCALLTYPE GetKey(
|
|
/* [in] */ ULONG uKeyIx,
|
|
/* [in] */ ULONG uFlags,
|
|
/* [out][in] */ ULONG __RPC_FAR *puKeyNameBufSize,
|
|
/* [in] */ LPWSTR pszKeyName,
|
|
/* [out][in] */ ULONG __RPC_FAR *puValueBufSize,
|
|
/* [in] */ LPWSTR pszValue);
|
|
|
|
HRESULT STDMETHODCALLTYPE RemoveKey(
|
|
/* [string][in] */ LPCWSTR pszName,
|
|
/* [in] */ ULONG uFlags){return m_pParent->RemoveKey(pszName,0);};
|
|
|
|
HRESULT STDMETHODCALLTYPE RemoveAllKeys(
|
|
/* [in] */ ULONG uFlags){return m_pParent->RemoveAllKeys(0);};
|
|
|
|
HRESULT STDMETHODCALLTYPE GetKeysInfo(
|
|
/* [in] */ ULONG uRequestedInfo,
|
|
/* [out] */ ULONGLONG __RPC_FAR *puResponse){return WBEM_E_NOT_AVAILABLE;};
|
|
|
|
protected:
|
|
IWbemPathKeyList * m_pParent;
|
|
};
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CLASS NAME:
|
|
//
|
|
// CParsedComponent
|
|
//
|
|
// DESCRIPTION:
|
|
//
|
|
// Each namespace, scope and the class is represented by an instance of this. It holds
|
|
// an array of CKeyRef objects and supports the IWbemPathKeyList interface.
|
|
//
|
|
//***************************************************************************
|
|
|
|
class CParsedComponent : public IWbemPathKeyList
|
|
{
|
|
public:
|
|
CParsedComponent(CRefCntCS *);
|
|
~CParsedComponent();
|
|
friend class CDefPathParser;
|
|
friend class CUmiPathParser;
|
|
void ClearKeys ();
|
|
HRESULT GetName(BSTR *pName);
|
|
HRESULT Unparse(BSTR *pKey, bool bGetQuotes, bool bUseClassName);
|
|
HRESULT GetComponentType(DWORD &dwType);
|
|
BOOL AddKeyRef(CKeyRef* pAcquireRef);
|
|
bool IsPossibleNamespace();
|
|
bool IsInstance();
|
|
HRESULT SetNS(LPCWSTR pName);
|
|
|
|
//IUnknown members
|
|
|
|
STDMETHODIMP QueryInterface(REFIID riid, PPVOID ppv)
|
|
{
|
|
*ppv=NULL;
|
|
|
|
if (IID_IUnknown==riid || IID_IWbemPathKeyList==riid)
|
|
*ppv=this;
|
|
else if (riid == IID_IMarshal && m_pFTM)
|
|
return m_pFTM->QueryInterface(riid, ppv);
|
|
//postponed till Blackcomb if (IID_IUmiURLKeyList==riid)
|
|
//postponed till Blackcomb *ppv = &m_UmiWrapper;
|
|
|
|
if (NULL!=*ppv)
|
|
{
|
|
AddRef();
|
|
return NOERROR;
|
|
}
|
|
|
|
return E_NOINTERFACE;
|
|
};
|
|
|
|
STDMETHODIMP_(ULONG) AddRef(void)
|
|
{
|
|
return InterlockedIncrement(&m_cRef);
|
|
};
|
|
STDMETHODIMP_(ULONG) Release(void)
|
|
{
|
|
long lRef = InterlockedDecrement(&m_cRef);
|
|
if (0L == lRef)
|
|
delete this;
|
|
return lRef;
|
|
};
|
|
|
|
HRESULT STDMETHODCALLTYPE GetCount(
|
|
/* [out] */ ULONG __RPC_FAR *puKeyCount);
|
|
|
|
HRESULT STDMETHODCALLTYPE SetKey(
|
|
/* [string][in] */ LPCWSTR wszName,
|
|
/* [in] */ ULONG uFlags,
|
|
/* [in] */ ULONG uCimType,
|
|
/* [in] */ LPVOID pKeyVal);
|
|
|
|
HRESULT STDMETHODCALLTYPE SetKey2(
|
|
/* [string][in] */ LPCWSTR wszName,
|
|
/* [in] */ ULONG uFlags,
|
|
/* [in] */ ULONG uCimType,
|
|
/* [in] */ VARIANT __RPC_FAR *pKeyVal);
|
|
|
|
HRESULT STDMETHODCALLTYPE GetKey(
|
|
/* [in] */ ULONG uKeyIx,
|
|
/* [in] */ ULONG uFlags,
|
|
/* [out][in] */ ULONG __RPC_FAR *puNameBufSize,
|
|
/* [out][in] */ LPWSTR pszKeyName,
|
|
/* [out][in] */ ULONG __RPC_FAR *puKeyValBufSize,
|
|
/* [out][in] */ LPVOID pKeyVal,
|
|
/* [out] */ ULONG __RPC_FAR *puApparentCimType);
|
|
|
|
HRESULT STDMETHODCALLTYPE GetKey2(
|
|
/* [in] */ ULONG uKeyIx,
|
|
/* [in] */ ULONG uFlags,
|
|
/* [out][in] */ ULONG __RPC_FAR *puNameBufSize,
|
|
/* [out][in] */ LPWSTR pszKeyName,
|
|
/* [out][in] */ VARIANT __RPC_FAR *pKeyValue,
|
|
/* [out] */ ULONG __RPC_FAR *puApparentCimType);
|
|
|
|
HRESULT STDMETHODCALLTYPE RemoveKey(
|
|
/* [string][in] */ LPCWSTR wszName,
|
|
/* [in] */ ULONG uFlags);
|
|
|
|
HRESULT STDMETHODCALLTYPE RemoveAllKeys(
|
|
/* [in] */ ULONG uFlags);
|
|
|
|
HRESULT STDMETHODCALLTYPE MakeSingleton( boolean bSet);
|
|
|
|
HRESULT STDMETHODCALLTYPE GetInfo(
|
|
/* [in] */ ULONG uRequestedInfo,
|
|
/* [out] */ ULONGLONG __RPC_FAR *puResponse);
|
|
|
|
HRESULT STDMETHODCALLTYPE GetText(
|
|
/* [in] */ long lFlags,
|
|
/* [out][in] */ ULONG __RPC_FAR *puBuffLength,
|
|
/* [string][out] */ LPWSTR pszText);
|
|
|
|
|
|
private:
|
|
|
|
BSTR m_sClassName;
|
|
CFlexArray m_Keys;
|
|
bool m_bSingleton;
|
|
long m_cRef;
|
|
// postponed till blackcomb CUmiParsedComponent m_UmiWrapper;
|
|
CRefCntCS * m_pCS;
|
|
IUnknown * m_pFTM;
|
|
|
|
};
|
|
|
|
|
|
|
|
//***************************************************************************
|
|
//
|
|
// CLASS NAME:
|
|
//
|
|
// CDefPathParser
|
|
//
|
|
// DESCRIPTION:
|
|
//
|
|
// Provides the default wmi path parser.
|
|
//
|
|
//***************************************************************************
|
|
|
|
class CDefPathParser : public IWbemPath, public IUmiURL
|
|
{
|
|
public:
|
|
CDefPathParser(void);
|
|
~CDefPathParser(void);
|
|
DWORD GetNumComponents();
|
|
bool IsEmpty(void);
|
|
long GetNumNamespaces();
|
|
void Empty(void);
|
|
enum Status {UNINITIALIZED, BAD_STRING, EXECEPTION_THROWN, OK, FAILED_TO_INIT};
|
|
BOOL ActualRelativeTest(LPWSTR wszMachine,
|
|
LPWSTR wszNamespace,
|
|
BOOL bChildrenOK);
|
|
void InitEmpty(){};
|
|
|
|
//IUnknown members
|
|
STDMETHODIMP QueryInterface(REFIID riid, PPVOID ppv)
|
|
{
|
|
*ppv=NULL;
|
|
|
|
if(m_dwStatus == FAILED_TO_INIT)
|
|
return WBEM_E_OUT_OF_MEMORY;
|
|
if(m_pCS == NULL)
|
|
return E_NOINTERFACE;
|
|
if (IID_IUnknown==riid || IID_IWbemPath==riid)
|
|
*ppv=(IWbemPath *)this;
|
|
else if (riid == IID_IMarshal && m_pFTM)
|
|
return m_pFTM->QueryInterface(riid, ppv);
|
|
|
|
// postponed till Blackcomb else if (IID_IUmiURL==riid)
|
|
// postponed till Blackcomb *ppv=(IUmiURL *)this;
|
|
|
|
if (NULL!=*ppv)
|
|
{
|
|
AddRef();
|
|
return NOERROR;
|
|
}
|
|
|
|
return E_NOINTERFACE;
|
|
};
|
|
|
|
STDMETHODIMP_(ULONG) AddRef(void)
|
|
{
|
|
return InterlockedIncrement(&m_cRef);
|
|
};
|
|
STDMETHODIMP_(ULONG) Release(void)
|
|
{
|
|
long lRef = InterlockedDecrement(&m_cRef);
|
|
if (0L == lRef)
|
|
delete this;
|
|
return lRef;
|
|
};
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE SetText(
|
|
/* [in] */ ULONG uMode,
|
|
/* [in] */ LPCWSTR pszPath);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetText(
|
|
/* [in] */ long lFlags,
|
|
/* [out][in] */ ULONG __RPC_FAR *puBuffLength,
|
|
/* [string][out] */ LPWSTR pszText);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetInfo(
|
|
/* [in] */ ULONG uRequestedInfo,
|
|
/* [out] */ ULONGLONG __RPC_FAR *puResponse);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE SetServer(
|
|
/* [string][in] */ LPCWSTR Name);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetServer(
|
|
/* [out][in] */ ULONG __RPC_FAR *puNameBufLength,
|
|
/* [string][out] */ LPWSTR pName);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetNamespaceCount(
|
|
/* [out] */ ULONG __RPC_FAR *puCount);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE SetNamespaceAt(
|
|
/* [in] */ ULONG uIndex,
|
|
/* [string][in] */ LPCWSTR pszName);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetNamespaceAt(
|
|
/* [in] */ ULONG uIndex,
|
|
/* [out][in] */ ULONG __RPC_FAR *puNameBufLength,
|
|
/* [string][out] */ LPWSTR pName);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE RemoveNamespaceAt(
|
|
/* [in] */ ULONG uIndex);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE RemoveAllNamespaces( void);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetScopeCount(
|
|
/* [out] */ ULONG __RPC_FAR *puCount);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE SetScope(
|
|
unsigned long,unsigned short *);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE SetScopeFromText(
|
|
/* [in] */ ULONG uIndex,
|
|
/* [in] */ LPWSTR pszText);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetScope(
|
|
/* [in] */ ULONG uIndex,
|
|
/* [out][in] */ ULONG __RPC_FAR *puClassNameBufSize,
|
|
/* [in] */ LPWSTR pszClass,
|
|
/* [out] */ IWbemPathKeyList __RPC_FAR *__RPC_FAR *pKeyList);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetScopeAsText(
|
|
/* [in] */ ULONG uIndex,
|
|
/* [out][in] */ ULONG __RPC_FAR *puTextBufSize,
|
|
/* [out][in] */ LPWSTR pszText);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE RemoveScope(
|
|
/* [in] */ ULONG uIndex);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE RemoveAllScopes( void);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE SetClassName(
|
|
/* [string][in] */ LPCWSTR Name);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetClassName(
|
|
/* [out][in] */ ULONG __RPC_FAR *puBuffLength,
|
|
/* [string][out] */ LPWSTR pszName);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetKeyList(
|
|
/* [out] */ IWbemPathKeyList __RPC_FAR *__RPC_FAR *pOut);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE CreateClassPart(
|
|
/* [in] */ long lFlags,
|
|
/* [string][in] */ LPCWSTR Name);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE DeleteClassPart(
|
|
/* [in] */ long lFlags);
|
|
|
|
virtual BOOL STDMETHODCALLTYPE IsRelative(
|
|
/* [string][in] */ LPWSTR wszMachine,
|
|
/* [string][in] */ LPWSTR wszNamespace);
|
|
|
|
virtual BOOL STDMETHODCALLTYPE IsRelativeOrChild(
|
|
/* [string][in] */ LPWSTR wszMachine,
|
|
/* [string][in] */ LPWSTR wszNamespace,
|
|
/* [in] */ long lFlags);
|
|
|
|
virtual BOOL STDMETHODCALLTYPE IsLocal(
|
|
/* [string][in] */ LPCWSTR wszMachine);
|
|
|
|
virtual BOOL STDMETHODCALLTYPE IsSameClassName(
|
|
/* [string][in] */ LPCWSTR wszClass);
|
|
|
|
// IUmiURL interface
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE Set(
|
|
/* [in] */ long lFlags,
|
|
/* [in] */ LPCWSTR pszText);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE Get(
|
|
/* [in] */ long lFlags,
|
|
/* [out][in] */ ULONG __RPC_FAR *puBufSize,
|
|
/* [string][in] */ LPWSTR pszDest);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetPathInfo(
|
|
/* [in] */ ULONG uRequestedInfo,
|
|
/* [out] */ ULONGLONG __RPC_FAR *puResponse);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE SetLocator(
|
|
/* [string][in] */ LPCWSTR Name);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetLocator(
|
|
/* [out][in] */ ULONG __RPC_FAR *puNameBufLength,
|
|
/* [string][in] */ LPWSTR pName);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE SetRootNamespace(
|
|
/* [string][in] */ LPCWSTR Name);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetRootNamespace(
|
|
/* [out][in] */ ULONG __RPC_FAR *puNameBufLength,
|
|
/* [string][out][in] */ LPWSTR pName);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetComponentCount(
|
|
/* [out] */ ULONG __RPC_FAR *puCount);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE SetComponent(
|
|
/* [in] */ ULONG uIndex,
|
|
/* [in] */ LPWSTR pszClass);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE SetComponentFromText(
|
|
/* [in] */ ULONG uIndex,
|
|
/* [in] */ LPWSTR pszText);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetComponent(
|
|
/* [in] */ ULONG uIndex,
|
|
/* [out][in] */ ULONG __RPC_FAR *puClassNameBufSize,
|
|
/* [out][in] */ LPWSTR pszClass,
|
|
/* [out] */ IUmiURLKeyList __RPC_FAR *__RPC_FAR *pKeyList);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetComponentAsText(
|
|
/* [in] */ ULONG uIndex,
|
|
/* [out][in] */ ULONG __RPC_FAR *puTextBufSize,
|
|
/* [out][in] */ LPWSTR pszText);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE RemoveComponent(
|
|
/* [in] */ ULONG uIndex);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE RemoveAllComponents( void);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE SetLeafName(
|
|
/* [string][in] */ LPCWSTR Name);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetLeafName(
|
|
/* [out][in] */ ULONG __RPC_FAR *puBuffLength,
|
|
/* [string][out][in] */ LPWSTR pszName);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE GetKeyList(
|
|
/* [out] */ IUmiURLKeyList __RPC_FAR *__RPC_FAR *pOut);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE CreateLeafPart(
|
|
/* [in] */ long lFlags,
|
|
/* [string][in] */ LPCWSTR Name);
|
|
|
|
virtual HRESULT STDMETHODCALLTYPE DeleteLeafPart(
|
|
/* [in] */ long lFlags);
|
|
|
|
|
|
|
|
|
|
|
|
HRESULT SetServer(LPCWSTR Name, bool m_bServerNameSetByDefault, bool bAcquire);
|
|
BOOL HasServer(){return m_pServer != NULL;};
|
|
LPWSTR GetPath(DWORD nStartAt, DWORD nStopAt,bool bGetServer = false);
|
|
BOOL AddNamespace(LPCWSTR wszNamespace);
|
|
BOOL AddClass(LPCWSTR lpClassName);
|
|
BOOL AddKeyRef(CKeyRef* pAcquireRef);
|
|
BOOL SetSingletonObj();
|
|
LPWSTR GetNamespacePart();
|
|
LPWSTR GetParentNamespacePart();
|
|
BOOL SortKeys();
|
|
CParsedComponent * GetLastComponent();
|
|
HRESULT GetComponentString(ULONG Index, BSTR * pUnparsed, WCHAR & wDelim);
|
|
HRESULT AddComponent(CParsedComponent * pComp);
|
|
CParsedComponent * GetClass();
|
|
CRefCntCS * GetRefCntCS(){return m_pCS;};
|
|
void * m_pGenLex; // for test purposes only
|
|
|
|
protected:
|
|
bool m_bSetViaUMIPath;
|
|
long m_cRef;
|
|
LPWSTR m_pServer; // NULL if no server
|
|
CFlexArray m_Components; // list of namespaces and scopes
|
|
// CParsedComponent * m_pClass; // the class
|
|
DWORD m_dwStatus;
|
|
bool m_bParent; // true if text is ".."
|
|
LPWSTR m_pRawPath; // temporary fix for Raja
|
|
CRefCntCS * m_pCS;
|
|
LPWSTR m_wszOriginalPath;
|
|
bool m_bServerNameSetByDefault;
|
|
IUnknown * m_pFTM;
|
|
DWORD m_dwException;
|
|
|
|
};
|
|
|
|
|
|
#endif
|