windows-nt/Source/XPSP1/NT/net/wlbs/nlbmgr/provider/tests/tmgr.cpp
2020-09-26 16:20:57 +08:00

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;
}