声明:以下代码本人未测试。
转自:https://blog.csdn.net/u013472838/article/details/44850685
本文主要介绍了 OpenSCManager、CreateService、OpenService、ControlService、DeleteService、RegisterServiceCtrlHandler、SetServiceStatus、StartServiceCtrlDispatcher等操作服务程序的主要几个API的用法,具体的函数参数大家可以查阅MSDN。
为什么要使用服务应该程序呢?服务程序就像系统的一些服务一样,能够自动地启动,并执行相应的操作;而且因为服务程序的在层次上和一般的应用程序不同,其能够在系统启动时就自动地运行,而不像一般的应用程序那样一定要在登陆后才能运行,这些就是服务的一些好处了,如果你也想你的程序具有这样的功能,那么你就可以建立一个服务应用程序了。下面就跟着我一步一步地教你怎么去创建一个服务应用程序吧。
一、建立 Win32 Application 应用程序(当然你也可以建立其它的应用程序,但服务一般是没有用户界面的),并命名为 ServiceTest。
二、定义全局函数变量。这里主要是设置服务句柄和状态。
BOOL IsInstalled();
BOOL Install();
BOOL Uninstall();
void LogEvent(LPCTSTR pszFormat, ...);
void WINAPI ServiceMain();
void WINAPI ServiceStrl(DWORD dwOpcode);
TCHAR szServiceName[] = _T("ServiceTest");
BOOL bInstall;
SERVICE_STATUS_HANDLE hServiceStatus;
SERVICE_STATUS status;
DWORD dwThreadID;
三、添加Init初始化函数。
void Init()
{
hServiceStatus = NULL;
status.dwServiceType = SERVICE_WIN32_OWN_PROCESS;
status.dwCurrentState = SERVICE_STOPPED;
tatus.dwControlsAccepted = SERVICE_ACCEPT_STOP;
status.dwWin32ExitCode = 0;
status.dwServiceSpecificExitCode = 0;
status.dwCheckPoint = 0;
status.dwWaitHint = 0;
}
四、添加安装和删除服务函数。这里主要是用到了四个函数 OpenSCManager 和 CreateService。OpenSCManager 用于打开服务控制管理器;CreateService 用于创建服务;OpenService用于打开已有的服务,返回该服务的句柄;ControlService则用于控制已打开的服务状态,这里是让服务停止后才删除;DeleteService 用于删除指定服务。
BOOL Install();
{
//这里列出主要的两个函数,其它的可以在代码里找。
//打开服务控制管理器
OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
//创建服务
SC_HANDLE hService = ::CreateService(
hSCM, szServiceName, szServiceName,
SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS,
SERVICE_DEMAND_START, SERVICE_ERROR_NORMAL,
szFilePath, NULL, NULL, _T(""), NULL, NULL);
::CloseServiceHandle(hService);
::CloseServiceHandle(hSCM);
}
BOOL Uninstall();
{
//这里列出主要的两个函数,其它的可以在代码里找。
//打开服务控制管理器
OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS);
//打开服务
OpenService(hSCM, szServiceName, SERVICE_STOP | DELETE);
//停止服务
ControlService(hService, SERVICE_CONTROL_STOP, &status);
//删除服务
DeleteService(hService);
…
}
五、添加服务主线程函数和控制函数。这里调用 RegisterServiceCtrlHandler 来注册服务的控制函数,这里要设置status.dwControlsAccepted 为 SERVICE_ACCEPT_STOP,否则你不能控制这个服务的状态。
void WINAPI ServiceMain()
{
// Register the control request handler
status.dwCurrentState = SERVICE_START_PENDING;
status.dwControlsAccepted = SERVICE_ACCEPT_STOP;//这个要使用,否则你不能控制
//注册服务控制
hServiceStatus = RegisterServiceCtrlHandler(szServiceName, ServiceStrl);
if (hServiceStatus == NULL)
{
LogEvent(_T("Handler not installed"));
return;
}
SetServiceStatus(hServiceStatus, &status);
status.dwWin32ExitCode = S_OK;
status.dwCheckPoint = 0;
status.dwWaitHint = 0;
status.dwCurrentState = SERVICE_RUNNING;
SetServiceStatus(hServiceStatus, &status);
//模拟服务的运行,10后自动退出。应用时将主要任务放于此即可
int i = 0;
while (i < 10)
{
Sleep(1000);
i++;
}
status.dwCurrentState = SERVICE_STOPPED;
SetServiceStatus(hServiceStatus, &status);
LogEvent(_T("Service stopped"));
}
六、在主线程函数里注册控制函数和程序执行主体。这里主要是说明这就是程序的执行体。
void WINAPI ServiceMain()
{
…
//模拟服务的运行,10后自动退出。应用时将主要任务放于此即可
int i = 0;
while (i < 10)
{
Sleep(1000);
i++;
}
}
七、最后,要在main函数里注册添加安装、删除、注册主函数。
int APIENTRY WinMain(HINSTANCE hInstance,
HINSTANCE hPrevInstance,
LPSTR lpCmdLine,
int nCmdShow)
{
Init();
dwThreadID = ::GetCurrentThreadId();
SERVICE_TABLE_ENTRY st[] =
{
{ szServiceName, (LPSERVICE_MAIN_FUNCTION)ServiceMain },
{ NULL, NULL }
};
if (stricmp(lpCmdLine, "/install") == 0)
{
Install();
}
else if (stricmp(lpCmdLine, "/uninstall") == 0)
{
Uninstall();
}
else
{
if (!::StartServiceCtrlDispatcher(st))
{
LogEvent(_T("Register Service Main Function Error!"));
}
}
return 0;
}
八、总结。其实做一个服务程序并不难,主要是懂得程序的执行体放于哪里?和注册程序的主函数和注册控制函数,如果这两个没有注册的话,你就程序就不知道如何去控制了。status.dwControlsAccepted = SERVICE_ACCEPT_STOP;这个也重要,如果你没有设置的话,那么服务就不会受你控制了。
下面是一个服务类:
转自:https://bbs.csdn.net/topics/380086285 16楼
#include
#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;
}