304 lines
8.5 KiB
C++
304 lines
8.5 KiB
C++
//+----------------------------------------------------------------------------
|
|
//
|
|
// File: cluster.cpp
|
|
//
|
|
// Module: WLBS API
|
|
//
|
|
// Description: Implement class CWlbsCluster
|
|
//
|
|
// Copyright (C) Microsoft Corporation. All rights reserved.
|
|
//
|
|
// Author: Created 3/9/00
|
|
//
|
|
//+----------------------------------------------------------------------------
|
|
#include "precomp.h"
|
|
|
|
#include <debug.h>
|
|
#include "cluster.h"
|
|
#include "control.h"
|
|
#include "param.h"
|
|
#include "cluster.tmh" // For event tracing
|
|
|
|
|
|
CWlbsCluster::CWlbsCluster(DWORD dwConfigIndex)
|
|
{
|
|
m_reload_required = false;
|
|
m_notify_adapter_required = false;
|
|
m_this_cl_addr = 0;
|
|
m_this_host_id = 0;
|
|
m_this_ded_addr = 0;
|
|
m_dwConfigIndex = dwConfigIndex;
|
|
}
|
|
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: CWlbsCluster::ReadConfig
|
|
//
|
|
// Description: Read cluster settings from registry
|
|
//
|
|
// Arguments: PWLBS_REG_PARAMS reg_data -
|
|
//
|
|
// Returns: DWORD -
|
|
//
|
|
// History: fengsun Created Header 1/25/00
|
|
//
|
|
//+----------------------------------------------------------------------------
|
|
DWORD CWlbsCluster::ReadConfig(PWLBS_REG_PARAMS reg_data)
|
|
{
|
|
TRACE_VERB("->%!FUNC!");
|
|
|
|
if (ParamReadReg(m_AdapterGuid, reg_data) == false)
|
|
{
|
|
TRACE_VERB("<-%!FUNC! return %d", WLBS_REG_ERROR);
|
|
return WLBS_REG_ERROR;
|
|
}
|
|
|
|
/* create a copy in the old_params structure. this will be required to
|
|
* determine whether a reload is needed or a reboot is needed for commit */
|
|
|
|
memcpy ( &m_reg_params, reg_data, sizeof (WLBS_REG_PARAMS));
|
|
|
|
// m_this_cl_addr = IpAddressFromAbcdWsz(m_reg_params.cl_ip_addr);
|
|
m_this_ded_addr = IpAddressFromAbcdWsz(m_reg_params.ded_ip_addr);
|
|
|
|
TRACE_VERB("<-%!FUNC! return %d", WLBS_OK);
|
|
return WLBS_OK;
|
|
}
|
|
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: CWlbsCluster::GetClusterIpOrIndex
|
|
//
|
|
// Description: Get the index or IP of the cluster. If the cluster IP is non-zero
|
|
// The IP is return.
|
|
// If the cluster IP is 0, the index is returned
|
|
//
|
|
// Arguments: CWlbsControl* pControl -
|
|
//
|
|
// Returns: DWORD -
|
|
//
|
|
// History: fengsun Created Header 7/3/00
|
|
//
|
|
//+----------------------------------------------------------------------------
|
|
DWORD CWlbsCluster::GetClusterIpOrIndex(CWlbsControl* pControl)
|
|
{
|
|
TRACE_VERB("->%!FUNC!");
|
|
|
|
DWORD dwIp = CWlbsCluster::GetClusterIp();
|
|
|
|
if (dwIp!=0)
|
|
{
|
|
//
|
|
// Return the cluster IP if non 0
|
|
//
|
|
TRACE_VERB("<-%!FUNC! return %d", dwIp);
|
|
return dwIp;
|
|
}
|
|
|
|
if (pControl->GetClusterNum() == 1)
|
|
{
|
|
//
|
|
// For backward compatibility, return 0 if only one cluster exists
|
|
//
|
|
|
|
TRACE_VERB("<-%!FUNC! return 0");
|
|
return 0;
|
|
}
|
|
|
|
//
|
|
// Ip address is in the reverse order
|
|
//
|
|
dwIp = (CWlbsCluster::m_dwConfigIndex) <<24;
|
|
TRACE_VERB("<-%!FUNC! returning IP address in reverse order %d", dwIp);
|
|
return dwIp;
|
|
}
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: CWlbsCluster::WriteConfig
|
|
//
|
|
// Description: Write cluster settings to registry
|
|
//
|
|
// Arguments: WLBS_REG_PARAMS* reg_data -
|
|
//
|
|
// Returns: DWORD -
|
|
//
|
|
// History: fengsun Created Header 3/9/00
|
|
//
|
|
//+----------------------------------------------------------------------------
|
|
DWORD CWlbsCluster::WriteConfig(WLBS_REG_PARAMS* reg_data)
|
|
{
|
|
TRACE_VERB("->%!FUNC!");
|
|
DWORD Status;
|
|
|
|
Status = ParamWriteConfig(m_AdapterGuid, reg_data, &m_reg_params, &m_reload_required, &m_notify_adapter_required);
|
|
|
|
TRACE_VERB("<-%!FUNC! return %d", Status);
|
|
return Status;
|
|
}
|
|
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: CWlbsCluster::CommitChanges
|
|
//
|
|
// Description: Notify wlbs driver or nic driver to pick up the changes
|
|
//
|
|
// Arguments: CWlbsControl* pWlbsControl -
|
|
//
|
|
// Returns: DWORD -
|
|
//
|
|
// History: fengsun Created Header 7/6/00
|
|
// chrisdar 07.31.01 Modified adapter notification code to not disable
|
|
// and enable the NIC. Just do property change now.
|
|
// KarthicN 08/28/01 Moved contents over to ParamCommitChanges
|
|
//
|
|
//+----------------------------------------------------------------------------
|
|
DWORD CWlbsCluster::CommitChanges(CWlbsControl* pWlbsControl)
|
|
{
|
|
|
|
TRACE_VERB("->%!FUNC!");
|
|
DWORD Status;
|
|
|
|
ASSERT(pWlbsControl);
|
|
Status = ParamCommitChanges(m_AdapterGuid,
|
|
pWlbsControl->GetDriverHandle(),
|
|
m_this_cl_addr,
|
|
m_this_ded_addr,
|
|
&m_reload_required,
|
|
&m_notify_adapter_required);
|
|
|
|
TRACE_VERB("<-%!FUNC! return %d", Status);
|
|
return Status;
|
|
}
|
|
|
|
|
|
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: CWlbsCluster::Initialize
|
|
//
|
|
// Description: Initialization
|
|
//
|
|
// Arguments: const GUID& AdapterGuid -
|
|
//
|
|
// Returns: bool - true if succeeded
|
|
//
|
|
// History: fengsun Created Header 3/9/00
|
|
//
|
|
//+----------------------------------------------------------------------------
|
|
bool CWlbsCluster::Initialize(const GUID& AdapterGuid)
|
|
{
|
|
TRACE_VERB("->%!FUNC!");
|
|
|
|
m_AdapterGuid = AdapterGuid;
|
|
m_notify_adapter_required = false;
|
|
m_reload_required = false;
|
|
|
|
ZeroMemory (& m_reg_params, sizeof (m_reg_params));
|
|
|
|
if (!ParamReadReg(m_AdapterGuid, &m_reg_params))
|
|
{
|
|
TRACE_CRIT("%!FUNC! ParamReadReg failed");
|
|
// This check was added for tracing. No abort was done previously on error, so don't do so now.
|
|
}
|
|
|
|
m_this_cl_addr = IpAddressFromAbcdWsz(m_reg_params.cl_ip_addr);
|
|
m_this_ded_addr = IpAddressFromAbcdWsz(m_reg_params.ded_ip_addr);
|
|
m_this_host_id = m_reg_params.host_priority;
|
|
|
|
TRACE_VERB("->%!FUNC! return true");
|
|
return true;
|
|
}
|
|
|
|
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: CWlbsCluster::ReInitialize
|
|
//
|
|
// Description: Reload settings from registry
|
|
//
|
|
// Arguments:
|
|
//
|
|
// Returns: bool - true if succeeded
|
|
//
|
|
// History: fengsun Created Header 3/9/00
|
|
//
|
|
//+----------------------------------------------------------------------------
|
|
bool CWlbsCluster::ReInitialize()
|
|
{
|
|
TRACE_VERB("->%!FUNC!");
|
|
|
|
if (ParamReadReg(m_AdapterGuid, &m_reg_params) == false)
|
|
{
|
|
TRACE_CRIT("!FUNC! failed reading nlb registry parameters");
|
|
TRACE_VERB("<-%!FUNC! return false");
|
|
return false;
|
|
}
|
|
|
|
//
|
|
// Do not change the ClusterIP if the changes has not been commited
|
|
//
|
|
if (!IsCommitPending())
|
|
{
|
|
m_this_cl_addr = IpAddressFromAbcdWsz(m_reg_params.cl_ip_addr);
|
|
m_this_host_id = m_reg_params.host_priority;
|
|
}
|
|
|
|
m_this_ded_addr = IpAddressFromAbcdWsz(m_reg_params.ded_ip_addr);
|
|
|
|
TRACE_VERB("<-%!FUNC! return true");
|
|
return true;
|
|
}
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: CWlbsCluster::GetPassword
|
|
//
|
|
// Description: Get remote control password for this cluster
|
|
//
|
|
// Arguments:
|
|
//
|
|
// Returns: DWORD - password
|
|
//
|
|
// History: fengsun Created Header 2/3/00
|
|
//
|
|
//+----------------------------------------------------------------------------
|
|
DWORD CWlbsCluster::GetPassword()
|
|
{
|
|
TRACE_VERB("->%!FUNC!");
|
|
|
|
HKEY key = NULL;
|
|
LONG status;
|
|
DWORD dwRctPassword = 0;
|
|
|
|
if (!(key = RegOpenWlbsSetting(m_AdapterGuid, true)))
|
|
{
|
|
TRACE_CRIT("%!FUNC! RegOpenWlbsSetting failed");
|
|
// This check was added for tracing. No abort was done previously on error, so don't do so now.
|
|
}
|
|
|
|
DWORD size = sizeof(dwRctPassword);
|
|
status = RegQueryValueEx (key, CVY_NAME_RCT_PASSWORD, 0L, NULL,
|
|
(LPBYTE) & dwRctPassword, & size);
|
|
if (status != ERROR_SUCCESS)
|
|
{
|
|
dwRctPassword = CVY_DEF_RCT_PASSWORD;
|
|
TRACE_CRIT("%!FUNC! registry read for %ls failed with %d", CVY_NAME_RCT_PASSWORD, status);
|
|
}
|
|
|
|
status = RegCloseKey(key);
|
|
if (ERROR_SUCCESS != status)
|
|
{
|
|
TRACE_CRIT("%!FUNC! registry close failed with %d", status);
|
|
}
|
|
|
|
TRACE_VERB("<-%!FUNC!");
|
|
return dwRctPassword;
|
|
}
|
|
|