742 lines
19 KiB
C++
742 lines
19 KiB
C++
|
#include <crtdbg.h>
|
||
|
#include <comdef.h>
|
||
|
#include <iostream>
|
||
|
#include <memory>
|
||
|
#include <string>
|
||
|
#include <wbemprov.h>
|
||
|
#include <genlex.h> //for wmi object path parser
|
||
|
#include <objbase.h>
|
||
|
#include <wlbsconfig.h>
|
||
|
#include <ntrkcomm.h>
|
||
|
|
||
|
using namespace std;
|
||
|
|
||
|
#include "objpath.h"
|
||
|
#include "debug.h"
|
||
|
#include "controlwrapper.h"
|
||
|
#include "clusterwrapper.h"
|
||
|
#include "utils.h"
|
||
|
#include "wlbsparm.h"
|
||
|
#include "control.h"
|
||
|
|
||
|
|
||
|
#define NLB_DEFAULT_TIMEOUT 10
|
||
|
|
||
|
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// FillResponseClass
|
||
|
//
|
||
|
// Purpose: This function is obsolete. It was used in old versions to fill a
|
||
|
// a response parameter that was returned from the control methods
|
||
|
// of the Node and Cluster classes.
|
||
|
//
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
void FillResponseClass
|
||
|
(
|
||
|
DWORD a_dwNumHosts,
|
||
|
const WLBS_RESPONSE a_Response[],
|
||
|
WLBS_RESPONSE a_cResponse[],
|
||
|
LPCSTR a_szcommand
|
||
|
)
|
||
|
{
|
||
|
|
||
|
ASSERT( a_szcommand );
|
||
|
|
||
|
if( a_Response && a_cResponse ) {
|
||
|
for( DWORD nIter = 0; nIter < a_dwNumHosts; nIter++ ) {
|
||
|
a_cResponse [nIter] = a_Response[nIter];
|
||
|
//GetStatusDesc( a_cResponse[nIter].status, a_szcommand, a_cResponse[nIter].m_szStatus );
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
void CWlbsControlWrapper::Initialize()
|
||
|
{
|
||
|
DWORD dwRet = m_WlbsControl.Initialize();
|
||
|
|
||
|
if( dwRet != WLBS_PRESENT && dwRet != WLBS_LOCAL_ONLY)
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsInit );
|
||
|
|
||
|
m_WlbsControl.WlbsTimeoutSet( WLBS_ALL_CLUSTERS, NLB_DEFAULT_TIMEOUT );
|
||
|
|
||
|
CWlbsCluster** ppCluster;
|
||
|
DWORD dwNumClusters = 0;
|
||
|
|
||
|
//
|
||
|
// Use the local password for local query
|
||
|
//
|
||
|
m_WlbsControl.EnumClusterObjects( ppCluster, &dwNumClusters);
|
||
|
|
||
|
for (int i=0;i<dwNumClusters;i++)
|
||
|
{
|
||
|
m_WlbsControl.WlbsCodeSet( ppCluster[i]->GetClusterIp(),
|
||
|
ppCluster[i]->GetPassword() );
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
void CWlbsControlWrapper::ReInitialize()
|
||
|
{
|
||
|
m_WlbsControl.ReInitialize();
|
||
|
|
||
|
CWlbsCluster** ppCluster;
|
||
|
DWORD dwNumClusters = 0;
|
||
|
|
||
|
//
|
||
|
// In case pasword is changed, use the local password for local query
|
||
|
//
|
||
|
m_WlbsControl.EnumClusterObjects( ppCluster, &dwNumClusters);
|
||
|
|
||
|
for (int i=0;i<dwNumClusters;i++)
|
||
|
{
|
||
|
//
|
||
|
// If the change is not commited, the driver will still has the old password
|
||
|
//
|
||
|
if (!ppCluster[i]->IsCommitPending())
|
||
|
{
|
||
|
m_WlbsControl.WlbsCodeSet( ppCluster[i]->GetClusterIp(),
|
||
|
ppCluster[i]->GetPassword() );
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
|
||
|
void CWlbsControlWrapper::EnumClusters(CWlbsClusterWrapper** & ppCluster, DWORD* pdwNumClusters)
|
||
|
{
|
||
|
m_WlbsControl.EnumClusterObjects( (CWlbsCluster** &) ppCluster, pdwNumClusters);
|
||
|
}
|
||
|
|
||
|
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// CWlbsControlWrapper::Disable
|
||
|
//
|
||
|
// Purpose: Disable ALL traffic handling for the rule containing the
|
||
|
// specified port on specified host or all cluster hosts. Only rules
|
||
|
// that are set for multiple host filtering mode are affected.
|
||
|
//
|
||
|
//
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
DWORD CWlbsControlWrapper::Disable
|
||
|
(
|
||
|
DWORD a_dwCluster ,
|
||
|
DWORD a_dwHost ,
|
||
|
WLBS_RESPONSE* a_pResponse ,
|
||
|
DWORD& a_dwNumHosts ,
|
||
|
DWORD a_dwVip ,
|
||
|
DWORD a_dwPort
|
||
|
)
|
||
|
{
|
||
|
DWORD dwRet;
|
||
|
|
||
|
BOOL bClusterWide = ( a_dwNumHosts == WLBS_ALL_HOSTS );
|
||
|
|
||
|
WLBS_RESPONSE Response[WLBS_MAX_HOSTS];
|
||
|
DWORD dwNumHosts = WLBS_MAX_HOSTS;
|
||
|
|
||
|
|
||
|
dwRet = m_WlbsControl.WlbsDisable( a_dwCluster ,
|
||
|
a_dwHost ,
|
||
|
Response ,
|
||
|
&dwNumHosts ,
|
||
|
a_dwVip ,
|
||
|
a_dwPort
|
||
|
);
|
||
|
|
||
|
//check for Winsock errors
|
||
|
if( dwRet > 10000 )
|
||
|
{
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsQuery, bClusterWide );
|
||
|
}
|
||
|
|
||
|
//check the return value and throw if an error occurred
|
||
|
switch( dwRet ) {
|
||
|
|
||
|
case WLBS_INIT_ERROR:
|
||
|
case WLBS_BAD_PASSW:
|
||
|
case WLBS_TIMEOUT:
|
||
|
case WLBS_LOCAL_ONLY:
|
||
|
case WLBS_REMOTE_ONLY:
|
||
|
case WLBS_IO_ERROR:
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsDisable, bClusterWide );
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
a_dwNumHosts = dwNumHosts;
|
||
|
|
||
|
FillResponseClass( dwNumHosts, Response, a_pResponse, "disable" );
|
||
|
|
||
|
return dwRet;
|
||
|
}
|
||
|
|
||
|
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// CWlbsControlWrapper::Enable
|
||
|
//
|
||
|
// Purpose: Enable traffic handling for rule containing the
|
||
|
// specified port on specified host or all cluster hosts. Only rules
|
||
|
// that are set for multiple host filtering mode are affected.
|
||
|
//
|
||
|
//
|
||
|
//
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
DWORD CWlbsControlWrapper::Enable
|
||
|
(
|
||
|
DWORD a_dwCluster ,
|
||
|
DWORD a_dwHost ,
|
||
|
WLBS_RESPONSE* a_pResponse ,
|
||
|
DWORD& a_dwNumHosts ,
|
||
|
DWORD a_dwVip ,
|
||
|
DWORD a_dwPort
|
||
|
)
|
||
|
{
|
||
|
DWORD dwRet;
|
||
|
|
||
|
BOOL bClusterWide = ( a_dwNumHosts == WLBS_ALL_HOSTS );
|
||
|
|
||
|
WLBS_RESPONSE Response[WLBS_MAX_HOSTS];
|
||
|
DWORD dwNumHosts = WLBS_MAX_HOSTS;
|
||
|
|
||
|
dwRet = m_WlbsControl.WlbsEnable( a_dwCluster ,
|
||
|
a_dwHost ,
|
||
|
Response ,
|
||
|
&dwNumHosts ,
|
||
|
a_dwVip ,
|
||
|
a_dwPort
|
||
|
);
|
||
|
|
||
|
//check for Winsock errors
|
||
|
if( dwRet > 10000 )
|
||
|
{
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsQuery, bClusterWide );
|
||
|
}
|
||
|
|
||
|
//check the return value and throw if an error occurred
|
||
|
switch(dwRet) {
|
||
|
|
||
|
case WLBS_INIT_ERROR:
|
||
|
case WLBS_BAD_PASSW:
|
||
|
case WLBS_TIMEOUT:
|
||
|
case WLBS_LOCAL_ONLY:
|
||
|
case WLBS_REMOTE_ONLY:
|
||
|
case WLBS_IO_ERROR:
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsEnable, bClusterWide );
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
a_dwNumHosts = dwNumHosts;
|
||
|
|
||
|
FillResponseClass( dwNumHosts, Response, a_pResponse, "enable" );
|
||
|
|
||
|
return dwRet;
|
||
|
}
|
||
|
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// CWlbsControlWrapper::Drain
|
||
|
//
|
||
|
// Purpose: Disable NEW traffic handling for rule containing the specified
|
||
|
// port on specified host or all cluster hosts. Only rules that are
|
||
|
// set for multiple host filtering mode are affected.
|
||
|
|
||
|
//
|
||
|
//
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
DWORD CWlbsControlWrapper::Drain
|
||
|
(
|
||
|
DWORD a_dwCluster ,
|
||
|
DWORD a_dwHost ,
|
||
|
WLBS_RESPONSE* a_pResponse ,
|
||
|
DWORD& a_dwNumHosts ,
|
||
|
DWORD a_dwVip ,
|
||
|
DWORD a_dwPort
|
||
|
)
|
||
|
{
|
||
|
DWORD dwRet;
|
||
|
|
||
|
BOOL bClusterWide = ( a_dwNumHosts == WLBS_ALL_HOSTS );
|
||
|
|
||
|
WLBS_RESPONSE Response[WLBS_MAX_HOSTS];
|
||
|
DWORD dwNumHosts = WLBS_MAX_HOSTS;
|
||
|
|
||
|
dwRet = m_WlbsControl.WlbsDrain( a_dwCluster ,
|
||
|
a_dwHost ,
|
||
|
Response ,
|
||
|
&dwNumHosts ,
|
||
|
a_dwVip ,
|
||
|
a_dwPort
|
||
|
);
|
||
|
|
||
|
//check for Winsock errors
|
||
|
if( dwRet > 10000 )
|
||
|
{
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsQuery, bClusterWide );
|
||
|
}
|
||
|
|
||
|
//check the return value and throw if an error occurred
|
||
|
switch(dwRet) {
|
||
|
|
||
|
case WLBS_INIT_ERROR:
|
||
|
case WLBS_BAD_PASSW:
|
||
|
case WLBS_TIMEOUT:
|
||
|
case WLBS_LOCAL_ONLY:
|
||
|
case WLBS_REMOTE_ONLY:
|
||
|
case WLBS_IO_ERROR:
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsDrain, bClusterWide );
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
a_dwNumHosts = dwNumHosts;
|
||
|
|
||
|
FillResponseClass( dwNumHosts, Response, a_pResponse, "drain" );
|
||
|
|
||
|
return dwRet;
|
||
|
}
|
||
|
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// CWlbsControlWrapper::DrainStop
|
||
|
//
|
||
|
// Purpose: Enter draining mode on specified host or all cluster hosts.
|
||
|
// New connections will not be accepted. Cluster mode will be stopped
|
||
|
// when all existing connections finish. While draining, host will
|
||
|
// participate in convergence and remain part of the cluster.
|
||
|
//
|
||
|
//
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
DWORD CWlbsControlWrapper::DrainStop
|
||
|
(
|
||
|
DWORD a_dwCluster ,
|
||
|
DWORD a_dwHost ,
|
||
|
WLBS_RESPONSE* a_pResponse ,
|
||
|
DWORD& a_dwNumHosts
|
||
|
)
|
||
|
{
|
||
|
DWORD dwRet;
|
||
|
|
||
|
BOOL bClusterWide = ( a_dwNumHosts == WLBS_ALL_HOSTS );
|
||
|
|
||
|
WLBS_RESPONSE Response[WLBS_MAX_HOSTS];
|
||
|
DWORD dwNumHosts = WLBS_MAX_HOSTS;
|
||
|
|
||
|
dwRet = m_WlbsControl.WlbsDrainStop( a_dwCluster ,
|
||
|
a_dwHost ,
|
||
|
Response ,
|
||
|
&dwNumHosts
|
||
|
);
|
||
|
|
||
|
//check for Winsock errors
|
||
|
if( dwRet > 10000 )
|
||
|
{
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsQuery, bClusterWide );
|
||
|
}
|
||
|
|
||
|
//check the return value and throw if an error occurred
|
||
|
switch(dwRet) {
|
||
|
|
||
|
case WLBS_INIT_ERROR:
|
||
|
case WLBS_BAD_PASSW:
|
||
|
case WLBS_TIMEOUT:
|
||
|
case WLBS_LOCAL_ONLY:
|
||
|
case WLBS_REMOTE_ONLY:
|
||
|
case WLBS_IO_ERROR:
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsDrainStop, bClusterWide );
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
a_dwNumHosts = dwNumHosts;
|
||
|
|
||
|
FillResponseClass( dwNumHosts, Response, a_pResponse, "drainstop" );
|
||
|
|
||
|
return dwRet;
|
||
|
}
|
||
|
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// CWlbsControlWrapper::Resume
|
||
|
//
|
||
|
// Purpose: Resume cluster operation control on specified host or all
|
||
|
// cluster hosts.
|
||
|
//
|
||
|
//
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
DWORD CWlbsControlWrapper::Resume
|
||
|
(
|
||
|
DWORD a_dwCluster ,
|
||
|
DWORD a_dwHost ,
|
||
|
WLBS_RESPONSE* a_pResponse ,
|
||
|
DWORD& a_dwNumHosts
|
||
|
)
|
||
|
{
|
||
|
DWORD dwRet;
|
||
|
|
||
|
BOOL bClusterWide = ( a_dwNumHosts == WLBS_ALL_HOSTS );
|
||
|
|
||
|
WLBS_RESPONSE Response[WLBS_MAX_HOSTS];
|
||
|
DWORD dwNumHosts = WLBS_MAX_HOSTS;
|
||
|
|
||
|
dwRet = m_WlbsControl.WlbsResume(
|
||
|
a_dwCluster ,
|
||
|
a_dwHost ,
|
||
|
Response ,
|
||
|
&dwNumHosts
|
||
|
);
|
||
|
|
||
|
//check for Winsock errors
|
||
|
if( dwRet > 10000 )
|
||
|
{
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsQuery, bClusterWide );
|
||
|
}
|
||
|
|
||
|
//check the return value and throw if an error occurred
|
||
|
switch(dwRet) {
|
||
|
|
||
|
case WLBS_INIT_ERROR:
|
||
|
case WLBS_BAD_PASSW:
|
||
|
case WLBS_TIMEOUT:
|
||
|
case WLBS_LOCAL_ONLY:
|
||
|
case WLBS_REMOTE_ONLY:
|
||
|
case WLBS_IO_ERROR:
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsResume, bClusterWide );
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
a_dwNumHosts = dwNumHosts;
|
||
|
|
||
|
FillResponseClass( dwNumHosts, Response, a_pResponse, "resume" );
|
||
|
|
||
|
return dwRet;
|
||
|
}
|
||
|
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// CWlbsControlWrapper::Start
|
||
|
//
|
||
|
// Purpose: Start cluster operations on specified host or all cluster hosts.
|
||
|
//
|
||
|
//
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
DWORD CWlbsControlWrapper::Start
|
||
|
(
|
||
|
DWORD a_dwCluster ,
|
||
|
DWORD a_dwHost ,
|
||
|
WLBS_RESPONSE* a_pResponse ,
|
||
|
DWORD& a_dwNumHosts
|
||
|
)
|
||
|
{
|
||
|
DWORD dwRet;
|
||
|
|
||
|
BOOL bClusterWide = ( a_dwNumHosts == WLBS_ALL_HOSTS );
|
||
|
|
||
|
WLBS_RESPONSE Response[WLBS_MAX_HOSTS];
|
||
|
DWORD dwNumHosts = WLBS_MAX_HOSTS;
|
||
|
|
||
|
dwRet = m_WlbsControl.WlbsStart( a_dwCluster ,
|
||
|
a_dwHost ,
|
||
|
Response ,
|
||
|
&dwNumHosts
|
||
|
);
|
||
|
|
||
|
//check for Winsock errors
|
||
|
if( dwRet > 10000 )
|
||
|
{
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsQuery, bClusterWide );
|
||
|
}
|
||
|
|
||
|
//check the return value and throw if an error occurred
|
||
|
switch(dwRet) {
|
||
|
|
||
|
case WLBS_INIT_ERROR:
|
||
|
case WLBS_BAD_PASSW:
|
||
|
case WLBS_TIMEOUT:
|
||
|
case WLBS_LOCAL_ONLY:
|
||
|
case WLBS_REMOTE_ONLY:
|
||
|
case WLBS_IO_ERROR:
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsStart, bClusterWide );
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
a_dwNumHosts = dwNumHosts;
|
||
|
|
||
|
FillResponseClass( dwNumHosts, Response, a_pResponse, "start" );
|
||
|
|
||
|
return dwRet;
|
||
|
}
|
||
|
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// CWlbsControlWrapper::Stop
|
||
|
//
|
||
|
// Purpose: Stop cluster operations on specified host or all cluster hosts.
|
||
|
//
|
||
|
//
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
DWORD CWlbsControlWrapper::Stop
|
||
|
(
|
||
|
DWORD a_dwCluster ,
|
||
|
DWORD a_dwHost ,
|
||
|
WLBS_RESPONSE* a_pResponse ,
|
||
|
DWORD& a_dwNumHosts
|
||
|
)
|
||
|
{
|
||
|
DWORD dwRet;
|
||
|
|
||
|
BOOL bClusterWide = ( a_dwNumHosts == WLBS_ALL_HOSTS );
|
||
|
|
||
|
WLBS_RESPONSE Response[WLBS_MAX_HOSTS];
|
||
|
DWORD dwNumHosts = WLBS_MAX_HOSTS;
|
||
|
|
||
|
dwRet = m_WlbsControl.WlbsStop(
|
||
|
a_dwCluster ,
|
||
|
a_dwHost ,
|
||
|
Response ,
|
||
|
&dwNumHosts
|
||
|
);
|
||
|
|
||
|
//check for Winsock errors
|
||
|
if( dwRet > 10000 )
|
||
|
{
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsQuery, bClusterWide );
|
||
|
}
|
||
|
|
||
|
//check the return value and throw if an error occurred
|
||
|
switch(dwRet) {
|
||
|
|
||
|
case WLBS_INIT_ERROR:
|
||
|
case WLBS_BAD_PASSW:
|
||
|
case WLBS_TIMEOUT:
|
||
|
case WLBS_LOCAL_ONLY:
|
||
|
case WLBS_REMOTE_ONLY:
|
||
|
case WLBS_IO_ERROR:
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsStop, bClusterWide );
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
a_dwNumHosts = dwNumHosts;
|
||
|
|
||
|
FillResponseClass( dwNumHosts, Response, a_pResponse, "stop" );
|
||
|
|
||
|
return dwRet;
|
||
|
}
|
||
|
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// CWlbsControlWrapper::Suspend
|
||
|
//
|
||
|
// Purpose: Suspend cluster operation control on specified host or
|
||
|
// all cluster hosts.
|
||
|
//
|
||
|
//
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
DWORD CWlbsControlWrapper::Suspend
|
||
|
(
|
||
|
DWORD a_dwCluster ,
|
||
|
DWORD a_dwHost ,
|
||
|
WLBS_RESPONSE* a_pResponse ,
|
||
|
DWORD& a_dwNumHosts
|
||
|
)
|
||
|
{
|
||
|
DWORD dwRet;
|
||
|
|
||
|
BOOL bClusterWide = ( a_dwNumHosts == WLBS_ALL_HOSTS );
|
||
|
|
||
|
WLBS_RESPONSE Response[WLBS_MAX_HOSTS];
|
||
|
DWORD dwNumHosts = WLBS_MAX_HOSTS;
|
||
|
|
||
|
dwRet = m_WlbsControl.WlbsSuspend( a_dwCluster ,
|
||
|
a_dwHost ,
|
||
|
Response ,
|
||
|
&dwNumHosts
|
||
|
);
|
||
|
|
||
|
//check for Winsock errors
|
||
|
if( dwRet > 10000 )
|
||
|
{
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsQuery, bClusterWide );
|
||
|
}
|
||
|
|
||
|
//check the return value and throw if an error occurred
|
||
|
switch(dwRet) {
|
||
|
|
||
|
case WLBS_INIT_ERROR:
|
||
|
case WLBS_BAD_PASSW:
|
||
|
case WLBS_TIMEOUT:
|
||
|
case WLBS_LOCAL_ONLY:
|
||
|
case WLBS_REMOTE_ONLY:
|
||
|
case WLBS_IO_ERROR:
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsSuspend, bClusterWide );
|
||
|
break;
|
||
|
}
|
||
|
|
||
|
a_dwNumHosts = dwNumHosts;
|
||
|
|
||
|
FillResponseClass( dwNumHosts, Response, a_pResponse, "suspend" );
|
||
|
|
||
|
return dwRet;
|
||
|
}
|
||
|
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// CWlbsControlWrapper::Query
|
||
|
//
|
||
|
// Purpose: This invokes WlbsQuery and returns a response structure along
|
||
|
// with other parameters if requested.
|
||
|
//
|
||
|
// Errors: The function throws CErrorWlbsControl.
|
||
|
//
|
||
|
// Return: Status value returned by the target host
|
||
|
//
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
DWORD CWlbsControlWrapper::Query
|
||
|
(
|
||
|
CWlbsClusterWrapper* pCluster,
|
||
|
DWORD a_dwHost ,
|
||
|
WLBS_RESPONSE* a_pResponse ,
|
||
|
PDWORD a_pdwNumHosts ,
|
||
|
PDWORD a_pdwHostMap
|
||
|
)
|
||
|
{
|
||
|
DWORD dwResponseSize = 0;
|
||
|
if (a_pdwNumHosts)
|
||
|
{
|
||
|
dwResponseSize = *a_pdwNumHosts;
|
||
|
*a_pdwNumHosts = 0;
|
||
|
}
|
||
|
|
||
|
|
||
|
DWORD dwRet;
|
||
|
BOOL bClusterWide = 0;
|
||
|
|
||
|
bClusterWide = ( a_dwHost == WLBS_ALL_HOSTS );
|
||
|
|
||
|
WLBS_RESPONSE Response[WLBS_MAX_HOSTS];
|
||
|
DWORD dwNumHosts = WLBS_MAX_HOSTS;
|
||
|
|
||
|
dwRet = m_WlbsControl.WlbsQuery(
|
||
|
(CWlbsCluster*)pCluster ,
|
||
|
a_dwHost ,
|
||
|
Response ,
|
||
|
&dwNumHosts ,
|
||
|
a_pdwHostMap,
|
||
|
NULL
|
||
|
);
|
||
|
|
||
|
string strOut;
|
||
|
|
||
|
//check for Winsock errors
|
||
|
if( dwRet > 10000 )
|
||
|
{
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsQuery, bClusterWide );
|
||
|
}
|
||
|
|
||
|
//check the return value and throw if an error occurred
|
||
|
switch( dwRet ) {
|
||
|
case WLBS_INIT_ERROR:
|
||
|
case WLBS_BAD_PASSW:
|
||
|
case WLBS_TIMEOUT:
|
||
|
case WLBS_LOCAL_ONLY:
|
||
|
case WLBS_REMOTE_ONLY:
|
||
|
case WLBS_IO_ERROR:
|
||
|
|
||
|
throw CErrorWlbsControl( dwRet, CmdWlbsQuery, bClusterWide );
|
||
|
}
|
||
|
|
||
|
//local queries do not return the dedicated IP
|
||
|
//get the dedicated IP and fill the structure
|
||
|
if( a_dwHost == WLBS_LOCAL_HOST )
|
||
|
{
|
||
|
Response[0].address = pCluster->GetDedicatedIp();
|
||
|
}
|
||
|
|
||
|
if( a_pdwNumHosts )
|
||
|
{
|
||
|
*a_pdwNumHosts = _MIN(dwResponseSize, dwNumHosts);
|
||
|
FillResponseClass( dwNumHosts, Response, a_pResponse, "query" );
|
||
|
}
|
||
|
|
||
|
return dwRet;
|
||
|
}
|
||
|
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
//
|
||
|
// CWlbsClusterWrapper::CheckMembership
|
||
|
//
|
||
|
// Purpose: This verifies that the local host is a member of the cluster
|
||
|
// specified by the Cluster IP in the registry. At the time this
|
||
|
// was written, there was a remote chance that a user can modify the
|
||
|
// IP address in the registry prior to the load of this DLL.
|
||
|
//
|
||
|
// Note, this call is only required for the Node, Cluster and linked
|
||
|
// associations and should not be invoked for any of the Setting classes.
|
||
|
//
|
||
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
void CWlbsControlWrapper::CheckMembership()
|
||
|
{
|
||
|
// todo: make sure the host is in at least one cluster
|
||
|
|
||
|
/*
|
||
|
WLBS_RESPONSE WlbsResponse;
|
||
|
DWORD dwResSize = 1;
|
||
|
|
||
|
//get the cluster and HostID
|
||
|
DWORD dwRes = pControl->Query( m_pWlbsCluster->GetClusterIp(),
|
||
|
WLBS_LOCAL_HOST,
|
||
|
&WlbsResponse,
|
||
|
&dwResSize,
|
||
|
NULL);
|
||
|
|
||
|
switch( dwRes ) {
|
||
|
case WLBS_SUSPENDED:
|
||
|
case WLBS_STOPPED:
|
||
|
case WLBS_CONVERGING:
|
||
|
case WLBS_DRAINING:
|
||
|
case WLBS_CONVERGED:
|
||
|
case WLBS_DEFAULT:
|
||
|
break;
|
||
|
default:
|
||
|
throw CErrorWlbsControl( dwRes, CmdWlbsQuery );
|
||
|
}
|
||
|
*/
|
||
|
// DWORD dwClusterIP;
|
||
|
// GetClusterIP( &dwClusterIP );
|
||
|
|
||
|
//if( dwClusterIP == 0 )
|
||
|
//throw _com_error( WBEM_E_NOT_FOUND );
|
||
|
|
||
|
//*******************************
|
||
|
//this section does not work when the remote control is disabled
|
||
|
//on the local host
|
||
|
//*******************************
|
||
|
|
||
|
//call the query function
|
||
|
// dwRes = WlbsQuery( dwClusterIP,
|
||
|
// WlbsResponse.id,
|
||
|
// NULL,
|
||
|
// NULL,
|
||
|
// NULL,
|
||
|
// NULL );
|
||
|
|
||
|
//analyze query results for errors
|
||
|
// switch( dwRes ) {
|
||
|
// case WLBS_OK:
|
||
|
// case WLBS_STOPPED:
|
||
|
// case WLBS_CONVERGING:
|
||
|
// case WLBS_CONVERGED:
|
||
|
// case WLBS_DEFAULT:
|
||
|
// case WLBS_DRAINING:
|
||
|
// case WLBS_SUSPENDED:
|
||
|
// return;
|
||
|
// default:
|
||
|
// throw CErrorWlbsControl( dwRes, CmdWlbsQuery );
|
||
|
// }
|
||
|
|
||
|
//*******************************
|
||
|
//this section does not work when the remote control is disabled
|
||
|
//on the local host
|
||
|
//*******************************
|
||
|
|
||
|
}
|