929 lines
25 KiB
C++
929 lines
25 KiB
C++
|
/*++
|
||
|
|
|
||
|
|
Copyright (c) 1998 Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
iislbc.cxx
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
Command line utility to set/get IIS load balancing configuration
|
||
|
|
|
||
|
|
Author:
|
||
|
|
|
||
|
|
Philippe Choquier (phillich)
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
//#define INITGUID
|
||
|
|
extern "C" {
|
||
|
|
|
||
|
|
#include <nt.h>
|
||
|
|
#include <ntrtl.h>
|
||
|
|
#include <nturtl.h>
|
||
|
|
#include <ntsecapi.h>
|
||
|
|
|
||
|
|
#include <windows.h>
|
||
|
|
#include <winsock2.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <time.h>
|
||
|
|
|
||
|
|
#include <ipnat.h>
|
||
|
|
|
||
|
|
#if 0
|
||
|
|
typedef NTSTATUS
|
||
|
|
(*PNAT_REGISTER_SESSION_CONTROL)(
|
||
|
|
IN ULONG Version
|
||
|
|
);
|
||
|
|
|
||
|
|
typedef NTSTATUS
|
||
|
|
(*PNAT_IOCTL_SESSION_CONTROL)(
|
||
|
|
IN PIRP Irp
|
||
|
|
);
|
||
|
|
#endif
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <ole2.h>
|
||
|
|
#include <pdh.h>
|
||
|
|
|
||
|
|
#include <IisLb.h>
|
||
|
|
#include <lbxbf.hxx>
|
||
|
|
#include <lbcnfg.hxx>
|
||
|
|
#include <IisLbs.hxx>
|
||
|
|
#include <bootexp.hxx>
|
||
|
|
|
||
|
|
#include "iisrc.h"
|
||
|
|
|
||
|
|
//
|
||
|
|
// HRESULTTOWIN32() maps an HRESULT to a Win32 error. If the facility code
|
||
|
|
// of the HRESULT is FACILITY_WIN32, then the code portion (i.e. the
|
||
|
|
// original Win32 error) is returned. Otherwise, the original HRESULT is
|
||
|
|
// returned unchagned.
|
||
|
|
//
|
||
|
|
|
||
|
|
#define HRESULTTOWIN32(hres) \
|
||
|
|
((HRESULT_FACILITY(hres) == FACILITY_WIN32) \
|
||
|
|
? HRESULT_CODE(hres) \
|
||
|
|
: (hres))
|
||
|
|
|
||
|
|
|
||
|
|
#define RETURNCODETOHRESULT(rc) \
|
||
|
|
(((rc) < 0x10000) \
|
||
|
|
? HRESULT_FROM_WIN32(rc) \
|
||
|
|
: (rc))
|
||
|
|
|
||
|
|
enum LB_CMD {
|
||
|
|
CMD_NONE,
|
||
|
|
CMD_ADD_SERVER,
|
||
|
|
CMD_ADD_IP,
|
||
|
|
CMD_ADD_PERFMON,
|
||
|
|
CMD_DEL_SERVER,
|
||
|
|
CMD_DEL_IP,
|
||
|
|
CMD_DEL_PERFMON,
|
||
|
|
#if 0
|
||
|
|
CMD_SET_STICKY,
|
||
|
|
#endif
|
||
|
|
CMD_SET_IP,
|
||
|
|
CMD_LIST,
|
||
|
|
CMD_LIST_IP_ENDPOINTS,
|
||
|
|
CMD_START_DRIVER,
|
||
|
|
CMD_STOP_DRIVER,
|
||
|
|
} ;
|
||
|
|
|
||
|
|
#define LEN_SERVER_NAME 16
|
||
|
|
#define LEN_IP (4*3+3+1+3)
|
||
|
|
#define LEN_PERF 40
|
||
|
|
|
||
|
|
extern CKernelIpMapHelper g_KernelIpMap;
|
||
|
|
extern WSADATA g_WSAData;
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
LocatePerfCounter(
|
||
|
|
CComputerPerfCounters* pPerfCounters,
|
||
|
|
LPWSTR pszSrv,
|
||
|
|
LPWSTR pszParam,
|
||
|
|
UINT* piPerfCounter
|
||
|
|
)
|
||
|
|
{
|
||
|
|
UINT i;
|
||
|
|
LPWSTR pszS;
|
||
|
|
LPWSTR psz;
|
||
|
|
DWORD dw;
|
||
|
|
|
||
|
|
if ( pszSrv && (*pszSrv == L'\0' || !wcscmp(pszSrv,L"*")) )
|
||
|
|
{
|
||
|
|
pszSrv = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
for ( i = 0 ;
|
||
|
|
pPerfCounters->EnumPerfCounter( i, &pszS, &psz, &dw ) ;
|
||
|
|
++i )
|
||
|
|
{
|
||
|
|
if ( ((pszS == NULL) == (pszSrv==NULL)) &&
|
||
|
|
!_wcsicmp( psz, pszParam ) )
|
||
|
|
{
|
||
|
|
*piPerfCounter = i;
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
SetLastError( ERROR_INVALID_PARAMETER );
|
||
|
|
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
LocateServer(
|
||
|
|
CIPMap* IpMap,
|
||
|
|
LPWSTR pszParam,
|
||
|
|
UINT* piServer
|
||
|
|
)
|
||
|
|
{
|
||
|
|
UINT i;
|
||
|
|
LPWSTR psz;
|
||
|
|
|
||
|
|
|
||
|
|
for ( i = 0 ;
|
||
|
|
IpMap->EnumComputer( i, &psz ) ;
|
||
|
|
++ i )
|
||
|
|
{
|
||
|
|
if ( !_wcsicmp( psz, pszParam ) )
|
||
|
|
{
|
||
|
|
*piServer = i;
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
SetLastError( ERROR_INVALID_PARAMETER );
|
||
|
|
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
LocatePublicIp(
|
||
|
|
CIPMap* IpMap,
|
||
|
|
LPWSTR pszParam,
|
||
|
|
UINT* piRes
|
||
|
|
)
|
||
|
|
{
|
||
|
|
UINT i;
|
||
|
|
LPWSTR psz;
|
||
|
|
LPWSTR pszName;
|
||
|
|
DWORD dwSticky;
|
||
|
|
DWORD dwAttr;
|
||
|
|
|
||
|
|
|
||
|
|
for ( i = 0 ;
|
||
|
|
IpMap->EnumIpPublic( i, &psz, &pszName, &dwSticky, &dwAttr ) ;
|
||
|
|
++i )
|
||
|
|
{
|
||
|
|
if ( !_wcsicmp( psz, pszParam ) )
|
||
|
|
{
|
||
|
|
*piRes = i;
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
SetLastError( ERROR_INVALID_PARAMETER );
|
||
|
|
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
VOID
|
||
|
|
DisplayErrorMessage(
|
||
|
|
DWORD dwErr
|
||
|
|
)
|
||
|
|
{
|
||
|
|
LPWSTR pErr;
|
||
|
|
|
||
|
|
if ( FormatMessageW( FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
|
||
|
|
NULL,
|
||
|
|
dwErr,
|
||
|
|
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
|
||
|
|
(LPWSTR)&pErr,
|
||
|
|
0,
|
||
|
|
NULL ) )
|
||
|
|
{
|
||
|
|
LPWSTR p;
|
||
|
|
|
||
|
|
if ( p = wcschr( pErr, L'\r' ) )
|
||
|
|
{
|
||
|
|
*p = L'\0';
|
||
|
|
}
|
||
|
|
fputws( pErr, stdout );
|
||
|
|
|
||
|
|
LocalFree( pErr );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
ChooseCounter(
|
||
|
|
LPWSTR pszCounterBuffer
|
||
|
|
)
|
||
|
|
{
|
||
|
|
PDH_BROWSE_DLG_CONFIG_W BrowseInfo;
|
||
|
|
PDH_STATUS pdhStatus;
|
||
|
|
LPWSTR pszPostServerName;
|
||
|
|
WCHAR achSelect[128];
|
||
|
|
|
||
|
|
if ( !LoadStringW( NULL, IDS_LBC_SP, achSelect, sizeof(achSelect) ) )
|
||
|
|
{
|
||
|
|
achSelect[0] = L'\0';
|
||
|
|
}
|
||
|
|
|
||
|
|
#if defined(_X86_) && 0
|
||
|
|
__asm int 3
|
||
|
|
#endif
|
||
|
|
|
||
|
|
BrowseInfo.bIncludeInstanceIndex = FALSE;
|
||
|
|
BrowseInfo.bSingleCounterPerAdd = TRUE;
|
||
|
|
BrowseInfo.bSingleCounterPerDialog = TRUE;
|
||
|
|
BrowseInfo.bLocalCountersOnly = FALSE;
|
||
|
|
BrowseInfo.bWildCardInstances = FALSE;
|
||
|
|
BrowseInfo.bHideDetailBox = FALSE;
|
||
|
|
BrowseInfo.bInitializePath = FALSE;
|
||
|
|
BrowseInfo.bDisableMachineSelection = FALSE;
|
||
|
|
BrowseInfo.bIncludeCostlyObjects = TRUE;
|
||
|
|
|
||
|
|
BrowseInfo.hWndOwner = GetActiveWindow();
|
||
|
|
BrowseInfo.szDataSource = NULL;
|
||
|
|
BrowseInfo.szReturnPathBuffer = pszCounterBuffer;
|
||
|
|
BrowseInfo.cchReturnPathLength = MAX_PATH;
|
||
|
|
BrowseInfo.pCallBack = NULL;
|
||
|
|
BrowseInfo.dwCallBackArg = 0;
|
||
|
|
BrowseInfo.CallBackStatus = ERROR_SUCCESS;
|
||
|
|
BrowseInfo.dwDefaultDetailLevel = PERF_DETAIL_WIZARD;
|
||
|
|
BrowseInfo.szDialogBoxCaption = achSelect;
|
||
|
|
|
||
|
|
if ( PdhBrowseCountersW (&BrowseInfo) == ERROR_SUCCESS )
|
||
|
|
{
|
||
|
|
if ( !memcmp( pszCounterBuffer, L"\\\\", 2*sizeof(WCHAR) ) &&
|
||
|
|
(pszPostServerName = wcschr( pszCounterBuffer+2, L'\\' )) )
|
||
|
|
{
|
||
|
|
memmove( pszCounterBuffer, pszPostServerName, (wcslen(pszPostServerName)+1)*sizeof(WCHAR) );
|
||
|
|
}
|
||
|
|
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
#if 0
|
||
|
|
|
||
|
|
VOID
|
||
|
|
TestDriver(
|
||
|
|
)
|
||
|
|
{
|
||
|
|
HMODULE hMod;
|
||
|
|
PNAT_REGISTER_SESSION_CONTROL pfnNatRegisterSessionControl;
|
||
|
|
PNAT_IOCTL_SESSION_CONTROL pfnNatIoctlSessionControl;
|
||
|
|
PNAT_DIRECTOR_QUERY_SESSION pfnNatQuerySessionControl;
|
||
|
|
IRP Irp;
|
||
|
|
ULONG PrivateAddr;
|
||
|
|
USHORT PrivatePort;
|
||
|
|
ULONG PublicAddr;
|
||
|
|
USHORT PublicPort;
|
||
|
|
ULONG RemoteAddr;
|
||
|
|
USHORT RemotePort;
|
||
|
|
CKernelServerDescription* pS;
|
||
|
|
UINT iS;
|
||
|
|
UINT iP;
|
||
|
|
IO_STACK_LOCATION IrpSp;
|
||
|
|
LPVOID pv;
|
||
|
|
|
||
|
|
#if 0
|
||
|
|
UINT x;
|
||
|
|
UINT c;
|
||
|
|
UINT f;
|
||
|
|
UINT y;
|
||
|
|
|
||
|
|
for ( c = 0, x = 2 ; x < 1000 ; ++x )
|
||
|
|
{
|
||
|
|
f = TRUE;
|
||
|
|
for ( y = 2 ; y * y < x ; ++y )
|
||
|
|
{
|
||
|
|
if ( x % y == 0 )
|
||
|
|
{
|
||
|
|
f = FALSE;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if ( f )
|
||
|
|
{
|
||
|
|
wprintf( L"%d\t", x );
|
||
|
|
if ( (++c & 7) == 7 )
|
||
|
|
{
|
||
|
|
wprintf(L"\n");
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
if ( (hMod = LoadLibraryW( L"iislbdr.dll" )) )
|
||
|
|
{
|
||
|
|
pfnNatRegisterSessionControl = (PNAT_REGISTER_SESSION_CONTROL)GetProcAddress( hMod,
|
||
|
|
"NatRegisterSessionControl" );
|
||
|
|
pfnNatIoctlSessionControl = (PNAT_IOCTL_SESSION_CONTROL)GetProcAddress( hMod,
|
||
|
|
"NatIoctlSessionControl" );
|
||
|
|
pfnNatQuerySessionControl = (PNAT_DIRECTOR_QUERY_SESSION)GetProcAddress( hMod,
|
||
|
|
"NatQuerySessionControl" );
|
||
|
|
|
||
|
|
if ( pfnNatRegisterSessionControl &&
|
||
|
|
pfnNatIoctlSessionControl &&
|
||
|
|
pfnNatQuerySessionControl )
|
||
|
|
{
|
||
|
|
if ( WSAStartup( MAKEWORD( 2, 0), &g_WSAData ) == 0 &&
|
||
|
|
InitGlobals() )
|
||
|
|
{
|
||
|
|
PublicAddr = 0x04030201;
|
||
|
|
PublicPort = 0x5000;
|
||
|
|
RemoteAddr = 0x0f0f0f0f;
|
||
|
|
RemotePort = 0xf000;
|
||
|
|
|
||
|
|
pfnNatRegisterSessionControl( 1 );
|
||
|
|
|
||
|
|
for ( iS = 0;
|
||
|
|
iS < g_KernelIpMap.ServerCount() ;
|
||
|
|
++iS )
|
||
|
|
{
|
||
|
|
pS = g_KernelIpMap.GetServerPtr( iS );
|
||
|
|
pS->m_dwLoadAvail = 100;
|
||
|
|
pS->m_LoadbalancedLoadAvail = pS->m_dwLoadAvail;
|
||
|
|
}
|
||
|
|
|
||
|
|
// refcount set to != 0
|
||
|
|
|
||
|
|
for ( iP = 0;
|
||
|
|
iP < g_KernelIpMap.PrivateIpCount() ;
|
||
|
|
++iP )
|
||
|
|
{
|
||
|
|
g_KernelIpMap.GetPrivateIpEndpoint( iP )->m_dwIndex = 1;
|
||
|
|
}
|
||
|
|
|
||
|
|
#if defined(_X86_)
|
||
|
|
__asm int 3
|
||
|
|
#endif
|
||
|
|
g_KernelIpMap.GetPublicIpPtr( 0 )->m_dwNotifyPort = 0x5000;
|
||
|
|
g_KernelIpMap.GetPublicIpPtr( 0 )->m_usUniquePort = 0x5000;
|
||
|
|
IoGetCurrentIrpStackLocation( &Irp ) = &IrpSp;
|
||
|
|
IrpSp.Parameters.DeviceIoControl.InputBufferLength = g_KernelIpMap.GetSize();
|
||
|
|
Irp.AssociatedIrp.SystemBuffer = g_KernelIpMap.GetBuffer();
|
||
|
|
pfnNatIoctlSessionControl( &Irp );
|
||
|
|
pfnNatIoctlSessionControl( &Irp );
|
||
|
|
|
||
|
|
pfnNatQuerySessionControl( NULL, NAT_PROTOCOL_TCP,
|
||
|
|
PublicAddr, PublicPort,
|
||
|
|
RemoteAddr, RemotePort,
|
||
|
|
&PrivateAddr, &PrivatePort,
|
||
|
|
&pv );
|
||
|
|
|
||
|
|
// test cache
|
||
|
|
|
||
|
|
pfnNatQuerySessionControl( NULL, NAT_PROTOCOL_TCP,
|
||
|
|
PublicAddr, PublicPort,
|
||
|
|
RemoteAddr, RemotePort,
|
||
|
|
&PrivateAddr, &PrivatePort,
|
||
|
|
&pv );
|
||
|
|
|
||
|
|
// test new public IP addr
|
||
|
|
|
||
|
|
PublicAddr = 0x05030201;
|
||
|
|
PublicPort = 0x5000;
|
||
|
|
pfnNatQuerySessionControl( NULL, NAT_PROTOCOL_TCP,
|
||
|
|
PublicAddr, PublicPort,
|
||
|
|
RemoteAddr, RemotePort,
|
||
|
|
&PrivateAddr, &PrivatePort,
|
||
|
|
&pv );
|
||
|
|
|
||
|
|
// test invalid public IP addr
|
||
|
|
|
||
|
|
PublicAddr = 0xff030201;
|
||
|
|
PublicPort = 0x5000;
|
||
|
|
pfnNatQuerySessionControl( NULL, NAT_PROTOCOL_TCP,
|
||
|
|
PublicAddr, PublicPort,
|
||
|
|
RemoteAddr, RemotePort,
|
||
|
|
&PrivateAddr, &PrivatePort,
|
||
|
|
&pv );
|
||
|
|
|
||
|
|
pfnNatIoctlSessionControl( &Irp );
|
||
|
|
}
|
||
|
|
|
||
|
|
TerminateGlobals();
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
DisplayErrorMessage( GetLastError() );
|
||
|
|
}
|
||
|
|
|
||
|
|
FreeLibrary( hMod );
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
DisplayErrorMessage( GetLastError() );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|
||
|
|
|
||
|
|
BYTE abSerialized[32768];
|
||
|
|
|
||
|
|
#define RC_BUFF achMsg
|
||
|
|
#define RC_PRINTF(a,b) if ( LoadStringW( NULL, a, achMsg, sizeof(achMsg) ) ) wprintf b;
|
||
|
|
|
||
|
|
LPWSTR
|
||
|
|
ToUnicode(
|
||
|
|
LPSTR psz
|
||
|
|
)
|
||
|
|
{
|
||
|
|
LPWSTR ps = (LPWSTR)LocalAlloc( LMEM_FIXED, (strlen(psz)+1)*sizeof(WCHAR) );
|
||
|
|
|
||
|
|
if ( ps )
|
||
|
|
{
|
||
|
|
if ( !MultiByteToWideChar( CP_ACP,
|
||
|
|
MB_PRECOMPOSED,
|
||
|
|
psz,
|
||
|
|
-1,
|
||
|
|
ps,
|
||
|
|
(strlen(psz)+1) ) )
|
||
|
|
{
|
||
|
|
*ps = L'\0';
|
||
|
|
}
|
||
|
|
|
||
|
|
return ps;
|
||
|
|
}
|
||
|
|
|
||
|
|
return L"";
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
int __cdecl main(
|
||
|
|
int argc,
|
||
|
|
CHAR* argv[]
|
||
|
|
)
|
||
|
|
{
|
||
|
|
int Status = 0;
|
||
|
|
HRESULT hRes;
|
||
|
|
LPSTR pszMachineName = NULL;
|
||
|
|
COSERVERINFO csiMachineName;
|
||
|
|
MULTI_QI rgmq;
|
||
|
|
IClassFactory* pcsfFactory = NULL;
|
||
|
|
IMSIisLb* pIisLb = NULL;
|
||
|
|
WCHAR awchComputer[64];
|
||
|
|
CIPMap IpMap;
|
||
|
|
CComputerPerfCounters PerfCounters;
|
||
|
|
CIpEndpointList IpEndpoints;
|
||
|
|
DWORD dwSticky;
|
||
|
|
DWORD dwReq;
|
||
|
|
LPBYTE pbSerialized;
|
||
|
|
int arg;
|
||
|
|
int cParam;
|
||
|
|
LPWSTR apszParam[16];
|
||
|
|
LB_CMD iCmd = CMD_NONE;
|
||
|
|
BOOL fUpdateIpMap = FALSE;
|
||
|
|
BOOL fUpdatePerfCounters = FALSE;
|
||
|
|
BOOL fUpdateSticky = FALSE;
|
||
|
|
UINT iServer;
|
||
|
|
UINT iPublicIp;
|
||
|
|
UINT iPerfCounter;
|
||
|
|
XBF xbf;
|
||
|
|
LPWSTR pszPublicIp;
|
||
|
|
LPWSTR pszPrivateIp;
|
||
|
|
LPWSTR pszServer;
|
||
|
|
LPWSTR pszName;
|
||
|
|
LPWSTR pszPerfCounter;
|
||
|
|
DWORD dwWeight;
|
||
|
|
WCHAR achCounterBuffer[MAX_PATH];
|
||
|
|
WCHAR achMsg[1024];
|
||
|
|
DWORD dwAttr;
|
||
|
|
|
||
|
|
if ( argc < 2 )
|
||
|
|
{
|
||
|
|
RC_PRINTF( IDS_LBC_HELP, (RC_BUFF) );
|
||
|
|
|
||
|
|
return 3;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
RC_PRINTF( IDS_LBC_COPYRIGHT, (RC_BUFF) );
|
||
|
|
}
|
||
|
|
|
||
|
|
cParam = 0;
|
||
|
|
|
||
|
|
for ( arg = 1 ; arg < argc ; ++arg )
|
||
|
|
{
|
||
|
|
if ( argv[arg][0] == '-' )
|
||
|
|
{
|
||
|
|
switch ( argv[arg][1] )
|
||
|
|
{
|
||
|
|
case 'm':
|
||
|
|
pszMachineName = argv[arg]+2;
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 'a':
|
||
|
|
switch ( argv[arg][2] )
|
||
|
|
{
|
||
|
|
case 's':
|
||
|
|
iCmd = CMD_ADD_SERVER; break;
|
||
|
|
case 'i':
|
||
|
|
iCmd = CMD_ADD_IP; break;
|
||
|
|
case 'p':
|
||
|
|
iCmd = CMD_ADD_PERFMON; break;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 'd':
|
||
|
|
switch ( argv[arg][2] )
|
||
|
|
{
|
||
|
|
case 's':
|
||
|
|
iCmd = CMD_DEL_SERVER; break;
|
||
|
|
case 'i':
|
||
|
|
iCmd = CMD_DEL_IP; break;
|
||
|
|
case 'p':
|
||
|
|
iCmd = CMD_DEL_PERFMON; break;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 's':
|
||
|
|
switch ( argv[arg][2] )
|
||
|
|
{
|
||
|
|
#if 0
|
||
|
|
case 's':
|
||
|
|
iCmd = CMD_SET_STICKY; break;
|
||
|
|
#endif
|
||
|
|
case 'i':
|
||
|
|
iCmd = CMD_SET_IP; break;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case 'l':
|
||
|
|
switch ( argv[arg][2] )
|
||
|
|
{
|
||
|
|
case 'e':
|
||
|
|
iCmd = CMD_LIST_IP_ENDPOINTS; break;
|
||
|
|
case 'c':
|
||
|
|
iCmd = CMD_LIST;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
#if 0
|
||
|
|
case 't':
|
||
|
|
TestDriver();
|
||
|
|
return 0;
|
||
|
|
#endif
|
||
|
|
|
||
|
|
case '0':
|
||
|
|
iCmd = CMD_STOP_DRIVER;
|
||
|
|
break;
|
||
|
|
|
||
|
|
case '1':
|
||
|
|
iCmd = CMD_START_DRIVER;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else if ( cParam < sizeof(apszParam)/sizeof(LPWSTR) )
|
||
|
|
{
|
||
|
|
apszParam[cParam++] = ToUnicode( argv[arg] );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//fill the structure for CoCreateInstanceEx
|
||
|
|
ZeroMemory( &csiMachineName, sizeof(csiMachineName) );
|
||
|
|
|
||
|
|
if ( pszMachineName )
|
||
|
|
{
|
||
|
|
if ( !MultiByteToWideChar( CP_ACP,
|
||
|
|
MB_PRECOMPOSED,
|
||
|
|
pszMachineName,
|
||
|
|
-1,
|
||
|
|
awchComputer,
|
||
|
|
sizeof(awchComputer) ) )
|
||
|
|
{
|
||
|
|
return FALSE;
|
||
|
|
}
|
||
|
|
|
||
|
|
csiMachineName.pwszName = awchComputer;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
csiMachineName.pwszName = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
CoInitializeEx( NULL, COINIT_MULTITHREADED );
|
||
|
|
|
||
|
|
#if 0
|
||
|
|
hRes = CoGetClassObject(CLSID_MSIisLb, CLSCTX_SERVER, &csiMachineName,
|
||
|
|
IID_IClassFactory, (void**) &pcsfFactory);
|
||
|
|
|
||
|
|
if ( SUCCEEDED( hRes ) )
|
||
|
|
{
|
||
|
|
hRes = pcsfFactory->CreateInstance(NULL, IID_IMSIisLb, (void **) &rgmq.pItf);
|
||
|
|
|
||
|
|
if ( SUCCEEDED( hRes ) )
|
||
|
|
#else
|
||
|
|
if ( 1 )
|
||
|
|
{
|
||
|
|
rgmq.pIID = &IID_IMSIisLb;
|
||
|
|
rgmq.hr = 0;
|
||
|
|
rgmq.pItf = NULL;
|
||
|
|
hRes = CoCreateInstanceEx( CLSID_MSIisLb, NULL, CLSCTX_SERVER, &csiMachineName, 1, &rgmq );
|
||
|
|
|
||
|
|
if ( SUCCEEDED( hRes ) && SUCCEEDED( rgmq.hr ) )
|
||
|
|
#endif
|
||
|
|
{
|
||
|
|
pIisLb = (IMSIisLb*)rgmq.pItf;
|
||
|
|
|
||
|
|
// get current configuration
|
||
|
|
|
||
|
|
if ( SUCCEEDED( hRes = pIisLb->GetIpList( sizeof(abSerialized), abSerialized, &dwReq ) ) )
|
||
|
|
{
|
||
|
|
pbSerialized = abSerialized;
|
||
|
|
hRes = IpMap.Unserialize( &pbSerialized, &dwReq ) ? S_OK : RETURNCODETOHRESULT(ERROR_BAD_CONFIGURATION);
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( SUCCEEDED( hRes ) &&
|
||
|
|
SUCCEEDED( hRes = pIisLb->GetPerfmonCounters( sizeof(abSerialized), abSerialized, &dwReq ) ) )
|
||
|
|
{
|
||
|
|
pbSerialized = abSerialized;
|
||
|
|
hRes = PerfCounters.Unserialize( &pbSerialized, &dwReq ) ? S_OK : RETURNCODETOHRESULT(ERROR_BAD_CONFIGURATION);
|
||
|
|
}
|
||
|
|
|
||
|
|
#if 0
|
||
|
|
if ( SUCCEEDED( hRes ) )
|
||
|
|
{
|
||
|
|
hRes = pIisLb->GetStickyDuration( &dwSticky );
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
if ( SUCCEEDED( hRes ) &&
|
||
|
|
SUCCEEDED( hRes = pIisLb->GetIpEndpointList( sizeof(abSerialized), abSerialized, &dwReq ) ) )
|
||
|
|
{
|
||
|
|
pbSerialized = abSerialized;
|
||
|
|
hRes = IpEndpoints.Unserialize( &pbSerialized, &dwReq ) ? S_OK : RETURNCODETOHRESULT(ERROR_BAD_CONFIGURATION);
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( FAILED(hRes) )
|
||
|
|
{
|
||
|
|
pIisLb->Release();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#if 0
|
||
|
|
pcsfFactory->Release();
|
||
|
|
#endif
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( SUCCEEDED( hRes ) )
|
||
|
|
{
|
||
|
|
switch ( iCmd )
|
||
|
|
{
|
||
|
|
case CMD_START_DRIVER:
|
||
|
|
hRes = pIisLb->SetDriverState( SERVICE_RUNNING );
|
||
|
|
break;
|
||
|
|
|
||
|
|
case CMD_STOP_DRIVER:
|
||
|
|
hRes = pIisLb->SetDriverState( SERVICE_STOPPED );
|
||
|
|
break;
|
||
|
|
|
||
|
|
case CMD_ADD_SERVER:
|
||
|
|
if ( cParam == 1 &&
|
||
|
|
IpMap.AddComputer( apszParam[0] ) )
|
||
|
|
{
|
||
|
|
fUpdateIpMap = TRUE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hRes = RETURNCODETOHRESULT(GetLastError());
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case CMD_ADD_IP:
|
||
|
|
if ( cParam == 3 &&
|
||
|
|
IpMap.AddIpPublic( apszParam[0], apszParam[1], _wtoi(apszParam[2]), 0 ) )
|
||
|
|
{
|
||
|
|
fUpdateIpMap = TRUE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hRes = RETURNCODETOHRESULT(GetLastError());
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case CMD_ADD_PERFMON:
|
||
|
|
if ( cParam >= 2 &&
|
||
|
|
!wcscmp( apszParam[1], L"*" ) )
|
||
|
|
{
|
||
|
|
if ( ChooseCounter( achCounterBuffer ) )
|
||
|
|
{
|
||
|
|
apszParam[1] = achCounterBuffer;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
cParam = 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
// server name : * for all
|
||
|
|
if ( cParam >= 1 &&
|
||
|
|
!wcscmp( apszParam[0], L"*" ) )
|
||
|
|
{
|
||
|
|
apszParam[0] = NULL;
|
||
|
|
}
|
||
|
|
if ( cParam == 3 &&
|
||
|
|
PerfCounters.AddPerfCounter( apszParam[0], apszParam[1], _wtoi(apszParam[2]) ) )
|
||
|
|
{
|
||
|
|
fUpdatePerfCounters = TRUE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hRes = RETURNCODETOHRESULT(GetLastError());
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case CMD_DEL_SERVER:
|
||
|
|
if ( cParam == 1 &&
|
||
|
|
LocateServer( &IpMap, apszParam[0], &iServer ) )
|
||
|
|
{
|
||
|
|
if ( IpMap.DeleteComputer( iServer ) )
|
||
|
|
{
|
||
|
|
fUpdateIpMap = TRUE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hRes = RETURNCODETOHRESULT( GetLastError() );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hRes = RETURNCODETOHRESULT(ERROR_INVALID_PARAMETER);
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case CMD_DEL_IP:
|
||
|
|
if ( cParam == 1 &&
|
||
|
|
LocatePublicIp( &IpMap, apszParam[0], &iPublicIp ) )
|
||
|
|
{
|
||
|
|
if ( IpMap.DeleteIpPublic( iPublicIp ) )
|
||
|
|
{
|
||
|
|
fUpdateIpMap = TRUE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hRes = RETURNCODETOHRESULT( GetLastError() );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hRes = RETURNCODETOHRESULT(ERROR_INVALID_PARAMETER);
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case CMD_DEL_PERFMON:
|
||
|
|
if ( cParam == 2 &&
|
||
|
|
LocatePerfCounter( &PerfCounters, apszParam[0], apszParam[1], &iPerfCounter ) )
|
||
|
|
{
|
||
|
|
if ( PerfCounters.DeletePerfCounter( iPerfCounter ) )
|
||
|
|
{
|
||
|
|
fUpdatePerfCounters = TRUE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hRes = RETURNCODETOHRESULT( GetLastError() );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hRes = RETURNCODETOHRESULT(ERROR_INVALID_PARAMETER);
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
#if 0
|
||
|
|
case CMD_SET_STICKY:
|
||
|
|
if ( cParam == 1 )
|
||
|
|
{
|
||
|
|
dwSticky = _wtoi( apszParam[0] );
|
||
|
|
fUpdateSticky = TRUE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hRes = RETURNCODETOHRESULT(ERROR_INVALID_PARAMETER);
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
#endif
|
||
|
|
|
||
|
|
case CMD_SET_IP:
|
||
|
|
if ( cParam == 4 &&
|
||
|
|
LocateServer( &IpMap, apszParam[0], &iServer ) &&
|
||
|
|
LocatePublicIp( &IpMap, apszParam[1], &iPublicIp ) )
|
||
|
|
{
|
||
|
|
if ( IpMap.SetIpPrivate( iServer, iPublicIp, apszParam[2], apszParam[3] ) )
|
||
|
|
{
|
||
|
|
fUpdateIpMap = TRUE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hRes = RETURNCODETOHRESULT( GetLastError() );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
hRes = RETURNCODETOHRESULT(ERROR_INVALID_PARAMETER);
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case CMD_LIST_IP_ENDPOINTS:
|
||
|
|
RC_PRINTF( IDS_LBC_IPE, (RC_BUFF) );
|
||
|
|
for ( iPublicIp = 0 ;
|
||
|
|
IpEndpoints.EnumIpEndpoint( iPublicIp, &pszPublicIp ) ;
|
||
|
|
++iPublicIp )
|
||
|
|
{
|
||
|
|
wprintf( L"%-*.*s\t", LEN_IP, LEN_IP, pszPublicIp );
|
||
|
|
}
|
||
|
|
wprintf( L"\n");
|
||
|
|
break;
|
||
|
|
|
||
|
|
case CMD_LIST:
|
||
|
|
RC_PRINTF( IDS_LBC_IPM, (RC_BUFF, LEN_SERVER_NAME, LEN_SERVER_NAME, "") );
|
||
|
|
for ( iPublicIp = 0 ;
|
||
|
|
IpMap.EnumIpPublic( iPublicIp, &pszPublicIp, &pszName, &dwSticky, &dwAttr ) ;
|
||
|
|
++iPublicIp )
|
||
|
|
{
|
||
|
|
wprintf( L"%-*.*s\t", LEN_IP, LEN_IP, pszPublicIp );
|
||
|
|
}
|
||
|
|
wprintf( L"\n");
|
||
|
|
|
||
|
|
for ( iServer = 0 ;
|
||
|
|
IpMap.EnumComputer( iServer, &pszServer ) ;
|
||
|
|
++iServer )
|
||
|
|
{
|
||
|
|
wprintf( L"%-*.*s\t", LEN_SERVER_NAME, LEN_SERVER_NAME, pszServer );
|
||
|
|
for ( iPublicIp = 0 ;
|
||
|
|
IpMap.EnumIpPublic( iPublicIp, &pszPublicIp, &pszName, &dwSticky, &dwAttr ) ;
|
||
|
|
++iPublicIp )
|
||
|
|
{
|
||
|
|
IpMap.GetIpPrivate( iServer, iPublicIp, &pszPrivateIp, &pszName );
|
||
|
|
wprintf( L"%-*.*s\t", LEN_IP, LEN_IP, pszPrivateIp );
|
||
|
|
}
|
||
|
|
wprintf( L"\n");
|
||
|
|
}
|
||
|
|
wprintf( L"\n");
|
||
|
|
|
||
|
|
RC_PRINTF( IDS_LBC_PM, (RC_BUFF) );
|
||
|
|
for ( iPerfCounter = 0 ;
|
||
|
|
PerfCounters.EnumPerfCounter( iPerfCounter, &pszServer, &pszPerfCounter, &dwWeight ) ;
|
||
|
|
++iPerfCounter )
|
||
|
|
{
|
||
|
|
wprintf( L"%-*.*s%-*.*s\t%u\n",
|
||
|
|
LEN_SERVER_NAME, LEN_SERVER_NAME, pszServer ? pszServer : L"*",
|
||
|
|
LEN_PERF, LEN_PERF, pszPerfCounter,
|
||
|
|
dwWeight );
|
||
|
|
}
|
||
|
|
wprintf( L"\n");
|
||
|
|
#if 0
|
||
|
|
RC_PRINTF( IDS_LBC_ST, (RC_BUFF, dwSticky) );
|
||
|
|
#endif
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
// update configuration
|
||
|
|
|
||
|
|
if ( SUCCEEDED( hRes ) &&
|
||
|
|
fUpdateIpMap &&
|
||
|
|
(xbf.Reset, IpMap.Serialize( &xbf )) )
|
||
|
|
{
|
||
|
|
hRes = pIisLb->SetIpList( xbf.GetUsed(), xbf.GetBuff() );
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( SUCCEEDED( hRes ) &&
|
||
|
|
fUpdatePerfCounters &&
|
||
|
|
(xbf.Reset, PerfCounters.Serialize( &xbf )) )
|
||
|
|
{
|
||
|
|
hRes = pIisLb->SetPerfmonCounters( xbf.GetUsed(), xbf.GetBuff() );
|
||
|
|
}
|
||
|
|
|
||
|
|
#if 0
|
||
|
|
if ( SUCCEEDED( hRes ) &&
|
||
|
|
fUpdateSticky )
|
||
|
|
{
|
||
|
|
hRes = pIisLb->SetStickyDuration( dwSticky );
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
pIisLb->Release();
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( FAILED( hRes ) )
|
||
|
|
{
|
||
|
|
DisplayErrorMessage( HRESULTTOWIN32( hRes ) );
|
||
|
|
}
|
||
|
|
|
||
|
|
CoUninitialize();
|
||
|
|
|
||
|
|
return Status;
|
||
|
|
}
|