|
|
#include "WLBS_Provider.h"
#include "WLBS_clustersetting.h"
#include "ClusterWrapper.h"
#include "ControlWrapper.h"
#include "utils.h"
#include "param.h"
////////////////////////////////////////////////////////////////////////////////
//
// CWLBS_ClusterSetting::CWLBS_ClusterSetting
//
// Purpose: Constructor
//
////////////////////////////////////////////////////////////////////////////////
CWLBS_ClusterSetting::CWLBS_ClusterSetting ( CWbemServices* a_pNameSpace, IWbemObjectSink* a_pResponseHandler ) : CWlbs_Root( a_pNameSpace, a_pResponseHandler ) { }
////////////////////////////////////////////////////////////////////////////////
//
// CWLBS_ClusterSetting::Create
//
// Purpose: This instantiates this class and is invoked from an array of
// function pointers.
//
////////////////////////////////////////////////////////////////////////////////
CWlbs_Root* CWLBS_ClusterSetting::Create ( CWbemServices* a_pNameSpace, IWbemObjectSink* a_pResponseHandler ) {
CWlbs_Root* pRoot = new CWLBS_ClusterSetting( a_pNameSpace, a_pResponseHandler );
if( !pRoot ) throw _com_error( WBEM_E_OUT_OF_MEMORY );
return pRoot; }
////////////////////////////////////////////////////////////////////////////////
//
// CWLBS_ClusterSetting::GetInstance
//
// Purpose:
//
////////////////////////////////////////////////////////////////////////////////
HRESULT CWLBS_ClusterSetting::GetInstance ( const ParsedObjectPath* a_pParsedPath, long /* a_lFlags */, IWbemContext* /* a_pIContex */ ) { IWbemClassObject* pWlbsInstance = NULL; HRESULT hRes = 0;
try {
wstring wstrHostName; //get the name key property and convert to wstring
//throws _com_error
wstrHostName = (*a_pParsedPath->m_paKeys)->m_vValue.bstrVal;
//get the cluster
CWlbsClusterWrapper* pCluster = GetClusterFromHostName(g_pWlbsControl, wstrHostName); if (pCluster == NULL) { throw _com_error( WBEM_E_NOT_FOUND ); }
//get the Wbem class instance
SpawnInstance( MOF_CLUSTERSETTING::szName, &pWlbsInstance );
//Convert status to string description
FillWbemInstance( pWlbsInstance, pCluster );
//send the results back to WinMgMt
m_pResponseHandler->Indicate( 1, &pWlbsInstance );
if( pWlbsInstance ) {
pWlbsInstance->Release(); pWlbsInstance = NULL;
}
m_pResponseHandler->SetStatus( 0, WBEM_S_NO_ERROR, NULL, NULL );
hRes = WBEM_S_NO_ERROR; }
catch(CErrorWlbsControl Err) {
IWbemClassObject* pWbemExtStat = NULL;
CreateExtendedStatus( m_pNameSpace, &pWbemExtStat, Err.Error(), (PWCHAR)(Err.Description()) );
m_pResponseHandler->SetStatus(0, WBEM_E_FAILED, NULL, pWbemExtStat);
if( pWbemExtStat ) pWbemExtStat->Release();
if( pWlbsInstance ) pWlbsInstance->Release();
//do not return WBEM_E_FAILED, this causes a race condition
hRes = WBEM_S_NO_ERROR; }
catch(_com_error HResErr ) {
m_pResponseHandler->SetStatus(0, HResErr.Error(), NULL, NULL);
if( pWlbsInstance ) pWlbsInstance->Release();
hRes = HResErr.Error(); //transform Win32 error to a WBEM error
if( hRes == ERROR_FILE_NOT_FOUND ) hRes = WBEM_E_NOT_FOUND; }
catch(...) {
if( pWlbsInstance ) pWlbsInstance->Release();
throw;
}
return hRes; }
////////////////////////////////////////////////////////////////////////////////
//
// CWLBS_ClusterSetting::EnumInstances
//
// Purpose: This function obtains the clustersetting data for the current host.
// The node does not have to be a member of a cluster for this
// to succeed. However, WLBS must be installed.
//
////////////////////////////////////////////////////////////////////////////////
HRESULT CWLBS_ClusterSetting::EnumInstances ( BSTR /* a_bstrClass */, long /* a_lFlags */, IWbemContext* /* a_pIContex */ ) { IWbemClassObject* pWlbsInstance = NULL; HRESULT hRes = 0;
try {
DWORD dwNumClusters = 0; CWlbsClusterWrapper** ppCluster = NULL;
g_pWlbsControl->EnumClusters(ppCluster, &dwNumClusters); if (dwNumClusters == 0) { throw _com_error( WBEM_E_NOT_FOUND ); }
for (DWORD i=0; i < dwNumClusters; i++) { //get the Wbem class instance
SpawnInstance( MOF_CLUSTERSETTING::szName, &pWlbsInstance );
//get the cluster configuration
FillWbemInstance( pWlbsInstance , ppCluster[i]);
//send the results back to WinMgMt
m_pResponseHandler->Indicate( 1, &pWlbsInstance );
if( pWlbsInstance ) {
pWlbsInstance->Release(); pWlbsInstance = NULL;
} }
m_pResponseHandler->SetStatus( 0, WBEM_S_NO_ERROR, NULL, NULL );
hRes = WBEM_S_NO_ERROR; }
catch(CErrorWlbsControl Err) {
IWbemClassObject* pWbemExtStat = NULL;
CreateExtendedStatus( m_pNameSpace, &pWbemExtStat, Err.Error(), (PWCHAR)(Err.Description()) );
m_pResponseHandler->SetStatus(0, WBEM_E_FAILED, NULL, pWbemExtStat);
if( pWbemExtStat ) pWbemExtStat->Release();
if( pWlbsInstance ) pWlbsInstance->Release();
//do not return WBEM_E_FAILED, this causes a race condition
hRes = WBEM_S_NO_ERROR; }
catch(_com_error HResErr ) {
m_pResponseHandler->SetStatus(0, HResErr.Error(), NULL, NULL);
if( pWlbsInstance ) pWlbsInstance->Release();
hRes = HResErr.Error(); //transform Win32 error to a WBEM error
if( hRes == ERROR_FILE_NOT_FOUND ) hRes = WBEM_E_NOT_FOUND ; }
catch(...) {
if( pWlbsInstance ) {
pWlbsInstance->Release(); pWlbsInstance = NULL;
}
throw;
}
return hRes; }
////////////////////////////////////////////////////////////////////////////////
//
// CWLBS_ClusterSetting::PutInstance
//
// Purpose: This function updates an instance of a MOF ClusterSetting
// class. The node does not have to be a member of a cluster. However,
// WLBS must be installed for this function to succeed.
//
////////////////////////////////////////////////////////////////////////////////
HRESULT CWLBS_ClusterSetting::PutInstance ( IWbemClassObject* a_pInstance, long /* a_lFlags */, IWbemContext* /* a_pIContex */ ) { HRESULT hRes = 0; VARIANT vHostName;
try {
VariantInit( &vHostName );
//get the host name value
hRes = a_pInstance->Get( _bstr_t( MOF_CLUSTERSETTING::pProperties[MOF_CLUSTERSETTING::NAME] ), 0, &vHostName, NULL, NULL );
if( FAILED( hRes ) ) throw _com_error( hRes );
CWlbsClusterWrapper* pCluster = GetClusterFromHostName(g_pWlbsControl, vHostName.bstrVal); if (pCluster == NULL) { throw _com_error( WBEM_E_NOT_FOUND ); }
//get the cluster IP value
_variant_t vClusterIp;
hRes = a_pInstance->Get( _bstr_t( MOF_CLUSTERSETTING::pProperties[MOF_CLUSTERSETTING::CLUSIPADDRESS] ), 0, &vClusterIp, NULL, NULL );
DWORD dwClusterIp = IpAddressFromAbcdWsz(vClusterIp.bstrVal); //
// Make sure the non-zero cluster IP is unique
//
if (dwClusterIp != 0) { CWlbsClusterWrapper* pTmpCluster = g_pWlbsControl->GetClusterFromIpOrIndex(dwClusterIp);
if (pTmpCluster && pCluster != pTmpCluster) { TRACE_ERROR1("CWLBS_ClusterSetting::PutInstance duplicate cluster IP found %ws", vClusterIp.bstrVal);
throw CErrorWlbsControl( WLBS_REG_ERROR, CmdWlbsWriteReg ); } } UpdateConfiguration( a_pInstance, pCluster );
// CLD: Need to check return code for error
if (S_OK != VariantClear( &vHostName )) throw _com_error( WBEM_E_FAILED );
m_pResponseHandler->SetStatus( 0, WBEM_S_NO_ERROR, NULL, NULL );
hRes = WBEM_S_NO_ERROR;
}
catch(CErrorWlbsControl Err) {
IWbemClassObject* pWbemExtStat = NULL;
CreateExtendedStatus( m_pNameSpace, &pWbemExtStat, Err.Error(), (PWCHAR)(Err.Description()) );
m_pResponseHandler->SetStatus(0, WBEM_E_FAILED, NULL, pWbemExtStat);
if( pWbemExtStat ) pWbemExtStat->Release();
// CLD: Need to check return code for error
if (S_OK != VariantClear( &vHostName )) throw _com_error( WBEM_E_FAILED );
//do not return WBEM_E_FAILED, this causes a race condition
hRes = WBEM_S_NO_ERROR; }
catch(_com_error HResErr ) {
m_pResponseHandler->SetStatus(0, HResErr.Error(), NULL, NULL);
// CLD: Need to check return code for error
// No throw here since we are already throwing an exception.
VariantClear( &vHostName );
hRes = HResErr.Error(); }
catch (...) {
// CLD: Need to check return code for error
// No throw here since we are already throwing an exception.
VariantClear( &vHostName );
throw; }
return hRes;
}
////////////////////////////////////////////////////////////////////////////////
//
// CWLBS_ClusterSetting::ExecMethod
//
// Purpose:
//
////////////////////////////////////////////////////////////////////////////////
HRESULT CWLBS_ClusterSetting::ExecMethod ( const ParsedObjectPath* a_pParsedPath, const BSTR& a_strMethodName, long /* a_lFlags */, IWbemContext* /* a_pIContex */, IWbemClassObject* a_pIInParams ) { IWbemClassObject* pOutputInstance = NULL; VARIANT vValue; HRESULT hRes = 0;
try {
VariantInit( &vValue );
CWlbsClusterWrapper* pCluster = NULL; if (a_pParsedPath->m_paKeys == NULL) { //
// No cluster IP specified
//
throw _com_error( WBEM_E_INVALID_PARAMETER ); } else { const wchar_t* wstrRequestedClusterName = (*a_pParsedPath->m_paKeys)->m_vValue.bstrVal;
pCluster = GetClusterFromHostName(g_pWlbsControl, wstrRequestedClusterName);
}
if (pCluster == NULL) throw _com_error( WBEM_E_NOT_FOUND );
//determine the method being executed
if( _wcsicmp( a_strMethodName, MOF_CLUSTERSETTING::pMethods[MOF_CLUSTERSETTING::SETPASS] ) == 0 ) {
//get the password
hRes = a_pIInParams->Get ( _bstr_t( MOF_PARAM::PASSW ), 0, &vValue, NULL, NULL );
if( vValue.vt != VT_BSTR ) throw _com_error ( WBEM_E_INVALID_PARAMETER );
pCluster->SetPassword( vValue.bstrVal );
} else if( _wcsicmp( a_strMethodName, MOF_CLUSTERSETTING::pMethods[MOF_CLUSTERSETTING::LDSETT] ) == 0 ) { //get the output object instance
GetMethodOutputInstance( MOF_CLUSTERSETTING::szName, a_strMethodName, &pOutputInstance);
DWORD dwReturnValue = pCluster->Commit(g_pWlbsControl); //set the return value
vValue.vt = VT_I4; vValue.lVal = static_cast<long>(dwReturnValue); hRes = pOutputInstance->Put(_bstr_t(L"ReturnValue"), 0, &vValue, 0);
if( FAILED( hRes ) ) { throw _com_error( hRes ); }
if( pOutputInstance ) { hRes = m_pResponseHandler->Indicate(1, &pOutputInstance);
if( FAILED( hRes ) ) { throw _com_error( hRes ); } }
} else if( _wcsicmp( a_strMethodName, MOF_CLUSTERSETTING::pMethods[MOF_CLUSTERSETTING::SETDEF] ) == 0 ) { pCluster->SetClusterDefaults(); } else { throw _com_error( WBEM_E_METHOD_NOT_IMPLEMENTED ); }
//get the parameters
//call the underlying API
//set the function return parameter
// CLD: Need to check return code for error
if (S_OK != VariantClear( &vValue )) throw _com_error( WBEM_E_FAILED );
if( pOutputInstance ) { pOutputInstance->Release(); pOutputInstance = NULL; }
m_pResponseHandler->SetStatus( 0, WBEM_S_NO_ERROR, NULL, NULL );
hRes = WBEM_S_NO_ERROR;
}
catch(CErrorWlbsControl Err) {
IWbemClassObject* pWbemExtStat = NULL;
CreateExtendedStatus( m_pNameSpace, &pWbemExtStat, Err.Error(), (PWCHAR)(Err.Description()) );
m_pResponseHandler->SetStatus(0, WBEM_E_FAILED, NULL, pWbemExtStat);
if( pWbemExtStat ) pWbemExtStat->Release();
// CLD: Need to check return code for error
// No throw here since we are already throwing an exception.
VariantClear( &vValue );
if( pOutputInstance ) { pOutputInstance->Release(); pOutputInstance = NULL; }
//do not return WBEM_E_FAILED, this causes a race condition
hRes = WBEM_S_NO_ERROR; }
catch(_com_error HResErr ) {
m_pResponseHandler->SetStatus(0, HResErr.Error(), NULL, NULL);
// CLD: Need to check return code for error
// No throw here since we are already throwing an exception.
VariantClear( &vValue );
if( pOutputInstance ) { pOutputInstance->Release(); }
hRes = HResErr.Error(); }
catch ( ... ) {
// CLD: Need to check return code for error
// No throw here since we are already throwing an exception.
VariantClear( &vValue );
if( pOutputInstance ) { pOutputInstance->Release(); }
throw; }
return hRes; }
////////////////////////////////////////////////////////////////////////////////
//
// CWLBS_ClusterSetting::FillWbemInstance
//
// Purpose: This function copies all of the data from a cluster configuration
// structure to a WBEM instance.
//
////////////////////////////////////////////////////////////////////////////////
void CWLBS_ClusterSetting::FillWbemInstance( IWbemClassObject* a_pWbemInstance, CWlbsClusterWrapper* pCluster) { namespace CLUSTER = MOF_CLUSTERSETTING;
ASSERT( a_pWbemInstance ); ASSERT(pCluster );
CClusterConfiguration ClusterConfig;
pCluster->GetClusterConfig( ClusterConfig );
wstring wstrHostName; ConstructHostName( wstrHostName, pCluster->GetClusterIpOrIndex(g_pWlbsControl), pCluster->GetHostID() );
//NAME
a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::NAME] ) , 0 , &_variant_t(wstrHostName.c_str()), NULL );
//CLUSNAME
a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::CLUSNAME] ), 0 , &_variant_t(ClusterConfig.szClusterName.c_str()), NULL );
//CLUSIPADDRESS
a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::CLUSIPADDRESS] ), 0 , &_variant_t(ClusterConfig.szClusterIPAddress.c_str()), NULL );
//CLUSNETMASK
a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::CLUSNETMASK] ), 0 , &_variant_t(ClusterConfig.szClusterNetworkMask.c_str()), NULL );
//CLUSMAC
a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::CLUSMAC] ), 0 , &_variant_t(ClusterConfig.szClusterMACAddress.c_str()), NULL );
//MULTIENABLE
a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::MULTIENABLE] ), 0 , &_variant_t(ClusterConfig.bMulticastSupportEnable), NULL );
//REMCNTEN
a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::REMCNTEN] ), 0 , &_variant_t(ClusterConfig.bRemoteControlEnabled), NULL );
//IGMPSUPPORT
a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::IGMPSUPPORT] ), 0 , &_variant_t(ClusterConfig.bIgmpSupport), NULL ); //CLUSTERIPTOMULTICASTIP
a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::CLUSTERIPTOMULTICASTIP] ), 0 , &_variant_t(ClusterConfig.bClusterIPToMulticastIP), NULL ); //MULTICASTIPADDRESS
a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::MULTICASTIPADDRESS] ), 0 , &_variant_t(ClusterConfig.szMulticastIPAddress.c_str()), NULL );
//ADAPTERGUID
GUID AdapterGuid = pCluster->GetAdapterGuid(); WCHAR szAdapterGuid[128]; StringFromGUID2(AdapterGuid, szAdapterGuid, sizeof(szAdapterGuid)/sizeof(szAdapterGuid[0]) ); a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::ADAPTERGUID] ), 0 , &_variant_t(szAdapterGuid), NULL );
//BDA Team Active
a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::BDATEAMACTIVE] ), 0 , &_variant_t(ClusterConfig.bBDATeamActive), NULL );
if (ClusterConfig.bBDATeamActive) { //BDA Team Id
a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::BDATEAMID] ), 0 , &_variant_t(ClusterConfig.szBDATeamId.c_str()), NULL );
//BDA Team Master
a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::BDATEAMMASTER] ), 0 , &_variant_t(ClusterConfig.bBDATeamMaster), NULL );
//BDA Reverse Hash
a_pWbemInstance->Put ( _bstr_t( CLUSTER::pProperties[CLUSTER::BDAREVERSEHASH] ), 0 , &_variant_t(ClusterConfig.bBDAReverseHash), NULL ); }
}
////////////////////////////////////////////////////////////////////////////////
//
// CWLBS_ClusterSetting::UpdateConfiguration
//
// Purpose: This function updates the configuration data for a member node or a
// potential WLBS cluster node.
//
////////////////////////////////////////////////////////////////////////////////
void CWLBS_ClusterSetting::UpdateConfiguration ( IWbemClassObject* a_pInstance, CWlbsClusterWrapper* pCluster ) { namespace CLUSTER = MOF_CLUSTERSETTING;
CClusterConfiguration NewConfiguration; CClusterConfiguration OldConfiguration;
pCluster->GetClusterConfig( OldConfiguration );
//Cluster Name
UpdateConfigProp ( NewConfiguration.szClusterName, OldConfiguration.szClusterName, CLUSTER::pProperties[CLUSTER::CLUSNAME], a_pInstance );
//Cluster IP
UpdateConfigProp ( NewConfiguration.szClusterIPAddress, OldConfiguration.szClusterIPAddress, CLUSTER::pProperties[CLUSTER::CLUSIPADDRESS], a_pInstance );
//Cluster Network Mask
UpdateConfigProp ( NewConfiguration.szClusterNetworkMask, OldConfiguration.szClusterNetworkMask, CLUSTER::pProperties[CLUSTER::CLUSNETMASK], a_pInstance );
//Cluster enable remote control
UpdateConfigProp ( NewConfiguration.bRemoteControlEnabled, OldConfiguration.bRemoteControlEnabled, CLUSTER::pProperties[CLUSTER::REMCNTEN], a_pInstance );
//Cluster enable multicast support
UpdateConfigProp ( NewConfiguration.bMulticastSupportEnable, OldConfiguration.bMulticastSupportEnable, CLUSTER::pProperties[CLUSTER::MULTIENABLE], a_pInstance );
//IGMPSUPPORT
UpdateConfigProp ( NewConfiguration.bIgmpSupport, OldConfiguration.bIgmpSupport, CLUSTER::pProperties[CLUSTER::IGMPSUPPORT], a_pInstance );
//CLUSTERIPTOMULTICASTIP
UpdateConfigProp ( NewConfiguration.bClusterIPToMulticastIP, OldConfiguration.bClusterIPToMulticastIP, CLUSTER::pProperties[CLUSTER::CLUSTERIPTOMULTICASTIP], a_pInstance );
//MULTICASTIPADDRESS
UpdateConfigProp ( NewConfiguration.szMulticastIPAddress, OldConfiguration.szMulticastIPAddress, CLUSTER::pProperties[CLUSTER::MULTICASTIPADDRESS], a_pInstance );
//BDA Teaming Active ?
UpdateConfigProp ( NewConfiguration.bBDATeamActive, OldConfiguration.bBDATeamActive, CLUSTER::pProperties[CLUSTER::BDATEAMACTIVE], a_pInstance );
// Set the other BDA properties only if the "Active" property is set
if (NewConfiguration.bBDATeamActive) { //BDA Team ID
UpdateConfigProp ( NewConfiguration.szBDATeamId, OldConfiguration.szBDATeamId, CLUSTER::pProperties[CLUSTER::BDATEAMID], a_pInstance );
//BDA Team Master
UpdateConfigProp ( NewConfiguration.bBDATeamMaster, OldConfiguration.bBDATeamMaster, CLUSTER::pProperties[CLUSTER::BDATEAMMASTER], a_pInstance );
//BDA Team Reverse Hash
UpdateConfigProp ( NewConfiguration.bBDAReverseHash, OldConfiguration.bBDAReverseHash, CLUSTER::pProperties[CLUSTER::BDAREVERSEHASH], a_pInstance );
}
pCluster->PutClusterConfig( NewConfiguration ); }
|