528 lines
10 KiB
C++
528 lines
10 KiB
C++
#include "precomp.h"
|
|
|
|
#include <debug.h>
|
|
#include "cluster.h"
|
|
#include "control.h"
|
|
#include "param.h"
|
|
|
|
CWlbsControl g_WlbsControl;
|
|
|
|
|
|
|
|
DWORD WINAPI WlbsInit
|
|
(
|
|
PTCHAR product,
|
|
DWORD version,
|
|
PVOID reserved
|
|
)
|
|
{
|
|
return g_WlbsControl.Initialize();
|
|
}
|
|
|
|
|
|
DWORD WINAPI WlbsQuery
|
|
(
|
|
DWORD cluster,
|
|
DWORD host,
|
|
PWLBS_RESPONSE response,
|
|
PDWORD num_hosts,
|
|
PDWORD host_map,
|
|
PVOID reserved
|
|
)
|
|
{
|
|
return g_WlbsControl.WlbsQuery(cluster, host, response,num_hosts, host_map,reserved);
|
|
}
|
|
|
|
DWORD WINAPI WlbsSuspend
|
|
(
|
|
DWORD cluster,
|
|
DWORD host,
|
|
PWLBS_RESPONSE response,
|
|
PDWORD num_hosts
|
|
)
|
|
{
|
|
return g_WlbsControl.WlbsSuspend(cluster, host, response, num_hosts);
|
|
}
|
|
|
|
DWORD WINAPI WlbsResume
|
|
(
|
|
DWORD cluster,
|
|
DWORD host,
|
|
PWLBS_RESPONSE response,
|
|
PDWORD num_hosts
|
|
)
|
|
{
|
|
return g_WlbsControl.WlbsResume(cluster, host, response, num_hosts);
|
|
}
|
|
|
|
DWORD WINAPI WlbsStart
|
|
(
|
|
DWORD cluster,
|
|
DWORD host,
|
|
PWLBS_RESPONSE response,
|
|
PDWORD num_hosts
|
|
)
|
|
{
|
|
return g_WlbsControl.WlbsStart(cluster, host, response, num_hosts);
|
|
}
|
|
|
|
DWORD WINAPI WlbsStop
|
|
(
|
|
DWORD cluster,
|
|
DWORD host,
|
|
PWLBS_RESPONSE response,
|
|
PDWORD num_hosts
|
|
)
|
|
{
|
|
return g_WlbsControl.WlbsStop(cluster, host, response, num_hosts);
|
|
}
|
|
|
|
DWORD WINAPI WlbsDrainStop
|
|
(
|
|
DWORD cluster,
|
|
DWORD host,
|
|
PWLBS_RESPONSE response,
|
|
PDWORD num_hosts
|
|
)
|
|
{
|
|
return g_WlbsControl.WlbsDrainStop(cluster, host, response, num_hosts);
|
|
}
|
|
|
|
DWORD WINAPI WlbsEnable
|
|
(
|
|
DWORD cluster,
|
|
DWORD host,
|
|
PWLBS_RESPONSE response,
|
|
PDWORD num_hosts,
|
|
DWORD vip,
|
|
DWORD port
|
|
)
|
|
{
|
|
return g_WlbsControl.WlbsEnable(cluster, host, response, num_hosts, vip, port);
|
|
}
|
|
|
|
DWORD WINAPI WlbsDisable
|
|
(
|
|
DWORD cluster,
|
|
DWORD host,
|
|
PWLBS_RESPONSE response,
|
|
PDWORD num_hosts,
|
|
DWORD vip,
|
|
DWORD port
|
|
)
|
|
{
|
|
return g_WlbsControl.WlbsDisable(cluster, host, response, num_hosts, vip, port);
|
|
}
|
|
|
|
DWORD WINAPI WlbsDrain
|
|
(
|
|
DWORD cluster,
|
|
DWORD host,
|
|
PWLBS_RESPONSE response,
|
|
PDWORD num_hosts,
|
|
DWORD vip,
|
|
DWORD port
|
|
)
|
|
{
|
|
return g_WlbsControl.WlbsDrain(cluster, host, response, num_hosts, vip, port);
|
|
}
|
|
|
|
extern "C" DWORD WINAPI WlbsAdjust
|
|
(
|
|
DWORD cluster,
|
|
DWORD host,
|
|
PWLBS_RESPONSE response,
|
|
PDWORD num_hosts,
|
|
DWORD port,
|
|
DWORD value
|
|
)
|
|
{
|
|
return g_WlbsControl.WlbsAdjust(cluster, host, response, num_hosts, port, value);
|
|
}
|
|
|
|
DWORD WINAPI WlbsResolve
|
|
(
|
|
const WCHAR* address
|
|
)
|
|
{
|
|
struct hostent * host;
|
|
char buf[256];
|
|
|
|
if (address == NULL)
|
|
return 0;
|
|
|
|
unsigned long addr = IpAddressFromAbcdWsz (address);
|
|
|
|
if (addr != INADDR_NONE)
|
|
{
|
|
return addr;
|
|
}
|
|
|
|
if (!WideCharToMultiByte(CP_ACP, 0, address, -1, buf, sizeof(buf), NULL, NULL))
|
|
return 0;
|
|
|
|
host = gethostbyname (buf);
|
|
|
|
if (host == NULL)
|
|
return 0;
|
|
|
|
return((struct in_addr *) (host -> h_addr)) -> s_addr;
|
|
|
|
} /* end WlbsResolve */
|
|
|
|
#define MAXIPSTRLEN WLBS_MAX_CL_IP_ADDR + 1
|
|
|
|
BOOL WINAPI WlbsAddressToString
|
|
(
|
|
DWORD address,
|
|
PTCHAR buf,
|
|
PDWORD lenp
|
|
)
|
|
{
|
|
PCHAR temp;
|
|
DWORD len;
|
|
|
|
if ( lenp == NULL )
|
|
return FALSE;
|
|
|
|
if (*lenp < MAXIPSTRLEN)
|
|
return FALSE;
|
|
|
|
AbcdWszFromIpAddress(address, buf);
|
|
*lenp=lstrlen(buf);
|
|
|
|
return TRUE;
|
|
|
|
} /* end WlbsAddressToString */
|
|
|
|
|
|
BOOL WINAPI WlbsAddressToName
|
|
(
|
|
DWORD address,
|
|
PTCHAR buf,
|
|
PDWORD lenp
|
|
)
|
|
{
|
|
struct hostent * name;
|
|
DWORD len;
|
|
|
|
if ( lenp == NULL )
|
|
return FALSE;
|
|
|
|
name = gethostbyaddr ((char *) & address, sizeof (DWORD), AF_INET);
|
|
|
|
if (name == NULL)
|
|
{
|
|
*lenp = 0;
|
|
return FALSE;
|
|
}
|
|
|
|
len = strlen (name -> h_name) + 1;
|
|
|
|
/* This is to prevent stprintf from breaking */
|
|
if (*lenp > 0)
|
|
{
|
|
name -> h_name [*lenp - 1] = 0;
|
|
#ifdef UNICODE
|
|
wsprintf (buf, L"%S", name -> h_name);
|
|
#else
|
|
sprintf (buf, "%s", name -> h_name);
|
|
#endif
|
|
}
|
|
|
|
if (len > *lenp)
|
|
{
|
|
*lenp = len;
|
|
return FALSE;
|
|
}
|
|
|
|
*lenp = len;
|
|
return TRUE;
|
|
|
|
} /* end WlbsAddressToName */
|
|
|
|
|
|
VOID WINAPI WlbsPortSet
|
|
(
|
|
DWORD cluster,
|
|
WORD port
|
|
)
|
|
{
|
|
g_WlbsControl.WlbsPortSet(cluster, port);
|
|
}
|
|
|
|
VOID WINAPI WlbsPasswordSet
|
|
(
|
|
DWORD cluster,
|
|
PTCHAR password
|
|
)
|
|
{
|
|
g_WlbsControl.WlbsPasswordSet(cluster, password);
|
|
}
|
|
|
|
VOID WINAPI WlbsCodeSet
|
|
(
|
|
DWORD cluster,
|
|
DWORD passw
|
|
)
|
|
{
|
|
g_WlbsControl.WlbsCodeSet(cluster, passw);
|
|
}
|
|
|
|
VOID WINAPI WlbsDestinationSet
|
|
(
|
|
DWORD cluster,
|
|
DWORD dest
|
|
)
|
|
{
|
|
g_WlbsControl.WlbsDestinationSet(cluster, dest);
|
|
}
|
|
|
|
VOID WINAPI WlbsTimeoutSet
|
|
(
|
|
DWORD cluster,
|
|
DWORD milliseconds
|
|
)
|
|
{
|
|
g_WlbsControl.WlbsTimeoutSet(cluster, milliseconds);
|
|
}
|
|
|
|
|
|
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: WlbsReadReg
|
|
//
|
|
// Description:
|
|
//
|
|
// Arguments: WORD cluster -
|
|
// PWLBS_REG_PARAMS reg_data
|
|
//
|
|
// Returns: DWORD WINAPI -
|
|
//
|
|
// History: fengsun Created Header 3/9/00
|
|
//
|
|
//+----------------------------------------------------------------------------
|
|
DWORD WINAPI WlbsReadReg(
|
|
DWORD cluster,
|
|
PWLBS_REG_PARAMS reg_data
|
|
)
|
|
{
|
|
CWlbsCluster* pCluster = g_WlbsControl.GetClusterFromIpOrIndex(cluster);
|
|
|
|
if(pCluster == NULL)
|
|
{
|
|
return WLBS_NOT_FOUND;
|
|
}
|
|
|
|
return pCluster->ReadConfig(reg_data);
|
|
}
|
|
|
|
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: WlbsWriteReg
|
|
//
|
|
// Description:
|
|
//
|
|
// Arguments: WORD cluster -
|
|
// PWLBS_REG_PARAMS reg_data
|
|
//
|
|
// Returns: DWORD WINAPI -
|
|
//
|
|
// History: fengsun Created Header 3/9/00
|
|
//
|
|
//+----------------------------------------------------------------------------
|
|
DWORD WINAPI WlbsWriteReg(
|
|
DWORD cluster,
|
|
const PWLBS_REG_PARAMS reg_data
|
|
)
|
|
{
|
|
CWlbsCluster* pCluster = g_WlbsControl.GetClusterFromIpOrIndex(cluster);
|
|
|
|
if(pCluster == NULL)
|
|
{
|
|
return WLBS_NOT_FOUND;
|
|
}
|
|
|
|
return pCluster->WriteConfig(reg_data);
|
|
}
|
|
|
|
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: WlbsCommitChanges
|
|
//
|
|
// Description:
|
|
//
|
|
// Arguments: DWORD cluster -
|
|
//
|
|
// Returns: DWORD WINAPI -
|
|
//
|
|
// History: fengsun Created Header 3/9/00
|
|
//
|
|
//+----------------------------------------------------------------------------
|
|
DWORD WINAPI WlbsCommitChanges(DWORD cluster)
|
|
{
|
|
CWlbsCluster* pCluster = g_WlbsControl.GetClusterFromIpOrIndex(cluster);
|
|
|
|
if(pCluster == NULL)
|
|
{
|
|
return WLBS_NOT_FOUND;
|
|
}
|
|
|
|
return pCluster->CommitChanges(&g_WlbsControl);
|
|
}
|
|
|
|
DWORD WINAPI WlbsSetDefaults
|
|
(
|
|
PWLBS_REG_PARAMS reg_data
|
|
)
|
|
{
|
|
return ParamSetDefaults(reg_data);
|
|
}
|
|
|
|
|
|
BOOL WINAPI WlbsFormatMessage
|
|
(
|
|
DWORD, // error,
|
|
WLBS_COMMAND, // command,
|
|
BOOL, // cluster
|
|
PTCHAR messagep,
|
|
PDWORD lenp
|
|
)
|
|
{
|
|
DWORD len;
|
|
PTSTR error_str = _TEXT("Not yet implemented");
|
|
|
|
#if 0
|
|
va_list arglist;
|
|
DWORD error;
|
|
|
|
|
|
va_start(arglist, id);
|
|
|
|
|
|
if (FormatMessage (FORMAT_MESSAGE_FROM_HMODULE, NULL, id, 0,
|
|
message, CVY_BUF_SIZE, & arglist) == 0)
|
|
{
|
|
error = GetLastError();
|
|
|
|
wprintf(L"Could not print error message due to: ");
|
|
|
|
if (FormatMessage (FORMAT_MESSAGE_FROM_SYSTEM, NULL, error, 0,
|
|
message, CVY_BUF_SIZE, NULL) == 0)
|
|
{
|
|
wprintf(L"%d\n", error);
|
|
}
|
|
else
|
|
wprintf(L"\n %ls\n", message);
|
|
}
|
|
else
|
|
{
|
|
wprintf(L"%ls", message);
|
|
}
|
|
#endif
|
|
|
|
|
|
|
|
len = _tcslen(error_str) + 1;
|
|
|
|
/* This is to prevent stprintf from breaking */
|
|
|
|
if (lenp == NULL)
|
|
return FALSE;
|
|
|
|
if (*lenp == 0)
|
|
{
|
|
*lenp = len;
|
|
return FALSE;
|
|
}
|
|
|
|
_tcsncpy(messagep, error_str, *lenp);
|
|
|
|
if (len > *lenp)
|
|
{
|
|
*lenp = len;
|
|
return FALSE;
|
|
}
|
|
|
|
*lenp = len;
|
|
return TRUE;
|
|
|
|
} /* end WlbsFormatMessage */
|
|
|
|
DWORD WINAPI WlbsEnumClusters(OUT DWORD* pdwAddresses, OUT DWORD* pdwNum)
|
|
{
|
|
return g_WlbsControl.EnumClusters(pdwAddresses, pdwNum);
|
|
}
|
|
|
|
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: WlbsGetAdapterGuid
|
|
//
|
|
// Description:
|
|
//
|
|
// Arguments: IN DWORD cluster -
|
|
// OUT GUID* pAdapterGuid -
|
|
//
|
|
// Returns: DWORD WINAPI -
|
|
//
|
|
// History: fengsun Created Header 3/10/00
|
|
//
|
|
//+----------------------------------------------------------------------------
|
|
DWORD WINAPI WlbsGetAdapterGuid(IN DWORD cluster, OUT GUID* pAdapterGuid)
|
|
{
|
|
ASSERT(pAdapterGuid);
|
|
|
|
if (pAdapterGuid == NULL)
|
|
{
|
|
return WLBS_BAD_PARAMS;
|
|
}
|
|
|
|
CWlbsCluster* pCluster = g_WlbsControl.GetClusterFromIpOrIndex(cluster);
|
|
|
|
if(pCluster == NULL)
|
|
{
|
|
return WLBS_NOT_FOUND;
|
|
}
|
|
|
|
*pAdapterGuid = pCluster->GetAdapterGuid();
|
|
|
|
return WLBS_OK;
|
|
}
|
|
|
|
|
|
|
|
//+----------------------------------------------------------------------------
|
|
//
|
|
// Function: WlbsNotifyConfigChange
|
|
//
|
|
// Description:
|
|
//
|
|
// Arguments: DWORD cluster -
|
|
//
|
|
// Returns: DWORD WINAPI -
|
|
//
|
|
// History: fengsun Created Header 7/3/00
|
|
//
|
|
//+----------------------------------------------------------------------------
|
|
DWORD WINAPI WlbsNotifyConfigChange(DWORD cluster)
|
|
{
|
|
CWlbsCluster* pCluster = g_WlbsControl.GetClusterFromIpOrIndex(cluster);
|
|
|
|
if(pCluster == NULL)
|
|
{
|
|
return WLBS_NOT_FOUND;
|
|
}
|
|
|
|
DWORD dwStatus = NotifyDriverConfigChanges(g_WlbsControl.GetDriverHandle(), pCluster->GetAdapterGuid());
|
|
|
|
return dwStatus;
|
|
}
|