mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1277 lines
38 KiB
1277 lines
38 KiB
/*
|
|
IisRsta.cxx
|
|
|
|
Command line utility for access to IIisServiceControl
|
|
|
|
FILE HISTORY:
|
|
Phillich 06-Oct-1998 Created
|
|
|
|
*/
|
|
|
|
#define INITGUID
|
|
#include <windows.h>
|
|
#include <tchar.h>
|
|
#include <stdio.h>
|
|
#include <ole2.h>
|
|
#include <locale.h>
|
|
#include "iisrsta.h"
|
|
#include "iisrstam.h"
|
|
|
|
typedef BOOL (*PFNQUERYSERVICECONFIG2)(SC_HANDLE,DWORD,LPBYTE,DWORD,LPDWORD) ;
|
|
typedef BOOL (*PFNCHANGESERVICECONFIG2)(SC_HANDLE,DWORD,LPVOID);
|
|
|
|
//
|
|
// 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))
|
|
|
|
|
|
//
|
|
// RETURNCODETOHRESULT() maps a return code to an HRESULT. If the return
|
|
// code is a Win32 error (identified by a zero high word) then it is mapped
|
|
// using the standard HRESULT_FROM_WIN32() macro. Otherwise, the return
|
|
// code is assumed to already be an HRESULT and is returned unchanged.
|
|
//
|
|
|
|
#define RETURNCODETOHRESULT(rc) \
|
|
(((rc) < 0x10000) \
|
|
? HRESULT_FROM_WIN32(rc) \
|
|
: (rc))
|
|
|
|
//
|
|
// Helper functions
|
|
//
|
|
|
|
|
|
VOID
|
|
DisplayErrorMessage(
|
|
DWORD dwErr
|
|
)
|
|
/*++
|
|
|
|
DisplayErrorMessage
|
|
|
|
Display error message associated with error code
|
|
|
|
Arguments:
|
|
|
|
dwErr - Win32 error code
|
|
|
|
Returns:
|
|
|
|
Nothing
|
|
|
|
--*/
|
|
{
|
|
LPTSTR pErr;
|
|
|
|
if ( FormatMessage( FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
|
|
NULL,
|
|
dwErr,
|
|
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
|
|
(LPTSTR)&pErr,
|
|
0,
|
|
NULL ) )
|
|
{
|
|
LPTSTR p;
|
|
|
|
if ( p = _tcschr( pErr, TEXT('\r') ) )
|
|
{
|
|
*p = TEXT('\0');
|
|
}
|
|
_fputts( pErr, stdout );
|
|
|
|
LocalFree( pErr );
|
|
}
|
|
}
|
|
|
|
|
|
VOID
|
|
PrintMessage(
|
|
DWORD dwId,
|
|
DWORD dwParams,
|
|
LPVOID* pParams
|
|
)
|
|
/*++
|
|
|
|
PrintMessage
|
|
|
|
Print message ( from message file ) with optional parameters
|
|
|
|
Arguments:
|
|
|
|
dwId - message ID
|
|
dwParams - # of params in pParams
|
|
pParams - ptr to array of parameters
|
|
|
|
Returns:
|
|
|
|
Nothing
|
|
|
|
--*/
|
|
{
|
|
LPTSTR pBuf;
|
|
|
|
if ( FormatMessage( FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_HMODULE|FORMAT_MESSAGE_ARGUMENT_ARRAY,
|
|
(LPCVOID)NULL, //GetModuleHandle(NULL),
|
|
dwId,
|
|
0,
|
|
(LPTSTR)&pBuf,
|
|
dwParams,
|
|
(va_list *)pParams ) )
|
|
{
|
|
_fputts( pBuf, stdout );
|
|
|
|
LocalFree( pBuf );
|
|
}
|
|
}
|
|
|
|
|
|
VOID
|
|
CmdError(
|
|
DWORD dwId,
|
|
HRESULT hRes
|
|
)
|
|
/*++
|
|
|
|
CmdError
|
|
|
|
Display message followed by error description ( error message + numerical code )
|
|
|
|
Arguments:
|
|
|
|
dwId - message ID
|
|
hRes - error code
|
|
|
|
Returns:
|
|
|
|
Nothing
|
|
|
|
--*/
|
|
{
|
|
TCHAR achBuf[128];
|
|
|
|
PrintMessage( dwId, 0, NULL );
|
|
|
|
if ( hRes == RETURNCODETOHRESULT( ERROR_RESOURCE_DISABLED ) )
|
|
{
|
|
PrintMessage( IRSTASTR_REMOTE_DISABLED, 0, NULL );
|
|
}
|
|
else if ( hRes == RETURNCODETOHRESULT( ERROR_ACCESS_DENIED ) )
|
|
{
|
|
PrintMessage( IRSTASTR_ACCESS_DENIED, 0, NULL );
|
|
}
|
|
else if ( hRes == RETURNCODETOHRESULT( ERROR_SERVICE_NOT_ACTIVE ) )
|
|
{
|
|
PrintMessage( IRSTASTR_IISADMIN_DISABLED, 0, NULL );
|
|
}
|
|
else
|
|
{
|
|
DisplayErrorMessage( HRESULTTOWIN32( hRes ) );
|
|
wsprintf( achBuf, _T(" (%u, %x)\n"), (DWORD)hRes, (DWORD)hRes );
|
|
_fputts( achBuf, stdout );
|
|
}
|
|
}
|
|
|
|
|
|
LPTSTR
|
|
GetString(
|
|
DWORD dwId
|
|
)
|
|
/*++
|
|
|
|
GetString
|
|
|
|
Retrieve message content
|
|
|
|
Arguments:
|
|
|
|
dwId - message ID
|
|
|
|
Returns:
|
|
|
|
Ptr to message. Must be freed using LocalFree()
|
|
|
|
--*/
|
|
{
|
|
#if 0
|
|
TCHAR aBuf[32];
|
|
|
|
aBuf[0] = TEXT('#');
|
|
_ultot( dwId, aBuf + 1, 10 );
|
|
|
|
return (LPTSTR)LoadResource( NULL, FindResourceEx( NULL, RT_MESSAGETABLE, aBuf, MAKELANGID(LANG_NEUTRAL,SUBLANG_NEUTRAL) ) );
|
|
#else
|
|
LPTSTR pBuf;
|
|
|
|
if ( FormatMessage( FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_FROM_HMODULE,
|
|
(LPCVOID)NULL,
|
|
dwId,
|
|
0,
|
|
(LPTSTR)&pBuf,
|
|
0,
|
|
NULL ) )
|
|
{
|
|
LPTSTR p;
|
|
|
|
if ( p = _tcschr( pBuf, TEXT('\r') ) )
|
|
{
|
|
*p = TEXT('\0');
|
|
}
|
|
return pBuf;
|
|
}
|
|
|
|
return NULL;
|
|
#endif
|
|
}
|
|
|
|
|
|
|
|
HRESULT
|
|
DeserializeEnumServiceBuffer(
|
|
LPBYTE pbInBuffer,
|
|
DWORD dwNumServices,
|
|
LPBYTE pbBuffer,
|
|
DWORD dwBufferSize,
|
|
LPDWORD pdwMDRequiredBufferSize
|
|
)
|
|
/*++
|
|
|
|
DeserializeEnumServiceBuffer
|
|
|
|
Deserialize array of SERIALIZED_ENUM_SERVICE_STATUS to buffer,
|
|
replacing offset in buffer by ptr
|
|
|
|
Arguments:
|
|
|
|
pbInBuffer - buffer containing serialized status as array of SERIALIZED_ENUM_SERVICE_STATUS
|
|
dwNumServices - # of entries in pbInBuffer
|
|
pbBuffer - buffer filled with deserialized status as array of ENUM_SERVICE_STATUS
|
|
dwBufferSize - maximum size of pbBuffer
|
|
pdwMDRequiredBufferSize - updated with required size if dwBufferSize too small
|
|
|
|
Returns:
|
|
ERROR_INSUFFICIENT_BUFFER if dwBufferSize too small
|
|
otherwise COM status
|
|
|
|
--*/
|
|
{
|
|
HRESULT hresReturn = S_OK;
|
|
DWORD dwMinSize = 0;
|
|
UINT i;
|
|
SERIALIZED_ENUM_SERVICE_STATUS* pessDependentServices = (SERIALIZED_ENUM_SERVICE_STATUS*)pbInBuffer;
|
|
|
|
if ( !pbBuffer )
|
|
{
|
|
dwBufferSize = 0;
|
|
}
|
|
|
|
dwMinSize = sizeof(ENUM_SERVICE_STATUS) * dwNumServices;
|
|
|
|
for ( i = 0 ;
|
|
i < dwNumServices ;
|
|
++i )
|
|
{
|
|
UINT cServiceName = _tcslen( (TCHAR*)(pbInBuffer + pessDependentServices[i].iServiceName) ) + 1;
|
|
UINT cDisplayName = _tcslen( (TCHAR*)(pbInBuffer + pessDependentServices[i].iDisplayName) ) + 1;
|
|
|
|
if ( dwBufferSize >= dwMinSize + ( cServiceName + cDisplayName ) * sizeof(TCHAR) )
|
|
{
|
|
((LPENUM_SERVICE_STATUS)pbBuffer)[i].ServiceStatus =
|
|
pessDependentServices[i].ServiceStatus;
|
|
|
|
memcpy( pbBuffer + dwMinSize, pbInBuffer + pessDependentServices[i].iServiceName, cServiceName * sizeof(TCHAR) );
|
|
((LPENUM_SERVICE_STATUS)pbBuffer)[i].lpServiceName = (TCHAR*)(pbBuffer + dwMinSize);
|
|
|
|
memcpy( pbBuffer + dwMinSize + cServiceName * sizeof(TCHAR), pbInBuffer + pessDependentServices[i].iDisplayName, cDisplayName * sizeof(TCHAR) );
|
|
((LPENUM_SERVICE_STATUS)pbBuffer)[i].lpDisplayName = (TCHAR*)(pbBuffer + dwMinSize) + cServiceName;
|
|
}
|
|
|
|
dwMinSize += ( cServiceName + cDisplayName ) * sizeof(TCHAR);
|
|
}
|
|
|
|
if ( dwBufferSize < dwMinSize )
|
|
{
|
|
*pdwMDRequiredBufferSize = dwMinSize;
|
|
|
|
hresReturn = RETURNCODETOHRESULT( ERROR_INSUFFICIENT_BUFFER );
|
|
}
|
|
|
|
return hresReturn;
|
|
}
|
|
|
|
|
|
HRESULT
|
|
SetEnableRemote(
|
|
DWORD dwValue
|
|
)
|
|
/*++
|
|
|
|
SetEnableRemote
|
|
|
|
set restart I/F enabled flag in registry
|
|
( HKLM\SOFTWARE\Microsoft\INetStp::EnableRestart::REG_DWORD )
|
|
|
|
Arguments:
|
|
|
|
dwValue - 0 to disable I/F, !0 to enable
|
|
|
|
Returns:
|
|
status
|
|
|
|
--*/
|
|
{
|
|
DWORD dwSt = NO_ERROR;
|
|
HKEY hKey;
|
|
|
|
|
|
//
|
|
// Check admin privilege by accessing IISADMIN key for write
|
|
//
|
|
|
|
if ( ( dwSt = RegOpenKeyEx( HKEY_LOCAL_MACHINE,
|
|
TEXT("SYSTEM\\CurrentControlSet\\Services\\IISADMIN"),
|
|
0,
|
|
KEY_WRITE,
|
|
&hKey ) ) == ERROR_SUCCESS )
|
|
{
|
|
RegCloseKey( hKey );
|
|
|
|
//
|
|
// Set IISCTL interface access flag in registry
|
|
//
|
|
|
|
if ( ( dwSt = RegOpenKeyEx( HKEY_LOCAL_MACHINE,
|
|
TEXT("SOFTWARE\\Microsoft\\INetStp"),
|
|
0,
|
|
KEY_WRITE,
|
|
&hKey ) ) == ERROR_SUCCESS )
|
|
{
|
|
if ( ( dwSt = RegSetValueEx( hKey,
|
|
TEXT("EnableRestart"),
|
|
0,
|
|
REG_DWORD,
|
|
(const BYTE*)&dwValue,
|
|
sizeof(DWORD) ) ) == ERROR_SUCCESS )
|
|
{
|
|
}
|
|
|
|
RegCloseKey( hKey );
|
|
}
|
|
}
|
|
|
|
return RETURNCODETOHRESULT( dwSt );
|
|
}
|
|
|
|
|
|
BOOL
|
|
GetNumeric(
|
|
LPSTR pszNumArg,
|
|
LPDWORD pdwVal
|
|
)
|
|
{
|
|
if ( !isdigit( (UCHAR)(*pszNumArg) ) )
|
|
{
|
|
return FALSE;
|
|
}
|
|
|
|
*pdwVal = atoi( pszNumArg );
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
BOOL
|
|
IsIIS6orGreater(
|
|
)
|
|
/*++
|
|
|
|
IsIIS6orGreater
|
|
|
|
According to Aaron we are guaranteed to have
|
|
the appropriate keys in the registry by the
|
|
time iisreset /scm is called.
|
|
|
|
Arguments:
|
|
|
|
None
|
|
|
|
Returns:
|
|
TRUE = IIS 6 or greater;
|
|
FALSE = anything below IIS 6;
|
|
|
|
--*/
|
|
{
|
|
// Assume it is not IIS 6.
|
|
BOOL IsIIS6 = FALSE;
|
|
|
|
HKEY hKey;
|
|
DWORD dwValue;
|
|
DWORD dwType;
|
|
DWORD dwSize;
|
|
|
|
// Check the registry of the major version setup number.
|
|
// If it is not there we assume it is not a 6.0 machine,
|
|
// since it's better to setup a 6.0 machine in 5.0 state
|
|
// instead of a 5.0 machine in 6.0 state.
|
|
if ( RegOpenKeyEx( HKEY_LOCAL_MACHINE,
|
|
TEXT("System\\CurrentControlSet\\Services\\W3SVC\\Parameters"),
|
|
0,
|
|
KEY_READ,
|
|
&hKey ) == ERROR_SUCCESS )
|
|
{
|
|
if ( RegQueryValueEx( hKey,
|
|
TEXT("MajorVersion"),
|
|
0,
|
|
&dwType,
|
|
(LPBYTE)&dwValue,
|
|
&dwSize ) == ERROR_SUCCESS )
|
|
{
|
|
|
|
if ( dwType == REG_DWORD )
|
|
{
|
|
if ( dwValue >= 6 )
|
|
{
|
|
// were are running on a system that does not
|
|
// have atleast IIS 6 on it
|
|
|
|
IsIIS6 = TRUE;
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
RegCloseKey( hKey );
|
|
|
|
}
|
|
|
|
|
|
return IsIIS6;
|
|
|
|
}
|
|
|
|
HRESULT
|
|
SetSCM(
|
|
LPCTSTR pszCmdline,
|
|
BOOL fEnable
|
|
)
|
|
/*++
|
|
|
|
SetSCM
|
|
|
|
set SCM Recovery configuration for IISADMIN service.
|
|
Does nothing on pre Windows 2000 systems.
|
|
|
|
Arguments:
|
|
|
|
pszCmdline - command line to invoke on service failure ( for IISAdmin ), can be NULL to invoke current EXE ( i.e. IISRESET )
|
|
fEnable - TRUE to enable IISRESET invocation on service failure, FALSE to disable
|
|
|
|
Returns:
|
|
status
|
|
|
|
--*/
|
|
{
|
|
PFNQUERYSERVICECONFIG2 pfnQueryServiceConfig2 = NULL;
|
|
PFNCHANGESERVICECONFIG2 pfnChangeServiceConfig2 = NULL;
|
|
SERVICE_FAILURE_ACTIONS sfaAction;
|
|
SC_ACTION saCmdline[3];
|
|
SC_HANDLE schSCM;
|
|
SC_HANDLE schSrv;
|
|
HRESULT hres = S_OK;
|
|
TCHAR achModuleName[MAX_PATH];
|
|
TCHAR achFailureCommand[MAX_PATH+32];
|
|
HINSTANCE hAdvapi;
|
|
|
|
|
|
if ( hAdvapi = LoadLibrary(_T("ADVAPI32.DLL")) )
|
|
{
|
|
pfnQueryServiceConfig2 = (PFNQUERYSERVICECONFIG2)GetProcAddress( hAdvapi, "QueryServiceConfig2W" );
|
|
pfnChangeServiceConfig2 = (PFNCHANGESERVICECONFIG2)GetProcAddress( hAdvapi, "ChangeServiceConfig2W" );
|
|
}
|
|
|
|
if ( pfnQueryServiceConfig2
|
|
&& pfnChangeServiceConfig2 )
|
|
{
|
|
if ( pszCmdline == NULL )
|
|
{
|
|
if ( !GetModuleFileName( NULL,
|
|
achModuleName,
|
|
sizeof(achModuleName)/sizeof(TCHAR) ) )
|
|
{
|
|
hres = RETURNCODETOHRESULT( GetLastError() );
|
|
}
|
|
else
|
|
{
|
|
pszCmdline = achModuleName;
|
|
}
|
|
}
|
|
|
|
if ( SUCCEEDED( hres ) )
|
|
{
|
|
schSCM = OpenSCManager(NULL,
|
|
NULL,
|
|
SC_MANAGER_ALL_ACCESS);
|
|
if ( schSCM == NULL )
|
|
{
|
|
hres = RETURNCODETOHRESULT( GetLastError() );
|
|
}
|
|
else
|
|
{
|
|
|
|
// We need to determine what IIS version we are running on
|
|
// in order to know the correct way to setup the server.
|
|
|
|
BOOL fIIS6orGreater = IsIIS6orGreater();
|
|
|
|
//
|
|
// In IIS 5.1 we want to use the default /restart
|
|
// setting of iisreset. In IIS 6 or greater we
|
|
// want to use the /start option. A customer
|
|
// can configure it to use the /restart option, but
|
|
// that is not the default for the scm.
|
|
//
|
|
if ( fIIS6orGreater )
|
|
{
|
|
wsprintf( achFailureCommand, _T("\"%s\" /start /fail=%%1%%"), pszCmdline );
|
|
}
|
|
else
|
|
{
|
|
wsprintf( achFailureCommand, _T("\"%s\" /fail=%%1%%"), pszCmdline );
|
|
}
|
|
|
|
sfaAction.lpCommand = achFailureCommand;
|
|
sfaAction.lpRebootMsg = _T("");
|
|
sfaAction.dwResetPeriod = 24 * 60 * 60;
|
|
|
|
if ( fEnable )
|
|
{
|
|
sfaAction.cActions = 3;
|
|
sfaAction.lpsaActions = saCmdline;
|
|
saCmdline[0].Type = SC_ACTION_RUN_COMMAND;
|
|
saCmdline[0].Delay = 1;
|
|
saCmdline[1].Type = SC_ACTION_RUN_COMMAND;
|
|
saCmdline[1].Delay = 1;
|
|
saCmdline[2].Type = SC_ACTION_RUN_COMMAND;
|
|
saCmdline[2].Delay = 1;
|
|
}
|
|
else
|
|
{
|
|
sfaAction.cActions = 3;
|
|
sfaAction.lpsaActions = saCmdline;
|
|
saCmdline[0].Type = SC_ACTION_NONE;
|
|
saCmdline[0].Delay = 0;
|
|
saCmdline[1].Type = SC_ACTION_NONE;
|
|
saCmdline[1].Delay = 0;
|
|
saCmdline[2].Type = SC_ACTION_NONE;
|
|
saCmdline[2].Delay = 0;
|
|
}
|
|
|
|
schSrv = OpenService( schSCM,
|
|
_T("IISADMIN"),
|
|
SERVICE_ALL_ACCESS);
|
|
|
|
if ( schSrv )
|
|
{
|
|
if ( !pfnChangeServiceConfig2( schSrv,
|
|
SERVICE_CONFIG_FAILURE_ACTIONS,
|
|
&sfaAction ) )
|
|
{
|
|
hres = RETURNCODETOHRESULT( GetLastError() );
|
|
}
|
|
|
|
CloseServiceHandle( schSrv );
|
|
}
|
|
else
|
|
{
|
|
hres = RETURNCODETOHRESULT( GetLastError() );
|
|
}
|
|
|
|
//
|
|
// Now that we have configured IISAdmin correctly, we also need to
|
|
// configure W3SVC.
|
|
//
|
|
|
|
if ( fIIS6orGreater )
|
|
{
|
|
|
|
sfaAction.lpCommand = NULL;
|
|
sfaAction.lpRebootMsg = _T("");
|
|
sfaAction.dwResetPeriod = 24 * 60 * 60;
|
|
|
|
if ( fEnable )
|
|
{
|
|
sfaAction.cActions = 3;
|
|
sfaAction.lpsaActions = saCmdline;
|
|
saCmdline[0].Type = SC_ACTION_RESTART;
|
|
saCmdline[0].Delay = 1;
|
|
saCmdline[1].Type = SC_ACTION_RESTART;
|
|
saCmdline[1].Delay = 1;
|
|
saCmdline[2].Type = SC_ACTION_RESTART;
|
|
saCmdline[2].Delay = 1;
|
|
}
|
|
else
|
|
{
|
|
sfaAction.cActions = 3;
|
|
sfaAction.lpsaActions = saCmdline;
|
|
saCmdline[0].Type = SC_ACTION_NONE;
|
|
saCmdline[0].Delay = 0;
|
|
saCmdline[1].Type = SC_ACTION_NONE;
|
|
saCmdline[1].Delay = 0;
|
|
saCmdline[2].Type = SC_ACTION_NONE;
|
|
saCmdline[2].Delay = 0;
|
|
}
|
|
|
|
schSrv = OpenService( schSCM,
|
|
_T("W3SVC"),
|
|
SERVICE_ALL_ACCESS);
|
|
|
|
if ( schSrv )
|
|
{
|
|
if ( !pfnChangeServiceConfig2( schSrv,
|
|
SERVICE_CONFIG_FAILURE_ACTIONS,
|
|
&sfaAction ) )
|
|
{
|
|
hres = RETURNCODETOHRESULT( GetLastError() );
|
|
}
|
|
|
|
CloseServiceHandle( schSrv );
|
|
}
|
|
else
|
|
{
|
|
hres = RETURNCODETOHRESULT( GetLastError() );
|
|
}
|
|
} // end of setting up restart on w3svc.
|
|
|
|
|
|
CloseServiceHandle( schSCM );
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
if ( hAdvapi )
|
|
{
|
|
FreeLibrary( hAdvapi );
|
|
}
|
|
|
|
return hres;
|
|
}
|
|
|
|
|
|
enum CMDS { CMD_NONE, CMD_START, CMD_STOP, CMD_REBOOT, CMD_RESTART, CMD_KILL };
|
|
|
|
|
|
int __cdecl
|
|
main(
|
|
int argc,
|
|
char*argv[]
|
|
)
|
|
/*++
|
|
|
|
main
|
|
|
|
main function
|
|
|
|
Arguments:
|
|
|
|
argc
|
|
argv
|
|
|
|
Returns:
|
|
|
|
0 if no error, otherwise error code
|
|
|
|
--*/
|
|
{
|
|
IIisServiceControl* pIf;
|
|
int iA;
|
|
int Status = 0;
|
|
HRESULT hRes;
|
|
CMDS iCmd = CMD_NONE;
|
|
DWORD dwStopTimeout = 60 * 1000;
|
|
DWORD dwRestartTimeout = 20 * 1000;
|
|
DWORD dwStartTimeout = 60 * 1000;
|
|
LPBYTE pbBuffer;
|
|
BYTE abBuffer[4096];
|
|
LPBYTE pbOutBuffer;
|
|
BYTE abOutBuffer[4096];
|
|
DWORD dwRequired;
|
|
DWORD dwNumServices;
|
|
LPVOID apvParams[8];
|
|
UINT i;
|
|
BOOL fNoCmd = FALSE;
|
|
BOOL fRebootOnError = FALSE;
|
|
BOOL fKillOnError = TRUE;
|
|
BOOL fForce = TRUE;
|
|
BOOL fRebootRestart = FALSE;
|
|
BOOL fStatus = FALSE;
|
|
COSERVERINFO csiMachineName;
|
|
MULTI_QI rgmq;
|
|
WCHAR awchComputer[64];
|
|
LPSTR pszMachineName = NULL;
|
|
BOOL fErrDisplayed = FALSE;
|
|
DWORD dwFailCount;
|
|
|
|
//
|
|
// Make sure international versions display text ok - RonaldM
|
|
//
|
|
_tsetlocale( LC_ALL, _T(".OCP") );
|
|
|
|
_fputts( _T("\n"), stdout );
|
|
|
|
//
|
|
// scan command line
|
|
//
|
|
|
|
for ( iA = 1 ;
|
|
iA < argc ;
|
|
++iA )
|
|
{
|
|
if ( argv[iA][0] == '-' || argv[iA][0] == '/' )
|
|
{
|
|
if ( !lstrcmpiA( argv[iA]+1, "ENABLE" ) )
|
|
{
|
|
hRes = SetEnableRemote( 1 );
|
|
|
|
if ( FAILED( hRes ) )
|
|
{
|
|
if ( hRes == RETURNCODETOHRESULT( ERROR_ACCESS_DENIED ) )
|
|
{
|
|
PrintMessage( IRSTASTR_ACCESS_DENIED, 0, NULL );
|
|
}
|
|
else
|
|
{
|
|
DisplayErrorMessage( HRESULTTOWIN32( hRes ) );
|
|
}
|
|
|
|
Status = HRESULTTOWIN32( hRes );
|
|
}
|
|
else
|
|
{
|
|
PrintMessage( IRSTASTR_ENABLED, 0, NULL );
|
|
}
|
|
|
|
fNoCmd = TRUE;
|
|
}
|
|
else if ( !lstrcmpiA( argv[iA]+1, "DISABLE" ) )
|
|
{
|
|
hRes = SetEnableRemote( 0 );
|
|
|
|
if ( FAILED( hRes ) )
|
|
{
|
|
if ( hRes == RETURNCODETOHRESULT( ERROR_ACCESS_DENIED ) )
|
|
{
|
|
PrintMessage( IRSTASTR_ACCESS_DENIED, 0, NULL );
|
|
}
|
|
else
|
|
{
|
|
DisplayErrorMessage( HRESULTTOWIN32( hRes ) );
|
|
}
|
|
|
|
Status = HRESULTTOWIN32( hRes );
|
|
}
|
|
else
|
|
{
|
|
PrintMessage( IRSTASTR_DISABLED, 0, NULL );
|
|
}
|
|
|
|
fNoCmd = TRUE;
|
|
}
|
|
else if ( !_strnicmp( argv[iA]+1, "SCM", sizeof("SCM")-1 ) )
|
|
{
|
|
if ( FAILED( hRes = SetSCM( NULL, TRUE ) ) )
|
|
{
|
|
DisplayErrorMessage( HRESULTTOWIN32( hRes ) );
|
|
|
|
Status = HRESULTTOWIN32( hRes );
|
|
}
|
|
goto Exit;
|
|
}
|
|
else if ( !_strnicmp( argv[iA]+1, "NOSCM", sizeof("NOSCM")-1 ) )
|
|
{
|
|
if ( FAILED( hRes = SetSCM( NULL, FALSE ) ) )
|
|
{
|
|
DisplayErrorMessage( HRESULTTOWIN32( hRes ) );
|
|
|
|
Status = HRESULTTOWIN32( hRes );
|
|
}
|
|
goto Exit;
|
|
}
|
|
else if ( !_strnicmp( argv[iA]+1, "STOPTIMEOUT:", sizeof("STOPTIMEOUT:")-1 ) )
|
|
{
|
|
if ( !GetNumeric( argv[iA]+ 1 + sizeof("STOPTIMEOUT:") - 1, &dwStopTimeout ) )
|
|
{
|
|
goto invalid_param;
|
|
}
|
|
dwStopTimeout *= 1000;
|
|
}
|
|
else if ( !_strnicmp( argv[iA]+1, "TIMEOUT:", sizeof("TIMEOUT:")-1 ) )
|
|
{
|
|
if ( !GetNumeric( argv[iA]+ 1 + sizeof("TIMEOUT:") - 1, &dwStopTimeout ) )
|
|
{
|
|
goto invalid_param;
|
|
}
|
|
dwStopTimeout *= 1000;
|
|
dwRestartTimeout = dwStopTimeout;
|
|
}
|
|
else if ( !_strnicmp( argv[iA]+1, "STARTTIMEOUT:", sizeof("STARTTIMEOUT:")-1 ) )
|
|
{
|
|
if ( !GetNumeric( argv[iA]+ 1 + sizeof("STARTTIMEOUT:") - 1, &dwStartTimeout ) )
|
|
{
|
|
goto invalid_param;
|
|
}
|
|
dwStartTimeout *= 1000;
|
|
}
|
|
else if ( !_strnicmp( argv[iA]+1, "RESTARTTIMEOUT:", sizeof("RESTARTTIMEOUT:")-1 ) )
|
|
{
|
|
if ( !GetNumeric( argv[iA]+ 1 + sizeof("RESTARTTIMEOUT:") - 1, &dwRestartTimeout ) )
|
|
{
|
|
goto invalid_param;
|
|
}
|
|
dwRestartTimeout *= 1000;
|
|
}
|
|
else if ( !_strnicmp( argv[iA]+1, "fail=", sizeof("fail=")-1 ) )
|
|
{
|
|
//
|
|
// SCM flag to control restart threshold. We restart only
|
|
// 50 times per SCM restart period.
|
|
//
|
|
// ... The SCM UI already has a way to address this: they add
|
|
// a "/fail=N" parameter to the command line
|
|
// of the restarter app for each time they fail within the time
|
|
// period and leave it up to the restarter app to interpret
|
|
// this parameter and throttle restarts approprately.
|
|
// So...(here's the change request) we need to capture this
|
|
// value, and if "N" is greater than our limit for 1 day of
|
|
// restarting (hardcoded to 50), we should just exit the
|
|
// command line app with a success code without doing anything,
|
|
// i.e. "IISRESET.EXE /fail=50" should restart IIS,
|
|
// but "IISRESET.EXE /fail=51" should be a no-op.
|
|
// For simplicity, we should hardcode this value of 50 in
|
|
// our app-- if a user wants a different value, he can write
|
|
// a batch file wrapper to do it! You can see
|
|
// the SCM recovery tab for more info.
|
|
//
|
|
|
|
dwFailCount = atoi( argv[iA]+ 1 + sizeof("fail=") - 1 );
|
|
|
|
if ( dwFailCount > 50 )
|
|
{
|
|
return 0;
|
|
}
|
|
}
|
|
else if ( !lstrcmpiA( argv[iA]+1, "START" ) )
|
|
{
|
|
if ( fRebootRestart )
|
|
{
|
|
}
|
|
else if ( iCmd == CMD_STOP )
|
|
{
|
|
iCmd = CMD_RESTART;
|
|
}
|
|
else
|
|
{
|
|
iCmd = CMD_START;
|
|
}
|
|
}
|
|
else if ( !lstrcmpiA( argv[iA]+1, "STOP" ) )
|
|
{
|
|
if ( fRebootRestart )
|
|
{
|
|
}
|
|
if ( iCmd == CMD_START )
|
|
{
|
|
iCmd = CMD_RESTART;
|
|
}
|
|
else
|
|
{
|
|
iCmd = CMD_STOP;
|
|
}
|
|
}
|
|
else if ( !lstrcmpiA( argv[iA]+1, "REBOOT" ) )
|
|
{
|
|
iCmd = CMD_REBOOT;
|
|
fRebootRestart = TRUE;
|
|
}
|
|
else if ( !lstrcmpiA( argv[iA]+1, "KILL" ) )
|
|
{
|
|
iCmd = CMD_KILL;
|
|
fRebootRestart = TRUE;
|
|
}
|
|
else if ( !lstrcmpiA( argv[iA]+1, "RESTART" ) )
|
|
{
|
|
iCmd = CMD_RESTART;
|
|
fRebootRestart = TRUE;
|
|
}
|
|
else if ( !lstrcmpiA( argv[iA]+1, "STATUS" ) )
|
|
{
|
|
fStatus = TRUE;
|
|
}
|
|
else if ( !lstrcmpiA( argv[iA]+1, "REBOOTONERROR" ) )
|
|
{
|
|
fRebootOnError = TRUE;
|
|
fKillOnError = FALSE;
|
|
}
|
|
else if ( !lstrcmpiA( argv[iA]+1, "NOFORCE" ) )
|
|
{
|
|
fKillOnError = FALSE;
|
|
fForce = FALSE;
|
|
}
|
|
else if ( !lstrcmpiA( argv[iA]+1, "HELP" ) )
|
|
{
|
|
PrintMessage( IRSTASTR_USAGE, 0, NULL );
|
|
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
invalid_param:
|
|
PrintMessage( IRSTASTR_USAGE, 0, NULL );
|
|
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
pszMachineName = argv[iA];
|
|
}
|
|
}
|
|
|
|
if ( iCmd == CMD_NONE && !fNoCmd && !fStatus )
|
|
{
|
|
iCmd = CMD_RESTART;
|
|
}
|
|
|
|
//
|
|
//fill the structure for CoCreateInstanceEx
|
|
//
|
|
|
|
ZeroMemory( &csiMachineName, sizeof(csiMachineName) );
|
|
|
|
if ( pszMachineName )
|
|
{
|
|
if ( !MultiByteToWideChar( CP_ACP,
|
|
MB_PRECOMPOSED,
|
|
pszMachineName,
|
|
-1,
|
|
awchComputer,
|
|
sizeof(awchComputer) / sizeof(WCHAR)) )
|
|
{
|
|
return GetLastError();
|
|
}
|
|
|
|
csiMachineName.pwszName = awchComputer;
|
|
}
|
|
else
|
|
{
|
|
csiMachineName.pwszName = NULL;
|
|
}
|
|
|
|
if ( fNoCmd )
|
|
{
|
|
iCmd = CMD_NONE;
|
|
fStatus = FALSE;
|
|
}
|
|
|
|
if ( iCmd != CMD_NONE || fStatus )
|
|
{
|
|
|
|
BOOL fCoInitialized = true;
|
|
//
|
|
// call method
|
|
//
|
|
|
|
rgmq.pIID = &IID_IIisServiceControl;
|
|
rgmq.pItf = NULL;
|
|
rgmq.hr = 0;
|
|
|
|
if (FAILED(hRes = CoInitializeEx( NULL, COINIT_MULTITHREADED ))) {
|
|
|
|
fCoInitialized = false;
|
|
}
|
|
else if ( SUCCEEDED( hRes = CoCreateInstanceEx( CLSID_IisServiceControl,
|
|
NULL,
|
|
CLSCTX_SERVER,
|
|
&csiMachineName,
|
|
1,
|
|
&rgmq ) ) &&
|
|
SUCCEEDED( hRes = rgmq.hr ) )
|
|
{
|
|
pIf = (IIisServiceControl*)rgmq.pItf;
|
|
|
|
switch ( iCmd )
|
|
{
|
|
case CMD_START:
|
|
PrintMessage( IRSTASTR_START_ATTEMPT, 0, NULL );
|
|
hRes = pIf->Start( dwStartTimeout );
|
|
if ( SUCCEEDED( hRes ) )
|
|
{
|
|
PrintMessage( IRSTASTR_START_SUCCESS, 0, NULL );
|
|
}
|
|
else
|
|
{
|
|
CmdError( IRSTASTR_START_FAILED, hRes );
|
|
fErrDisplayed = TRUE;
|
|
}
|
|
break;
|
|
|
|
case CMD_STOP:
|
|
PrintMessage( IRSTASTR_STOP_ATTEMPT, 0, NULL );
|
|
hRes = pIf->Stop( dwStopTimeout, fKillOnError );
|
|
if ( SUCCEEDED( hRes ) )
|
|
{
|
|
PrintMessage( IRSTASTR_STOP_SUCCESS, 0, NULL );
|
|
}
|
|
else
|
|
{
|
|
CmdError( IRSTASTR_STOP_FAILED, hRes );
|
|
fErrDisplayed = TRUE;
|
|
}
|
|
break;
|
|
|
|
case CMD_REBOOT:
|
|
PrintMessage( IRSTASTR_REBOOT_ATTEMPT, 0, NULL );
|
|
hRes = pIf->Reboot( dwRestartTimeout, fForce );
|
|
if ( SUCCEEDED( hRes ) )
|
|
{
|
|
PrintMessage( IRSTASTR_REBOOT_SUCCESS, 0, NULL );
|
|
}
|
|
else
|
|
{
|
|
CmdError( IRSTASTR_REBOOT_FAILED, hRes );
|
|
fErrDisplayed = TRUE;
|
|
}
|
|
break;
|
|
|
|
case CMD_KILL:
|
|
PrintMessage( IRSTASTR_KILL_ON_ERROR, 0, NULL );
|
|
hRes = pIf->Kill();
|
|
if ( SUCCEEDED( hRes ) )
|
|
{
|
|
PrintMessage( IRSTASTR_KILL_SUCCESS, 0, NULL );
|
|
}
|
|
else
|
|
{
|
|
CmdError( IRSTASTR_KILL_FAILED, hRes );
|
|
fErrDisplayed = TRUE;
|
|
}
|
|
break;
|
|
|
|
case CMD_RESTART:
|
|
|
|
PrintMessage( IRSTASTR_STOP_ATTEMPT, 0, NULL );
|
|
hRes = pIf->Stop( dwRestartTimeout, fKillOnError );
|
|
|
|
if ( SUCCEEDED( hRes ) )
|
|
{
|
|
PrintMessage( IRSTASTR_STOP_SUCCESS, 0, NULL );
|
|
|
|
PrintMessage( IRSTASTR_START_ATTEMPT, 0, NULL );
|
|
hRes = pIf->Start( dwStartTimeout );
|
|
|
|
if ( SUCCEEDED( hRes ) )
|
|
{
|
|
PrintMessage( IRSTASTR_RESTART_SUCCESS, 0, NULL );
|
|
}
|
|
}
|
|
if ( FAILED( hRes ) )
|
|
{
|
|
CmdError( IRSTASTR_RESTART_FAILED, hRes );
|
|
fErrDisplayed = TRUE;
|
|
}
|
|
break;
|
|
}
|
|
|
|
if ( fStatus )
|
|
{
|
|
pbBuffer = NULL;
|
|
fErrDisplayed = FALSE;
|
|
|
|
if ( FAILED( hRes = pIf->Status( sizeof(abBuffer),
|
|
abBuffer,
|
|
&dwRequired,
|
|
&dwNumServices ) ) )
|
|
{
|
|
if ( RETURNCODETOHRESULT( ERROR_INSUFFICIENT_BUFFER ) == hRes )
|
|
{
|
|
if ( (pbBuffer = (LPBYTE)LocalAlloc( LMEM_FIXED, dwRequired )) == NULL )
|
|
{
|
|
hRes = RETURNCODETOHRESULT( ERROR_NOT_ENOUGH_MEMORY );
|
|
}
|
|
else
|
|
{
|
|
hRes = pIf->Status( dwRequired,
|
|
pbBuffer,
|
|
&dwRequired,
|
|
&dwNumServices );
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
pbBuffer = abBuffer;
|
|
}
|
|
|
|
if ( SUCCEEDED( hRes ) )
|
|
{
|
|
pbOutBuffer = NULL;
|
|
|
|
if ( FAILED( hRes = DeserializeEnumServiceBuffer( pbBuffer,
|
|
dwNumServices,
|
|
abOutBuffer,
|
|
sizeof(abOutBuffer),
|
|
&dwRequired ) ) )
|
|
{
|
|
if ( RETURNCODETOHRESULT( ERROR_INSUFFICIENT_BUFFER ) == hRes )
|
|
{
|
|
if ( (pbOutBuffer = (LPBYTE)LocalAlloc( LMEM_FIXED, dwRequired )) == NULL )
|
|
{
|
|
hRes = RETURNCODETOHRESULT( ERROR_NOT_ENOUGH_MEMORY );
|
|
}
|
|
else
|
|
{
|
|
hRes = DeserializeEnumServiceBuffer( pbBuffer,
|
|
dwNumServices,
|
|
pbOutBuffer,
|
|
dwRequired,
|
|
&dwRequired );
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
pbOutBuffer = abOutBuffer;
|
|
}
|
|
}
|
|
|
|
if ( SUCCEEDED( hRes ) )
|
|
{
|
|
for ( i = 0 ; i < dwNumServices ; ++i )
|
|
{
|
|
apvParams[0] = ((LPENUM_SERVICE_STATUS)pbOutBuffer)[i].lpDisplayName;
|
|
apvParams[1] = ((LPENUM_SERVICE_STATUS)pbOutBuffer)[i].lpServiceName;
|
|
|
|
switch ( ((LPENUM_SERVICE_STATUS)pbOutBuffer)[i].ServiceStatus.dwCurrentState )
|
|
{
|
|
case SERVICE_STOPPED:
|
|
apvParams[2] = GetString(IRSTASTR_SERVICE_STOPPED); break;
|
|
|
|
case SERVICE_START_PENDING:
|
|
apvParams[2] = GetString(IRSTASTR_SERVICE_START_PENDING); break;
|
|
|
|
case SERVICE_STOP_PENDING:
|
|
apvParams[2] = GetString(IRSTASTR_SERVICE_STOP_PENDING); break;
|
|
|
|
case SERVICE_RUNNING:
|
|
apvParams[2] = GetString(IRSTASTR_SERVICE_RUNNING); break;
|
|
|
|
case SERVICE_CONTINUE_PENDING:
|
|
apvParams[2] = GetString(IRSTASTR_SERVICE_CONTINUE_PENDING); break;
|
|
|
|
case SERVICE_PAUSE_PENDING:
|
|
apvParams[2] = GetString(IRSTASTR_SERVICE_PAUSE_PENDING); break;
|
|
|
|
case SERVICE_PAUSED:
|
|
apvParams[2] = GetString(IRSTASTR_SERVICE_PAUSED); break;
|
|
|
|
default:
|
|
apvParams[2] = GetString(IRSTASTR_SERVICE_DEFAULT); break;
|
|
}
|
|
|
|
PrintMessage( IRSTASTR_STATUS_ITEM, 3, apvParams );
|
|
}
|
|
}
|
|
|
|
if ( pbBuffer != NULL && pbBuffer != abBuffer )
|
|
{
|
|
LocalFree( pbBuffer );
|
|
}
|
|
|
|
if ( pbOutBuffer != NULL && pbOutBuffer != abOutBuffer )
|
|
{
|
|
LocalFree( pbOutBuffer );
|
|
}
|
|
|
|
}
|
|
|
|
if ( FAILED( hRes ) && fRebootOnError
|
|
&& ( iCmd == CMD_STOP || iCmd == CMD_RESTART ) )
|
|
{
|
|
fErrDisplayed = FALSE;
|
|
PrintMessage( IRSTASTR_REBOOT_ON_ERROR, 0, NULL );
|
|
|
|
hRes = pIf->Reboot( 0, fForce );
|
|
|
|
if ( SUCCEEDED( hRes ) )
|
|
{
|
|
PrintMessage( IRSTASTR_REBOOT_SUCCESS, 0, NULL );
|
|
}
|
|
}
|
|
#if 0
|
|
if ( FAILED( hRes ) && fKillOnError
|
|
&& ( iCmd == CMD_STOP || iCmd == CMD_RESTART ) )
|
|
{
|
|
fErrDisplayed = FALSE;
|
|
PrintMessage( IRSTASTR_KILL_ON_ERROR, 0, NULL );
|
|
|
|
hRes = pIf->Kill();
|
|
|
|
if ( SUCCEEDED( hRes ) )
|
|
{
|
|
PrintMessage( IRSTASTR_KILL_SUCCESS, 0, NULL );
|
|
}
|
|
}
|
|
#endif
|
|
pIf->Release();
|
|
|
|
if ( FAILED( hRes ) )
|
|
{
|
|
if ( !fErrDisplayed )
|
|
{
|
|
if ( hRes == RETURNCODETOHRESULT( ERROR_RESOURCE_DISABLED ) )
|
|
{
|
|
PrintMessage( IRSTASTR_REMOTE_DISABLED, 0, NULL );
|
|
}
|
|
else if ( hRes == RETURNCODETOHRESULT( ERROR_ACCESS_DENIED ) )
|
|
{
|
|
PrintMessage( IRSTASTR_ACCESS_DENIED, 0, NULL );
|
|
}
|
|
else
|
|
{
|
|
DisplayErrorMessage( HRESULTTOWIN32( hRes ) );
|
|
}
|
|
}
|
|
|
|
Status = HRESULTTOWIN32( hRes );
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if ( hRes == RETURNCODETOHRESULT( ERROR_ACCESS_DENIED ) )
|
|
{
|
|
PrintMessage( IRSTASTR_ACCESS_DENIED, 0, NULL );
|
|
}
|
|
else
|
|
{
|
|
DisplayErrorMessage( HRESULTTOWIN32( hRes ) );
|
|
}
|
|
|
|
Status = HRESULTTOWIN32( hRes );
|
|
}
|
|
if (fCoInitialized) {
|
|
CoUninitialize();
|
|
}
|
|
}
|
|
|
|
Exit:
|
|
|
|
return Status;
|
|
}
|