179 lines
4 KiB
C++
179 lines
4 KiB
C++
|
/*++
|
||
|
|
|
||
|
|
Copyright(c) 2001 Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
NLB Manager
|
||
|
|
|
||
|
|
File Name:
|
||
|
|
|
||
|
|
nlbhost.cpp
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
Implementation of class NLBHost
|
||
|
|
|
||
|
|
NLBHost is responsible for connecting to an NLB host and getting/setting
|
||
|
|
its NLB-related configuration.
|
||
|
|
|
||
|
|
History:
|
||
|
|
|
||
|
|
03/31/01 JosephJ Created
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
|
||
|
|
#include "tprov.h"
|
||
|
|
|
||
|
|
|
||
|
|
VOID
|
||
|
|
MyLogger(
|
||
|
|
PVOID Context,
|
||
|
|
const WCHAR * Text
|
||
|
|
)
|
||
|
|
{
|
||
|
|
wprintf(L"LOG: %s", Text);
|
||
|
|
wprintf(L"\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
void test_tmgr(int argc, WCHAR* argv[])
|
||
|
|
{
|
||
|
|
|
||
|
|
WCHAR wszBuf[1024];
|
||
|
|
NLBHost * pHost = NULL;
|
||
|
|
WBEMSTATUS Status;
|
||
|
|
NLBHost::HostInformation *pHostInfo = NULL;
|
||
|
|
|
||
|
|
pHost = new NLBHost(
|
||
|
|
L"JOSEPHJ4E",
|
||
|
|
L"JosephJ's Dev Machine",
|
||
|
|
MyLogger,
|
||
|
|
NULL // Logger context
|
||
|
|
);
|
||
|
|
|
||
|
|
if (pHost == NULL)
|
||
|
|
{
|
||
|
|
MyLogger(NULL, L"Could not create an instance of NLBHost.");
|
||
|
|
goto end;
|
||
|
|
}
|
||
|
|
|
||
|
|
UINT PingStatus = pHost->Ping();
|
||
|
|
|
||
|
|
if (PingStatus!=ERROR_SUCCESS) goto end;
|
||
|
|
|
||
|
|
|
||
|
|
Status = pHost->GetHostInformation(
|
||
|
|
&pHostInfo
|
||
|
|
);
|
||
|
|
|
||
|
|
if (FAILED(Status))
|
||
|
|
{
|
||
|
|
pHostInfo = NULL;
|
||
|
|
goto end;
|
||
|
|
}
|
||
|
|
|
||
|
|
int NumNics = pHostInfo->NumNics;
|
||
|
|
|
||
|
|
wsprintf(
|
||
|
|
wszBuf,
|
||
|
|
L"Processing host with MachineName %s(NumNics=%d)",
|
||
|
|
(LPCWSTR)pHostInfo->MachineName,
|
||
|
|
NumNics
|
||
|
|
);
|
||
|
|
MyLogger(NULL, wszBuf);
|
||
|
|
|
||
|
|
for( int i = 0; i < NumNics; i++)
|
||
|
|
{
|
||
|
|
NLBHost::NicInformation *pNicInfo;
|
||
|
|
NLB_EXTENDED_CLUSTER_CONFIGURATION ClusterConfig;
|
||
|
|
UINT GenerationId=0;
|
||
|
|
UINT RequestId;
|
||
|
|
|
||
|
|
pNicInfo = &pHostInfo->nicInformation[i];
|
||
|
|
wsprintf(
|
||
|
|
wszBuf,
|
||
|
|
L"Processing NIC %s (%s)\n",
|
||
|
|
(LPCWSTR) pNicInfo->adapterGuid,
|
||
|
|
(LPCWSTR) pNicInfo->friendlyName
|
||
|
|
);
|
||
|
|
MyLogger(NULL, wszBuf);
|
||
|
|
|
||
|
|
Status = pHost->GetClusterConfiguration(
|
||
|
|
pNicInfo->adapterGuid,
|
||
|
|
&ClusterConfig
|
||
|
|
);
|
||
|
|
if (FAILED(Status))
|
||
|
|
{
|
||
|
|
// Failure...
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
Status = pHost->SetClusterConfiguration(
|
||
|
|
pNicInfo->adapterGuid,
|
||
|
|
&ClusterConfig,
|
||
|
|
GenerationId,
|
||
|
|
&RequestId
|
||
|
|
);
|
||
|
|
|
||
|
|
if (Status == WBEM_S_PENDING)
|
||
|
|
{
|
||
|
|
INT Delay = 1000; // 1 sec
|
||
|
|
|
||
|
|
// Give some time for the remote host's connectivity to be lost...
|
||
|
|
//
|
||
|
|
MyLogger(NULL, L"Sleeping for 5 seconds.");
|
||
|
|
Sleep(5000);
|
||
|
|
|
||
|
|
// Now wait until we can ping the host, then query for the async
|
||
|
|
// result of the SetClusterConfiguration operation.
|
||
|
|
// We keep polling for the result, sleeping increasing amounts
|
||
|
|
// in between.
|
||
|
|
//
|
||
|
|
MyLogger(NULL, L"Pinging the host ....");
|
||
|
|
while (pHost->Ping() == ERROR_SUCCESS)
|
||
|
|
{
|
||
|
|
UINT ResultCode;
|
||
|
|
_bstr_t ResultText;
|
||
|
|
|
||
|
|
MyLogger(NULL, L"Checking for asynchronous completion...");
|
||
|
|
Status = pHost->GetAsyncResult(
|
||
|
|
RequestId,
|
||
|
|
&GenerationId,
|
||
|
|
&ResultCode,
|
||
|
|
&ResultText
|
||
|
|
);
|
||
|
|
if (Status == WBEM_S_PENDING)
|
||
|
|
{
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
MyLogger(NULL, L"Operation still pending...");
|
||
|
|
MyLogger(NULL, L"Waiting to try again ....");
|
||
|
|
Sleep(Delay);
|
||
|
|
MyLogger(NULL, L"Pinging the host ....");
|
||
|
|
Delay *=2;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
end:
|
||
|
|
|
||
|
|
if (pHostInfo != NULL)
|
||
|
|
{
|
||
|
|
delete pHostInfo;
|
||
|
|
pHostInfo = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (pHost != NULL)
|
||
|
|
{
|
||
|
|
delete pHost;
|
||
|
|
pHost = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
return;
|
||
|
|
|
||
|
|
}
|