587 lines
13 KiB
C++
587 lines
13 KiB
C++
#ifndef _MNLBSETTING_H
|
|
#define _MNLBSETTING_H
|
|
//
|
|
// Copyright (c) Microsoft. All Rights Reserved
|
|
//
|
|
// THIS IS UNPUBLISHED PROPRIETARY SOURCE CODE OF Microsoft.
|
|
// The copyright notice above does not evidence any
|
|
// actual or intended publication of such source code.
|
|
//
|
|
// OneLiner : MNLBSetting interface.
|
|
// DevUnit : wlbstest
|
|
// Author : Murtaza Hakim
|
|
//
|
|
// Description:
|
|
// -----------
|
|
|
|
|
|
// Include Files
|
|
|
|
|
|
#include "MNLBPortRule.h"
|
|
|
|
#include "Common.h"
|
|
|
|
#include "MWmiObject.h"
|
|
|
|
#include "MWmiInstance.h"
|
|
|
|
#include <vector>
|
|
#include <memory>
|
|
|
|
#include <comdef.h>
|
|
|
|
using namespace std;
|
|
|
|
class MNLBSetting
|
|
{
|
|
public:
|
|
enum MNLBSetting_Error
|
|
{
|
|
MNLBSetting_SUCCESS = 0,
|
|
|
|
COM_FAILURE = 1,
|
|
|
|
UNCONSTRUCTED = 2,
|
|
|
|
MACHINE_FAILURE = 3,
|
|
|
|
BOUND = 4,
|
|
|
|
UNBOUND = 5,
|
|
|
|
NO_SUCH_NIC = 6,
|
|
|
|
NLBS_NOT_INSTALLED = 7,
|
|
|
|
NO_SUCH_IP = 8,
|
|
|
|
INVALID_RULE = 9,
|
|
};
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// constructor.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// ip IN : ip address of machine to configure/admin nlbs on.
|
|
// fullNICName IN : full name of nic to use.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// none.
|
|
//
|
|
|
|
MNLBSetting( _bstr_t ip,
|
|
_bstr_t fullNICName );
|
|
|
|
|
|
// for local machine.
|
|
MNLBSetting( _bstr_t fullNICName );
|
|
|
|
|
|
// NOT IMPLEMENTED.
|
|
//
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// default constructor.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// none.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// none.
|
|
|
|
MNLBSetting();
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// copy constructor.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// objToCopy IN : object to copy.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// none.
|
|
|
|
MNLBSetting(const MNLBSetting& objToCopy );
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// assignment operator
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// rhs IN : object to assign.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// self
|
|
|
|
MNLBSetting&
|
|
operator=( const MNLBSetting& rhs );
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// destructor
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// none
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// none.
|
|
|
|
~MNLBSetting();
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// gets the nlbs host properties.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// hp OUT : host properties.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
getHostProperties( HostProperties* hp );
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// gets the nlbs cluster properties.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// cp OUT : cluster properties.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
getClusterProperties( ClusterProperties* cp );
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// sets the nlbs host properties.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// hp IN : host properties to set.
|
|
// retVal OUT : return value of method ran.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
setHostProperties( const HostProperties& hp,
|
|
unsigned long* retVal );
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// sets the nlbs cluster properties.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// cp IN : cluster properties to set.
|
|
// retVal OUT : return value of method ran.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
setClusterProperties( const ClusterProperties& cp,
|
|
unsigned long* retVal );
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// get the load balanced port rules associated with this machine.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// portsLB OUT : load balanced port rules associated with this machine.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
|
|
MNLBSetting_Error
|
|
getPortRulesLoadBalanced( vector<MNLBPortRuleLoadBalanced>* portsLB );
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// get the failover port rules associated with this machine.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// portsF OUT : failover port rules associated with this machine.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
getPortRulesFailover( vector<MNLBPortRuleFailover>* portsF );
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// get the disabled port rules associated with this machine.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// portsD OUT : failover port rules associated with this machine.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
getPortRulesDisabled( vector<MNLBPortRuleDisabled>* portsD );
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// adds a load balanced port rule to the machine.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// portRuleLB IN : load balanced port rule to add.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
addPortRuleLoadBalanced( const MNLBPortRuleLoadBalanced& portRuleLB );
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// adds a failover port rule to the machine.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// portRuleF IN : failover port rule to add.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
addPortRuleFailover( const MNLBPortRuleFailover& portRuleF );
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// adds a disabled port rule to the machine.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// portRuleD IN : disabled port rule to add.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
addPortRuleDisabled( const MNLBPortRuleDisabled& portRuleD );
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// removes a load balanced port rule to the machine.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// portRuleLB IN : load balanced port rule to remove.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
removePortRuleLoadBalanced( const MNLBPortRuleLoadBalanced& portRuleLB );
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// removes a failover port rule to the machine.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// portRuleF IN : failover port rule to remove.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
removePortRuleFailover( const MNLBPortRuleFailover& portRuleF );
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// removes a disabled port rule to the machine.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// portRuleD IN : disabled port rule to remove.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
removePortRuleDisabled( const MNLBPortRuleDisabled& portRuleD );
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// starts host/cluster operations.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// hostID IN : host id or Common::ALL_HOSTS for cluster wide
|
|
// or Common::THIS_HOST for connected machine.
|
|
// retVal OUT : return value of method ran.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
start( int hostID, unsigned long* retVal );
|
|
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// stops host/cluster operations.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// hostID IN : host id or Common::ALL_HOSTS for cluster wide or
|
|
// Common::THIS_HOST for connected machine.
|
|
// retVal OUT : return value of method ran.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
stop( int hostID, unsigned long* retVal );
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// resume control over host/cluster operations.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// hostID IN : host id or Common::ALL_HOSTS for cluster wide or
|
|
// Common::THIS_HOST for connected machine.
|
|
// retVal OUT : return value of method ran.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
resume( int hostID, unsigned long* retVal );
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// suspend control over host/cluster operations.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// hostID IN : host id or Common::ALL_HOSTS for cluster wide or
|
|
// Common::THIS_HOST for connected machine.
|
|
// retVal OUT : return value of method ran.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
suspend( int hostID, unsigned long* retVal );
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// finishes all existing connections and
|
|
// stops host/cluster operations.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// hostID IN : host id or Common::ALL_HOSTS for cluster wide or
|
|
// Common::THIS_HOST for connected machine.
|
|
// retVal OUT : return value of method ran.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
drainstop( int hostID, unsigned long* retVal );
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// enables traffic for port rule for host/cluster.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// hostID IN : host id or Common::ALL_HOSTS for cluster wide or
|
|
// Common::THIS_HOST for connected machine.
|
|
// retVal OUT : return value of method ran.
|
|
// port IN : port to affect or default is all ports.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
enable( int hostID, unsigned long* retVal, unsigned long port = Common::ALL_PORTS );
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// disables ALL traffic for port rule for host/cluster.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// hostID IN : host id or Common::ALL_HOSTS for cluster wide or
|
|
// Common::THIS_HOST for connected machine.
|
|
// retVal OUT : return value of method ran.
|
|
// port IN : port to affect or default is all ports.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
disable( int hostID, unsigned long* retVal, unsigned long port = Common::ALL_PORTS);
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// disables NEW traffic for port rule for host/cluster.
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// hostID IN : host id or Common::ALL_HOSTS for cluster wide or
|
|
// Common::THIS_HOST for connected machine.
|
|
// retVal OUT : return value of method ran.
|
|
// port IN : port to affect or default is all ports.
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
|
|
MNLBSetting_Error
|
|
drain( int hostID, unsigned long* retVal, unsigned long port = Common::ALL_PORTS );
|
|
|
|
//
|
|
// Description:
|
|
// -----------
|
|
// refreshes the wmi connection to the host
|
|
//
|
|
// Parameters:
|
|
// ----------
|
|
// none
|
|
//
|
|
// Returns:
|
|
// -------
|
|
// SUCCESS else error code.
|
|
MNLBSetting_Error
|
|
refreshConnection();
|
|
|
|
MNLBSetting_Error
|
|
reload( unsigned long* retVal );
|
|
|
|
|
|
private:
|
|
|
|
// data
|
|
auto_ptr<MWmiObject> p_machine;
|
|
|
|
_bstr_t mIP;
|
|
|
|
_bstr_t nic;
|
|
|
|
// functions
|
|
|
|
MNLBSetting_Error
|
|
connectToMachine();
|
|
|
|
MNLBSetting_Error
|
|
checkNicBinding(_bstr_t& guid );
|
|
|
|
MNLBSetting_Error
|
|
getInstanceToRunMethodOn( int hostID, vector <MWmiInstance>* instanceStore );
|
|
|
|
|
|
MNLBSetting::MNLBSetting_Error
|
|
MNLBSetting::getPortRulesLoadBalanced_private( vector<MNLBPortRuleLoadBalanced>* portsLB,
|
|
vector<MWmiInstance>* instances );
|
|
|
|
MNLBSetting::MNLBSetting_Error
|
|
MNLBSetting::getPortRulesFailover_private( vector<MNLBPortRuleFailover>* portsF,
|
|
vector<MWmiInstance>* instances );
|
|
|
|
MNLBSetting::MNLBSetting_Error
|
|
MNLBSetting::getPortRulesDisabled_private( vector<MNLBPortRuleDisabled>* portsD,
|
|
vector<MWmiInstance>* instances );
|
|
|
|
MNLBSetting_Error
|
|
getClusterIPAndHostID( _bstr_t& clusterIP,
|
|
_bstr_t& hostID );
|
|
|
|
};
|
|
|
|
|
|
// ensure type safety
|
|
|
|
typedef class MNLBSetting MNLBSetting;
|
|
|
|
#endif
|
|
|