// metatool.cpp : implementation file // // some common tools used for "smart" writing to the metabase #include "stdafx.h" #define _COMIMPORT #include #include #include #include "wrapmb.h" #include "metatool.h" //---------------------------------------------------------------- // open the metabase with an option to create the directory if it doesn't // exist. It would be nice to move this into wrapmb, but that is too big // a change for now. Maybe we can do that later. BOOL OpenAndCreate( CWrapMetaBase* pmb, LPCTSTR pszTarget, DWORD perm, BOOL fCreate ) { BOOL f; CString szTarget = pszTarget; // start by just trying to open it. easy easy. if ( pmb->Open(szTarget, perm) ) return TRUE; // if requested, try to create the key if it doesn't exist if ( fCreate ) { // find the nearest openable parent directory and open it CString szPartial; CString szBase = szTarget; do { szBase = szBase.Left( szBase.ReverseFind(_T('/')) ); szPartial = szTarget.Right( szTarget.GetLength() - szBase.GetLength() - 1 ); f = pmb->Open( szBase, METADATA_PERMISSION_WRITE | perm ); } while (!f && !szBase.IsEmpty()); // if all that failed, fail if ( !f ) return FALSE; // create the key that we really want f = pmb->AddObject( szPartial ); pmb->Close(); // if all that failed, fail if ( !f ) return FALSE; // try again if ( pmb->Open(szTarget, perm) ) return TRUE; } // total washout return FALSE; } //---------------------------------------------------------------- // starting at the root, check for values set on sub-keys that may need to be overridden // and propmt the user for what to do void CheckInheritence( LPCTSTR pszServer, LPCTSTR pszInheritRoot, DWORD dwMDIdentifier, DWORD dwMDDataType, DWORD dwMDUserType = IIS_MD_UT_SERVER, DWORD dwMDAttributes = METADATA_INHERIT) { // // Build a generic title in case this property is custom // CString strTitle; strTitle.Format(IDS_GENERIC_INHERITANCE_TITLE, dwMDIdentifier); CComAuthInfo auth(pszServer); CInheritanceDlg dlgInherit( TRUE, // Look in table first dwMDIdentifier, dwMDAttributes, dwMDUserType, dwMDDataType, strTitle, FROM_WRITE_PROPERTY, &auth, pszInheritRoot ); // if it worked, then run the dialog if ( !dlgInherit.IsEmpty() ) dlgInherit.DoModal(); } // notice that the dwords and generic blobs are handled seperately even though // we count route the dwords through the blob mechanisms. This is done for two // reasone. 1) Handling dwords is much more efficient than handling blobs. // and 2) Most of the values are dwords. //---------------------------------------------------------------- // opens the metabase, writes out the value, then uses the inheritence // checking functionality from the iisui.dll to check for the inherited // properties and propt the user for what to do BOOL SetMetaDword(IMSAdminBase* pMB, LPCTSTR pszServer, LPCTSTR pszMetaRoot, LPCTSTR pszSub, DWORD idData, DWORD iType, DWORD dwValue, BOOL fCheckInheritence) { BOOL fAnswer = FALSE; BOOL fChanged = TRUE; DWORD dword; CWrapMetaBase mbWrap; if ( !mbWrap.FInit(pMB) ) return FALSE; // open the target if ( OpenAndCreate( &mbWrap, pszMetaRoot, METADATA_PERMISSION_WRITE | METADATA_PERMISSION_READ, TRUE ) ) { // attempt to get the current value - no inheritence if ( mbWrap.GetDword(pszSub, idData, iType, &dword, 0) ) { // set the changed flag fChanged = (dwValue != dword); } // save it out, if it changed or is not there if ( fChanged ) fAnswer = mbWrap.SetDword( pszSub, idData, iType, dwValue ); // close the metabase mbWrap.Close(); } else fChanged = FALSE; // set up and run the inheritence checking dialog if ( fCheckInheritence && fChanged ) { CString szInheritRoot = pszMetaRoot; szInheritRoot += pszSub; CheckInheritence( pszServer, szInheritRoot, idData, DWORD_METADATA, iType); } return fAnswer; } //---------------------------------------------------------------- // assumes that the metabase is actually open to the parent of the one we are interested // and that the real target name is passed into szSub BOOL SetMetaData(IMSAdminBase* pMB, LPCTSTR pszServer, LPCTSTR pszMetaRoot, LPCTSTR pszSub, DWORD idData, DWORD iType, DWORD iDataType, PVOID pData, DWORD cbData, BOOL fCheckInheritence, BOOL fSecure ) { BOOL fAnswer = FALSE; BOOL fChanged = TRUE; DWORD cbTestData; DWORD flags = METADATA_INHERIT; CWrapMetaBase mbWrap; if ( !mbWrap.FInit(pMB) ) return FALSE; // open the target if ( OpenAndCreate( &mbWrap, pszMetaRoot, METADATA_PERMISSION_WRITE | METADATA_PERMISSION_READ, TRUE ) ) { // attempt to get the current value - no inheritence PVOID pTestData = mbWrap.GetData( pszSub, idData, iType, iDataType, &cbTestData, 0 ); if ( pTestData ) { // set the changed flag if ( cbData == cbTestData ) { fChanged = (memcmp(pData, pTestData, cbData) != 0); } mbWrap.FreeWrapData( pTestData ); } // set security if requested if ( fSecure ) flags |= METADATA_SECURE; // save it out, if it changed or is not there if ( fChanged ) fAnswer = mbWrap.SetData( pszSub, idData, iType, iDataType, pData, cbData, flags ); // close the metabase mbWrap.Close(); } else fChanged = FALSE; // set up and run the inheritence checking dialog if ( fCheckInheritence && fChanged ) { CString szInheritRoot = pszMetaRoot; szInheritRoot += pszSub; CheckInheritence( pszServer, szInheritRoot, idData , iDataType, iType); } return fAnswer; } //---------------------------------------------------------------- BOOL SetMetaString(IMSAdminBase* pMB, LPCTSTR pszServer, LPCTSTR pszMetaRoot, LPCTSTR pszSub, DWORD idData, DWORD iType, CString sz, BOOL fCheckInheritence, BOOL fSecure) { return SetMetaData(pMB, pszServer, pszMetaRoot, pszSub, idData, iType, STRING_METADATA, (LPTSTR)(LPCTSTR)sz, (sz.GetLength()+1)*sizeof(TCHAR), fCheckInheritence, fSecure ); } //---------------------------------------------------------------- BOOL SetMetaMultiSz(IMSAdminBase* pMB, LPCTSTR pszServer, LPCTSTR pszMetaRoot, LPCTSTR pszSub, DWORD idData, DWORD iType, PVOID pData, DWORD cchmsz, BOOL fCheckInheritence ) { return SetMetaData(pMB, pszServer, pszMetaRoot, pszSub, idData, iType, MULTISZ_METADATA, pData, cchmsz*2, fCheckInheritence, FALSE ); } //---------------------------------------------------------------- BOOL SetMBData(CWrapMetaBase* pMB, CCheckInheritList* pInheritList, LPCTSTR pszSub, DWORD idData, DWORD iType, DWORD iDataType, PVOID pData, DWORD cbData, BOOL fSecure ) { BOOL fAnswer = FALSE; BOOL fChanged = TRUE; DWORD cbTestData; DWORD flags = METADATA_INHERIT; // attempt to get the current value - no inheritence PVOID pTestData = pMB->GetData( pszSub, idData, iType, iDataType, &cbTestData, 0 ); if ( pTestData ) { // set the changed flag if ( cbData == cbTestData ) { fChanged = (memcmp(pData, pTestData, cbData) != 0); } pMB->FreeWrapData( pTestData ); } // set security if requested if ( fSecure ) flags |= METADATA_SECURE; // save it out, if it changed or is not there if ( fChanged ) { fAnswer = pMB->SetData( pszSub, idData, iType, iDataType, pData, cbData, flags ); } // add it to the change list if ( pInheritList && fChanged && fAnswer ) { // prep the inheritence check record pInheritList->Add( idData, iDataType, iType, flags ); } return fAnswer; } //---------------------------------------------------------------- BOOL SetMBDword(CWrapMetaBase* pMB, CCheckInheritList* pInheritList, LPCTSTR pszSub, DWORD idData, DWORD iType, DWORD dwValue) { return SetMBData(pMB, pInheritList, pszSub, idData, iType, DWORD_METADATA, &dwValue, sizeof(DWORD), FALSE ); } //---------------------------------------------------------------- BOOL SetMBString(CWrapMetaBase* pMB, CCheckInheritList* pInheritList, LPCTSTR pszSub, DWORD idData, DWORD iType, CString sz, BOOL fSecure) { return SetMBData(pMB, pInheritList, pszSub, idData, iType, STRING_METADATA, (LPTSTR)(LPCTSTR)sz, (sz.GetLength()+1)*sizeof(TCHAR), fSecure ); } //---------------------------------------------------------------- BOOL SetMBMultiSz(CWrapMetaBase* pMB, CCheckInheritList* pInheritList, LPCTSTR pszSub, DWORD idData, DWORD iType, PVOID pData, DWORD cchmsz ) { return SetMBData(pMB, pInheritList, pszSub, idData, iType, MULTISZ_METADATA, pData, cchmsz*2, FALSE ); } //------------------------------------------------------------- void CCheckInheritList::CheckInheritence( LPCTSTR pszServer, LPCTSTR pszInheritRoot ) { // get the number of items to check DWORD cItems = (DWORD)rgbItems.GetSize(); // loop through the items, checking each for ( DWORD iItem = 0; iItem < cItems; iItem++ ) { // check the inheritence on the item ::CheckInheritence( pszServer, pszInheritRoot, rgbItems[iItem].dwMDIdentifier, rgbItems[iItem].dwMDDataType, rgbItems[iItem].dwMDUserType, rgbItems[iItem].dwMDAttributes ); } } //------------------------------------------------------------- INT CCheckInheritList::Add( DWORD dwMDIdentifier, DWORD dwMDDataType, DWORD dwMDUserType, DWORD dwMDAttributes ) { INHERIT_CHECK_ITEM item; item.dwMDIdentifier = dwMDIdentifier; item.dwMDDataType = dwMDDataType; item.dwMDUserType = dwMDUserType; item.dwMDAttributes = dwMDAttributes; return (INT)rgbItems.Add( item ); }