/*++ 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 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 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 spObj; CComPtr spClass; TSmartPointerArray 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 spEnum; ULONG lFetch; CComBSTR bstrTrustee; IDispatch* pDisp = NULL; CComPtr 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, m_metabase); 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 spObj; CComPtr spClass; TSmartPointerArray 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, m_metabase); 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 spObj; CComPtr spClass; TSmartPointerArray 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); }