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.
 
 
 
 
 
 

1812 lines
44 KiB

//
// compdata.cpp : Implementation of ComponentData
//
// This COM object is primarily concerned with
// the scope pane items.
//
// Cory West <[email protected]>
// Copyright (c) Microsoft Corporation 1997
//
#include "stdafx.h"
#include "macros.h"
USE_HANDLE_MACROS("SCHMMGMT(compdata.cpp)")
#include "dataobj.h"
#include "compdata.h"
#include "cookie.h"
#include "snapmgr.h"
#include "schmutil.h"
#include "cache.h"
#include "relation.h"
#include "attrpage.h"
#include "advui.h"
#include "aclpage.h"
#include "select.h"
#include "classgen.hpp"
#include "newclass.hpp"
#include "newattr.hpp"
//
// CComponentData implementation.
//
#include "stdcdata.cpp"
//
// ComponentData
//
ComponentData::ComponentData()
:
m_pRootCookie( NULL ),
m_pPathname( NULL ),
m_hItem( NULL )
{
//
// We must refcount the root cookie, since a dataobject for it
// might outlive the IComponentData. JonN 9/2/97
//
m_pRootCookie = new Cookie( SCHMMGMT_SCHMMGMT );
ASSERT(NULL != m_pRootCookie);
m_pRootCookie->AddRef();
g_SchemaCache.SetScopeControl( this );
SetHtmlHelpFileName (L"schmmgmt.chm");
SetCanChangeOperationsMaster( );
SetCanCreateClass( );
SetCanCreateAttribute( );
}
ComponentData::~ComponentData()
{
SAFE_RELEASE(m_pRootCookie);
SAFE_RELEASE(m_pPathname);
}
DEFINE_FORWARDS_MACHINE_NAME( ComponentData, m_pRootCookie )
CCookie&
ComponentData::QueryBaseRootCookie()
{
ASSERT(NULL != m_pRootCookie);
return (CCookie&)*m_pRootCookie;
}
STDMETHODIMP
ComponentData::Initialize( LPUNKNOWN pUnknown )
{
HRESULT hr = CComponentData::Initialize( pUnknown );
if( SUCCEEDED(hr) )
{
ASSERT( !m_pPathname );
// create Pathname Object
if( FAILED( CoCreateInstance(CLSID_Pathname,
NULL,
CLSCTX_INPROC_SERVER,
IID_IADsPathname,
(void**)&m_pPathname) ) )
{
// in case of an error, ignore and later provide no escaping.
ASSERT( FALSE );
SAFE_RELEASE( m_pPathname );
}
}
return hr;
}
STDMETHODIMP
ComponentData::CreateComponent( LPCOMPONENT* ppComponent )
{
MFC_TRY;
ASSERT(ppComponent != NULL);
CComObject<Component>* pObject;
CComObject<Component>::CreateInstance(&pObject);
ASSERT(pObject != NULL);
pObject->SetComponentDataPtr( (ComponentData*)this );
return pObject->QueryInterface( IID_IComponent,
reinterpret_cast<void**>(ppComponent) );
MFC_CATCH;
}
HRESULT
ComponentData::LoadIcons(
LPIMAGELIST pImageList,
BOOL
)
/***
This routine loads the icon resources that MMC will use.
We use the image list member ImageListSetIcon to make these
resources available to MMC.
***/
{
HICON hIcon;
HRESULT hr = S_OK;
if( !IsErrorSet() )
{
//
// Set the generic and the last icon in case they are used.
//
hIcon = LoadIcon(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDI_GENERIC));
ASSERT(hIcon != NULL);
hr = pImageList->ImageListSetIcon((PLONG_PTR)hIcon,iIconGeneric);
ASSERT(SUCCEEDED(hr));
hr = pImageList->ImageListSetIcon((PLONG_PTR)hIcon,iIconLast);
ASSERT(SUCCEEDED(hr));
//
// Set the closed folder icon.
//
hIcon = LoadIcon(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDI_FOLDER_CLOSED));
ASSERT(hIcon != NULL);
hr = pImageList->ImageListSetIcon((PLONG_PTR)hIcon,iIconFolder);
ASSERT(SUCCEEDED(hr));
//
// Set the class, attribute, and display specifier icon.
//
hIcon = LoadIcon(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDI_CLASS));
ASSERT(hIcon != NULL);
hr = pImageList->ImageListSetIcon((PLONG_PTR)hIcon,iIconClass);
ASSERT(SUCCEEDED(hr));
hIcon = LoadIcon(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDI_ATTRIBUTE));
ASSERT(hIcon != NULL);
hr = pImageList->ImageListSetIcon((PLONG_PTR)hIcon,iIconAttribute);
ASSERT(SUCCEEDED(hr));
}
return S_OK;
}
HRESULT
ComponentData::OnNotifyExpand(
LPDATAOBJECT lpDataObject,
BOOL bExpanding,
HSCOPEITEM hParent
)
/***
This routine is called in response to IComponentData:Notify with
the MMCN_EXPAND notification. The argument bExpanding tells us
whether the node is expanding or contracting.
***/
{
ASSERT( NULL != lpDataObject &&
NULL != hParent &&
NULL != m_pConsoleNameSpace );
//
// Do nothing on a contraction - we will get notifications to
// destroy the disappearing nodes.
//
if ( !bExpanding )
return S_OK;
//
// This code will not work if SchmMgmt becomes an extension
// since the RawCookie format will not be available.
//
CCookie* pBaseParentCookie = NULL;
HRESULT hr = ExtractData( lpDataObject,
CSchmMgmtDataObject::m_CFRawCookie,
reinterpret_cast<PBYTE>(&pBaseParentCookie),
sizeof(pBaseParentCookie) );
ASSERT( SUCCEEDED(hr) );
Cookie* pParentCookie = ActiveCookie(pBaseParentCookie);
ASSERT( NULL != pParentCookie );
//
// If this node already has children then this expansion is a
// re-expansion. We should complain, but not do anything.
//
if ( !((pParentCookie->m_listScopeCookieBlocks).IsEmpty()) ) {
ASSERT(FALSE);
return S_OK;
}
switch ( pParentCookie->m_objecttype ) {
case SCHMMGMT_SCHMMGMT:
// expanding the root, need to bind
hr = _InitBasePaths();
if( SUCCEEDED(hr) )
{
CheckSchemaPermissions( ); // ignoring errors
}
else
{
SetError( IDS_ERR_ERROR, IDS_ERR_NO_SCHEMA_PATH );
SetDelayedRefreshOnShow( hParent );
return S_OK;
}
InitializeRootTree( hParent, pParentCookie );
break;
case SCHMMGMT_CLASSES:
return FastInsertClassScopeCookies(
pParentCookie,
hParent );
break;
//
// These node types have no scope children.
//
case SCHMMGMT_CLASS:
case SCHMMGMT_ATTRIBUTE:
case SCHMMGMT_ATTRIBUTES:
return S_OK;
//
// We received an unknown node type.
//
default:
TRACE( "ComponentData::EnumerateScopeChildren bad parent type.\n" );
ASSERT( FALSE );
return S_OK;
}
return S_OK;
}
HRESULT
ComponentData::OnNotifyRelease(
LPDATAOBJECT,
HSCOPEITEM
) {
//
// Since we are not a valid extension snap in,
// we don't need to do anything here.
//
return S_OK;
}
HRESULT
ComponentData::OnNotifyDelete(
LPDATAOBJECT lpDataObject)
{
CCookie* pBaseParentCookie = NULL;
HRESULT hr = ExtractData( lpDataObject,
CSchmMgmtDataObject::m_CFRawCookie,
reinterpret_cast<PBYTE>(&pBaseParentCookie),
sizeof(pBaseParentCookie) );
ASSERT( SUCCEEDED(hr) );
Cookie* pParentCookie = ActiveCookie(pBaseParentCookie);
ASSERT( NULL != pParentCookie );
UINT promptID = 0;
LPARAM updateType = SCHMMGMT_CLASS;
if (pParentCookie->m_objecttype == SCHMMGMT_CLASS)
{
promptID = IDS_DELETE_CLASS_PROMPT;
updateType = SCHMMGMT_CLASS;
}
else if (pParentCookie->m_objecttype == SCHMMGMT_ATTRIBUTE)
{
promptID = IDS_DELETE_ATTR_PROMPT;
updateType = SCHMMGMT_ATTRIBUTES;
}
else
{
// We should never get called to delete anything but
// class and attribute nodes
ASSERT(FALSE);
return E_FAIL;
}
if( IDYES == AfxMessageBox( promptID, MB_YESNO | MB_ICONWARNING ))
{
hr = DeleteClass( pParentCookie );
if ( SUCCEEDED(hr) )
{
// Remove the node from the UI
m_pConsoleNameSpace->DeleteItem( pParentCookie->m_hScopeItem, TRUE );
// Remove the node from the list
bool result = g_ClassCookieList.DeleteCookie(pParentCookie);
ASSERT(result);
}
else
{
CString szDeleteError;
szDeleteError.Format(IDS_ERRMSG_DELETE_FAILED_CLASS, GetErrorMessage(hr, TRUE));
DoErrMsgBox( ::GetActiveWindow(), TRUE, szDeleteError );
}
}
return hr;
}
HRESULT
ComponentData::DeleteClass(
Cookie* pcookie
)
/***
This deletes an attribute from the schema
***/
{
HRESULT hr = S_OK;
do
{
if ( !pcookie )
{
hr = E_INVALIDARG;
break;
}
SchemaObject* pObject = g_SchemaCache.LookupSchemaObjectByCN(
pcookie->strSchemaObject,
SCHMMGMT_CLASS );
if ( !pObject )
{
hr = E_FAIL;
break;
}
CString szAdsPath;
GetSchemaObjectPath( pObject->commonName, szAdsPath );
hr = DeleteObject( szAdsPath, pcookie, g_ClassFilter );
} while (false);
return hr;
}
//
// This is where we store the string handed back to GetDisplayInfo().
//
CString g_strResultColumnText;
BSTR
ComponentData::QueryResultColumnText(
CCookie& basecookieref,
int nCol
) {
#ifndef UNICODE
#error not ANSI-enabled
#endif
BSTR strDisplayText = NULL;
Cookie& cookieref = (Cookie&)basecookieref;
SchemaObject *pSchemaObject = NULL;
SchemaObject *pSrcSchemaObject = NULL;
switch ( cookieref.m_objecttype ) {
//
// These only have first column textual data.
//
case SCHMMGMT_SCHMMGMT:
if ( COLNUM_SCHEMA_NAME == nCol )
strDisplayText = const_cast<BSTR>(((LPCTSTR)g_strSchmMgmt));
break;
case SCHMMGMT_CLASSES:
if ( COLNUM_SCHEMA_NAME == nCol )
strDisplayText = const_cast<BSTR>(((LPCTSTR)g_strClasses));
break;
case SCHMMGMT_ATTRIBUTES:
if ( COLNUM_SCHEMA_NAME == nCol )
strDisplayText = const_cast<BSTR>(((LPCTSTR)g_strAttributes));
break;
case SCHMMGMT_CLASS:
//
// These display names come out of the schema cache objects.
//
pSchemaObject = g_SchemaCache.LookupSchemaObjectByCN(
cookieref.strSchemaObject,
SCHMMGMT_CLASS );
//
// If there is no cache object, we just have to return UNKNOWN.
//
if ( !pSchemaObject ) {
ASSERT(FALSE);
strDisplayText = const_cast<BSTR>( (LPCTSTR)g_Unknown );
break;
}
//
// Otherwise, return the appropriate text for the column.
//
if ( COLNUM_CLASS_NAME == nCol ) {
strDisplayText = const_cast<BSTR>((LPCTSTR)pSchemaObject->ldapDisplayName);
} else if ( COLNUM_CLASS_TYPE == nCol ) {
switch ( pSchemaObject->dwClassType ) {
case 0:
strDisplayText = const_cast<BSTR>(((LPCTSTR)g_88Class));
break;
case 1:
strDisplayText = const_cast<BSTR>(((LPCTSTR)g_StructuralClass));
break;
case 2:
strDisplayText = const_cast<BSTR>(((LPCTSTR)g_AbstractClass));
break;
case 3:
strDisplayText = const_cast<BSTR>(((LPCTSTR)g_AuxClass));
break;
default:
strDisplayText = const_cast<BSTR>(((LPCTSTR)g_Unknown));
break;
}
} else if ( COLNUM_CLASS_STATUS == nCol ) {
if ( pSchemaObject->isDefunct ) {
strDisplayText = const_cast<BSTR>( (LPCTSTR)g_Defunct );
} else {
strDisplayText = const_cast<BSTR>( (LPCTSTR)g_Active );
}
} else if ( COLNUM_CLASS_DESCRIPTION == nCol ) {
strDisplayText = const_cast<BSTR>((LPCTSTR)pSchemaObject->description);
}
break;
case SCHMMGMT_ATTRIBUTE:
pSchemaObject = g_SchemaCache.LookupSchemaObjectByCN(
cookieref.strSchemaObject,
SCHMMGMT_ATTRIBUTE );
//
// If there is no cache object, we just have to return UNKNOWN.
//
if ( !pSchemaObject ) {
ASSERT(FALSE);
strDisplayText = const_cast<BSTR>( (LPCTSTR)g_Unknown );
break;
}
//
// Otherwise, return the appropriate text for the column.
//
if ( COLNUM_ATTRIBUTE_NAME == nCol ) {
strDisplayText = const_cast<BSTR>((LPCTSTR)pSchemaObject->ldapDisplayName);
} else if ( COLNUM_ATTRIBUTE_TYPE == nCol ) {
//
// If the parent cookie is the attributes folder, then
// this column is the syntax. Otherwise, this column
// is the optional/mandatory status of the attribute.
//
if ( (cookieref.pParentCookie)->m_objecttype == SCHMMGMT_ATTRIBUTES ) {
strDisplayText = const_cast<BSTR>(
(LPCTSTR)g_Syntax[pSchemaObject->SyntaxOrdinal].m_strSyntaxName
);
} else {
if ( cookieref.Mandatory ) {
strDisplayText = const_cast<BSTR>(((LPCTSTR)g_MandatoryAttribute));
} else {
strDisplayText = const_cast<BSTR>(((LPCTSTR)g_OptionalAttribute));
}
}
} else if ( COLNUM_ATTRIBUTE_STATUS == nCol) {
if ( pSchemaObject->isDefunct ) {
strDisplayText = const_cast<BSTR>( (LPCTSTR)g_Defunct );
} else {
strDisplayText = const_cast<BSTR>( (LPCTSTR)g_Active );
}
} else if ( COLNUM_ATTRIBUTE_SYSTEM == nCol ) {
//
// If the parent is the attributes folder, this is the description.
// Otherwise, it's the system status.
//
if ( (cookieref.pParentCookie)->m_objecttype == SCHMMGMT_ATTRIBUTES ) {
strDisplayText = const_cast<BSTR>((LPCTSTR)pSchemaObject->description);
} else {
if ( cookieref.System ) {
strDisplayText = const_cast<BSTR>(((LPCTSTR)g_Yes));
} else {
strDisplayText = const_cast<BSTR>(((LPCTSTR)g_No));
}
}
} else if ( COLNUM_ATTRIBUTE_DESCRIPTION == nCol ) {
strDisplayText = const_cast<BSTR>((LPCTSTR)pSchemaObject->description);
} else if ( COLNUM_ATTRIBUTE_PARENT == nCol ) {
pSrcSchemaObject = g_SchemaCache.LookupSchemaObjectByCN(
cookieref.strSrcSchemaObject,
SCHMMGMT_CLASS );
//
// If there is no cache object, we just have to return UNKNOWN.
//
if ( !pSchemaObject ) {
ASSERT(FALSE);
strDisplayText = const_cast<BSTR>( (LPCTSTR)g_Unknown );
break;
}
//
// Otherwise, return the appropriate text for the column.
//
strDisplayText = const_cast<BSTR>((LPCTSTR)pSrcSchemaObject->ldapDisplayName);
}
break;
default:
TRACE( "ComponentData::QueryResultColumnText bad cookie.\n" );
ASSERT( FALSE );
break;
}
//
// Release the schema cache references.
//
if ( pSchemaObject ) {
g_SchemaCache.ReleaseRef( pSchemaObject );
}
if ( pSrcSchemaObject ) {
g_SchemaCache.ReleaseRef( pSrcSchemaObject );
}
//
// Return the appropriate display string.
//
if ( strDisplayText ) {
return strDisplayText;
}
return L"";
}
//
// Given a cookie, this returns the icon to display for that cookie.
//
int
ComponentData::QueryImage(
CCookie& basecookieref,
BOOL )
{
Cookie& cookieref = (Cookie&)basecookieref;
switch ( cookieref.m_objecttype ) {
case SCHMMGMT_SCHMMGMT:
case SCHMMGMT_CLASSES:
case SCHMMGMT_ATTRIBUTES:
return iIconFolder;
break;
case SCHMMGMT_CLASS:
return iIconClass;
break;
case SCHMMGMT_ATTRIBUTE:
return iIconAttribute;
break;
default:
TRACE( "ComponentData::QueryImage bad parent type.\n" );
ASSERT( FALSE );
break;
}
return iIconGeneric;
}
//
// This routines tells MMC whether or not there are
// property pages and menus for this item.
//
HRESULT
_InsertMenuHelper(
LPCONTEXTMENUCALLBACK piCallback,
long lInsertionPointID,
int index,
BOOL fEnabled = TRUE );
HRESULT
_InsertMenuHelper(
LPCONTEXTMENUCALLBACK piCallback,
long lInsertionPointID,
int index,
BOOL fEnabled /* = TRUE */ )
{
CONTEXTMENUITEM MenuItem;
MenuItem.lInsertionPointID = lInsertionPointID;
MenuItem.fFlags = fEnabled ? 0 : MF_GRAYED ;
MenuItem.fSpecialFlags = 0;
MenuItem.strName = const_cast<BSTR>(
(LPCTSTR)g_MenuStrings[index] );
MenuItem.strStatusBarText = const_cast<BSTR>(
(LPCTSTR)g_StatusStrings[index] );
MenuItem.lCommandID = index;
return piCallback->AddItem( &MenuItem );
}
STDMETHODIMP
ComponentData::AddMenuItems(
LPDATAOBJECT piDataObject,
LPCONTEXTMENUCALLBACK piCallback,
long* )
{
CCookie* pBaseParentCookie = NULL;
HRESULT hr = ExtractData( piDataObject,
CSchmMgmtDataObject::m_CFRawCookie,
reinterpret_cast<PBYTE>(&pBaseParentCookie),
sizeof(pBaseParentCookie) );
ASSERT( SUCCEEDED(hr) );
Cookie* pParentCookie = ActiveCookie(pBaseParentCookie);
ASSERT( NULL != pParentCookie );
AFX_MANAGE_STATE(AfxGetStaticModuleState());
LoadGlobalCookieStrings();
switch (pParentCookie->m_objecttype)
{
case SCHMMGMT_SCHMMGMT: // Root Folder
{
VERIFY(
SUCCEEDED(
_InsertMenuHelper(
piCallback,
CCM_INSERTIONPOINTID_PRIMARY_TOP,
SCHEMA_RETARGET)));
VERIFY(
SUCCEEDED(
_InsertMenuHelper(
piCallback,
CCM_INSERTIONPOINTID_PRIMARY_TOP,
SCHEMA_EDIT_FSMO,
!IsErrorSet() && IsSchemaLoaded())));
VERIFY(
SUCCEEDED(
_InsertMenuHelper(
piCallback,
CCM_INSERTIONPOINTID_PRIMARY_TOP,
SCHEMA_SECURITY,
!IsErrorSet() && IsSchemaLoaded())));
VERIFY(
SUCCEEDED(
_InsertMenuHelper(
piCallback,
CCM_INSERTIONPOINTID_PRIMARY_TOP,
SCHEMA_REFRESH,
!IsErrorSet() && IsSchemaLoaded())));
break;
}
case SCHMMGMT_CLASSES: // classes folder
{
// 285448, 293449
VERIFY(
SUCCEEDED(
_InsertMenuHelper(
piCallback,
CCM_INSERTIONPOINTID_PRIMARY_TOP,
NEW_CLASS,
!IsErrorSet() && CanCreateClass())));
if( !IsErrorSet() && CanCreateClass() ) // add only if enabsed
VERIFY(
SUCCEEDED(
_InsertMenuHelper(
piCallback,
CCM_INSERTIONPOINTID_PRIMARY_NEW,
CLASSES_CREATE_CLASS)));
break;
}
case SCHMMGMT_ATTRIBUTES: // attributes folder
{
VERIFY(
SUCCEEDED(
_InsertMenuHelper(
piCallback,
CCM_INSERTIONPOINTID_PRIMARY_TOP,
NEW_ATTRIBUTE,
!IsErrorSet() && CanCreateAttribute())));
if( !IsErrorSet() && CanCreateAttribute() ) // add only if enabsed
VERIFY(
SUCCEEDED(
_InsertMenuHelper(
piCallback,
CCM_INSERTIONPOINTID_PRIMARY_NEW,
ATTRIBUTES_CREATE_ATTRIBUTE)));
break;
}
default:
{
// could be class or attribute item
}
} // switch
return S_OK;
}
STDMETHODIMP
ComponentData::Command(long lCommandID, LPDATAOBJECT piDataObject)
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
switch ( lCommandID )
{
case NEW_ATTRIBUTE:
case ATTRIBUTES_CREATE_ATTRIBUTE:
{
CDialog warn(IDD_CREATE_WARN);
if (IDOK == warn.DoModal())
{
CreateAttributeDialog(this, piDataObject).DoModal();
}
break;
}
case NEW_CLASS:
case CLASSES_CREATE_CLASS:
{
CDialog warn(IDD_CREATE_WARN);
if (IDOK == warn.DoModal())
{
DoNewClassDialog(*this);
}
break;
}
case SCHEMA_RETARGET:
_OnRetarget(piDataObject);
break;
case SCHEMA_EDIT_FSMO:
_OnEditFSMO();
break;
case SCHEMA_REFRESH:
if( FAILED(_OnRefresh(piDataObject)) )
DoErrMsgBox(::GetActiveWindow(), TRUE, IDS_ERR_NO_UPDATE);
break;
case SCHEMA_SECURITY:
_OnSecurity();
break;
default:
break;
}
return S_OK;
}
STDMETHODIMP
ComponentData::QueryPagesFor(
LPDATAOBJECT pDataObject )
{
MFC_TRY;
if ( NULL == pDataObject ) {
ASSERT(FALSE);
return E_POINTER;
}
HRESULT hr;
CCookie* pBaseParentCookie = NULL;
hr = ExtractData( pDataObject,
CSchmMgmtDataObject::m_CFRawCookie,
reinterpret_cast<PBYTE>(&pBaseParentCookie),
sizeof(pBaseParentCookie) );
ASSERT( SUCCEEDED(hr) );
Cookie* pParentCookie = ActiveCookie(pBaseParentCookie);
ASSERT( NULL != pParentCookie );
if ( pParentCookie->m_objecttype == SCHMMGMT_CLASS ) {
return S_OK;
}
return S_FALSE;
MFC_CATCH;
}
//
// This adds pages to the property sheet if appropriate.
// The handle parameter must be saved in the property page
// object to notify the parent when modified.
//
STDMETHODIMP
ComponentData::CreatePropertyPages(
LPPROPERTYSHEETCALLBACK pCallBack,
LONG_PTR,
LPDATAOBJECT pDataObject )
{
MFC_TRY;
CWaitCursor wait;
//
// Validate the parameters.
//
if ( ( NULL == pCallBack ) ||
( NULL == pDataObject ) ) {
ASSERT(FALSE);
return E_POINTER;
}
//
// Make sure this is a class object that we are calling up.
//
CCookie* pBaseParentCookie = NULL;
HRESULT hr = ExtractData( pDataObject,
CSchmMgmtDataObject::m_CFRawCookie,
reinterpret_cast<PBYTE>(&pBaseParentCookie),
sizeof(pBaseParentCookie) );
ASSERT( SUCCEEDED(hr) );
hr = S_OK;
Cookie* pParentCookie = ActiveCookie(pBaseParentCookie);
ASSERT( NULL != pParentCookie );
ASSERT( pParentCookie->m_objecttype == SCHMMGMT_CLASS );
//
// Create the page.
//
HPROPSHEETPAGE hPage;
ClassGeneralPage *pGeneralPage = new ClassGeneralPage( this );
ClassRelationshipPage *pRelationshipPage =
new ClassRelationshipPage( this, pDataObject );
ClassAttributePage *pAttributesPage =
new ClassAttributePage( this, pDataObject );
if ( pGeneralPage ) {
pGeneralPage->Load( *pParentCookie );
MMCPropPageCallback( &pGeneralPage->m_psp );
hPage = CreatePropertySheetPage( &pGeneralPage->m_psp );
pCallBack->AddPage(hPage);
}
if ( pRelationshipPage ) {
pRelationshipPage->Load( *pParentCookie );
MMCPropPageCallback( &pRelationshipPage->m_psp );
hPage = CreatePropertySheetPage( &pRelationshipPage->m_psp );
pCallBack->AddPage(hPage);
}
if ( pAttributesPage ) {
pAttributesPage->Load( *pParentCookie );
MMCPropPageCallback( &pAttributesPage->m_psp );
hPage = CreatePropertySheetPage( &pAttributesPage->m_psp );
pCallBack->AddPage(hPage);
}
//
// Add the ACL editor page.
//
SchemaObject * pObject = NULL;
CAclEditorPage * pAclPage = NULL;
CString szAdsPath;
pObject = g_SchemaCache.LookupSchemaObjectByCN(
pParentCookie->strSchemaObject,
SCHMMGMT_CLASS );
if ( pObject ) {
GetSchemaObjectPath( pObject->commonName, szAdsPath );
if ( !szAdsPath.IsEmpty() ) {
hr = CAclEditorPage::CreateInstance( &pAclPage, szAdsPath,
pParentCookie->strSchemaObject );
if ( SUCCEEDED(hr) )
{
ASSERT( pAclPage );
pCallBack->AddPage( pAclPage->CreatePage() );
}
else
{
DoErrMsgBox( ::GetActiveWindow(), TRUE, GetErrorMessage(hr) );
hr = S_FALSE; // tell mmc to cancel "show prop pages"
}
}
}
return hr;
MFC_CATCH;
}
HRESULT
ComponentData::FastInsertClassScopeCookies(
Cookie* pParentCookie,
HSCOPEITEM hParentScopeItem
)
/***
On an expand for the "Classes" node, this inserts the
class scope items from the cache.
pParentCookie is the cookie for the parent object.
hParentScopeItem is the HSCOPEITEM for the parent.
****/
{
HRESULT hr;
SCOPEDATAITEM ScopeItem;
Cookie* pNewCookie;
LPCWSTR lpcszMachineName = pParentCookie->QueryNonNULLMachineName();
SchemaObject *pObject, *pHead;
//
// Initialize the scope item.
//
::ZeroMemory( &ScopeItem, sizeof(ScopeItem) );
ScopeItem.mask =
SDI_STR
| SDI_IMAGE
| SDI_OPENIMAGE
| SDI_STATE
| SDI_PARAM
| SDI_PARENT
| SDI_CHILDREN;
ScopeItem.displayname = MMC_CALLBACK;
ScopeItem.relativeID = hParentScopeItem;
ScopeItem.nState = TVIS_EXPANDED;
ScopeItem.cChildren = 0;
//
// Remember the parent cookie and scope item; we
// may need to insert later as a refresh.
//
g_ClassCookieList.pParentCookie = pParentCookie;
g_ClassCookieList.hParentScopeItem = hParentScopeItem;
//
// Rather than having a clean class interface to the cache, we
// walk the cache data structures ourselves. This isn't super
// clean, but it's simple.
//
// Since we do this, we have to make sure that the cache is loaded.
//
// This is just like in
// Component::FastInsertAttributeResultCookies
//
g_SchemaCache.LoadCache();
pObject = g_SchemaCache.pSortedClasses;
//
// If there's no sorted list, we can't insert anything!!!!
// We must return an error or else the console will never
// ask us again for scope items.
//
if ( !pObject ) {
ASSERT( FALSE );
// @@ spb: bad message, this could occur if the schema
// queries were empty...
DoErrMsgBox( ::GetActiveWindow(), TRUE, IDS_ERR_NO_SCHEMA_PATH );
return E_FAIL;
}
//
// Do the insert.
//
pHead = pObject;
do {
//
// Insert this result.
//
pNewCookie= new Cookie( SCHMMGMT_CLASS,
lpcszMachineName );
if ( pNewCookie ) {
pNewCookie->pParentCookie = pParentCookie;
pNewCookie->strSchemaObject = pObject->commonName;
pParentCookie->m_listScopeCookieBlocks.AddHead(
(CBaseCookieBlock*)pNewCookie
);
ScopeItem.lParam = reinterpret_cast<LPARAM>((CCookie*)pNewCookie);
ScopeItem.nImage = QueryImage( *pNewCookie, FALSE );
ScopeItem.nOpenImage = QueryImage( *pNewCookie, TRUE );
hr = m_pConsoleNameSpace->InsertItem(&ScopeItem);
if ( SUCCEEDED(hr) ) {
pNewCookie->m_hScopeItem = ScopeItem.ID;
g_ClassCookieList.AddCookie( pNewCookie, ScopeItem.ID );
} else {
delete pNewCookie;
}
}
pObject = pObject->pSortedListFlink;
} while ( pObject != pHead );
return S_OK;
}
//
// Refreshing the scope pane view.
//
VOID
ComponentData::RefreshScopeView(
VOID
)
/***
When we reload the schema cache and the "Classes"
folder has been opened, this routine deletes all
the scope items from view and re-inserts them.
This makes new classes visible to the user.
***/
{
HRESULT hr;
CCookieListEntry *pHead = g_ClassCookieList.pHead;
CCookieListEntry *pCurrent;
if ( pHead != NULL ) {
//
// Remove all the scope pane items.
//
pCurrent = pHead;
while ( pCurrent->pNext != pHead ) {
hr = m_pConsoleNameSpace->DeleteItem( pCurrent->pNext->hScopeItem, TRUE );
ASSERT( SUCCEEDED( hr ));
//
// This should cause the cookie to be deleted.
//
pCurrent->pNext->pCookie->Release();
pCurrent = pCurrent->pNext;
}
//
// Remove the head of the list.
//
hr = m_pConsoleNameSpace->DeleteItem( pHead->hScopeItem, TRUE );
ASSERT( SUCCEEDED( hr ));
pHead->pCookie->Release();
//
// Delete the class cookie list.
//
g_ClassCookieList.DeleteAll();
}
//
// Re-insert them from the cache if this node has
// been expanded at some point. We have to do this
// because the console doesn't ever seem to ask
// again.
//
if ( g_ClassCookieList.pParentCookie ) {
FastInsertClassScopeCookies(
g_ClassCookieList.pParentCookie,
g_ClassCookieList.hParentScopeItem
);
}
return;
}
void ComponentData::_OnRetarget(LPDATAOBJECT lpDataObject)
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
MyBasePathsInfo oldBasePathsInfo;
MyBasePathsInfo newBasePathsInfo;
HRESULT hr = S_OK;
HWND hWndParent = NULL;
BOOL fWasErrorSet = IsErrorSet();
m_pConsole->GetMainWindow(&hWndParent);
CChangeDCDialog dlg(GetBasePathsInfo(), hWndParent);
do
{
if (IDOK != dlg.DoModal())
break;
CWaitCursor wait;
// save the old path just in case
oldBasePathsInfo.InitFromInfo( GetBasePathsInfo() );
// attempt to bind
hr = newBasePathsInfo.InitFromName(dlg.GetNewDCName());
BREAK_ON_FAILED_HRESULT(hr);
// switch focus
GetBasePathsInfo()->InitFromInfo(&newBasePathsInfo);
// invalidate the whole tree
hr = _OnRefresh(lpDataObject);
BREAK_ON_FAILED_HRESULT(hr);
SetError( 0, 0 );
// Reset the ResultViewType
if( IsErrorSet() != fWasErrorSet )
{
ASSERT( SCHMMGMT_SCHMMGMT == QueryRootCookie().m_objecttype );
hr = m_pConsole->SelectScopeItem( QueryRootCookie().m_hScopeItem );
ASSERT_BREAK_ON_FAILED_HRESULT(hr);
//
// Add children nodes if needed
//
if ( (QueryRootCookie().m_listScopeCookieBlocks).IsEmpty() )
{
InitializeRootTree( QueryRootCookie().m_hScopeItem, &QueryRootCookie() );
}
}
// Update the root display name
if (GetBasePathsInfo()->GetServerName())
{
CString strDisplayName;
strDisplayName.LoadString(IDS_SCOPE_SCHMMGMT);
strDisplayName += L" [";
strDisplayName += GetBasePathsInfo()->GetServerName();
strDisplayName += L"]";
SCOPEDATAITEM RootItem;
::ZeroMemory( &RootItem, sizeof(RootItem));
RootItem.mask = SDI_STR | SDI_PARAM;
RootItem.displayname = const_cast<PWSTR>((PCWSTR)strDisplayName);
RootItem.ID = QueryRootCookie().m_hScopeItem;
hr = m_pConsoleNameSpace->SetItem(&RootItem);
ASSERT(SUCCEEDED(hr));
}
} while( FALSE );
if( FAILED(hr) )
{
DoErrMsgBox(::GetActiveWindow(), TRUE, IDS_ERR_CANT_RETARGET);
// restoring...
GetBasePathsInfo()->InitFromInfo(&oldBasePathsInfo);
}
}
void ComponentData::_OnEditFSMO()
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
HWND hWndParent;
// Enable hourglass
CWaitCursor wait;
m_pConsole->GetMainWindow(&hWndParent);
CComPtr<IDisplayHelp> spIDisplayHelp;
m_pConsole->QueryInterface (IID_IDisplayHelp, (void **)&spIDisplayHelp);
ASSERT(spIDisplayHelp != NULL);
CEditFsmoDialog dlg(GetBasePathsInfo(), hWndParent, spIDisplayHelp, CanChangeOperationsMaster() );
dlg.DoModal();
}
HRESULT ComponentData::_OnRefresh(LPDATAOBJECT lpDataObject)
{
CWaitCursor wait;
HRESULT hr = S_OK;
do
{
//
// Force the ds to update the schema cache.
//
hr = ForceDsSchemaCacheUpdate();
// error here means Schema is read-only, cannot force refresh. Ignoring...
hr = S_OK;
// nothing shuld fail after this point
g_SchemaCache.FreeAll();
g_SchemaCache.LoadCache();
RefreshScopeView();
m_pConsole->UpdateAllViews(
lpDataObject,
SCHMMGMT_SCHMMGMT,
SCHMMGMT_UPDATEVIEW_REFRESH);
} while( FALSE );
return hr;
}
void ComponentData::_OnSecurity()
{
HRESULT hr = S_OK;
HWND hWndParent = NULL;
CString szSchemaPath;
do
{
// Enable hourglass
CWaitCursor wait;
CComPtr<IADs> spSchemaContainer;
GetBasePathsInfo()->GetSchemaPath(szSchemaPath);
hr = ADsGetObject( (LPWSTR)(LPCWSTR)szSchemaPath,
IID_IADs,
(void**) &spSchemaContainer);
BREAK_ON_FAILED_HRESULT(hr);
CComBSTR path;
hr = spSchemaContainer->get_ADsPath(&path);
BREAK_ON_FAILED_HRESULT(hr);
CComBSTR classname;
hr = spSchemaContainer->get_Class(&classname);
BREAK_ON_FAILED_HRESULT(hr);
m_pConsole->GetMainWindow(&hWndParent);
// Determine if the registry is accessible & schema modifications are allowed
PWSTR pszFsmoOwnerServerName = 0;
MyBasePathsInfo fsmoOwnerInfo;
HRESULT hr = FindFsmoOwner(GetBasePathsInfo(), SCHEMA_FSMO, &fsmoOwnerInfo, &pszFsmoOwnerServerName);
// If we have the server name, try to read the registry
BOOL fMarkReadOnly = ( FAILED(hr) || pszFsmoOwnerServerName == 0 || pszFsmoOwnerServerName[0] == 0);
// Ignore hr code.
hr =
DSEditSecurity(
hWndParent,
path,
classname,
fMarkReadOnly ? DSSI_READ_ONLY : 0,
NULL,
NULL,
NULL,
0);
}
while (0);
if (FAILED(hr))
{
m_pConsole->GetMainWindow(&hWndParent);
if( szSchemaPath.IsEmpty() )
DoErrMsgBox( hWndParent, TRUE, IDS_ERR_NO_SCHEMA_PATH );
else
DoErrMsgBox( hWndParent, TRUE, GetErrorMessage(hr,TRUE) );
}
}
HRESULT ComponentData::_InitBasePaths()
{
CWaitCursor wait;
// try to get a bind to a generic DC
HRESULT hr = GetBasePathsInfo()->InitFromName(NULL);
if (FAILED(hr))
return hr; // ADSI failed, cannot get any server
// from the current server, try to bind to the schema FSMO owner
MyBasePathsInfo fsmoBasePathsInfo;
PWSTR pszFsmoOwnerServerName = 0;
hr = FindFsmoOwner(GetBasePathsInfo(), SCHEMA_FSMO, &fsmoBasePathsInfo,
&pszFsmoOwnerServerName);
delete[] pszFsmoOwnerServerName;
pszFsmoOwnerServerName = 0;
if (FAILED(hr))
return S_OK; // still good keep what we have (even though not the FSMO owner)
// got it, we switch focus
return GetBasePathsInfo()->InitFromInfo(&fsmoBasePathsInfo);
}
STDMETHODIMP ComponentData::GetLinkedTopics(LPOLESTR* lpCompiledHelpFile)
{
if (lpCompiledHelpFile == NULL)
return E_INVALIDARG;
CString szHelpFilePath;
LPTSTR lpszBuffer = szHelpFilePath.GetBuffer(2*MAX_PATH);
UINT nLen = ::GetSystemWindowsDirectory(lpszBuffer, 2*MAX_PATH);
if (nLen == 0)
return E_FAIL;
szHelpFilePath.ReleaseBuffer();
szHelpFilePath += L"\\help\\ADconcepts.chm";
UINT nBytes = (szHelpFilePath.GetLength()+1) * sizeof(WCHAR);
*lpCompiledHelpFile = (LPOLESTR)::CoTaskMemAlloc(nBytes);
if (*lpCompiledHelpFile != NULL)
{
memcpy(*lpCompiledHelpFile, (LPCWSTR)szHelpFilePath, nBytes);
}
return S_OK;
}
const WCHAR CN_EQUALS[] = L"CN=";
HRESULT ComponentData::GetSchemaObjectPath( const CString & strCN,
CString & strPath,
ADS_FORMAT_ENUM formatType /* = ADS_FORMAT_X500 */ )
{
HRESULT hr = E_FAIL;
do
{
if( !m_pPathname )
break;
CComBSTR bstr;
// Escape it
hr = m_pPathname->GetEscapedElement( 0,
CComBSTR( CString(CN_EQUALS) + strCN ),
&bstr );
BREAK_ON_FAILED_HRESULT(hr);
// set the dn without the leaf node.
hr = m_pPathname->Set(
CComBSTR(
CString( GetBasePathsInfo()->GetProviderAndServerName())
+ GetBasePathsInfo()->GetSchemaNamingContext() ),
ADS_SETTYPE_FULL );
BREAK_ON_FAILED_HRESULT(hr);
// add new leaf element
hr = m_pPathname->AddLeafElement( bstr );
BREAK_ON_FAILED_HRESULT(hr);
// the result should be property escaped
hr = m_pPathname->put_EscapedMode( ADS_ESCAPEDMODE_DEFAULT );
BREAK_ON_FAILED_HRESULT(hr);
// the full thing is needed
hr = m_pPathname->SetDisplayType( ADS_DISPLAY_FULL );
BREAK_ON_FAILED_HRESULT(hr);
// get the final result.
hr = m_pPathname->Retrieve( formatType, &bstr );
BREAK_ON_FAILED_HRESULT(hr);
strPath = bstr;
} while( FALSE );
ASSERT( SUCCEEDED(hr) );
return hr;
}
HRESULT ComponentData::GetLeafObjectFromDN( const BSTR bstrDN,
CString & strCN )
{
HRESULT hr = E_FAIL;
do
{
if( !m_pPathname )
break;
// set the full DN.
hr = m_pPathname->Set( bstrDN, ADS_SETTYPE_DN );
BREAK_ON_FAILED_HRESULT(hr);
// the result should not be escaped
hr = m_pPathname->put_EscapedMode( ADS_ESCAPEDMODE_OFF_EX );
BREAK_ON_FAILED_HRESULT(hr);
// just the value please
hr = m_pPathname->SetDisplayType( ADS_DISPLAY_VALUE_ONLY );
BREAK_ON_FAILED_HRESULT(hr);
CComBSTR bstrCN;
// extract the leaf node
hr = m_pPathname->Retrieve( ADS_FORMAT_LEAF, &bstrCN );
BREAK_ON_FAILED_HRESULT(hr);
strCN = bstrCN;
} while( FALSE );
ASSERT( SUCCEEDED(hr) ); // this function should never fail (in theory)
return hr;
}
//
// Determine what operations are allowed. Optionally returns IADs * to Schema Container
// if the path is not present, the returned value is E_FAIL
//
HRESULT ComponentData::CheckSchemaPermissions( IADs ** ppADs /* = NULL */ )
{
HRESULT hr = S_OK;
CComPtr<IADs> ipADs;
CString szSchemaContainerPath;
CStringList strlist;
ASSERT( !ppADs || !(*ppADs) ); // if present, must point to NULL
do
{
//
// Disable new attrib/class menu items
//
SetCanCreateClass( FALSE );
SetCanCreateAttribute( FALSE );
SetCanChangeOperationsMaster( FALSE );
//
// Get the schema container path.
//
GetBasePathsInfo()->GetSchemaPath( szSchemaContainerPath );
if( szSchemaContainerPath.IsEmpty() )
{
hr = E_FAIL;
break;
}
//
// Open the schema container.
//
hr = ADsGetObject( (LPWSTR)(LPCWSTR)szSchemaContainerPath,
IID_IADs,
(void **)&ipADs );
BREAK_ON_FAILED_HRESULT(hr);
// extract the list of allowed attributes
hr = GetStringListElement( ipADs, &g_allowedAttributesEffective, strlist );
if( SUCCEEDED(hr) )
{
// search for needed attributes
for( POSITION pos = strlist.GetHeadPosition(); pos != NULL; )
{
CString * pstr = &strlist.GetNext( pos );
if( !pstr->CompareNoCase( g_fsmoRoleOwner ) )
{
SetCanChangeOperationsMaster( TRUE );
break;
}
}
}
// extract the list of allowed classes
hr = GetStringListElement( ipADs, &g_allowedChildClassesEffective, strlist );
if( SUCCEEDED(hr) )
{
// search for needed attributes
for( POSITION pos = strlist.GetHeadPosition(); pos != NULL; )
{
CString * pstr = &strlist.GetNext( pos );
if( !pstr->CompareNoCase( g_AttributeFilter ) )
{
SetCanCreateAttribute( TRUE );
if( CanCreateClass() )
break;
}
else if( !pstr->CompareNoCase( g_ClassFilter ) )
{
SetCanCreateClass( TRUE );
if( CanCreateAttribute() )
break;
}
}
}
} while( FALSE );
if( ppADs )
{
*ppADs = ipADs;
if( *ppADs )
(*ppADs)->AddRef();
}
return hr;
}
////////////////////////////////////////////////////////////////////
//
// Error handling
//
// Set's error title & body text. Call it with 0, 0 to remove
void ComponentData::SetError( UINT idsErrorTitle, UINT idsErrorText )
{
if( idsErrorTitle )
m_sErrorTitle.LoadString( idsErrorTitle );
else
m_sErrorTitle.Empty();
if( idsErrorText )
m_sErrorText.LoadString( idsErrorText );
else
m_sErrorText.Empty();
}
VOID ComponentData::InitializeRootTree( HSCOPEITEM hParent, Cookie * pParentCookie )
{
//
// This node has the two static nodes
// for Classes and Attributes.
//
HRESULT hr = S_OK;
LPCWSTR lpcszMachineName = pParentCookie->QueryNonNULLMachineName();
// Update the name of the root to contain the servername we are bound to
if (GetBasePathsInfo()->GetServerName())
{
CString strDisplayName;
strDisplayName.LoadString(IDS_SCOPE_SCHMMGMT);
strDisplayName += L" [";
strDisplayName += GetBasePathsInfo()->GetServerName();
strDisplayName += L"]";
SCOPEDATAITEM RootItem;
::ZeroMemory( &RootItem, sizeof(RootItem));
RootItem.mask = SDI_STR | SDI_PARAM;
RootItem.displayname = const_cast<PWSTR>((PCWSTR)strDisplayName);
RootItem.ID = hParent;
hr = m_pConsoleNameSpace->SetItem(&RootItem);
ASSERT(SUCCEEDED(hr));
}
SCOPEDATAITEM ScopeItem;
::ZeroMemory( &ScopeItem, sizeof(ScopeItem) );
ScopeItem.mask = SDI_STR | SDI_IMAGE | SDI_OPENIMAGE | SDI_STATE | SDI_PARAM | SDI_PARENT;
ScopeItem.displayname = MMC_CALLBACK;
ScopeItem.relativeID = hParent;
ScopeItem.nState = 0;
LoadGlobalCookieStrings();
//
// Create new cookies for the static scope items.
// We're doing something funky with the cookie cast.
//
Cookie* pNewCookie;
pNewCookie= new Cookie( SCHMMGMT_CLASSES,
lpcszMachineName );
pParentCookie->m_listScopeCookieBlocks.AddHead(
(CBaseCookieBlock*)pNewCookie );
ScopeItem.lParam = reinterpret_cast<LPARAM>((CCookie*)pNewCookie);
ScopeItem.nImage = QueryImage( *pNewCookie, FALSE );
ScopeItem.nOpenImage = QueryImage( *pNewCookie, TRUE );
hr = m_pConsoleNameSpace->InsertItem(&ScopeItem);
ASSERT(SUCCEEDED(hr));
pNewCookie->m_hScopeItem = ScopeItem.ID;
pNewCookie = new Cookie( SCHMMGMT_ATTRIBUTES,
lpcszMachineName );
pParentCookie->m_listScopeCookieBlocks.AddHead(
(CBaseCookieBlock*)pNewCookie );
ScopeItem.lParam = reinterpret_cast<LPARAM>((CCookie*)pNewCookie);
ScopeItem.nImage = QueryImage( *pNewCookie, FALSE );
ScopeItem.nOpenImage = QueryImage( *pNewCookie, TRUE );
// turn of the + on the Attributes node
ScopeItem.mask |= SDI_CHILDREN;
ScopeItem.cChildren = 0;
hr = m_pConsoleNameSpace->InsertItem(&ScopeItem);
ASSERT(SUCCEEDED(hr));
pNewCookie->m_hScopeItem = ScopeItem.ID;
//
// Force Cache load (if not done already)
//
g_SchemaCache.LoadCache();
}