Leaked source code of windows server 2003
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.
 
 
 
 
 
 

889 lines
25 KiB

/////////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 2002 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:
// <name> (<e-mail name>) Mmmm DD, 2002
//
// Revision History:
//
// Notes:
//
/////////////////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "VSSTaskEx.h"
#include "ExtObj.h"
#include "ResProp.h"
class CStackException : public CException
{
public:
CStackException( BOOL bAutoDelete )
: CException( bAutoDelete )
{
} //*** CStackException()
virtual ~CStackException( void )
{
} //*** ~CStackException()
}; //*** class CStackException
/////////////////////////////////////////////////////////////////////////////
// Global Variables
/////////////////////////////////////////////////////////////////////////////
const WCHAR g_wszResourceTypeNames[] =
CLUS_RESTYPE_NAME_VSSTASK L"\0"
;
const DWORD g_cchResourceTypeNames = sizeof( g_wszResourceTypeNames ) / sizeof( WCHAR );
static CRuntimeClass * g_rgprtcResPSPages[] = {
RUNTIME_CLASS( CVSSTaskParamsPage ),
NULL
};
static CRuntimeClass ** g_rgpprtcResPSPages[] = {
g_rgprtcResPSPages,
};
static CRuntimeClass ** g_rgpprtcResWizPages[] = {
g_rgprtcResPSPages,
};
/////////////////////////////////////////////////////////////////////////////
// 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;
CStackException _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;
CStackException _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;
CStackException _exc( FALSE /*bAutoDelete*/ );
const CString * _pstrResTypeName;
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;
// 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;
default:
_hr = E_NOTIMPL;
throw &_exc;
} // switch: object type
PodObjDataRW()->m_cot = _cot;
_hr = HrGetObjectName( _piGcoi );
} // try
catch ( CException * _pe )
{
_pe->Delete();
} // catch: CException
_piGcoi->Release();
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()