windows-nt/Source/XPSP1/NT/base/cluster/mgmt/cluscfg/w2kproxy/cproxycfgclusterinfo.cpp
2020-09-26 16:20:57 +08:00

915 lines
20 KiB
C++

//////////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 2000-2001 Microsoft Corporation
//
// Module Name:
// ConfigClusApi.cpp
//
// Description:
// CConfigClusApi implementation.
//
// Maintained By:
// Galen Barbee (GalenB) 02-AUG-2000
//
//////////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////////
// Include Files
//////////////////////////////////////////////////////////////////////////////
#include "pch.h"
#include "CProxyCfgClusterInfo.h"
#include "CClusCfgCredentials.h"
#include "CProxyCfgNetworkInfo.h"
//////////////////////////////////////////////////////////////////////////////
// Constant Definitions
//////////////////////////////////////////////////////////////////////////////
DEFINE_THISCLASS("CProxyCfgClusterInfo")
//
//
//
HRESULT
CProxyCfgClusterInfo::S_HrCreateInstance(
IUnknown ** ppunkOut,
IUnknown * punkOuterIn,
HCLUSTER * phClusterIn,
CLSID * pclsidMajorIn,
LPCWSTR pcszDomainIn
)
{
TraceFunc( "" );
HRESULT hr = S_OK;
CProxyCfgClusterInfo * pcc = NULL;
if ( ppunkOut == NULL )
{
hr = THR( E_POINTER );
goto Cleanup;
} // if:
pcc = new CProxyCfgClusterInfo;
if ( pcc == NULL )
{
hr = THR( E_OUTOFMEMORY );
goto Cleanup;
} // if:
hr = THR( pcc->HrInit( punkOuterIn, phClusterIn, pclsidMajorIn, pcszDomainIn ) );
if ( FAILED( hr ) )
goto Cleanup;
hr = THR( pcc->TypeSafeQI( IUnknown, ppunkOut ) );
Cleanup:
if ( pcc != NULL )
{
pcc->Release();
} // if:
HRETURN( hr );
} //*** CProxyCfgClusterInfo::S_HrCreateInstance()
//
//
//
CProxyCfgClusterInfo::CProxyCfgClusterInfo( void )
: m_cRef( 1 )
{
TraceFunc( "" );
InterlockedIncrement( &g_cObjects );
Assert( m_cRef == 1 );
Assert( m_punkOuter == NULL );
Assert( m_pcccb == NULL );
Assert( m_phCluster == NULL );
Assert( m_pclsidMajor == NULL );
Assert( m_bstrClusterName == NULL);
Assert( m_ulIPAddress == 0 );
Assert( m_ulSubnetMask == 0 );
Assert( m_bstrNetworkName == NULL);
Assert( m_pccc == NULL );
Assert( m_bstrBindingString == NULL );
TraceFuncExit();
} //*** CProxyCfgClusterInfo::CProxyCfgClusterInfo()
//
//
//
CProxyCfgClusterInfo::~CProxyCfgClusterInfo( void )
{
TraceFunc( "" );
// m_cRef - noop
if ( m_punkOuter != NULL )
{
m_punkOuter->Release( );
}
if ( m_pcccb != NULL )
{
m_pcccb->Release();
} // if:
// m_phCluster - DO NOT CLOSE!
// m_pclsidMajor - noop
TraceSysFreeString( m_bstrClusterName );
// m_ulIPAddress
// m_ulSubnetMask
TraceSysFreeString( m_bstrNetworkName );
TraceSysFreeString( m_bstrBindingString );
if ( m_pccc != NULL )
{
m_pccc->Release( );
}
InterlockedDecrement( &g_cObjects );
TraceFuncExit();
} //*** CProxyCfgClusterInfo::~CProxyCfgClusterInfo()
//
//
//
HRESULT
CProxyCfgClusterInfo::HrInit(
IUnknown * punkOuterIn,
HCLUSTER * phClusterIn,
CLSID * pclsidMajorIn,
LPCWSTR pcszDomainIn
)
{
TraceFunc( "" );
HRESULT hr;
DWORD sc;
DWORD cch;
BSTR bstrClusterName = NULL;
CLUSTERVERSIONINFO cvi;
if ( punkOuterIn != NULL )
{
m_punkOuter = punkOuterIn;
m_punkOuter->AddRef( );
}
if ( phClusterIn == NULL )
goto InvalidArg;
m_phCluster = phClusterIn;
if ( pclsidMajorIn != NULL )
{
m_pclsidMajor = pclsidMajorIn;
}
else
{
m_pclsidMajor = (CLSID *) &TASKID_Major_Client_And_Server_Log;
}
if ( punkOuterIn != NULL )
{
hr = THR( punkOuterIn->TypeSafeQI( IClusCfgCallback, &m_pcccb ) );
if ( FAILED( hr ) )
goto Cleanup;
}
//
// Get the cluster's name and version info.
//
cvi.dwVersionInfoSize = sizeof(cvi);
cch = 64; // arbitary starting buffer size
bstrClusterName = TraceSysAllocStringLen( NULL, cch );
if ( bstrClusterName == NULL )
goto OutOfMemory;
cch ++; // SysAllocStringLen allocates an extra character
sc = GetClusterInformation( *m_phCluster, bstrClusterName, &cch, &cvi );
if ( sc == ERROR_MORE_DATA )
{
TraceSysFreeString( bstrClusterName );
bstrClusterName = TraceSysAllocStringLen( NULL, cch );
if ( bstrClusterName == NULL )
goto OutOfMemory;
cch ++; // SysAllocStringLen allocates an extra character
sc = GetClusterInformation( *m_phCluster, bstrClusterName, &cch, &cvi );
}
if ( sc != ERROR_SUCCESS )
{
hr = HRESULT_FROM_WIN32( TW32( sc ) );
SSR_W2KPROXY_STATUS( TASKID_Major_Client_And_Server_Log, TASKID_Minor_HrInit_GetClusterInformation_Failed, hr );
goto Cleanup;
}
// Give up ownership
m_bstrClusterName = TraceSysAllocStringLen( NULL, SysStringLen( bstrClusterName ) + 1 + (UINT) wcslen( pcszDomainIn ) );
if ( m_bstrClusterName == NULL )
goto OutOfMemory;
wcscpy( m_bstrClusterName, bstrClusterName );
wcscat( m_bstrClusterName, L"." );
wcscat( m_bstrClusterName, pcszDomainIn );
Assert( m_bstrNetworkName == NULL );
hr = THR( HrGetIPAddressOfCluster( *m_phCluster, &m_ulIPAddress, &m_ulSubnetMask, &m_bstrNetworkName ) );
if ( FAILED( hr ) )
goto Cleanup;
hr = THR( HrLoadCredentials( ) );
if ( FAILED( hr ) )
goto Cleanup;
hr = S_OK;
Cleanup:
TraceSysFreeString( bstrClusterName );
HRETURN( hr );
InvalidArg:
hr = THR( E_INVALIDARG );
SSR_W2KPROXY_STATUS( TASKID_Major_Client_And_Server_Log, TASKID_Minor_W2KProxy_ClusterInfo_HrInit_InvalidArg, hr );
goto Cleanup;
OutOfMemory:
hr = E_OUTOFMEMORY;
SSR_W2KPROXY_STATUS( TASKID_Major_Client_And_Server_Log, TASKID_Minor_W2KProxy_ClusterInfo_HrInit_OutOfMemory, hr );
goto Cleanup;
} //*** CProxyCfgClusterInfo::HrInit()
//*************************************************************************//
/////////////////////////////////////////////////////////////////////////////
// CProxyCfgClusterInfo -- IUknkown interface.
/////////////////////////////////////////////////////////////////////////////
//
//
//
STDMETHODIMP
CProxyCfgClusterInfo::QueryInterface(
REFIID riid,
LPVOID *ppv
)
{
TraceQIFunc( riid, ppv );
HRESULT hr = E_NOINTERFACE;
if ( IsEqualIID( riid, IID_IUnknown ) )
{
*ppv = static_cast< IClusCfgClusterInfo * >( this );
hr = S_OK;
} // if: IUnknown
else if ( IsEqualIID( riid, IID_IClusCfgClusterInfo ) )
{
*ppv = TraceInterface( __THISCLASS__, IClusCfgClusterInfo, this, 0 );
hr = S_OK;
} // else if:
if ( SUCCEEDED( hr ) )
{
((IUnknown*) *ppv)->AddRef();
} // if: success
QIRETURN_IGNORESTDMARSHALLING( hr, riid );
} //*** CConfigClusApi::QueryInterface()
//////////////////////////////////////////////////////////////////////////////
//++
//
// STDMETHODIMP_( ULONG )
// CProxyCfgClusterInfo:: [IUNKNOWN] AddRef()
//
// Description:
// Increment the reference count of this object by one.
//
// Arguments:
// None.
//
// Return Value:
// The new reference count.
//
// Remarks:
// None.
//
//--
//////////////////////////////////////////////////////////////////////////////
STDMETHODIMP_(ULONG)
CProxyCfgClusterInfo::AddRef( void )
{
TraceFunc( "[IUnknown]" );
InterlockedIncrement( &m_cRef );
RETURN( m_cRef );
} //*** CProxyCfgClusterInfo::AddRef()
//////////////////////////////////////////////////////////////////////////////
//++
//
// STDMETHODIMP_( ULONG )
// CProxyCfgClusterInfo:: [IUNKNOWN] Release()
//
// Description:
// Decrement the reference count of this object by one.
//
// Arguments:
// None.
//
// Return Value:
// The new reference count.
//
// Remarks:
// None.
//
//--
//////////////////////////////////////////////////////////////////////////////
STDMETHODIMP_(ULONG)
CProxyCfgClusterInfo::Release( void )
{
TraceFunc( "[IUnknown]" );
InterlockedDecrement( &m_cRef );
if ( m_cRef )
RETURN( m_cRef );
TraceDo( delete this );
RETURN( 0 );
} //*** CProxyCfgClusterInfo::Release()
//****************************************************************************
//
// IClusCfgClusterInfo
//
//****************************************************************************
//
//
//
STDMETHODIMP
CProxyCfgClusterInfo::GetName(
BSTR * pbstrNameOut
)
{
TraceFunc( "[IClusCfgClusterInfo]" );
HRESULT hr = S_OK;
if ( pbstrNameOut == NULL )
goto InvalidPointer;
*pbstrNameOut = SysAllocString( m_bstrClusterName );
if ( *pbstrNameOut == NULL )
{
hr = THR( E_OUTOFMEMORY );
}
CharLower( *pbstrNameOut );
Cleanup:
HRETURN( hr );
InvalidPointer:
hr = THR( E_POINTER );
SSR_W2KPROXY_STATUS( TASKID_Major_Client_And_Server_Log, TASKID_Minor_W2kProxy_ClusterInfo_GetName_InvalidPointer, hr );
goto Cleanup;
} //*** CProxyCfgClusterInfo::GetName()
//
//
//
STDMETHODIMP
CProxyCfgClusterInfo::GetIPAddress(
DWORD * pdwIPAddress
)
{
TraceFunc( "[IClusCfgClusterInfo]" );
HRESULT hr = S_OK;
if ( pdwIPAddress == NULL )
goto InvalidPointer;
*pdwIPAddress = m_ulIPAddress;
Cleanup:
HRETURN( hr );
InvalidPointer:
hr = THR( E_POINTER );
SSR_W2KPROXY_STATUS( TASKID_Major_Client_And_Server_Log, TASKID_Minor_W2KProxy_ClusterInfo_GetIPAddress_InvalidPointer, hr );
goto Cleanup;
} //*** CProxyCfgClusterInfo::GetIPAddress()
//
//
//
STDMETHODIMP
CProxyCfgClusterInfo::GetSubnetMask(
DWORD * pdwNetMask
)
{
TraceFunc( "[IClusCfgClusterInfo]" );
HRESULT hr = S_OK;
if ( pdwNetMask == NULL )
goto InvalidPointer;
*pdwNetMask = m_ulSubnetMask;
Cleanup:
HRETURN( hr );
InvalidPointer:
hr = THR( E_POINTER );
SSR_W2KPROXY_STATUS( TASKID_Major_Client_And_Server_Log, TASKID_Minor_W2KProxy_ClusterInfo_GetSubnetMask_InvalidPointer, hr );
goto Cleanup;
} //*** CProxyCfgClusterInfo::GetSubnetMask()
//
//
//
STDMETHODIMP
CProxyCfgClusterInfo::GetNetworkInfo(
IClusCfgNetworkInfo ** ppICCNetInfoOut
)
{
TraceFunc( "[IClusCfgClusterInfo]" );
HRESULT hr;
IUnknown * punk = NULL;
if ( ppICCNetInfoOut == NULL )
goto InvalidPointer;
//
// Create the network info object.
//
hr = THR( CProxyCfgNetworkInfo::S_HrCreateInstance( &punk,
m_punkOuter,
m_phCluster,
m_pclsidMajor,
m_bstrNetworkName
) );
if ( FAILED( hr ) )
goto Cleanup;
hr = THR( punk->TypeSafeQI( IClusCfgNetworkInfo, ppICCNetInfoOut ) );
Cleanup:
if ( punk != NULL )
{
punk->Release();
}
HRETURN( hr );
InvalidPointer:
hr = THR( E_POINTER );
SSR_W2KPROXY_STATUS( TASKID_Major_Client_And_Server_Log, TASKID_Minor_GetNetworkInfo_InvalidPointer, hr );
goto Cleanup;
} //*** CProxyCfgClusterInfo::GetNetworkInfo()
//
//
//
STDMETHODIMP
CProxyCfgClusterInfo::GetClusterServiceAccountCredentials(
IClusCfgCredentials ** ppICCCredentialsOut
)
{
TraceFunc( "[IClusCfgClusterInfo]" );
HRESULT hr;
if ( ppICCCredentialsOut == NULL )
goto InvalidPointer;
hr = THR( m_pccc->TypeSafeQI( IClusCfgCredentials, ppICCCredentialsOut ) );
if ( FAILED( hr ) )
goto Cleanup;
Cleanup:
HRETURN( hr );
InvalidPointer:
hr = THR( E_POINTER );
SSR_W2KPROXY_STATUS( TASKID_Major_Client_And_Server_Log, TASKID_Minor_GetClusterServiceAccountCredentials_InvalidPointer, hr );
goto Cleanup;
} //*** CProxyCfgClusterInfo::GetClusterServiceAccountCredentials()
//
//
//
STDMETHODIMP
CProxyCfgClusterInfo::GetBindingString(
BSTR * pbstrBindingStringOut
)
{
TraceFunc( "[IClusCfgClusterInfo]" );
HRESULT hr = S_OK;
if ( pbstrBindingStringOut == NULL )
goto InvalidPointer;
*pbstrBindingStringOut = SysAllocString( m_bstrBindingString );
if ( *pbstrBindingStringOut == NULL )
{
hr = THR( E_OUTOFMEMORY );
}
Cleanup:
HRETURN( hr );
InvalidPointer:
hr = THR( E_POINTER );
SSR_W2KPROXY_STATUS( TASKID_Major_Client_And_Server_Log, TASKID_Minor_W2kProxy_ClusterInfo_GetBindingString_InvalidPointer, hr );
goto Cleanup;
} //*** CProxyCfgClusterInfo::GetBindingString()
//
//
//
STDMETHODIMP
CProxyCfgClusterInfo::SetCommitMode( ECommitMode ecmNewModeIn )
{
TraceFunc( "[IClusCfgClusterInfo]" );
Assert( ecmNewModeIn != cmUNKNOWN );
HRESULT hr = THR( E_NOTIMPL );
HRETURN( hr );
} //*** CProxyCfgClusterInfo::SetCommitMode()
//
//
//
STDMETHODIMP
CProxyCfgClusterInfo::GetCommitMode( ECommitMode * pecmCurrentModeOut )
{
TraceFunc( "[IClusCfgClusterInfo]" );
Assert( pecmCurrentModeOut != NULL );
HRESULT hr = S_FALSE;
HRETURN( hr );
} //*** CProxyCfgClusterInfo::GetCommitMode()
//
//
//
STDMETHODIMP
CProxyCfgClusterInfo::SetName( LPCWSTR pcszNameIn )
{
TraceFunc1( "[IClusCfgClusterInfo] pcszNameIn = '%ls'", pcszNameIn );
HRESULT hr = THR( E_NOTIMPL );
HRETURN( hr );
} //*** CProxyCfgClusterInfo::SetName()
//
//
//
STDMETHODIMP
CProxyCfgClusterInfo::SetIPAddress( DWORD dwIPAddressIn )
{
TraceFunc( "[IClusCfgClusterInfo]" );
HRESULT hr = S_OK;
if ( dwIPAddressIn != m_ulIPAddress )
{
hr = THR( E_FAIL );
}
HRETURN( hr );
} //*** CProxyCfgClusterInfo::SetIPAddress()
//
//
//
STDMETHODIMP
CProxyCfgClusterInfo::SetSubnetMask( DWORD dwNetMaskIn )
{
TraceFunc( "[IClusCfgClusterInfo]" );
HRESULT hr = S_OK;
if ( dwNetMaskIn != m_ulSubnetMask )
{
hr = THR( E_FAIL );
}
HRETURN( hr );
} //*** CProxyCfgClusterInfo::SetSubnetMask()
//
//
//
STDMETHODIMP
CProxyCfgClusterInfo::SetNetworkInfo( IClusCfgNetworkInfo * pICCNetInfoIn )
{
TraceFunc( "[IClusCfgClusterInfo]" );
HRESULT hr = THR( E_NOTIMPL );
HRETURN( hr );
} //*** CProxyCfgClusterInfo::SetNetworkInfo()
//
//
//
STDMETHODIMP
CProxyCfgClusterInfo::SetBindingString( LPCWSTR pcszBindingStringIn )
{
TraceFunc1( "[IClusCfgClusterInfo] pcszBindingStringIn = '%ls'", pcszBindingStringIn );
HRESULT hr = S_OK;
BSTR bstr = NULL;
if ( pcszBindingStringIn == NULL )
{
hr = THR( E_INVALIDARG );
goto Cleanup;
} // if:
bstr = TraceSysAllocString( pcszBindingStringIn );
if ( bstr == NULL )
{
hr = THR( E_OUTOFMEMORY );
goto Cleanup;
} // if:
TraceSysFreeString( m_bstrBindingString );
m_bstrBindingString = bstr;
Cleanup:
HRETURN( hr );
} //*** CProxyCfgClusterInfo::SetName()
//****************************************************************************
//
// IClusCfgCallback
//
//****************************************************************************
//////////////////////////////////////////////////////////////////////////////
//++
//
// CProxyCfgClusterInfo::SendStatusReport()
//
// Description:
//
// Arguments:
//
// Return Value:
//
// Remarks:
// None.
//
//--
//////////////////////////////////////////////////////////////////////////////
STDMETHODIMP
CProxyCfgClusterInfo::SendStatusReport(
LPCWSTR pcszNodeNameIn
, CLSID clsidTaskMajorIn
, CLSID clsidTaskMinorIn
, ULONG ulMinIn
, ULONG ulMaxIn
, ULONG ulCurrentIn
, HRESULT hrStatusIn
, LPCWSTR pcszDescriptionIn
, FILETIME * pftTimeIn
, LPCWSTR pcszReferenceIn
)
{
TraceFunc( "[IClusCfgCallback]" );
HRESULT hr = S_OK;
if ( m_pcccb != NULL )
{
hr = THR( m_pcccb->SendStatusReport( pcszNodeNameIn,
clsidTaskMajorIn,
clsidTaskMinorIn,
ulMinIn,
ulMaxIn,
ulCurrentIn,
hrStatusIn,
pcszDescriptionIn,
pftTimeIn,
pcszReferenceIn
) );
} // if:
HRETURN( hr );
} //*** CProxyCfgClusterInfo::SendStatusReport()
//****************************************************************************
//
// IClusCfgCallback
//
//****************************************************************************
//////////////////////////////////////////////////////////////////////////////
//++
//
// CProxyCfgClusterInfo::HrLoadCredentials()
//
// Description:
//
//
// Arguments:
//
//
// Return Value:
//
//
// Remarks:
// None.
//
//--
//////////////////////////////////////////////////////////////////////////////
HRESULT
CProxyCfgClusterInfo::HrLoadCredentials( void )
{
TraceFunc( "" );
HRESULT hr = S_OK;
SC_HANDLE schSCM = NULL;
SC_HANDLE schClusSvc = NULL;
DWORD sc;
DWORD cbpqsc = 128;
DWORD cbRequired;
QUERY_SERVICE_CONFIG * pqsc = NULL;
IUnknown * punk = NULL;
IClusCfgSetCredentials * piccsc = NULL;
schSCM = OpenSCManager( m_bstrClusterName, NULL, GENERIC_READ );
if ( schSCM == NULL )
{
hr = HRESULT_FROM_WIN32( TW32( GetLastError() ) );
SSR_W2KPROXY_STATUS( TASKID_Major_Client_And_Server_Log, TASKID_Minor_HrLoadCredentials_OpenSCManager_Failed, hr );
goto Cleanup;
} // if:
schClusSvc = OpenService( schSCM, L"ClusSvc", GENERIC_READ );
if ( schClusSvc == NULL )
{
hr = HRESULT_FROM_WIN32( TW32( GetLastError() ) );
SSR_W2KPROXY_STATUS( TASKID_Major_Client_And_Server_Log, TASKID_Minor_HrLoadCredentials_OpenService_Failed, hr );
goto Cleanup;
} // if:
for ( ; ; )
{
BOOL fRet;
pqsc = (QUERY_SERVICE_CONFIG *) TraceAlloc( 0, cbpqsc );
if ( pqsc == NULL )
goto OutOfMemory;
fRet = QueryServiceConfig( schClusSvc, pqsc, cbpqsc, &cbRequired );
if ( !fRet )
{
sc = GetLastError();
if ( sc == ERROR_INSUFFICIENT_BUFFER )
{
TraceFree( pqsc );
pqsc = NULL;
cbpqsc = cbRequired;
continue;
} // if:
else
{
hr = HRESULT_FROM_WIN32( TW32( sc ) );
SSR_W2KPROXY_STATUS( TASKID_Major_Client_And_Server_Log, TASKID_Minor_HrLoadCredentials_QueryServiceConfig_Failed, hr );
goto Cleanup;
} // else:
} // if:
else
{
break;
} // else:
} // for:
Assert( m_pccc == NULL );
hr = THR( CClusCfgCredentials::S_HrCreateInstance( &punk ) );
if ( FAILED( hr ) )
goto Cleanup;
hr = THR( punk->TypeSafeQI( IClusCfgCredentials, &m_pccc ) );
if ( FAILED( hr ) )
goto Cleanup;
hr = THR( m_pccc->TypeSafeQI( IClusCfgSetCredentials, &piccsc ) );
if ( FAILED( hr ) )
{
goto Cleanup;
} // if:
hr = THR( piccsc->SetDomainCredentials( pqsc->lpServiceStartName ) );
Cleanup:
if ( punk != NULL )
{
punk->Release( );
}
if ( schClusSvc != NULL )
{
CloseServiceHandle( schClusSvc );
} // if:
if ( schSCM != NULL )
{
CloseServiceHandle( schSCM );
} // if:
if ( pqsc != NULL )
{
TraceFree( pqsc );
} // if:
if ( piccsc != NULL )
{
piccsc->Release();
} // if:
HRETURN( hr );
OutOfMemory:
hr = E_OUTOFMEMORY;
SSR_W2KPROXY_STATUS( TASKID_Major_Client_And_Server_Log, TASKID_Minor_HrLoadCredentials_OutOfMemory, hr );
goto Cleanup;
} //*** CProxyCfgClusterInfo::HrLoadCredentials()