聲明:以下代碼本人未測試。
轉自: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 <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;
}