///////////////////////////////////////////////////////////////////////////// // // Copyright (c) 1999 Microsoft Corporation // // Module Name: // ExtObj.cpp // // Description: // Implementation of the CExtObject class, which implements the // extension interfaces required by a Microsoft Windows NT Cluster // Administrator Extension DLL. // // Author: // David Potter (DavidP) March 24, 1999 // // Revision History: // // Notes: // ///////////////////////////////////////////////////////////////////////////// #include "stdafx.h" #include "ClNetResEx.h" #include "ExtObj.h" #include "Dhcp.h" #include "Wins.h" ///////////////////////////////////////////////////////////////////////////// // Global Variables ///////////////////////////////////////////////////////////////////////////// const WCHAR g_wszResourceTypeNames[] = L"DHCP Service\0" L"WINS Service\0" ; const DWORD g_cchResourceTypeNames = sizeof( g_wszResourceTypeNames ) / sizeof( WCHAR ); static CRuntimeClass * g_rgprtcDhcpResPSPages[] = { RUNTIME_CLASS( CDhcpParamsPage ), NULL }; static CRuntimeClass * g_rgprtcWinsResPSPages[] = { RUNTIME_CLASS( CWinsParamsPage ), NULL }; static CRuntimeClass ** g_rgpprtcResPSPages[] = { g_rgprtcDhcpResPSPages, g_rgprtcWinsResPSPages, }; static CRuntimeClass ** g_rgpprtcResWizPages[] = { g_rgprtcDhcpResPSPages, g_rgprtcWinsResPSPages, }; ///////////////////////////////////////////////////////////////////////////// // CExtObject ///////////////////////////////////////////////////////////////////////////// ///////////////////////////////////////////////////////////////////////////// //++ // // CExtObject::CExtObject // // Description: // Default constructor. // // Arguments: // None. // // Return Value: // None. // //-- ///////////////////////////////////////////////////////////////////////////// CExtObject::CExtObject( void ) { m_piData = NULL; m_piWizardCallback = NULL; m_bWizard = FALSE; m_istrResTypeName = 0; m_lcid = NULL; m_hfont = NULL; m_hicon = NULL; m_hcluster = NULL; m_cobj = 0; m_podObjData = NULL; } //*** CExtObject::CExtObject() ///////////////////////////////////////////////////////////////////////////// //++ // // CExtObject::~CExtObject // // Description: // Destructor. // // Arguments: // None. // // Return Value: // None. // //-- ///////////////////////////////////////////////////////////////////////////// CExtObject::~CExtObject( void ) { // Release the data interface. if ( PiData() != NULL ) { PiData()->Release(); m_piData = NULL; } // if: we have a data interface pointer // Release the wizard callback interface. if ( PiWizardCallback() != NULL ) { PiWizardCallback()->Release(); m_piWizardCallback = NULL; } // if: we have a wizard callback interface pointer // Delete the pages. { POSITION pos; pos = Lpg().GetHeadPosition(); while ( pos != NULL ) { delete Lpg().GetNext(pos); } // while: more pages in the list } // Delete the pages delete m_podObjData; } //*** CExtObject::~CExtObject() ///////////////////////////////////////////////////////////////////////////// // ISupportErrorInfo Implementation ///////////////////////////////////////////////////////////////////////////// //++ // // CExtObject::InterfaceSupportsErrorInfo (ISupportErrorInfo) // // Routine Description: // Indicates whether an interface suportes the IErrorInfo interface. // This interface is provided by ATL. // // Arguments: // riid Interface ID. // // Return Value: // S_OK Interface supports IErrorInfo. // S_FALSE Interface does not support IErrorInfo. // //-- ///////////////////////////////////////////////////////////////////////////// STDMETHODIMP CExtObject::InterfaceSupportsErrorInfo( REFIID riid ) { static const IID * _rgiid[] = { &IID_IWEExtendPropertySheet, &IID_IWEExtendWizard, }; int _iiid; for ( _iiid = 0 ; _iiid < sizeof( _rgiid ) / sizeof( _rgiid[ 0 ] ) ; _iiid++ ) { if ( InlineIsEqualGUID( *_rgiid[ _iiid ], riid ) ) { return S_OK; } // if: found a matching IID } return S_FALSE; } //*** CExtObject::InterfaceSupportsErrorInfo() ///////////////////////////////////////////////////////////////////////////// // IWEExtendPropertySheet Implementation ///////////////////////////////////////////////////////////////////////////// //++ // // CExtObject::CreatePropertySheetPages (IWEExtendPropertySheet) // // Description: // Create property sheet pages and add them to the sheet. // // Arguments: // piData [IN] // IUnkown pointer from which to obtain interfaces for obtaining data // describing the object for which the sheet is being displayed. // // piCallback [IN] // Pointer to an IWCPropertySheetCallback interface for adding pages // to the sheet. // // Return Value: // NOERROR Pages added successfully. // E_INVALIDARG Invalid arguments to the function. // E_OUTOFMEMORY Error allocating memory. // E_FAIL Error creating a page. // E_NOTIMPL Not implemented for this type of data. // _hr Any error codes from HrGetUIInfo() or HrSaveData(). // //-- ///////////////////////////////////////////////////////////////////////////// STDMETHODIMP CExtObject::CreatePropertySheetPages( IN IUnknown * piData, IN IWCPropertySheetCallback * piCallback ) { HRESULT _hr = NOERROR; CException _exc( FALSE /*bAutoDelete*/ ); CRuntimeClass ** _pprtc = NULL; int _irtc; CBasePropertyPage * _ppage; AFX_MANAGE_STATE( AfxGetStaticModuleState() ); // Validate the parameters. if ( (piData == NULL) || (piCallback == NULL) ) { return E_INVALIDARG; } // if: all interfaces not specified try { // Get info about displaying UI. _hr = HrGetUIInfo( piData ); if ( _hr != NOERROR ) { throw &_exc; } // if: error getting UI info // Save the data. _hr = HrSaveData( piData ); if ( _hr != NOERROR ) { throw &_exc; } // if: error saving data from host // Delete any previous pages. { POSITION pos; pos = Lpg().GetHeadPosition(); while ( pos != NULL ) { delete Lpg().GetNext( pos ); } // while: more pages in the list Lpg().RemoveAll(); } // Delete any previous pages // Create property pages. ASSERT( PodObjData() != NULL ); switch ( PodObjData()->m_cot ) { case CLUADMEX_OT_RESOURCE: _pprtc = g_rgpprtcResPSPages[ IstrResTypeName() ]; break; default: _hr = E_NOTIMPL; throw &_exc; break; } // switch: object type // Create each page. for ( _irtc = 0 ; _pprtc[ _irtc ] != NULL ; _irtc++ ) { // Create the page. _ppage = static_cast< CBasePropertyPage * >( _pprtc[ _irtc ]->CreateObject() ); ASSERT( _ppage->IsKindOf( _pprtc[ _irtc ] ) ); // Add it to the list. Lpg().AddTail( _ppage ); // Initialize the property page. _hr = _ppage->HrInit( this ); if ( FAILED( _hr ) ) { throw &_exc; } // if: error initializing the page // Create the page. _hr = _ppage->HrCreatePage(); if ( FAILED( _hr ) ) { throw &_exc; } // if: error creating the page // Add it to the property sheet. _hr = piCallback->AddPropertySheetPage( reinterpret_cast< LONG * >( _ppage->Hpage() ) ); if ( _hr != NOERROR ) { throw &_exc; } // if: error adding the page to the sheet } // for: each page in the list } // try catch ( CMemoryException * _pme ) { TRACE( _T("CExtObject::CreatePropetySheetPages() - Failed to add property page\n") ); _pme->Delete(); _hr = E_OUTOFMEMORY; } // catch: anything catch ( CException * _pe ) { TRACE( _T("CExtObject::CreatePropetySheetPages() - Failed to add property page\n") ); _pe->Delete(); if ( _hr == NOERROR ) { _hr = E_FAIL; } // if: _hr hasn't beeen set yet } // catch: anything if ( _hr != NOERROR ) { piData->Release(); m_piData = NULL; } // if: error occurred piCallback->Release(); return _hr; } //*** CExtObject::CreatePropertySheetPages() ///////////////////////////////////////////////////////////////////////////// // IWEExtendWizard Implementation ///////////////////////////////////////////////////////////////////////////// //++ // // CExtObject::CreateWizardPages (IWEExtendWizard) // // Description: // Create property sheet pages and add them to the wizard. // // Arguments: // piData [IN] // IUnkown pointer from which to obtain interfaces for obtaining data // describing the object for which the wizard is being displayed. // // piCallback [IN] // Pointer to an IWCPropertySheetCallback interface for adding pages // to the sheet. // // Return Value: // NOERROR Pages added successfully. // E_INVALIDARG Invalid arguments to the function. // E_OUTOFMEMORY Error allocating memory. // E_FAIL Error creating a page. // E_NOTIMPL Not implemented for this type of data. // _hr Any error codes from HrGetUIInfo() or HrSaveData(). // //-- ///////////////////////////////////////////////////////////////////////////// STDMETHODIMP CExtObject::CreateWizardPages( IN IUnknown * piData, IN IWCWizardCallback * piCallback ) { HRESULT _hr = NOERROR; CException _exc( FALSE /*bAutoDelete*/ ); CRuntimeClass ** _pprtc = NULL; int _irtc; CBasePropertyPage * _ppage; AFX_MANAGE_STATE( AfxGetStaticModuleState() ); // Validate the parameters. if ( (piData == NULL) || (piCallback == NULL) ) { return E_INVALIDARG; } // if: all interfaces not specified try { // Get info about displaying UI. _hr = HrGetUIInfo( piData ); if ( _hr != NOERROR ) { throw &_exc; } // if: error getting UI info // Save the data. _hr = HrSaveData( piData ); if ( _hr != NOERROR ) { throw &_exc; } // if: error saving data from host // Delete any previous pages. { POSITION pos; pos = Lpg().GetHeadPosition(); while ( pos != NULL ) { delete Lpg().GetNext( pos ); } // while: more pages in the list Lpg().RemoveAll(); } // Delete any previous pages m_piWizardCallback = piCallback; m_bWizard = TRUE; // Create property pages. ASSERT( PodObjData() != NULL ); switch ( PodObjData()->m_cot ) { case CLUADMEX_OT_RESOURCE: _pprtc = g_rgpprtcResWizPages[ IstrResTypeName() ]; break; default: _hr = E_NOTIMPL; throw &_exc; break; } // switch: object type // Create each page. for ( _irtc = 0 ; _pprtc[ _irtc ] != NULL ; _irtc++ ) { // Create the page. _ppage = static_cast< CBasePropertyPage * >( _pprtc[ _irtc ]->CreateObject() ); ASSERT( _ppage->IsKindOf( _pprtc[ _irtc ] ) ); // Add it to the list. Lpg().AddTail( _ppage ); // Initialize the property page. _hr = _ppage->HrInit( this ); if ( FAILED( _hr ) ) { throw &_exc; } // if: error initializing the page // Create the page. _hr = _ppage->HrCreatePage(); if ( FAILED( _hr ) ) { throw &_exc; } // if: error creating the page // Add it to the property sheet. _hr = piCallback->AddWizardPage( reinterpret_cast< LONG * >( _ppage->Hpage() ) ); if ( _hr != NOERROR ) { throw &_exc; } // if: error adding the page to the sheet } // for: each page in the list } // try catch ( CMemoryException * _pme ) { TRACE( _T("CExtObject::CreateWizardPages() - Failed to add wizard page\n") ); _pme->Delete(); _hr = E_OUTOFMEMORY; } // catch: anything catch ( CException * _pe ) { TRACE( _T("CExtObject::CreateWizardPages() - Failed to add wizard page\n") ); _pe->Delete(); if ( _hr == NOERROR ) { _hr = E_FAIL; } // if: _hr hasn't beeen set yet } // catch: anything if ( _hr != NOERROR ) { piCallback->Release(); if ( m_piWizardCallback == piCallback ) { m_piWizardCallback = NULL; } // if: already saved interface pointer piData->Release(); m_piData = NULL; } // if: error occurred return _hr; } //*** CExtObject::CreateWizardPages() ///////////////////////////////////////////////////////////////////////////// //++ // // CExtObject::HrGetUIInfo // // Description: // Get info about displaying UI. // // Arguments: // piData [IN] // IUnkown pointer from which to obtain interfaces for obtaining data // describing the object. // // Return Value: // NOERROR // Data saved successfully. // // E_NOTIMPL // Not implemented for this type of data. // // _hr // Any error codes from IUnknown::QueryInterface(), // HrGetObjectName(), or HrGetResourceName(). // //-- ///////////////////////////////////////////////////////////////////////////// HRESULT CExtObject::HrGetUIInfo( IN IUnknown * piData ) { HRESULT _hr = NOERROR; ASSERT( piData != NULL ); // Save info about all types of objects. { IGetClusterUIInfo * _pi; _hr = piData->QueryInterface( IID_IGetClusterUIInfo, reinterpret_cast< LPVOID * >( &_pi ) ); if ( _hr != NOERROR ) { return _hr; } // if: error querying for interface m_lcid = _pi->GetLocale(); m_hfont = _pi->GetFont(); m_hicon = _pi->GetIcon(); _pi->Release(); } // Save info about all types of objects return _hr; } //*** CExtObject::HrGetUIInfo() ///////////////////////////////////////////////////////////////////////////// //++ // // CExtObject::HrSaveData // // Routine Description: // Save data from the object so that it can be used for the life // of the object. // // Arguments: // piData [IN] // IUnkown pointer from which to obtain interfaces for obtaining data // describing the object. // // Return Value: // NOERROR // Data saved successfully. // // E_NOTIMPL // Not implemented for this type of data. // // _hr // Any error codes from IUnknown::QueryInterface(), // HrGetObjectName(), or HrGetResourceName(). // //-- ///////////////////////////////////////////////////////////////////////////// HRESULT CExtObject::HrSaveData( IN IUnknown * piData ) { HRESULT _hr = NOERROR; ASSERT( piData != NULL ); if ( piData != m_piData ) { if ( m_piData != NULL ) { m_piData->Release(); } // if: interface queried for previously m_piData = piData; } // if: different data interface pointer // Save info about all types of objects. { IGetClusterDataInfo * _pi; _hr = piData->QueryInterface( IID_IGetClusterDataInfo, reinterpret_cast< LPVOID * >( &_pi ) ); if ( _hr != NOERROR ) { return _hr; } // if: error querying for interface m_hcluster = _pi->GetClusterHandle(); m_cobj = _pi->GetObjectCount(); if ( Cobj() != 1 ) // Only have support for one selected object. { _hr = E_NOTIMPL; } // if: too many objects for us to handle _pi->Release(); if ( _hr != NOERROR ) { return _hr; } // if: error occurred before here } // Save info about all types of objects // Save info about this object. _hr = HrGetObjectInfo(); return _hr; } //*** CExtObject::HrSaveData() ///////////////////////////////////////////////////////////////////////////// //++ // // CExtObject::HrGetObjectInfo // // Description: // Get information about the object. // // Arguments: // None. // // Return Value: // NOERROR // Data saved successfully. // // E_OUTOFMEMORY /// Error allocating memory. // // E_NOTIMPL // Not implemented for this type of data. // // _hr // Any error codes from IUnknown::QueryInterface(), // HrGetObjectName(), or HrGetResourceTypeName(). // //-- ///////////////////////////////////////////////////////////////////////////// HRESULT CExtObject::HrGetObjectInfo( void ) { HRESULT _hr = NOERROR; IGetClusterObjectInfo * _piGcoi; CLUADMEX_OBJECT_TYPE _cot = CLUADMEX_OT_NONE; CException _exc( FALSE /*bAutoDelete*/ ); const CString * _pstrResTypeName = NULL; ASSERT( PiData() != NULL ); // Get object info. { // Get an IGetClusterObjectInfo interface pointer. _hr = PiData()->QueryInterface( IID_IGetClusterObjectInfo, reinterpret_cast< LPVOID * >( &_piGcoi ) ); if ( _hr != NOERROR ) { return _hr; } // if: error querying for interface // Read the object data. try { // Delete the previous object data. delete m_podObjData; m_podObjData = NULL; // Get the type of the object. _cot = _piGcoi->GetObjectType( 0 ); switch ( _cot ) { case CLUADMEX_OT_RESOURCE: { IGetClusterResourceInfo * _pi; m_podObjData = new CResData; if ( m_podObjData == NULL ) { _hr = E_OUTOFMEMORY; throw &_exc; } // if: error allocating memory // Get an IGetClusterResourceInfo interface pointer. _hr = PiData()->QueryInterface( IID_IGetClusterResourceInfo, reinterpret_cast< LPVOID * >( &_pi ) ); if ( _hr != NOERROR ) { throw &_exc; } // if: error querying for interface PrdResDataRW()->m_hresource = _pi->GetResourceHandle( 0 ); ASSERT( PrdResDataRW()->m_hresource != NULL ); if ( PrdResDataRW()->m_hresource == NULL ) { _hr = E_INVALIDARG; } // if invalid resource handle else { _hr = HrGetResourceTypeName( _pi ); } // else: resource handle is valid _pi->Release(); if ( _hr != NOERROR ) { throw &_exc; } // if: error occurred above _pstrResTypeName = &PrdResDataRW()->m_strResTypeName; } // if: object is a resource break; case CLUADMEX_OT_RESOURCETYPE: m_podObjData = new CObjData; if ( m_podObjData == NULL ) { _hr = E_OUTOFMEMORY; throw &_exc; } _pstrResTypeName = &PodObjDataRW()->m_strName; break; default: _hr = E_NOTIMPL; throw &_exc; } // switch: object type PodObjDataRW()->m_cot = _cot; _hr = HrGetObjectName( _piGcoi ); } // try catch ( CException * _pe ) { if ( !FAILED (_hr) ) { _hr = E_FAIL; } _pe->Delete(); } // catch: CException _piGcoi->Release(); // If we failed to initialize _pstrResTypeName, then bail. // We are doing this because of PREFIX, which assumes that the // new operator for OT_RESOURCETYPE above can fail - // but it should never happen that the new should perform a throw. if ( _pstrResTypeName == NULL ) { _hr = E_OUTOFMEMORY; } if ( _hr != NOERROR ) { return _hr; } // if: error occurred above } // Get object info // If this is a resource or resource type, see if we know about this type. if ( ( (_cot == CLUADMEX_OT_RESOURCE) || (_cot == CLUADMEX_OT_RESOURCETYPE) ) && (_hr == NOERROR) ) { LPCWSTR _pwszResTypeName; // Find the resource type name in our list. // Save the index for use in other arrays. for ( m_istrResTypeName = 0, _pwszResTypeName = g_wszResourceTypeNames ; *_pwszResTypeName != L'\0' ; m_istrResTypeName++, _pwszResTypeName += lstrlenW( _pwszResTypeName ) + 1 ) { if ( _pstrResTypeName->CompareNoCase( _pwszResTypeName ) == 0 ) { break; } // if: found resource type name } // for: each resource type in the list if ( *_pwszResTypeName == L'\0' ) { _hr = E_NOTIMPL; } // if: resource type name not found } // See if we know about this resource type return _hr; } //*** CExtObject::HrGetObjectInfo() ///////////////////////////////////////////////////////////////////////////// //++ // // CExtObject::HrGetObjectName // // Description: // Get the name of the object. // // Arguments: // piData [IN] // IGetClusterObjectInfo interface pointer for getting the object // name. // // Return Value: // NOERROR // Data saved successfully. // // E_OUTOFMEMORY // Error allocating memory. // // E_NOTIMPL // Not implemented for this type of data. // // _hr // Any error codes from IGetClusterObjectInfo::GetObjectInfo(). // //-- ///////////////////////////////////////////////////////////////////////////// HRESULT CExtObject::HrGetObjectName( IN IGetClusterObjectInfo * pi ) { HRESULT _hr = NOERROR; WCHAR * _pwszName = NULL; LONG _cchName; ASSERT( pi != NULL ); _hr = pi->GetObjectName( 0, NULL, &_cchName ); if ( _hr != NOERROR ) { return _hr; } // if: error getting object name try { _pwszName = new WCHAR[ _cchName ]; _hr = pi->GetObjectName( 0, _pwszName, &_cchName ); if ( _hr != NOERROR ) { delete [] _pwszName; _pwszName = NULL; } // if: error getting object name PodObjDataRW()->m_strName = _pwszName; } // try catch ( CMemoryException * _pme ) { _pme->Delete(); _hr = E_OUTOFMEMORY; } // catch: CMemoryException delete [] _pwszName; return _hr; } //*** CExtObject::HrGetObjectName() ///////////////////////////////////////////////////////////////////////////// //++ // // CExtObject::HrGetResourceTypeName // // Routine Description: // Get the name of the resource's type. // // Arguments: // piData [IN] // IGetClusterResourceInfo interface pointer for getting the resource // type name. // // Return Value: // NOERROR // Data saved successfully. // // E_OUTOFMEMORY // Error allocating memory. // // E_NOTIMPL // Not implemented for this type of data. // // _hr // Any error codes from IGetClusterResourceInfo // ::GetResourceTypeName(). // //-- ///////////////////////////////////////////////////////////////////////////// HRESULT CExtObject::HrGetResourceTypeName( IN IGetClusterResourceInfo * pi ) { HRESULT _hr = NOERROR; WCHAR * _pwszName = NULL; LONG _cchName; ASSERT( pi != NULL ); _hr = pi->GetResourceTypeName( 0, NULL, &_cchName ); if ( _hr != NOERROR ) { return _hr; } // if: error getting resource type name try { _pwszName = new WCHAR[ _cchName ]; _hr = pi->GetResourceTypeName( 0, _pwszName, &_cchName ); if ( _hr != NOERROR ) { delete [] _pwszName; _pwszName = NULL; } // if: error getting resource type name PrdResDataRW()->m_strResTypeName = _pwszName; } // try catch ( CMemoryException * _pme ) { _pme->Delete(); _hr = E_OUTOFMEMORY; } // catch: CMemoryException delete [] _pwszName; return _hr; } //*** CExtObject::HrGetResourceTypeName()