創建windows服務

聲明:以下代碼本人未測試。

 

轉自: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;
}

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章