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.
 
 
 
 
 
 

1475 lines
42 KiB

/////////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 1997-2002 Microsoft Corporation
//
// Module Name:
// ClusResT.cpp
//
// Description:
// Implementation of the resource type classes for the MSCLUS
// automation classes.
//
// Author:
// Charles Stacy Harris (stacyh) 28-Feb-1997
// Galen Barbee (galenb) July 1998
//
// Revision History:
// July 1998 GalenB Maaaaaajjjjjjjjjoooooorrrr clean up
//
// Notes:
//
/////////////////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "ClusterObject.h"
#include "property.h"
#include "clusres.h"
#include "clusresg.h"
#include "clusneti.h"
#include "clusnode.h"
#include "clusrest.h"
/////////////////////////////////////////////////////////////////////////////
// Global variables
/////////////////////////////////////////////////////////////////////////////
static const IID * iidCClusResourceType[] =
{
&IID_ISClusResType
};
static const IID * iidCClusResourceTypes[] =
{
&IID_ISClusResTypes
};
//*************************************************************************//
/////////////////////////////////////////////////////////////////////////////
// CClusResType class
/////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::CClusResType
//
// Description:
// Constructor.
//
// Arguments:
// None.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
CClusResType::CClusResType( void )
{
m_pClusRefObject = NULL;
m_pCommonProperties = NULL;
m_pPrivateProperties = NULL;
m_pCommonROProperties = NULL;
m_pPrivateROProperties = NULL;
m_pClusterResTypeResources = NULL;
m_piids = (const IID *) iidCClusResourceType;
m_piidsSize = ARRAYSIZE( iidCClusResourceType );
} //*** CClusResType::CClusResType()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::~CClusResType
//
// Description:
// Destructor.
//
// Arguments:
// None.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
CClusResType::~CClusResType( void )
{
if ( m_pCommonProperties != NULL )
{
m_pCommonProperties->Release();
m_pCommonProperties = NULL;
} // if: release the property collection
if ( m_pPrivateProperties != NULL )
{
m_pPrivateProperties->Release();
m_pPrivateProperties = NULL;
} // if: release the property collection
if ( m_pCommonROProperties != NULL )
{
m_pCommonROProperties->Release();
m_pCommonROProperties = NULL;
} // if: release the property collection
if ( m_pPrivateROProperties != NULL )
{
m_pPrivateROProperties->Release();
m_pPrivateROProperties = NULL;
} // if: release the property collection
if ( m_pClusterResTypeResources != NULL )
{
m_pClusterResTypeResources->Release();
m_pClusterResTypeResources = NULL;
}
if ( m_pClusRefObject != NULL )
{
m_pClusRefObject->Release();
m_pClusRefObject = NULL;
}
} //*** CClusResType::~CClusResType()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::GetProperties
//
// Description:
// Creates a property collection for this object type (Resource Type).
//
// Arguments:
// ppProperties [OUT] - Catches the newly created collection.
// bPrivate [IN] - Are these private properties? Or Common?
// bReadOnly [IN] - Are these read only properties?
//
// Return Value:
// S_OK if successful, or other HRESULT error.
//
//--
/////////////////////////////////////////////////////////////////////////////
HRESULT CClusResType::GetProperties(
OUT ISClusProperties ** ppProperties,
IN BOOL bPrivate,
IN BOOL bReadOnly
)
{
//ASSERT( ppProperties != NULL );
HRESULT _hr = E_POINTER;
if ( ppProperties != NULL )
{
*ppProperties = NULL;
CComObject< CClusProperties > * pProperties = NULL;
_hr = CComObject< CClusProperties >::CreateInstance( &pProperties );
if ( SUCCEEDED( _hr ) )
{
CSmartPtr< CComObject< CClusProperties > > ptrProperties( pProperties );
_hr = ptrProperties->Create( this, bPrivate, bReadOnly );
if ( SUCCEEDED( _hr ) )
{
_hr = ptrProperties->Refresh();
if ( SUCCEEDED( _hr ) )
{
_hr = ptrProperties->QueryInterface( IID_ISClusProperties, (void **) ppProperties );
if ( SUCCEEDED( _hr ) )
{
ptrProperties->AddRef();
if ( bPrivate )
{
if ( bReadOnly )
{
m_pPrivateROProperties = pProperties;
}
else
{
m_pPrivateProperties = pProperties;
}
}
else
{
if ( bReadOnly )
{
m_pCommonROProperties = pProperties;
}
else
{
m_pCommonProperties = pProperties;
}
}
}
}
}
}
}
return _hr;
} //*** CClusResType::GetProperties()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::Create
//
// Description:
// Create a new Resource Type and add it to the cluster.
//
// Arguments:
// pClusRefObject [IN] - Cluster handle wrapper.
// bstrResourceTypeName [IN] - Name of the new resource type.
// bstrDisplayName [IN] - Resource type display name.
// bstrResourceTypeDll [IN] - Resource type implementation dll.
// dwLooksAlivePollInterval [IN] - Looks alive poll interval.
// dwIsAlivePollInterval [IN] - Is alive poll interval.
//
// Return Value:
// S_OK if successful, or Win32 error wrapped in HRESULT.
//
//--
/////////////////////////////////////////////////////////////////////////////
HRESULT CClusResType::Create(
IN ISClusRefObject * pClusRefObject,
IN BSTR bstrResourceTypeName,
IN BSTR bstrDisplayName,
IN BSTR bstrResourceTypeDll,
IN long dwLooksAlivePollInterval,
IN long dwIsAlivePollInterval
)
{
ASSERT( pClusRefObject != NULL );
ASSERT( bstrResourceTypeName != NULL );
ASSERT( bstrDisplayName != NULL );
ASSERT( bstrResourceTypeDll != NULL );
HRESULT _hr = E_POINTER;
if ( ( pClusRefObject != NULL ) &&
( pClusRefObject != NULL ) &&
( bstrResourceTypeName != NULL ) &&
( bstrDisplayName != NULL ) )
{
DWORD _sc = 0;
HCLUSTER hCluster = NULL;
m_pClusRefObject= pClusRefObject;
m_pClusRefObject->AddRef();
_hr = m_pClusRefObject->get_Handle( (ULONG_PTR *) &hCluster);
if ( SUCCEEDED( _hr ) )
{
_sc = CreateClusterResourceType(
hCluster,
bstrResourceTypeName,
bstrDisplayName,
bstrResourceTypeDll,
dwLooksAlivePollInterval,
dwIsAlivePollInterval
);
if ( _sc == ERROR_SUCCESS )
{
m_bstrResourceTypeName = bstrResourceTypeName ;
}
_hr = HRESULT_FROM_WIN32( _sc );
}
}
return _hr;
} //*** CClusResType::Create()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::Open
//
// Description:
// Create a resource type object from an existing object in the cluster.
//
// Arguments:
// pClusRefObject [IN] - Cluster handle wrapper.
// bstrResourceTypeName [IN] - Name of the resource type to open.
//
// Return Value:
// S_OK if successful, or other HRESULT error.
//
//--
/////////////////////////////////////////////////////////////////////////////
HRESULT CClusResType::Open(
IN ISClusRefObject * pClusRefObject,
IN BSTR bstrResourceTypeName
)
{
ASSERT( pClusRefObject != NULL );
//ASSERT( bstrResourceTypeName != NULL );
HRESULT _hr = E_POINTER;
if ( ( pClusRefObject != NULL ) && ( bstrResourceTypeName != NULL ) )
{
m_pClusRefObject = pClusRefObject;
m_pClusRefObject->AddRef();
m_bstrResourceTypeName = bstrResourceTypeName;
return S_OK;
}
return _hr;
} //*** CClusResType::Open()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::get_Name
//
// Description:
// Return the name of this object (Resource Type).
//
// Arguments:
// pbstrTypeName [OUT] - Catches the name of this object.
//
// Return Value:
// S_OK if successful, E_POINTER, or other HRESULT error.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResType::get_Name( OUT BSTR * pbstrTypeName )
{
//ASSERT( pbstrTypeName != NULL );
HRESULT _hr = E_POINTER;
if ( pbstrTypeName != NULL )
{
*pbstrTypeName = m_bstrResourceTypeName.Copy();
_hr = S_OK;
}
return _hr;
} //*** CClusResType::get_Name()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::Delete
//
// Description:
// Removes this object (Resource Type) from the cluster.
//
// Arguments:
// None.
//
// Return Value:
// S_OK if successful, E_POINTER, or other HRESULT error.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResType::Delete( void )
{
ASSERT( m_bstrResourceTypeName != NULL );
HRESULT _hr = E_POINTER;
if ( m_bstrResourceTypeName != NULL )
{
HCLUSTER hCluster = NULL;
_hr = m_pClusRefObject->get_Handle( (ULONG_PTR *) &hCluster );
if ( SUCCEEDED( _hr ) )
{
DWORD _sc = DeleteClusterResourceType( hCluster, m_bstrResourceTypeName );
_hr = HRESULT_FROM_WIN32( _sc );
}
}
return _hr;
} //*** CClusResType::Delete()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::get_CommonProperties
//
// Description:
// Get this object's (Resource Type) common properties collection.
//
// Arguments:
// ppProperties [OUT] - Catches the properties collection.
//
// Return Value:
// S_OK if successful, or other HRESULT error.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResType::get_CommonProperties(
OUT ISClusProperties ** ppProperties
)
{
//ASSERT( ppProperties != NULL );
HRESULT _hr = E_POINTER;
if ( ppProperties != NULL )
{
if ( m_pCommonProperties != NULL )
{
_hr = m_pCommonProperties->QueryInterface( IID_ISClusProperties, (void **) ppProperties );
}
else
{
_hr = GetProperties( ppProperties, FALSE, FALSE );
}
}
return _hr;
} //*** CClusResType::get_CommonProperties()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::get_PrivateProperties
//
// Description:
// Get this object's (Resource Type) private properties collection.
//
// Arguments:
// ppProperties [OUT] - Catches the properties collection.
//
// Return Value:
// S_OK if successful, or other HRESULT error.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResType::get_PrivateProperties(
OUT ISClusProperties ** ppProperties
)
{
//ASSERT( ppProperties != NULL );
HRESULT _hr = E_POINTER;
if ( ppProperties != NULL )
{
if ( m_pPrivateProperties != NULL )
{
_hr = m_pPrivateProperties->QueryInterface( IID_ISClusProperties, (void **) ppProperties );
}
else
{
_hr = GetProperties( ppProperties, TRUE, FALSE );
}
}
return _hr;
} //*** CClusResType::get_PrivateProperties()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::get_CommonROProperties
//
// Description:
// Get this object's (Resource Type) common read only properties collection.
//
// Arguments:
// ppProperties [OUT] - Catches the properties collection.
//
// Return Value:
// S_OK if successful, or other HRESULT error.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResType::get_CommonROProperties(
OUT ISClusProperties ** ppProperties
)
{
//ASSERT( ppProperties != NULL );
HRESULT _hr = E_POINTER;
if ( ppProperties != NULL )
{
if ( m_pCommonROProperties != NULL )
{
_hr = m_pCommonROProperties->QueryInterface( IID_ISClusProperties, (void **) ppProperties );
}
else
{
_hr = GetProperties( ppProperties, FALSE, TRUE );
}
}
return _hr;
} //*** CClusResType::get_CommonROProperties()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::get_PrivateROProperties
//
// Description:
// Get this object's (Resource Type) private readonly properties collection.
//
// Arguments:
// ppProperties [OUT] - Catches the properties collection.
//
// Return Value:
// S_OK if successful, or other HRESULT error.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResType::get_PrivateROProperties(
OUT ISClusProperties ** ppProperties
)
{
//ASSERT( ppProperties != NULL );
HRESULT _hr = E_POINTER;
if ( ppProperties != NULL )
{
if ( m_pPrivateROProperties != NULL )
{
_hr = m_pPrivateROProperties->QueryInterface( IID_ISClusProperties, (void **) ppProperties );
}
else
{
_hr = GetProperties( ppProperties, TRUE, TRUE );
}
}
return _hr;
} //*** CClusResType::get_PrivateROProperties()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::get_Cluster
//
// Description:
//
//
// Arguments:
//
//
// Return Value:
// S_OK if successful, or other HRESULT error.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResType::get_Cluster(
ISCluster ** ppCluster
)
{
return ::HrGetCluster( ppCluster, m_pClusRefObject );
} //*** CClusResType::get_Cluster()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::HrLoadProperties
//
// Description:
// This virtual function does the actual load of the property list from
// the cluster.
//
// Arguments:
// rcplPropList [IN OUT] - The property list to load.
// bReadOnly [IN] - Load the read only properties?
// bPrivate [IN] - Load the common or the private properties?
//
// Return Value:
// S_OK if successful, or other HRESULT error.
//
//--
/////////////////////////////////////////////////////////////////////////////
HRESULT CClusResType::HrLoadProperties(
IN OUT CClusPropList & rcplPropList,
IN BOOL bReadOnly,
IN BOOL bPrivate
)
{
ASSERT( m_pClusRefObject != NULL );
HRESULT _hr = E_POINTER;
if ( m_pClusRefObject != NULL )
{
HCLUSTER hCluster = NULL;
_hr = m_pClusRefObject->get_Handle( (ULONG_PTR *) &hCluster );
if ( SUCCEEDED( _hr ) )
{
DWORD _dwControlCode = 0;
DWORD _sc = ERROR_SUCCESS;
if ( bReadOnly )
{
_dwControlCode = bPrivate
? CLUSCTL_RESOURCE_TYPE_GET_RO_PRIVATE_PROPERTIES
: CLUSCTL_RESOURCE_TYPE_GET_RO_COMMON_PROPERTIES;
}
else
{
_dwControlCode = bPrivate
? CLUSCTL_RESOURCE_TYPE_GET_PRIVATE_PROPERTIES
: CLUSCTL_RESOURCE_TYPE_GET_COMMON_PROPERTIES;
}
_sc = rcplPropList.ScGetResourceTypeProperties( hCluster, m_bstrResourceTypeName, _dwControlCode );
_hr = HRESULT_FROM_WIN32( _sc );
} // if:
} // if:
return _hr;
} //*** CClusResType::HrLoadProperties()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::ScWriteProperties
//
// Description:
// This virtual function does the actual saving of the property list to
// the cluster.
//
// Arguments:
// rcplPropList [IN] - The property list to save.
// bPrivate [IN] - Save the common or the private properties?
//
// Return Value:
// ERROR_SUCCESS if successful, or other Win32 error if not.
//
//--
/////////////////////////////////////////////////////////////////////////////
DWORD CClusResType::ScWriteProperties(
const CClusPropList & rcplPropList,
BOOL bPrivate
)
{
ASSERT( m_pClusRefObject != NULL );
DWORD _sc = ERROR_BAD_ARGUMENTS;
if ( m_pClusRefObject != NULL )
{
HCLUSTER hCluster = NULL;
if ( S_OK == m_pClusRefObject->get_Handle( (ULONG_PTR *) &hCluster ) )
{
DWORD dwControlCode = bPrivate
? CLUSCTL_RESOURCE_TYPE_SET_PRIVATE_PROPERTIES
: CLUSCTL_RESOURCE_TYPE_SET_COMMON_PROPERTIES;
DWORD nBytesReturned = 0;
_sc = ClusterResourceTypeControl(
hCluster,
m_bstrResourceTypeName,
NULL,
dwControlCode,
rcplPropList,
rcplPropList.CbBufferSize(),
0,
0,
&nBytesReturned
);
} // if:
} // if:
return _sc;
} //*** CClusResType::ScWriteProperties()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::get_Resources
//
// Description:
// Create a collection of the resources of this type.
//
// Arguments:
// ppClusterResTypeResources [OUT] - Catches the collection.
//
// Return Value:
// S_OK if successful, E_POINTER, or Win32 error as HRESULT.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResType::get_Resources(
OUT ISClusResTypeResources ** ppClusterResTypeResources
)
{
return ::HrCreateResourceCollection< CClusResTypeResources, ISClusResTypeResources, CComBSTR >(
&m_pClusterResTypeResources,
m_bstrResourceTypeName,
ppClusterResTypeResources,
IID_ISClusResTypeResources,
m_pClusRefObject
);
} //*** CClusResType::get_Resources()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::get_PossibleOwnerNodes
//
// Description:
// Create a collection of possible owner nodes for this resource type.
//
// Arguments:
// ppOwnerNodes [OUT] - Catches the collection.
//
// Return Value:
// S_OK if successful, E_POINTER, or Win32 error as HRESULT.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResType::get_PossibleOwnerNodes(
ISClusResTypePossibleOwnerNodes ** ppOwnerNodes
)
{
//
// KB: Can't use one of the template functions here! There is an internal compiler error(C1001).
// I am assuming that the type names are too long. Very sad for 1999 ;-)
//
//ASSERT( ppOwnerNodes != NULL );
HRESULT _hr = E_POINTER;
#if CLUSAPI_VERSION >= 0x0500
if ( ppOwnerNodes != NULL )
{
CComObject< CClusResTypePossibleOwnerNodes > * pClusterNodes = NULL;
*ppOwnerNodes = NULL;
_hr = CComObject< CClusResTypePossibleOwnerNodes >::CreateInstance( &pClusterNodes );
if ( SUCCEEDED( _hr ) )
{
CSmartPtr< ISClusRefObject > ptrRefObject( m_pClusRefObject );
CSmartPtr< CComObject< CClusResTypePossibleOwnerNodes > > ptrClusterNodes( pClusterNodes );
_hr = ptrClusterNodes->Create( ptrRefObject, m_bstrResourceTypeName );
if ( SUCCEEDED( _hr ) )
{
_hr = ptrClusterNodes->Refresh();
if ( SUCCEEDED( _hr ) )
{
_hr = ptrClusterNodes->QueryInterface( IID_ISClusResTypePossibleOwnerNodes, (void **) ppOwnerNodes );
}
}
}
}
#else
_hr = E_NOTIMPL;
#endif // CLUSAPI_VERSION >= 0x0500
return _hr;
} //*** CClusResType::get_PossibleOwnerNodes()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResType::get_AvailableDisks
//
// Description:
// Get the collection of available disks.
//
// Arguments:
// ppAvailableDisk [OUT] - catches the available disks collection.
//
// Return Value:
// S_OK if successful, E_POINTER, or Win32 error as HRESULT.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResType::get_AvailableDisks(
OUT ISClusDisks ** ppAvailableDisks
)
{
return ::HrCreateResourceCollection< CClusDisks, ISClusDisks, CComBSTR >(
ppAvailableDisks,
m_bstrResourceTypeName,
IID_ISClusDisks,
m_pClusRefObject
);
} //*** CClusResType::get_AvailableDisks()
//*************************************************************************//
/////////////////////////////////////////////////////////////////////////////
// CClusResTypes class
/////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResTypes::CClusResTypes
//
// Description:
// Constructor.
//
// Arguments:
// None.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
CClusResTypes::CClusResTypes( void )
{
m_pClusRefObject = NULL;
m_piids = (const IID *) iidCClusResourceTypes;
m_piidsSize = ARRAYSIZE( iidCClusResourceTypes );
} //*** CClusResTypes::CClusResTypes()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResTypes::~CClusResTypes
//
// Description:
// Destructor.
//
// Arguments:
// None.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
CClusResTypes::~CClusResTypes( void )
{
Clear();
if ( m_pClusRefObject != NULL )
{
m_pClusRefObject->Release();
m_pClusRefObject = NULL;
}
} //*** CClusResTypes::~CClusResTypes()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResTypes::Create
//
// Description:
// Finish the heavy weight construction.
//
// Arguments:
// pClusRefObject [IN] - Wraps the cluster handle.
//
// Return Value:
// S_OK if successful, or E_POINTER if not.
//
//--
/////////////////////////////////////////////////////////////////////////////
HRESULT CClusResTypes::Create( IN ISClusRefObject * pClusRefObject )
{
ASSERT( pClusRefObject != NULL );
HRESULT _hr = E_POINTER;
if ( pClusRefObject != NULL )
{
m_pClusRefObject = pClusRefObject;
m_pClusRefObject->AddRef();
_hr = S_OK;
} // if: args are not NULL
return _hr;
} //*** CClusResTypes::Create()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResTypes::get_Count
//
// Description:
// Returns the count of elements (ResTypes) in the collection.
//
// Arguments:
// plCount [OUT] - Catches the count.
//
// Return Value:
// S_OK if successful, or E_POINTER if not.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResTypes::get_Count( OUT long * plCount )
{
//ASSERT( plCount != NULL );
HRESULT _hr = E_POINTER;
if ( plCount != NULL )
{
*plCount = m_ResourceTypes.size();
_hr = S_OK;
}
return _hr;
} //*** CClusResTypes::get_Count()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResTypes::Clear
//
// Description:
// Clean out the vector of ClusResGroup objects.
//
// Arguments:
// None.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
void CClusResTypes::Clear( void )
{
::ReleaseAndEmptyCollection< ResourceTypeList, CComObject< CClusResType > >( m_ResourceTypes );
} //*** CClusResTypes::Clear()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResTypes::FindItem
//
// Description:
// Find the passed in resource type in the collection.
//
// Arguments:
// pszResourceTypeName [IN] - The name of the resource type to find.
// pnIndex [OUT] - Catches the index of the group.
//
// Return Value:
// S_OK if successful, E_POINTER, or E_INVALIDARG if the resource type
// was not found.
//
//--
/////////////////////////////////////////////////////////////////////////////
HRESULT CClusResTypes::FindItem(
IN LPWSTR pszResourceTypeName,
OUT UINT * pnIndex
)
{
//ASSERT( pszResourceTypeName != NULL );
//ASSERT( pnIndex != NULL );
HRESULT _hr = E_POINTER;
if ( ( pszResourceTypeName != NULL ) && ( pnIndex != NULL ) )
{
CComObject<CClusResType> * pResourceType = NULL;
int nMax = m_ResourceTypes.size();
_hr = E_INVALIDARG;
for( int i = 0; i < nMax; i++ )
{
pResourceType = m_ResourceTypes[i];
if ( pResourceType && ( lstrcmpi( pszResourceTypeName, pResourceType->Name() ) == 0 ) )
{
*pnIndex = i;
_hr = S_OK;
break;
}
}
}
return _hr;
} //*** CClusResTypes::FindItem( pszResourceTypeName )
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResTypes::FindItem
//
// Description:
// Find the passed in resource type in the collection.
//
// Arguments:
// pszResourceTypeName [IN] - The name of the resource type to find.
// pnIndex [OUT] - Catches the index of the group.
//
// Return Value:
// S_OK if successful, E_POINTER, or E_INVALIDARG if the resource type
// was not found.
//
//--
/////////////////////////////////////////////////////////////////////////////
HRESULT CClusResTypes::FindItem(
IN ISClusResType * pResourceType,
OUT UINT * pnIndex
)
{
//ASSERT( pResourceType != NULL );
//ASSERT( pnIndex != NULL );
HRESULT _hr = E_POINTER;
if ( ( pResourceType != NULL ) && ( pnIndex != NULL ) )
{
CComBSTR bstrName;
_hr = pResourceType->get_Name( &bstrName );
if ( SUCCEEDED( _hr ) )
{
_hr = FindItem( bstrName, pnIndex );
}
}
return _hr;
} //*** CClusResTypes::FindItem( pResourceType )
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResTypes::RemoveAt
//
// Description:
// Remove the object (ResType) at the passed in index/position from the
// collection.
//
// Arguments:
// nPos [IN] - Index of the object to remove.
//
// Return Value:
// S_OK if successful, or E_INVALIDARG is the index is out of range.
//
//--
/////////////////////////////////////////////////////////////////////////////
HRESULT CClusResTypes::RemoveAt( IN size_t pos )
{
CComObject<CClusResType> * pResourceType = NULL;
ResourceTypeList::iterator first = m_ResourceTypes.begin();
ResourceTypeList::iterator last = m_ResourceTypes.end();
HRESULT _hr = E_INVALIDARG;
for ( size_t t = 0; ( t < pos ) && ( first != last ); t++, first++ );
if ( first != last )
{
pResourceType = *first;
if ( pResourceType )
{
pResourceType->Release();
}
m_ResourceTypes.erase( first );
_hr = S_OK;
}
return _hr;
} //*** CClusResTypes::RemoveAt()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResTypes::GetIndex
//
// Description:
// Convert the passed in variant index into the real index in the
// collection.
//
// Arguments:
// varIndex [IN] - The index to convert.
// pnIndex [OUT] - Catches the index.
//
// Return Value:
// S_OK if successful, E_POINTER, or E_INVALIDARG.
//
//--
/////////////////////////////////////////////////////////////////////////////
HRESULT CClusResTypes::GetIndex(
IN VARIANT varIndex,
OUT UINT * pnIndex
)
{
//ASSERT( pnIndex != NULL );
HRESULT _hr = E_POINTER;
if ( pnIndex != NULL )
{
CComVariant v;
UINT nIndex = 0;
*pnIndex = 0;
v.Copy( &varIndex );
// Check to see if the index is a number.
_hr = v.ChangeType( VT_I4 );
if ( SUCCEEDED( _hr ) )
{
nIndex = v.lVal;
nIndex--; // Adjust index to be 0 relative instead of 1 relative
}
else
{
// Check to see if the index is a string.
_hr = v.ChangeType( VT_BSTR );
if ( SUCCEEDED( _hr ) )
{
// Search for the string.
_hr = FindItem( v.bstrVal, &nIndex );
}
}
// We found an index, now check the range.
if ( SUCCEEDED( _hr ) )
{
if ( nIndex < m_ResourceTypes.size() )
{
*pnIndex = nIndex;
}
else
{
_hr = E_INVALIDARG;
}
}
}
return _hr;
} //*** CClusResTypes::GetIndex()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResTypes::get_Item
//
// Description:
// Returns the object (Group) at the passed in index.
//
// Arguments:
// varIndex [IN] - Hold the index. This is a one based number, or
// a string that is the name of the group to get.
// ppResourceType [OUT] - Catches the resource type object.
//
// Return Value:
// S_OK if successful, E_POINTER, or E_INVALIDARG if the index is out
// of range, or other HRESULT error.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResTypes::get_Item(
IN VARIANT varIndex,
OUT ISClusResType ** ppResourceType
)
{
//ASSERT( ppResourceType != NULL );
HRESULT _hr = E_POINTER;
if ( ppResourceType != NULL )
{
CComObject<CClusResType> * pResourceType = NULL;
UINT nIndex = 0;
*ppResourceType = NULL;
_hr = GetIndex( varIndex, &nIndex );
if ( SUCCEEDED( _hr ) )
{
pResourceType = m_ResourceTypes[ nIndex ];
_hr = pResourceType->QueryInterface( IID_ISClusResType, (void **) ppResourceType );
}
}
return _hr;
} //*** CClusResTypes::get_Item()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResTypes::get__NewEnum
//
// Description:
// Create and return a new enumeration for this collection.
//
// Arguments:
// ppunk [OUT] - Catches the new enumeration.
//
// Return Value:
// S_OK if successful, E_POINTER, or other HRESULT error.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResTypes::get__NewEnum( OUT IUnknown ** ppunk )
{
return ::HrNewIDispatchEnum< ResourceTypeList, CComObject< CClusResType > >( ppunk, m_ResourceTypes );
} //*** CClusResTypes::get__NewEnum()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResTypes::CreateItem
//
// Description:
// Create a new object (ResType) and add it to the collection.
//
// Arguments:
// bstrResourceGroupName [IN] - The name of the new group.
// ppResourceGroup [OUT] - Catches the new object.
//
// Return Value:
// S_OK if successful, E_POINTER, or Win32 error as HRESULT.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResTypes::CreateItem(
BSTR bstrResTypeName,
BSTR bstrDisplayName,
BSTR bstrResTypeDll,
long nLooksAliveInterval,
long nIsAliveInterval,
ISClusResType ** ppResourceType
)
{
//ASSERT( bstrResTypeName != NULL );
//ASSERT( bstrDisplayName != NULL );
//ASSERT( bstrResTypeDll != NULL );
//ASSERT( ppResourceType != NULL );
HRESULT _hr = E_POINTER;
if ( ( bstrResTypeName != NULL ) &&
( bstrDisplayName != NULL ) &&
( bstrResTypeDll != NULL ) &&
( ppResourceType != NULL ) )
{
UINT nIndex;
*ppResourceType = NULL;
_hr = FindItem( bstrResTypeName, &nIndex );
if ( FAILED( _hr ) )
{
CComObject<CClusResType> * pClusterResourceType = NULL;
_hr = CComObject< CClusResType >::CreateInstance( &pClusterResourceType );
if ( SUCCEEDED( _hr ) )
{
CSmartPtr< ISClusRefObject > ptrRefObject( m_pClusRefObject );
CSmartPtr< CComObject< CClusResType > > ptrResourceType( pClusterResourceType );
_hr = ptrResourceType->Create( ptrRefObject, bstrResTypeName, bstrDisplayName, bstrResTypeDll, nLooksAliveInterval, nIsAliveInterval );
if ( SUCCEEDED( _hr ) )
{
_hr = ptrResourceType->QueryInterface( IID_ISClusResType, (void **) ppResourceType );
if ( SUCCEEDED( _hr ) )
{
ptrResourceType->AddRef();
m_ResourceTypes.insert( m_ResourceTypes.end(), ptrResourceType );
}
}
}
}
else
{
CComObject<CClusResType> * pClusterResourceType = NULL;
pClusterResourceType = m_ResourceTypes[ nIndex ];
_hr = pClusterResourceType->QueryInterface( IID_ISClusResType, (void **) ppResourceType );
}
}
return _hr;
} //*** CClusResTypes::CreateItem()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResTypes::DeleteItem
//
// Description:
// Deletes the object (ResType) at the passed in index.
//
// Arguments:
// varIndex [IN] - The index of the object to delete.
//
// Return Value:
// S_OK if successful, E_INVALIDARG, or Win32 error as HRESULT.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResTypes::DeleteItem( VARIANT varIndex )
{
HRESULT _hr = S_OK;
UINT nIndex = 0;
_hr = GetIndex( varIndex, &nIndex );
if ( SUCCEEDED( _hr ) )
{
ISClusResType * pResourceType = (ISClusResType *) m_ResourceTypes[ nIndex ];
// delete the resource type
_hr = pResourceType->Delete();
if ( SUCCEEDED( _hr ) )
{
RemoveAt( nIndex );
} // if: the restype was deleted from the cluster
} // if: we have a valid index
return _hr;
} //*** CClusResTypes::DeleteItem()
/////////////////////////////////////////////////////////////////////////////
//++
//
// CClusResTypes::Refresh
//
// Description:
// Load the collection from the cluster.
//
// Arguments:
// None.
//
// Return Value:
// S_OK is successful, or Win32 error as HRESULT if not.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CClusResTypes::Refresh( void )
{
HCLUSTER hCluster = NULL;
HRESULT _hr;
ASSERT( m_pClusRefObject != NULL );
_hr = m_pClusRefObject->get_Handle( (ULONG_PTR *) &hCluster );
if ( SUCCEEDED( _hr ) )
{
HCLUSENUM hEnum = NULL;
DWORD _sc = ERROR_SUCCESS;
hEnum = ::ClusterOpenEnum( hCluster, CLUSTER_ENUM_RESTYPE );
if ( hEnum != NULL )
{
int _nIndex = 0;
DWORD dwType = 0;
LPWSTR pwszName = NULL;
CComObject< CClusResType > * pClusterResourceType = NULL;
Clear();
for( _nIndex = 0, _hr = S_OK; SUCCEEDED( _hr ); _nIndex++ )
{
_sc = ::WrapClusterEnum( hEnum, _nIndex, &dwType, &pwszName );
if ( _sc == ERROR_NO_MORE_ITEMS )
{
_hr = S_OK;
break;
}
else if ( _sc == ERROR_SUCCESS )
{
_hr = CComObject< CClusResType >::CreateInstance( &pClusterResourceType );
if ( SUCCEEDED( _hr ) )
{
CSmartPtr< ISClusRefObject > ptrRefObject( m_pClusRefObject );
CSmartPtr< CComObject< CClusResType > > ptrType( pClusterResourceType );
BSTR bstr = NULL;
bstr = SysAllocString( pwszName );
if ( bstr == NULL )
{
_hr = E_OUTOFMEMORY;
}
else
{
_hr = ptrType->Open( ptrRefObject, bstr );
if ( SUCCEEDED( _hr ) )
{
ptrType->AddRef();
m_ResourceTypes.insert( m_ResourceTypes.end(), ptrType );
}
else if ( HRESULT_CODE( _hr ) == ERROR_RESOURCE_TYPE_NOT_FOUND )
{
//
// It is possible for the resource type to have been deleted from the cluster
// in the time between creating the enum and opening the resourcetype . When
// that happens we need to simply skip that resource type and continue
// enumerating.
//
_hr = S_FALSE; // success code to keep us in the loop
} // else if: the cluster resource type was not found
SysFreeString( bstr );
}
}
::LocalFree( pwszName );
pwszName = NULL;
}
else
{
_hr = HRESULT_FROM_WIN32( _sc );
}
} // for:
::ClusterCloseEnum( hEnum );
}
else
{
_sc = GetLastError();
_hr = HRESULT_FROM_WIN32( _sc );
}
}
return _hr;
} //*** CClusResTypes::Refresh()