|
|
/////////////////////////////////////////////////////////////////////////////
//
// 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()
|