windows-nt/Source/XPSP1/NT/base/cluster/admin/msclus/msclus.cpp

500 lines
15 KiB
C++
Raw Normal View History

2020-09-26 03:20:57 -05:00
/////////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 1997-1999 Microsoft Corporation
//
// Module Name:
// MSClus.cpp
//
// Description:
// Implementation of the DLL Exports 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 <atlimpl.cpp>
#include <initguid.h>
#include <ClusRtl.h>
#include "ClusterObject.h"
#include "property.h"
#include "ClusNeti.h"
#include "ClusNetw.h"
#include "ClusRes.h"
#include "ClusRest.h"
#include "ClusResg.h"
#include "ClusNode.h"
#include "Version.h"
#include "ClusApp.h"
#include "Cluster.h"
#define IID_DEFINED
#include "msclus_i.c"
#undef IID_DEFINED
CComModule _Module;
BEGIN_OBJECT_MAP(ObjectMap)
OBJECT_ENTRY(CLSID_ClusApplication, CClusApplication)
OBJECT_ENTRY(CLSID_Cluster, CCluster)
OBJECT_ENTRY(CLSID_ClusVersion, CClusVersion)
OBJECT_ENTRY(CLSID_DomainNames, CDomainNames)
OBJECT_ENTRY(CLSID_ClusResGroupPreferredOwnerNodes, CClusResGroupPreferredOwnerNodes)
OBJECT_ENTRY(CLSID_ClusterNames, CClusterNames)
OBJECT_ENTRY(CLSID_ClusNetInterface, CClusNetInterface)
OBJECT_ENTRY(CLSID_ClusNetInterfaces, CClusNetInterfaces)
OBJECT_ENTRY(CLSID_ClusNetwork, CClusNetwork)
OBJECT_ENTRY(CLSID_ClusNetworks, CClusNetworks)
OBJECT_ENTRY(CLSID_ClusNetworkNetInterfaces, CClusNetworkNetInterfaces)
OBJECT_ENTRY(CLSID_ClusNode, CClusNode)
OBJECT_ENTRY(CLSID_ClusNodes, CClusNodes)
OBJECT_ENTRY(CLSID_ClusNodeNetInterfaces, CClusNodeNetInterfaces)
OBJECT_ENTRY(CLSID_ClusProperty, CClusProperty)
OBJECT_ENTRY(CLSID_ClusProperties, CClusProperties)
OBJECT_ENTRY(CLSID_ClusRefObject, CClusRefObject)
OBJECT_ENTRY(CLSID_ClusResDependencies, CClusResDependencies)
OBJECT_ENTRY(CLSID_ClusResGroup, CClusResGroup)
OBJECT_ENTRY(CLSID_ClusResGroups, CClusResGroups)
OBJECT_ENTRY(CLSID_ClusResource, CClusResource)
OBJECT_ENTRY(CLSID_ClusResources, CClusResources)
OBJECT_ENTRY(CLSID_ClusResPossibleOwnerNodes, CClusResPossibleOwnerNodes)
OBJECT_ENTRY(CLSID_ClusResType, CClusResType)
OBJECT_ENTRY(CLSID_ClusResTypes, CClusResTypes)
OBJECT_ENTRY(CLSID_ClusResTypeResources, CClusResTypeResources)
OBJECT_ENTRY(CLSID_ClusResGroupResources, CClusResGroupResources)
#if CLUSAPI_VERSION >= 0x0500
OBJECT_ENTRY(CLSID_ClusResTypePossibleOwnerNodes, CClusResTypePossibleOwnerNodes)
#endif // CLUSAPI_VERSION >= 0x0500
OBJECT_ENTRY(CLSID_ClusPropertyValue, CClusPropertyValue)
OBJECT_ENTRY(CLSID_ClusPropertyValues, CClusPropertyValues)
OBJECT_ENTRY(CLSID_ClusPropertyValueData, CClusPropertyValueData)
OBJECT_ENTRY(CLSID_ClusPartition, CClusPartition)
OBJECT_ENTRY(CLSID_ClusPartitions, CClusPartitions)
OBJECT_ENTRY(CLSID_ClusDisk, CClusDisk)
OBJECT_ENTRY(CLSID_ClusDisks, CClusDisks)
OBJECT_ENTRY(CLSID_ClusScsiAddress, CClusScsiAddress)
OBJECT_ENTRY(CLSID_ClusRegistryKeys, CClusResourceRegistryKeys)
#if CLUSAPI_VERSION >= 0x0500
OBJECT_ENTRY(CLSID_ClusCryptoKeys, CClusResourceCryptoKeys)
#endif // CLUSAPI_VERSION >= 0x0500
END_OBJECT_MAP()
/////////////////////////////////////////////////////////////////////////////
// Forward function declarations
/////////////////////////////////////////////////////////////////////////////
static void RegisterRegistryCleanUp( void );
static void UnregisterRegistryCleanUp( void );
static const LPWSTR g_ptszRegisterRegistryNodesToDelete[] =
{
_T( "software\\classes\\MSCluster.Application" ),
_T( "software\\classes\\MSCluster.Application.2" ),
_T( "software\\classes\\MSCluster.Cluster.2" ),
_T( "software\\classes\\MSCluster.ClusGroupResources" ),
_T( "software\\classes\\MSCluster.ClusGroupResources.1" ),
_T( "software\\classes\\MSCluster.ClusGroupOwners" ),
_T( "software\\classes\\MSCluster.ClusGroupOwners.1" ),
_T( "software\\classes\\MSCluster.ClusResOwners" ),
_T( "software\\classes\\MSCluster.ClusResOwners.1" ),
_T( "software\\classes\\CLSID\\{f2e60717-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e60718-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\CLSID\\{f2e60719-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e6071a-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606e0-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606e1-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606e3-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606e5-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606e7-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606e9-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606eb-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606ed-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606ef-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606f3-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606f5-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606f7-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606f9-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606fb-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606fd-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606fe-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606ff-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e60700-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e60702-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e60704-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\CLSID\\{f2e606f2-2631-11d1-89f1-00a0c90d061e}" ),
NULL
};
/*
static const LPWSTR g_ptszUnregisterRegistryNodesToDelete[] =
{
// _T( "software\\classes\\typelib\\{f2e606e0-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606e2-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606e4-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606e6-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606e8-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606ea-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606ec-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606ee-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606f0-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606f2-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606f4-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606f6-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606f8-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606fa-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606fc-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e606fe-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e60700-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e60702-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e60704-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e60706-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e60708-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e6070a-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e6070c-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e6070e-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e60710-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e60712-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e60714-2631-11d1-89f1-00a0c90d061e}" ),
_T( "software\\classes\\interface\\{f2e60716-2631-11d1-89f1-00a0c90d061e}" ),
NULL
};
*/
/////////////////////////////////////////////////////////////////////////////
//++
//
// DllMain
//
// Description:
// DLL Entry Point.
//
// Arguments:
// hInstance [IN] - Out instance handle.
// dwReason [IN] - The reason we are being called.
// lpReserved [IN] - Don't rightly know what this is...
//
// Return Value:
// TRUE if successful, FALSE if not.
//
//--
/////////////////////////////////////////////////////////////////////////////
extern "C" BOOL WINAPI DllMain(
IN HINSTANCE hInstance,
IN DWORD dwReason,
IN LPVOID //lpReserved
)
{
if ( dwReason == DLL_PROCESS_ATTACH )
{
#ifdef _DEBUG
_CrtSetDbgFlag( _CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF );
_CrtSetReportMode( _CRT_ERROR, _CRTDBG_MODE_DEBUG );
#endif
_Module.Init( ObjectMap, hInstance );
DisableThreadLibraryCalls( hInstance );
}
else if ( dwReason == DLL_PROCESS_DETACH )
{
_Module.Term();
}
return TRUE; // ok
} //*** DllMain()
/////////////////////////////////////////////////////////////////////////////
//++
//
// DllCanUnloadNow
//
// Description:
// Used to determine whether the DLL can be unloaded by OLE.
//
// Arguments:
// None.
//
// Return Value:
// S_OK if we can unload, S_FALSE if we cannot.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDAPI DllCanUnloadNow( void )
{
return ( _Module.GetLockCount() == 0 ) ? S_OK : S_FALSE;
} //*** DllCanUnloadNow()
/////////////////////////////////////////////////////////////////////////////
//++
//
// DllGetClassObject
//
// Description:
// Retrieves the class object from a DLL object handler or object
// application. DllGetClassObject is called from within the
// CoGetClassObject function when the class context is a DLL.
//
// Arguments:
// rclsid [IN] - CLSID that will associate the correct data and code.
// riid [IN] - Reference to the identifier of the interface that the
// caller is to use to communicate with the class object.
// Usually, this is IID_IClassFactory (defined in the OLE
// headers as the interface identifier for IClassFactory).
// ppv [OUT] - Address of pointer variable that receives the interface
// pointer requested in riid. Upon successful return, *ppv
// contains the requested interface pointer. If an error
// occurs, the interface pointer is NULL.
//
// Return Value:
// S_OK if successful, or CLASS_E_CLASSNOTAVAILABLE if not.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDAPI DllGetClassObject(
IN REFCLSID rclsid,
IN REFIID riid,
OUT LPVOID * ppv
)
{
return _Module.GetClassObject( rclsid, riid, ppv );
} //*** DllGetClassObject()
/////////////////////////////////////////////////////////////////////////////
//++
//
// DllRegisterServer
//
// Description:
// Add entries to the system registry.
//
// Arguments:
// None.
//
// Return Value:
// S_OK if successful, or other HRESULT error if not.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDAPI DllRegisterServer( void )
{
RegisterRegistryCleanUp();
//
// Registers object, typelib and all interfaces in typelib
//
return _Module.RegisterServer( TRUE );
} //*** DllRegisterServer()
/////////////////////////////////////////////////////////////////////////////
//++
//
// DllUnregisterServer
//
// Description:
// Removes entries from the system registry.
//
// Arguments:
// None.
//
// Return Value:
// S_OK if successful, or other HRESULT error if not.
//
//--
/////////////////////////////////////////////////////////////////////////////
STDAPI DllUnregisterServer( void )
{
HRESULT _hr = S_FALSE;
UnregisterRegistryCleanUp();
//
// Unregisters object, typelib and all interfaces in typelib
//
_hr = _Module.UnregisterServer();
if ( SUCCEEDED( _hr ) )
{
#if _WIN32_WINNT >= 0x0400
_hr = UnRegisterTypeLib( LIBID_MSClusterLib, 1, 0, LOCALE_NEUTRAL, SYS_WIN32 );
#endif
} //if: server was unregistered
return _hr;
} //*** DllUnregisterServer()
/////////////////////////////////////////////////////////////////////////////
//++
//
// HrGetCluster
//
// Description:
// Common implementation of creating a new cluster object.
//
// Arguments:
// ppCluster [OUT] - Catches the newly created object.
// pClusRefObject [IN] - Object that wraps the cluster handle.
//
// Return Value:
// S_OK for success
//
//--
/////////////////////////////////////////////////////////////////////////////
HRESULT HrGetCluster(
OUT ISCluster ** ppCluster,
IN ISClusRefObject * pClusRefObject
)
{
//ASSERT( ppCluster != NULL );
ASSERT( pClusRefObject != NULL );
HRESULT _hr = E_POINTER;
if ( ( ppCluster != NULL ) && ( pClusRefObject != NULL ) )
{
HCLUSTER hCluster = NULL;
_hr = pClusRefObject->get_Handle( (ULONG_PTR *) &hCluster );
if ( SUCCEEDED( _hr ) )
{
CComObject< CCluster > * pCluster = NULL;
_hr = CComObject< CCluster >::CreateInstance( &pCluster );
if ( SUCCEEDED( _hr ) )
{
pCluster->ClusRefObject( pClusRefObject );
pCluster->Hcluster( hCluster );
_hr = pCluster->QueryInterface( IID_ISCluster, (void **) ppCluster );
}
}
}
return _hr;
} //*** HrGetCluster()
/////////////////////////////////////////////////////////////////////////////
//++
//
// RegisterRegistryCleanUp
//
// Description:
// Clean up the registry during registration
//
// Arguments:
// None.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
static void RegisterRegistryCleanUp( void )
{
int nIndex;;
for ( nIndex = 0; ; nIndex++ )
{
if ( g_ptszRegisterRegistryNodesToDelete[ nIndex ] == NULL )
{
break;
} // if:
RegDelnode( HKEY_LOCAL_MACHINE, g_ptszRegisterRegistryNodesToDelete[ nIndex ] );
} // for:
} //*** RegisterRegistryCleanUp()
/////////////////////////////////////////////////////////////////////////////
//++
//
// UnregisterRegistryCleanUp
//
// Description:
// Clean up the registry during unregistration
//
// Arguments:
// None.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
static void UnregisterRegistryCleanUp( void )
{
return;
} //*** UnregisterRegistryCleanUp()
/////////////////////////////////////////////////////////////////////////////
//++
//
// ClearIDispatchEnum
//
// Description:
// Cleans up an Enum of IDispatch pointers.
//
// Arguments:
// ppVarVect [IN OUT] - The enum to clean up.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
void ClearIDispatchEnum(
IN OUT CComVariant ** ppvarVect
)
{
if ( ppvarVect != NULL )
{
size_t cCount = ARRAYSIZE( *ppvarVect );
size_t iIndex;
for ( iIndex = 0; iIndex < cCount; iIndex++ )
{
(*ppvarVect[iIndex]).pdispVal->Release();
} // for:
delete [] *ppvarVect;
*ppvarVect = NULL;
}
} //*** ClearIDispatchEnum()
/////////////////////////////////////////////////////////////////////////////
//++
//
// ClearVariantEnum
//
// Description:
// Cleans up an Enum of variant values.
//
// Arguments:
// ppVarVect [IN OUT] - The enum to clean up.
//
// Return Value:
// None.
//
//--
/////////////////////////////////////////////////////////////////////////////
void ClearVariantEnum(
IN OUT CComVariant ** ppvarVect
)
{
if ( ppvarVect != NULL )
{
delete [] *ppvarVect;
*ppvarVect = NULL;
}
} //*** ClearVariantEnum()