// metatool.cpp : implementation file
//
// some common tools used for "smart" writing to the metabase
#include "stdafx.h"
#define _COMSTATIC
#include <comprop.h>
#include <idlg.h>
#include <resource.h>
#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);
CInheritanceDlg dlgInherit(
TRUE, // Look in table first
dwMDIdentifier,
dwMDAttributes,
dwMDUserType,
dwMDDataType,
strTitle,
FROM_WRITE_PROPERTY,
pszServer,
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 );
}