Source code of Windows XP (NT5)
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.
 
 
 
 
 
 

765 lines
21 KiB

/*++
Copyright (c) 1998-2001 Microsoft Corporation
Module Name:
enum.cpp
Abstract:
Enumerates metabase tree.
Author:
???
Revision History:
Mohit Srivastava 18-Dec-00
--*/
#include "iisprov.h"
#include "enum.h"
#include "ipsecurity.h"
#include "adminacl.h"
#include "WbemObjectSink.h"
#include "instancehelper.h"
#include "SmartPointer.h"
#include <adserr.h>
extern CDynSchema* g_pDynSch;
///////////////////////////////////////
//
// CEnum class
//
///////////////////////////////////////
CEnum::CEnum()
{
m_pInstMgr = NULL;
m_pNamespace = NULL;
m_pAssociation = NULL;
m_pParsedObject = NULL;
m_hKey = NULL;
}
CEnum::~CEnum()
{
if(m_hKey)
m_metabase.CloseKey(m_hKey);
delete m_pInstMgr;
}
void CEnum::Init(
IWbemObjectSink FAR* a_pHandler,
CWbemServices* a_pNamespace,
ParsedObjectPath* a_pParsedObject,
LPWSTR a_pszKey,
WMI_ASSOCIATION* a_pAssociation,
SQL_LEVEL_1_RPN_EXPRESSION_EXT* a_pExp) // default(NULL)
{
if (!a_pHandler || !a_pNamespace || !a_pParsedObject)
throw WBEM_E_FAILED;
m_pInstMgr = new CWbemObjectSink(a_pHandler);
if(!m_pInstMgr)
{
THROW_ON_ERROR(WBEM_E_OUT_OF_MEMORY);
}
m_pNamespace = a_pNamespace;
m_pAssociation = a_pAssociation;
m_pParsedObject = a_pParsedObject;
m_pExp = a_pExp;
m_hKey = m_metabase.OpenKey(a_pszKey, false); // read only
}
void CEnum::SetObjectPath(
LPCWSTR a_pszPropertyName,
LPCWSTR a_pszObjectPath,
IWbemClassObject* a_pObj
)
{
_bstr_t bstr(a_pszPropertyName);
_variant_t v(a_pszObjectPath);
HRESULT hr = a_pObj->Put(bstr, 0, &v, 0);
THROW_ON_ERROR(hr);
}
void CEnum::PingObject()
{
CComPtr<IWbemClassObject> spObj;
CInstanceHelper InstanceHelper(m_pParsedObject, m_pNamespace);
DBG_ASSERT(!InstanceHelper.IsAssoc());
try
{
InstanceHelper.GetInstance(false, &m_metabase, &spObj, m_pExp);
}
catch(HRESULT hr)
{
if(hr != E_ADS_PROPERTY_NOT_SUPPORTED && hr != WBEM_E_NOT_FOUND)
{
throw;
}
}
if(spObj != NULL)
{
m_pInstMgr->Indicate(spObj);
}
}
void CEnum::PingAssociation(
LPCWSTR a_pszLeftKeyPath
)
{
HRESULT hr;
CComPtr<IWbemClassObject> spObj;
CComPtr<IWbemClassObject> spClass;
TSmartPointerArray<WCHAR> swszObjectPath;
CObjectPathParser PathParser(e_ParserAcceptRelativeNamespace);
if( m_pAssociation->pType != &WMI_ASSOCIATION_TYPE_DATA::s_ElementSetting &&
m_pAssociation->pType != &WMI_ASSOCIATION_TYPE_DATA::s_Component )
{
return;
}
hr = m_pNamespace->GetObject(
m_pAssociation->pszAssociationName,
0,
NULL,
&spClass,
NULL
);
THROW_ON_ERROR(hr);
hr = spClass->SpawnInstance(0, &spObj);
THROW_ON_ERROR(hr);
//
// first right side
//
if (!m_pParsedObject->SetClassName(m_pAssociation->pcRight->pszClassName))
throw WBEM_E_FAILED;
if (PathParser.Unparse(m_pParsedObject,&swszObjectPath))
throw WBEM_E_FAILED;
SetObjectPath(m_pAssociation->pType->pszRight, swszObjectPath, spObj);
swszObjectPath.Delete();
//
// then left side
//
if (m_pAssociation->pType == &WMI_ASSOCIATION_TYPE_DATA::s_Component)
{
// clear keyref first
m_pParsedObject->ClearKeys();
// add a keyref
_variant_t vt;
if(m_pAssociation->pcLeft->pkt == &METABASE_KEYTYPE_DATA::s_IIsComputer)
vt = L"LM"; // IIsComputer.Name = "LM"
else
vt = a_pszLeftKeyPath;
THROW_ON_FALSE(m_pParsedObject->AddKeyRef(m_pAssociation->pcLeft->pszKeyName,&vt));
}
if (!m_pParsedObject->SetClassName(m_pAssociation->pcLeft->pszClassName))
throw WBEM_E_FAILED;
if (PathParser.Unparse(m_pParsedObject,&swszObjectPath))
throw WBEM_E_FAILED;
SetObjectPath(m_pAssociation->pType->pszLeft, swszObjectPath, spObj);
swszObjectPath.Delete();
m_pInstMgr->Indicate(spObj);
}
void CEnum::DoPing(
LPCWSTR a_pszKeyName,
LPCWSTR a_pszKeyPath,
LPCWSTR a_pszParentKeyPath
)
{
// add keyref
_variant_t v(a_pszKeyPath);
THROW_ON_FALSE(m_pParsedObject->AddKeyRef(a_pszKeyName,&v));
// ping
if (!m_pAssociation)
PingObject();
else
PingAssociation(a_pszParentKeyPath);
// clear keyref
m_pParsedObject->ClearKeys();
}
void CEnum::Recurse(
LPCWSTR a_pszMetabasePath, // Current metabase location relative to m_hKey
METABASE_KEYTYPE* a_pktParentKeyType,// Current keytype
LPCWSTR a_pszLeftPath,
LPCWSTR a_pszWmiPrimaryKey,// "Name" - WMI only - nothing to do with MB
METABASE_KEYTYPE* a_pktSearch // the kt we are looking for
)
{
DWORD i = 0;
HRESULT hr;
WCHAR szSubKey[METADATA_MAX_NAME_LEN];
METABASE_KEYTYPE* pktCurrent;
DBGPRINTF((DBG_CONTEXT, "Recurse (Path = %ws, Left = %ws)\n", a_pszMetabasePath, a_pszLeftPath));
do
{
pktCurrent = a_pktSearch;
//
// Enumerate all subkeys of a_pszMetabasePath until we find a potential
// (grand*)parent of pktCurrent
//
hr = m_metabase.EnumKeys(
m_hKey,
a_pszMetabasePath,
szSubKey,
&i,
pktCurrent
);
i++;
if( hr == ERROR_SUCCESS)
{
_bstr_t bstrMetabasePath;
if(a_pszMetabasePath)
{
bstrMetabasePath = a_pszMetabasePath;
bstrMetabasePath += L"/";
}
bstrMetabasePath += szSubKey;
//
// With the exception of AdminACL, AdminACE, IPSecurity, we will only
// ping the object if we have found a keytype match in the metabase.
//
if( pktCurrent == a_pktSearch &&
!( m_pAssociation &&
m_pAssociation->pType == &WMI_ASSOCIATION_TYPE_DATA::s_Component &&
m_pAssociation->pcLeft->pkt != a_pktParentKeyType
)
)
{
DoPing(a_pszWmiPrimaryKey, bstrMetabasePath, a_pszLeftPath);
}
else if( a_pktSearch == &METABASE_KEYTYPE_DATA::s_TYPE_AdminACL || // AdminACL
a_pktSearch == &METABASE_KEYTYPE_DATA::s_TYPE_AdminACE
)
{
if( (m_pAssociation == NULL || // should never be
m_pAssociation->pType != &WMI_ASSOCIATION_TYPE_DATA::s_AdminACL ||
m_pAssociation->pcLeft->pkt == pktCurrent ||
m_pAssociation == &WMI_ASSOCIATION_DATA::s_AdminACLToACE) )
{
DoPingAdminACL(a_pktSearch, a_pszWmiPrimaryKey, bstrMetabasePath);
}
}
else if( a_pktSearch == &METABASE_KEYTYPE_DATA::s_TYPE_IPSecurity ) // IPSecurity
{
if( !(m_pAssociation &&
m_pAssociation->pType == &WMI_ASSOCIATION_TYPE_DATA::s_IPSecurity &&
m_pAssociation->pcLeft->pkt != pktCurrent
)
)
{
DoPingIPSecurity(a_pktSearch, a_pszWmiPrimaryKey, bstrMetabasePath);
}
}
// recusive
if(ContinueRecurse(pktCurrent, a_pktSearch))
{
Recurse(
bstrMetabasePath,
pktCurrent,
bstrMetabasePath,
a_pszWmiPrimaryKey,
a_pktSearch);
}
}
}while(hr == ERROR_SUCCESS);
DBGPRINTF((DBG_CONTEXT, "Recurse Exited\n"));
}
// DESC: You are looking for a_eKeyType by traversing thru the tree. You are
// currently at a_eParentKeyType and need to determine if you should keep
// on going.
// COMM: This seems very similar to CMetabase::CheckKey
bool CEnum::ContinueRecurse(
METABASE_KEYTYPE* a_pktParentKeyType,
METABASE_KEYTYPE* a_pktKeyType
)
{
bool bRet = false;
if( a_pktKeyType == &METABASE_KEYTYPE_DATA::s_TYPE_AdminACL ||
a_pktKeyType == &METABASE_KEYTYPE_DATA::s_TYPE_AdminACE ||
a_pktKeyType == &METABASE_KEYTYPE_DATA::s_TYPE_IPSecurity )
{
return true;
}
return g_pDynSch->IsContainedUnder(a_pktParentKeyType, a_pktKeyType);
/*switch(a_pktKeyType)
{
case &METABASE_KEYTYPE_DATA::s_IIsLogModule:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsLogModules )
bRet = true;
break;
case &METABASE_KEYTYPE_DATA::s_IIsFtpInfo:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsFtpService )
bRet = true;
break;
case &METABASE_KEYTYPE_DATA::s_IIsFtpServer:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsFtpService )
bRet = true;
break;
case &METABASE_KEYTYPE_DATA::s_IIsFtpVirtualDir:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsFtpService ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsFtpServer ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsFtpVirtualDir
)
bRet = true;
break;
case &METABASE_KEYTYPE_DATA::s_IIsWebInfo:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebService )
bRet = true;
break;
case &METABASE_KEYTYPE_DATA::s_IIsFilters:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebService ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebServer
)
bRet = true;
break;
case &METABASE_KEYTYPE_DATA::s_IIsFilter:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebService ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebServer ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsFilters
)
bRet = true;
break;
case &METABASE_KEYTYPE_DATA::s_IIsCompressionSchemes:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebService ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebServer ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsFilters
)
bRet = true;
break;
case &METABASE_KEYTYPE_DATA::s_IIsCompressionScheme:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebService ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebServer ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsFilters ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsCompressionSchemes )
bRet = true;
break;
case &METABASE_KEYTYPE_DATA::s_IIsWebServer:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebService )
bRet = true;
break;
case &METABASE_KEYTYPE_DATA::s_IIsCertMapper:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebService ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebServer
)
bRet = true;
break;
case &METABASE_KEYTYPE_DATA::s_IIsWebVirtualDir:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebService ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebServer ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebVirtualDir ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebDirectory
)
bRet = true;
break;
case &METABASE_KEYTYPE_DATA::s_IIsWebDirectory:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebService ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebServer ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebVirtualDir ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebDirectory
)
bRet = true;
break;
case &METABASE_KEYTYPE_DATA::s_IIsWebFile:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebService ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebServer ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebVirtualDir ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebDirectory
)
bRet = true;
break;
case TYPE_AdminACL:
case TYPE_AdminACE:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebService ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebServer ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebVirtualDir ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebDirectory ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsFtpService ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsFtpServer ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsFtpVirtualDir
)
bRet = true;
break;
case TYPE_IPSecurity:
if( a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebService ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebServer ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebVirtualDir ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsWebDirectory ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsFtpService ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsFtpServer ||
a_pktParentKeyType == &METABASE_KEYTYPE_DATA::s_IIsFtpVirtualDir
)
bRet = true;
break;
default:
break;
}*/
//return bRet;
//return true;
}
void CEnum::DoPingAdminACL(
METABASE_KEYTYPE* a_pktKeyType, // Search key
LPCWSTR a_pszKeyName, // Wmi Primary key - nothing to do with MB
LPCWSTR a_pszKeyPath // Current metabase path relative to m_hKey
)
{
// add keyref
_variant_t v(a_pszKeyPath);
THROW_ON_FALSE(m_pParsedObject->AddKeyRef(a_pszKeyName,&v));
if(a_pktKeyType == &METABASE_KEYTYPE_DATA::s_TYPE_AdminACE)
{
EnumACE(a_pszKeyPath);
}
else if(a_pktKeyType == &METABASE_KEYTYPE_DATA::s_TYPE_AdminACL)
{
// ping
if (!m_pAssociation)
PingObject();
else
PingAssociationAdminACL(a_pszKeyPath);
}
// clear keyref
m_pParsedObject->ClearKeys();
}
// for AdminACL
void CEnum::EnumACE(
LPCWSTR pszKeyPath
)
{
HRESULT hr = S_OK;
_variant_t var;
CComPtr<IEnumVARIANT> spEnum;
ULONG lFetch;
CComBSTR bstrTrustee;
IDispatch* pDisp = NULL;
CComPtr<IADsAccessControlEntry> spACE;
_bstr_t bstrMbPath;
WMI_CLASS* pWMIClass;
// get the metabase path of the object
BOOL fClass = FALSE;
if(m_pAssociation)
fClass = CUtils::GetClass(m_pAssociation->pcLeft->pszClassName,&pWMIClass);
else
fClass = CUtils::GetClass(m_pParsedObject->m_pClass,&pWMIClass);
if(!fClass)
return;
CUtils::GetMetabasePath(NULL,m_pParsedObject,pWMIClass,bstrMbPath);
// open ADSI
CAdminACL objACL;
hr = objACL.OpenSD(bstrMbPath);
if(SUCCEEDED(hr))
hr = objACL.GetACEEnum(&spEnum);
if ( FAILED(hr) )
return;
//////////////////////////////////////////////
// Enumerate ACEs
//////////////////////////////////////////////
hr = spEnum->Next( 1, &var, &lFetch );
while( hr == S_OK )
{
if ( lFetch == 1 )
{
if ( VT_DISPATCH != V_VT(&var) )
{
break;
}
pDisp = V_DISPATCH(&var);
/////////////////////////////
// Get the individual ACE
/////////////////////////////
hr = pDisp->QueryInterface(
IID_IADsAccessControlEntry,
(void**)&spACE
);
if ( SUCCEEDED(hr) )
{
hr = spACE->get_Trustee(&bstrTrustee);
if( SUCCEEDED(hr) )
{
// add keyref
_variant_t v(bstrTrustee);
//m_pParsedObject->RemoveKeyRef(L"Trustee");
THROW_ON_FALSE(m_pParsedObject->AddKeyRefEx(L"Trustee",&v));
// ping
if (!m_pAssociation)
PingObject();
else
PingAssociationAdminACL(pszKeyPath);
}
bstrTrustee = (LPWSTR)NULL;
spACE = NULL;
}
}
hr = spEnum->Next( 1, &var, &lFetch );
}
}
void CEnum::PingAssociationAdminACL(
LPCWSTR a_pszLeftKeyPath
)
{
HRESULT hr;
CComPtr<IWbemClassObject> spObj;
CComPtr<IWbemClassObject> spClass;
TSmartPointerArray<WCHAR> swszObjectPath;
CObjectPathParser PathParser(e_ParserAcceptRelativeNamespace);
_bstr_t bstrMbPath;
WMI_CLASS* pWMIClass;
if(m_pAssociation->pType != &WMI_ASSOCIATION_TYPE_DATA::s_AdminACL)
{
return;
}
// get the metabase path of the object
if (CUtils::GetClass(m_pAssociation->pcLeft->pszClassName,&pWMIClass))
{
CUtils::GetMetabasePath(NULL,m_pParsedObject,pWMIClass,bstrMbPath);
}
else
{
return;
}
// check if AdminACL existed
CAdminACL objACL;
hr = objACL.OpenSD(bstrMbPath);
objACL.CloseSD();
if(FAILED(hr))
{
return;
}
hr = m_pNamespace->GetObject(
m_pAssociation->pszAssociationName,
0,
NULL,
&spClass,
NULL
);
THROW_ON_ERROR(hr);
hr = spClass->SpawnInstance(0, &spObj);
THROW_ON_ERROR(hr);
//
// first right side
//
if (!m_pParsedObject->SetClassName(m_pAssociation->pcRight->pszClassName))
{
throw WBEM_E_FAILED;
}
if (PathParser.Unparse(m_pParsedObject,&swszObjectPath))
{
throw WBEM_E_FAILED;
}
SetObjectPath(m_pAssociation->pType->pszRight, swszObjectPath, spObj);
swszObjectPath.Delete();
//
// then left side
//
if(m_pAssociation == &WMI_ASSOCIATION_DATA::s_AdminACLToACE)
{
// clear keyref first
m_pParsedObject->ClearKeys();
// add a keyref
_variant_t vt = a_pszLeftKeyPath;
THROW_ON_FALSE(m_pParsedObject->AddKeyRef(m_pAssociation->pcLeft->pszKeyName,&vt));
}
if (!m_pParsedObject->SetClassName(m_pAssociation->pcLeft->pszClassName))
{
throw WBEM_E_FAILED;
}
if (PathParser.Unparse(m_pParsedObject,&swszObjectPath))
{
throw WBEM_E_FAILED;
}
SetObjectPath(m_pAssociation->pType->pszLeft, swszObjectPath, spObj);
swszObjectPath.Delete();
m_pInstMgr->Indicate(spObj);
}
// for IPSecurity
void CEnum::DoPingIPSecurity(
METABASE_KEYTYPE* a_pktKeyType,
LPCWSTR a_pszKeyName,
LPCWSTR a_pszKeyPath
)
{
// add keyref
_variant_t v(a_pszKeyPath);
THROW_ON_FALSE(m_pParsedObject->AddKeyRef(a_pszKeyName,&v));
// ping
if (!m_pAssociation)
PingObject();
else
PingAssociationIPSecurity(a_pszKeyPath);
// clear keyref
m_pParsedObject->ClearKeys();
}
// for IPSecurity
void CEnum::PingAssociationIPSecurity(
LPCWSTR a_pszLeftKeyPath
)
{
HRESULT hr;
CComPtr<IWbemClassObject> spObj;
CComPtr<IWbemClassObject> spClass;
TSmartPointerArray<WCHAR> swszObjectPath;
CObjectPathParser PathParser(e_ParserAcceptRelativeNamespace);
_bstr_t bstrMbPath;
WMI_CLASS* pWMIClass;
if(m_pAssociation->pType != &WMI_ASSOCIATION_TYPE_DATA::s_IPSecurity)
return;
// get the metabase path of the object
if (CUtils::GetClass(m_pAssociation->pcLeft->pszClassName,&pWMIClass))
{
CUtils::GetMetabasePath(NULL,m_pParsedObject,pWMIClass,bstrMbPath);
}
else
return;
// check if IPSecurity existed
CIPSecurity objIPsec;
hr = objIPsec.OpenSD(bstrMbPath, m_metabase);
objIPsec.CloseSD();
if(FAILED(hr))
return;
hr = m_pNamespace->GetObject(
m_pAssociation->pszAssociationName,
0,
NULL,
&spClass,
NULL
);
THROW_ON_ERROR(hr);
hr = spClass->SpawnInstance(0, &spObj);
THROW_ON_ERROR(hr);
//
// first right side
//
if (!m_pParsedObject->SetClassName(m_pAssociation->pcRight->pszClassName))
throw WBEM_E_FAILED;
if (PathParser.Unparse(m_pParsedObject,&swszObjectPath))
{
throw WBEM_E_FAILED;
}
SetObjectPath(m_pAssociation->pType->pszRight, swszObjectPath, spObj);
swszObjectPath.Delete();
//
// then left side
//
if (!m_pParsedObject->SetClassName(m_pAssociation->pcLeft->pszClassName))
throw WBEM_E_FAILED;
if (PathParser.Unparse(m_pParsedObject,&swszObjectPath))
throw WBEM_E_FAILED;
SetObjectPath(m_pAssociation->pType->pszLeft, swszObjectPath, spObj);
swszObjectPath.Delete();
m_pInstMgr->Indicate(spObj);
}