65,210
社区成员
发帖
与我相关
我的任务
分享
#include <windows.h>
#include "tchar.h"
class CServiceManage
{
typedef enum {NOINSTALL, RUNNING, STOPPED,OperateFail=-1};
public:
CServiceManage();
virtual ~CServiceManage();
public:
//安装服务
BOOL InstatllService(char *szServiceName,char *szServiceParamName);
//卸载服务
BOOL UninstallService(char *szServiceName,char *szServiceParamName);
//启动服务
BOOL StartupService(char *szServiceName);
//停止服务
BOOL StopService(char *szServiceName);
//获取服务当前状态
DWORD GetServiceState(char *szServiceName);
//判断服务是否安装
BOOL ServiceIsInstalled(char *szServiceName);
};
//CPP文件
#include "stdafx.h"
#include "ServiceManage.h"
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
CServiceManage::CServiceManage()
{
}
CServiceManage::~CServiceManage()
{
}
/*****************************************************************
** 函数名称: BOOL InstatllService(char *szServiceName,char *szServiceParamName)
**
** 功能描述: 安装服务
**
** 入 参: szServiceName 服务名称 szServiceParamName 服务程序名称
**
** 出 参: 无
**
** 返 回 值: BOOL TRUE成功 FALSE失败
*****************************************************************/
BOOL CServiceManage::InstatllService(char *szServiceName,char *szServiceParamName)
{
if (ServiceIsInstalled(szServiceName))
return TRUE;
TCHAR szFilePath[_MAX_PATH];
::GetModuleFileName(NULL, szFilePath, _MAX_PATH);
(_tcsrchr(szFilePath, '\\'))[1] = 0;
_tcscat(szFilePath, szServiceParamName);
ShellExecute(NULL, "open", szFilePath, "/Service", NULL, SW_HIDE);
return TRUE;
}
/*****************************************************************
** 函数名称: BOOL UninstallService(char *szServiceName,char *szServiceParamName)
**
** 功能描述: 卸载服务
**
** 入 参: szServiceName 服务名称 szServiceParamName 服务程序名称
**
** 出 参: 无
**
** 返 回 值: BOOL TRUE成功 FALSE失败
*****************************************************************/
BOOL CServiceManage::UninstallService(char *szServiceName,char *szServiceParamName)
{
if (!ServiceIsInstalled(szServiceName))
return TRUE;
TCHAR szFilePath[_MAX_PATH];
::GetModuleFileName(NULL, szFilePath, _MAX_PATH);
(_tcsrchr(szFilePath, '\\'))[1] = 0;
_tcscat(szFilePath, szServiceParamName);
ShellExecute(NULL, "open", szFilePath, "/UnregServer", NULL, SW_HIDE);
return TRUE;
}
/*****************************************************************
** 函数名称: BOOL StartupService(char *szServiceName)
**
** 功能描述: 启动服务
**
** 入 参: szServiceName 服务名称
**
** 出 参: 无
**
** 返 回 值: BOOL TRUE成功 FALSE失败
*****************************************************************/
BOOL CServiceManage::StartupService(char *szServiceName)
{
if (!ServiceIsInstalled(szServiceName))
return FALSE;
SC_HANDLE hSCM = ::OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
if (hSCM == NULL)
{
MessageBox(NULL,"Couldn't open service manager", szServiceName, MB_OK);
return FALSE;
}
SC_HANDLE hService = ::OpenService(hSCM, szServiceName, SERVICE_ALL_ACCESS);
if (hService == NULL)
{
::CloseServiceHandle(hSCM);
MessageBox(NULL,"Couldn't open service", szServiceName, MB_OK);
return FALSE;
}
SERVICE_STATUS status;
if (!::QueryServiceStatus(hService, &status))
{
::CloseServiceHandle(hService);
::CloseServiceHandle(hSCM);
MessageBox(NULL,"Couldn't Query Service Status", szServiceName, MB_OK);
return FALSE;
}
if (status.dwCurrentState == SERVICE_STOPPED)
{
if (!::StartService(hService, NULL, NULL))
{
::CloseServiceHandle(hService);
::CloseServiceHandle(hSCM);
MessageBox(NULL,"Couldn't Startup Service", szServiceName, MB_OK);
return FALSE;
}
}
::CloseServiceHandle(hService);
::CloseServiceHandle(hSCM);
return TRUE;
}
/*****************************************************************
** 函数名称: BOOL StopService(char *szServiceName)
**
** 功能描述: 停止服务
**
** 入 参: szServiceName 服务名称
**
** 出 参: 无
**
** 返 回 值: BOOL TRUE成功 FALSE失败
*****************************************************************/
BOOL CServiceManage::StopService(char *szServiceName)
{
if (!ServiceIsInstalled(szServiceName))
return FALSE;
SC_HANDLE hSCM = ::OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
if (hSCM == NULL)
{
MessageBox(NULL,"Couldn't open service manager", szServiceName, MB_OK);
return FALSE;
}
SC_HANDLE hService = ::OpenService(hSCM, szServiceName, SERVICE_ALL_ACCESS);
if (hService == NULL)
{
::CloseServiceHandle(hSCM);
MessageBox(NULL,"Couldn't open service", szServiceName, MB_OK);
return FALSE;
}
SERVICE_STATUS status;
if (!::QueryServiceStatus(hService, &status))
{
::CloseServiceHandle(hService);
::CloseServiceHandle(hSCM);
MessageBox(NULL,"Couldn't Query Service Status", szServiceName, MB_OK);
return FALSE;
}
if (status.dwCurrentState != SERVICE_STOPPED)
{
if (!::ControlService(hService, SERVICE_CONTROL_STOP, &status))
{
::CloseServiceHandle(hService);
::CloseServiceHandle(hSCM);
MessageBox(NULL,"Couldn't Stop Service", szServiceName, MB_OK);
return FALSE;
}
}
::CloseServiceHandle(hService);
::CloseServiceHandle(hSCM);
return TRUE;
}
/*****************************************************************
** 函数名称: DWORD GetServiceState(char *szServiceName)
**
** 功能描述: 获取服务当前状态
**
** 入 参: szServiceName 服务名称
**
** 出 参: 无
**
** 返 回 值: DWORD 服务状态
*****************************************************************/
DWORD CServiceManage::GetServiceState(char *szServiceName)
{
if (!ServiceIsInstalled(szServiceName))
return NOINSTALL;
SC_HANDLE hSCM = ::OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
if (hSCM == NULL)
{
MessageBox(NULL,"Couldn't open service manager", szServiceName, MB_OK);
return OperateFail;
}
SC_HANDLE hService = ::OpenService(hSCM, szServiceName, SERVICE_ALL_ACCESS);
if (hService == NULL)
{
::CloseServiceHandle(hSCM);
MessageBox(NULL,"Couldn't open service", szServiceName, MB_OK);
return OperateFail;
}
SERVICE_STATUS status;
if (!::QueryServiceStatus(hService, &status))
{
::CloseServiceHandle(hService);
::CloseServiceHandle(hSCM);
MessageBox(NULL,"Couldn't Query Service Status", szServiceName, MB_OK);
return OperateFail;
}
::CloseServiceHandle(hService);
::CloseServiceHandle(hSCM);
switch(status.dwCurrentState)
{
case SERVICE_RUNNING:
return RUNNING;
case SERVICE_STOPPED:
return STOPPED;
default:
return OperateFail;
}
return RUNNING;
}
/*****************************************************************
** 函数名称: BOOL ServiceIsInstalled(char *szServiceName)
**
** 功能描述: 判断服务是否安装
**
** 入 参: szServiceName 服务名称
**
** 出 参: 无
**
** 返 回 值: BOOL TRUE成功 FALSE失败
*****************************************************************/
BOOL CServiceManage::ServiceIsInstalled(char *szServiceName)
{
BOOL bResult = FALSE;
SC_HANDLE hSCM = ::OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
if (hSCM != NULL)
{
SC_HANDLE hService = ::OpenService(hSCM, szServiceName, SERVICE_QUERY_CONFIG);
if (hService != NULL)
{
bResult = TRUE;
::CloseServiceHandle(hService);
}
::CloseServiceHandle(hSCM);
}
return bResult;
}