310 lines
8.1 KiB
C++
310 lines
8.1 KiB
C++
|
/*++
|
||
|
|
|
||
|
|
Copyright (C) 1997-2001 Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
SERVUTIL.CPP
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
Defines various service utilities.
|
||
|
|
|
||
|
|
History:
|
||
|
|
|
||
|
|
a-davj 04-Mar-97 Created.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
#include "precomp.h"
|
||
|
|
#include "servutil.h"
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// BOOL InstallService
|
||
|
|
//
|
||
|
|
// DESCRIPTION:
|
||
|
|
//
|
||
|
|
// Installs a service
|
||
|
|
//
|
||
|
|
// PARAMETERS:
|
||
|
|
//
|
||
|
|
// pServiceName short service name
|
||
|
|
// pDisplayName name displayed to the user
|
||
|
|
// pBinary full path to the binary
|
||
|
|
//
|
||
|
|
// RETURN VALUE:
|
||
|
|
//
|
||
|
|
// TRUE if it worked
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
BOOL InstallService(
|
||
|
|
IN LPCTSTR pServiceName,
|
||
|
|
IN LPCTSTR pDisplayName,
|
||
|
|
IN LPCTSTR pBinary)
|
||
|
|
{
|
||
|
|
SC_HANDLE schService = NULL;
|
||
|
|
SC_HANDLE schSCManager;
|
||
|
|
|
||
|
|
schSCManager = OpenSCManager(
|
||
|
|
NULL, // machine (NULL == local)
|
||
|
|
NULL, // database (NULL == default)
|
||
|
|
SC_MANAGER_ALL_ACCESS // access required
|
||
|
|
);
|
||
|
|
if ( schSCManager )
|
||
|
|
{
|
||
|
|
schService = CreateService(
|
||
|
|
schSCManager, // SCManager database
|
||
|
|
pServiceName, // name of service
|
||
|
|
pDisplayName, // name to display
|
||
|
|
SERVICE_ALL_ACCESS, // desired access
|
||
|
|
SERVICE_WIN32_OWN_PROCESS, // service type
|
||
|
|
SERVICE_DEMAND_START, // start type
|
||
|
|
SERVICE_ERROR_NORMAL, // error control type
|
||
|
|
pBinary, // service's binary
|
||
|
|
NULL, // no load ordering group
|
||
|
|
NULL, // no tag identifier
|
||
|
|
NULL, // dependencies
|
||
|
|
NULL, // LocalSystem account
|
||
|
|
NULL); // no password
|
||
|
|
|
||
|
|
if ( schService )
|
||
|
|
{
|
||
|
|
CloseServiceHandle(schService);
|
||
|
|
}
|
||
|
|
|
||
|
|
CloseServiceHandle(schSCManager);
|
||
|
|
}
|
||
|
|
return schService != NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// BOOL RemoveService
|
||
|
|
//
|
||
|
|
// DESCRIPTION:
|
||
|
|
//
|
||
|
|
// Stops and then removes the service.
|
||
|
|
//
|
||
|
|
// PARAMETERS:
|
||
|
|
//
|
||
|
|
// pServiceName Short service name
|
||
|
|
//
|
||
|
|
// RETURN VALUE:
|
||
|
|
//
|
||
|
|
// TRUE if it worked
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
BOOL RemoveService(
|
||
|
|
IN LPCTSTR pServiceName)
|
||
|
|
{
|
||
|
|
SC_HANDLE schService;
|
||
|
|
BOOL bRet = FALSE;
|
||
|
|
SC_HANDLE schSCManager;
|
||
|
|
StopService(pServiceName, 15);
|
||
|
|
|
||
|
|
schSCManager = OpenSCManager(
|
||
|
|
NULL, // machine (NULL == local)
|
||
|
|
NULL, // database (NULL == default)
|
||
|
|
SC_MANAGER_ALL_ACCESS // access required
|
||
|
|
);
|
||
|
|
if ( schSCManager )
|
||
|
|
{
|
||
|
|
schService = OpenService(schSCManager, pServiceName, SERVICE_ALL_ACCESS);
|
||
|
|
|
||
|
|
if (schService)
|
||
|
|
{
|
||
|
|
bRet = DeleteService(schService);
|
||
|
|
CloseServiceHandle(schService);
|
||
|
|
}
|
||
|
|
|
||
|
|
CloseServiceHandle(schSCManager);
|
||
|
|
}
|
||
|
|
return bRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// BOOL StopService
|
||
|
|
//
|
||
|
|
// DESCRIPTION:
|
||
|
|
//
|
||
|
|
// Stops and then removes the service.
|
||
|
|
//
|
||
|
|
// PARAMETERS:
|
||
|
|
//
|
||
|
|
// pServiceName short service name
|
||
|
|
// dwMaxWait max time in seconds to wait
|
||
|
|
//
|
||
|
|
// RETURN VALUE:
|
||
|
|
//
|
||
|
|
// TRUE if it worked
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
BOOL StopService(
|
||
|
|
IN LPCTSTR pServiceName,
|
||
|
|
IN DWORD dwMaxWait)
|
||
|
|
{
|
||
|
|
BOOL bRet = FALSE;
|
||
|
|
SC_HANDLE schService;
|
||
|
|
SC_HANDLE schSCManager;
|
||
|
|
DWORD dwCnt;
|
||
|
|
SERVICE_STATUS ssStatus; // current status of the service
|
||
|
|
|
||
|
|
schSCManager = OpenSCManager(
|
||
|
|
NULL, // machine (NULL == local)
|
||
|
|
NULL, // database (NULL == default)
|
||
|
|
SC_MANAGER_ALL_ACCESS // access required
|
||
|
|
);
|
||
|
|
if ( schSCManager )
|
||
|
|
{
|
||
|
|
schService = OpenService(schSCManager, pServiceName, SERVICE_ALL_ACCESS);
|
||
|
|
|
||
|
|
if (schService)
|
||
|
|
{
|
||
|
|
// try to stop the service
|
||
|
|
if ( bRet = ControlService( schService, SERVICE_CONTROL_STOP, &ssStatus ) )
|
||
|
|
{
|
||
|
|
for(dwCnt=0; dwCnt < dwMaxWait &&
|
||
|
|
QueryServiceStatus( schService, &ssStatus ); dwCnt++)
|
||
|
|
{
|
||
|
|
if ( ssStatus.dwCurrentState == SERVICE_STOP_PENDING )
|
||
|
|
Sleep( 1000 );
|
||
|
|
else
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
CloseServiceHandle(schService);
|
||
|
|
}
|
||
|
|
|
||
|
|
CloseServiceHandle(schSCManager);
|
||
|
|
}
|
||
|
|
return bRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// BOOL StartService
|
||
|
|
//
|
||
|
|
// DESCRIPTION:
|
||
|
|
//
|
||
|
|
// Starts the service runnig
|
||
|
|
//
|
||
|
|
// PARAMETERS:
|
||
|
|
//
|
||
|
|
// pServiceName short service name
|
||
|
|
// dwMaxWait max time in seconds to wait
|
||
|
|
//
|
||
|
|
// RETURN VALUE:
|
||
|
|
//
|
||
|
|
// TRUE if it worked
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
BOOL StartService(
|
||
|
|
IN LPCTSTR pServiceName,
|
||
|
|
IN DWORD dwMaxWait)
|
||
|
|
{
|
||
|
|
DWORD dwCnt;
|
||
|
|
BOOL bRet = FALSE;
|
||
|
|
SC_HANDLE schService;
|
||
|
|
SC_HANDLE schSCManager;
|
||
|
|
SERVICE_STATUS ssStatus; // current status of the service
|
||
|
|
|
||
|
|
schSCManager = OpenSCManager(
|
||
|
|
NULL, // machine (NULL == local)
|
||
|
|
NULL, // database (NULL == default)
|
||
|
|
SC_MANAGER_ALL_ACCESS // access required
|
||
|
|
);
|
||
|
|
if ( schSCManager )
|
||
|
|
{
|
||
|
|
schService = OpenService(schSCManager, pServiceName, SERVICE_ALL_ACCESS);
|
||
|
|
|
||
|
|
if (schService)
|
||
|
|
{
|
||
|
|
// try to stop the service
|
||
|
|
if ( bRet = StartService( schService, 0, NULL ) )
|
||
|
|
{
|
||
|
|
for(dwCnt = 0; dwCnt < dwMaxWait &&
|
||
|
|
QueryServiceStatus( schService, &ssStatus ); dwCnt++)
|
||
|
|
{
|
||
|
|
if ( ssStatus.dwCurrentState != SERVICE_RUNNING )
|
||
|
|
Sleep( 1000 );
|
||
|
|
else
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
CloseServiceHandle(schService);
|
||
|
|
}
|
||
|
|
|
||
|
|
CloseServiceHandle(schSCManager);
|
||
|
|
}
|
||
|
|
return bRet;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//***************************************************************************
|
||
|
|
//
|
||
|
|
// BOOL SetDependency
|
||
|
|
//
|
||
|
|
// DESCRIPTION:
|
||
|
|
//
|
||
|
|
// Sets a service's dependency list.
|
||
|
|
//
|
||
|
|
// PARAMETERS:
|
||
|
|
//
|
||
|
|
// pServiceName short service name
|
||
|
|
// pDependency dependency list
|
||
|
|
//
|
||
|
|
// RETURN VALUE:
|
||
|
|
//
|
||
|
|
// TRUE if it worked
|
||
|
|
//
|
||
|
|
//***************************************************************************
|
||
|
|
|
||
|
|
BOOL SetDependency(
|
||
|
|
IN LPCTSTR pServiceName,
|
||
|
|
IN LPCTSTR pDependency)
|
||
|
|
{
|
||
|
|
BOOL bRet = FALSE;
|
||
|
|
SC_HANDLE schService;
|
||
|
|
SC_HANDLE schSCManager;
|
||
|
|
|
||
|
|
schSCManager = OpenSCManager(
|
||
|
|
NULL, // machine (NULL == local)
|
||
|
|
NULL, // database (NULL == default)
|
||
|
|
SC_MANAGER_ALL_ACCESS // access required
|
||
|
|
);
|
||
|
|
if ( schSCManager )
|
||
|
|
{
|
||
|
|
schService = OpenService(schSCManager, pServiceName, SERVICE_ALL_ACCESS);
|
||
|
|
|
||
|
|
if (schService)
|
||
|
|
{
|
||
|
|
bRet = ChangeServiceConfig(
|
||
|
|
schService,
|
||
|
|
SERVICE_NO_CHANGE,
|
||
|
|
SERVICE_NO_CHANGE,
|
||
|
|
SERVICE_NO_CHANGE,
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
pDependency,
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
NULL);
|
||
|
|
CloseServiceHandle(schService);
|
||
|
|
}
|
||
|
|
|
||
|
|
CloseServiceHandle(schSCManager);
|
||
|
|
}
|
||
|
|
return bRet;
|
||
|
|
}
|