C++程序服务化封装

2024-06-02 22:32
文章标签 c++ 程序 封装 服务化

本文主要是介绍C++程序服务化封装，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

之前有一篇“C++exe做系统服务，自动加载”是C语言风格的方式，使用是没有问题，但是很杂乱，这里对之封装，简化使用方式。

WindowsService.h

#pragma once
#include <string>
using namespace std;//注入Windows服务类
class CWindowsService
{
public:CWindowsService();//构造函数CWindowsService(const string strServiceName);//构造函数~CWindowsService();//析构函数void SetServiceName(const string strServiceName);//设置服务名称(所有操作前必须调用)void Run();//一键自助安装服务   private:bool Install(int & dwCurrentState);//安装服务bool Uninstall();//卸载服务bool Start();//启动服务bool Stop();//停止服务static void ServiceMain(DWORD argc, LPTSTR *argv);//创建服务主要处理逻辑static void ServiceCtrlHandler(DWORD nControlCode);//接受servicecontrol 命令BOOL UpdateServiceStatus(DWORD dwCurrentState, DWORD dwWin32ExitCode, DWORD dwServiceSpecificExitCode, DWORD dwCheckPoint, DWORD dwWaitHint);//更新服务状态 更新失败即结束服务BOOL StartServiceThread();//创建主窗口线程static DWORD ServiceExecutionThread(LPDWORD param);//创建服务主线程void KillService();// 停止服务 即结束主窗口线程 设置状态失败或是收到STOP SERVICE 命令调用此函数private:HANDLE m_hServiceThread;SERVICE_STATUS_HANDLE m_nServiceStatusHandle;HANDLE m_EventkillService;HWND m_hMain;BOOL m_bServiceRunning;DWORD m_dwServiceCurrentStatus;string m_ServiceDisplayName;string m_ServiceName;
};extern CWindowsService *g_pWindowsService;

WindowsService.cpp

#include "stdafx.h"
#include "WindowsService.h"
#include "server.h"CWindowsService *g_pWindowsService = NULL;//构造函数
CWindowsService::CWindowsService()
{m_hServiceThread = 0;m_nServiceStatusHandle = 0;m_EventkillService = NULL;m_hMain = NULL;m_bServiceRunning = FALSE;m_dwServiceCurrentStatus = 0;
}CWindowsService::CWindowsService(const string strServiceName)
{CWindowsService();m_ServiceDisplayName = strServiceName;m_ServiceName = strServiceName;
}//析构函数
CWindowsService::~CWindowsService()
{
}//设置服务名称
void CWindowsService::SetServiceName(const string strServiceName)
{m_ServiceDisplayName = strServiceName;m_ServiceName = strServiceName;
}//一键自助安装服务
void CWindowsService::Run()
{BOOL bNT = FALSE;BOOL bInstalled = FALSE;int dwCurrentState = 0;SC_HANDLE hService, hScm;hScm = OpenSCManager(0, 0, SC_MANAGER_CONNECT);if (hScm){bNT = TRUE;hService = OpenService(hScm, m_ServiceName.c_str(), GENERIC_READ);if (hService){bInstalled = TRUE;SERVICE_STATUS ServiceStatus;if (QueryServiceStatus(hService, &ServiceStatus)){dwCurrentState = ServiceStatus.dwCurrentState;if (dwCurrentState == SERVICE_START_PENDING){CloseServiceHandle(hService);CloseServiceHandle(hScm);char chServiceName[256] = "";strcpy_s(chServiceName, m_ServiceName.c_str());const SERVICE_TABLE_ENTRY servicetable[] ={{ chServiceName, (LPSERVICE_MAIN_FUNCTION)ServiceMain },{ NULL,NULL }};BOOL success;success = StartServiceCtrlDispatcher(servicetable);if (!success){int nError = GetLastError();TCHAR buffer[1000];_stprintf_s(buffer, _T("启动服务出错，错误码：%d"), nError);MessageBox(0, buffer, m_ServiceName.c_str(), MB_OK);}return ;}}CloseServiceHandle(hService);}CloseServiceHandle(hScm);}if (!bInstalled){if (MessageBox(0, _T("是否安装服务?"), m_ServiceName.c_str(), MB_YESNO | MB_ICONQUESTION) == IDYES){if (!Install(dwCurrentState)){MessageBox(0, _T("安装服务失败"), m_ServiceName.c_str(), MB_OK);return;}}elsereturn ;}if (dwCurrentState == SERVICE_STOPPED && (MessageBox(0, _T("启动服务?"), m_ServiceName.c_str(), MB_YESNO | MB_ICONQUESTION) == IDYES)){if (!Start()){MessageBox(0, _T("启动服务失败"), m_ServiceName.c_str(), MB_OK);}return;}if (dwCurrentState == SERVICE_STOPPED && (MessageBox(0, _T("卸载服务?"), m_ServiceName.c_str(), MB_YESNO | MB_ICONQUESTION) == IDYES)){if (!Uninstall()){MessageBox(0, _T("卸载服务失败"), m_ServiceName.c_str(), MB_OK);}return ;}if (dwCurrentState != SERVICE_STOPPED && (MessageBox(0, _T("停止服务?"), m_ServiceName.c_str(), MB_YESNO | MB_ICONQUESTION) == IDYES)){if (!Stop()){MessageBox(0, _T("停止服务失败"), m_ServiceName.c_str(), MB_OK);}return;}if (dwCurrentState == SERVICE_STOPPED)return ;
}//安装服务
bool CWindowsService::Install(int & dwCurrentState)
{SC_HANDLE hService, hScm;hScm = OpenSCManager(0, 0, SC_MANAGER_CREATE_SERVICE);if (!hScm){return false;}int nStartMode = SERVICE_AUTO_START;if (MessageBox(0, _T("自动运行服务?"), m_ServiceName.c_str(), MB_YESNO | MB_ICONQUESTION) == IDYES){nStartMode = SERVICE_AUTO_START;}TCHAR buffer[MAX_PATH + 3];buffer[0] = '"';DWORD written = GetModuleFileName(0, buffer + 1, MAX_PATH);if (!written){CloseServiceHandle(hScm);MessageBox(0, _T("创建服务失败，未能获取应用程序当前的路径"), m_ServiceName.c_str(), MB_ICONSTOP);return false;}buffer[written + 1] = '"';buffer[written + 2] = 0;hService = CreateService(hScm, m_ServiceName.c_str(),m_ServiceDisplayName.c_str(),SERVICE_ALL_ACCESS, SERVICE_WIN32_OWN_PROCESS | SERVICE_INTERACTIVE_PROCESS, nStartMode,SERVICE_ERROR_NORMAL,buffer,0, 0, NULL, 0, 0);if (!hService){CloseServiceHandle(hScm);return false;}SERVICE_DESCRIPTION description;memset(&description, 0, sizeof(SERVICE_DESCRIPTION));TCHAR szdescription[100];memset(szdescription, 0, sizeof(TCHAR) * 100);_stprintf_s(szdescription, 100, _T("%s"), m_ServiceName.c_str());//服务名称description.lpDescription = szdescription;ChangeServiceConfig2(hService, SERVICE_CONFIG_DESCRIPTION, &description);CloseServiceHandle(hService);CloseServiceHandle(hScm);dwCurrentState = SERVICE_STOPPED;return true;
}//卸载服务
bool CWindowsService::Uninstall()
{SC_HANDLE hService, hScm;hScm = OpenSCManager(0, 0, SC_MANAGER_CREATE_SERVICE);if (!hScm){return false;}hService = OpenService(hScm, m_ServiceName.c_str(), SERVICE_ALL_ACCESS);if (!hService){CloseServiceHandle(hScm);return false;}DeleteService(hService);CloseServiceHandle(hService);CloseServiceHandle(hScm);return true;
}//启动服务
bool CWindowsService::Start()
{SC_HANDLE hService, hScm;hScm = OpenSCManager(0, 0, SC_MANAGER_ALL_ACCESS);if (!hScm){return false;}hService = OpenService(hScm, m_ServiceName.c_str(), SERVICE_ALL_ACCESS);if (!hService){CloseServiceHandle(hScm);return false;}if (StartService(hService, 0, NULL) == 0){DWORD dwerr = GetLastError();Sleep(1);}CloseServiceHandle(hService);CloseServiceHandle(hScm);return true;
}//停止服务
bool CWindowsService::Stop()
{SC_HANDLE hService, hScm;hScm = OpenSCManager(0, 0, SC_MANAGER_ALL_ACCESS);if (!hScm){return false;}hService = OpenService(hScm, m_ServiceName.c_str(), SERVICE_ALL_ACCESS);if (!hService){CloseServiceHandle(hScm);return false;}SERVICE_STATUS status;ControlService(hService, SERVICE_CONTROL_STOP, &status);CloseServiceHandle(hService);CloseServiceHandle(hScm);return true;
}//创建服务主要处理逻辑
void CWindowsService::ServiceMain(DWORD argc, LPTSTR * argv)
{BOOL success;g_pWindowsService->m_nServiceStatusHandle = RegisterServiceCtrlHandler(g_pWindowsService->m_ServiceName.c_str(), (LPHANDLER_FUNCTION)(ServiceCtrlHandler));if (!g_pWindowsService->m_nServiceStatusHandle)return;success = g_pWindowsService->UpdateServiceStatus(SERVICE_START_PENDING, NO_ERROR, 0, 1, 3000);if (!success)return;g_pWindowsService->m_EventkillService = CreateEvent(0, TRUE, FALSE, 0);//创建时间等待结束if (g_pWindowsService->m_EventkillService == NULL)return;success = g_pWindowsService->UpdateServiceStatus(SERVICE_START_PENDING, NO_ERROR, 0, 2, 1000);if (!success)return;success = g_pWindowsService->StartServiceThread(); //开启主线程if (!success)return;g_pWindowsService->m_dwServiceCurrentStatus = SERVICE_RUNNING;success = g_pWindowsService->UpdateServiceStatus(SERVICE_RUNNING, NO_ERROR, 0, 0, 0);if (!success)return;WaitForSingleObject(g_pWindowsService->m_EventkillService, INFINITE); //等待结束CloseHandle(g_pWindowsService->m_EventkillService);WaitForSingleObject(g_pWindowsService->m_hServiceThread, INFINITE); // 更待结束线程结束CloseHandle(g_pWindowsService->m_hServiceThread);g_pWindowsService->UpdateServiceStatus(SERVICE_STOPPED, NO_ERROR, 0, 0, 0);
}//接受servicecontrol 命令
void CWindowsService::ServiceCtrlHandler(DWORD nControlCode)
{BOOL success;switch (nControlCode){case SERVICE_CONTROL_SHUTDOWN:case SERVICE_CONTROL_STOP:g_pWindowsService->m_dwServiceCurrentStatus = SERVICE_STOP_PENDING;success = g_pWindowsService->UpdateServiceStatus(SERVICE_STOP_PENDING, NO_ERROR, 0, 1, 3000);g_pWindowsService->KillService();return;default:break;}g_pWindowsService->UpdateServiceStatus(g_pWindowsService->m_dwServiceCurrentStatus, NO_ERROR, 0, 0, 0);
}//更新服务状态 更新失败即结束服务
BOOL CWindowsService::UpdateServiceStatus(DWORD dwCurrentState, DWORD dwWin32ExitCode, DWORD dwServiceSpecificExitCode, DWORD dwCheckPoint, DWORD dwWaitHint)
{BOOL success;SERVICE_STATUS nServiceStatus;nServiceStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS;nServiceStatus.dwCurrentState = dwCurrentState;if (dwCurrentState == SERVICE_START_PENDING){nServiceStatus.dwControlsAccepted = 0;}else{nServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP| SERVICE_ACCEPT_SHUTDOWN;}if (dwServiceSpecificExitCode == 0){nServiceStatus.dwWin32ExitCode = dwWin32ExitCode;}else{nServiceStatus.dwWin32ExitCode = ERROR_SERVICE_SPECIFIC_ERROR;}nServiceStatus.dwServiceSpecificExitCode = dwServiceSpecificExitCode;nServiceStatus.dwCheckPoint = dwCheckPoint;nServiceStatus.dwWaitHint = dwWaitHint;success = SetServiceStatus(m_nServiceStatusHandle, &nServiceStatus);if (!success){KillService();return success;}else{return success;}
}//创建主窗口线程
BOOL CWindowsService::StartServiceThread()
{DWORD id;m_hServiceThread = CreateThread(0, 0, (LPTHREAD_START_ROUTINE)ServiceExecutionThread, 0, 0, &id);if (m_hServiceThread == 0){return FALSE;}else{m_bServiceRunning = TRUE;return true;}
}//创建服务主线程(应用主要启动)
DWORD CWindowsService::ServiceExecutionThread(LPDWORD param)
{BOOL res = TRUE;//启动CServer *pServer = new CServer;g_pWindowsService->m_bServiceRunning = pServer->Create();if (!g_pWindowsService->m_bServiceRunning){return 0;}else{g_pWindowsService->m_hMain = pServer->GetMainHwnd();}MSG msg;while (GetMessage(&msg, 0, 0, 0)){TranslateMessage(&msg);DispatchMessage(&msg);}SetEvent(g_pWindowsService->m_EventkillService);return 0;
}// 停止服务 即结束主窗口线程 设置状态失败或是收到STOP SERVICE 命令调用此函数
void CWindowsService::KillService()
{if (m_hMain != NULL)PostMessage(m_hMain, WM_CLOSE, 0, 0); //结束窗口m_bServiceRunning = FALSE;
}
  • 需要修改地方
    WindowsService.cpp
    //创建服务主线程(应用主要启动)
    DWORD CWindowsService::ServiceExecutionThread(LPDWORD param)
    修改该函数里面的
    //启动CServer *pServer = new CServer;g_pWindowsService->m_bServiceRunning = pServer->Create();

这里就是服务真正需要运行的代码，对应修改。

  • 调用方式

#include "stdafx.h"
#include "WindowsService.h"#ifndef _DEBUG
#pragma comment( linker, "/subsystem:windows /entry:mainCRTStartup" )
#endifint _tmain(int argc, _TCHAR* argv[])
{
#ifdef _DEBUG//调试//调试使用
#else//服务（发布使用）//设置服务名称string strServiceName = "ServiceName";g_pWindowsService = new CWindowsService(strServiceName);g_pWindowsService->Run();//注册服务，启动
#endifreturn 0;
}

这篇关于C++程序服务化封装的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！


http://www.chinasem.cn/article/1025273。

相关文章

C++使用栈实现括号匹配的代码详解

C++使用栈实现括号匹配的代码详解

《C++使用栈实现括号匹配的代码详解》在编程中,括号匹配是一个常见问题,尤其是在处理数学表达式、编译器解析等任务时,栈是一种非常适合处理此类问题的数据结构,能够精确地管理括号的匹配问题,本文将通过C+... 目录引言问题描述代码讲解代码解析栈的状态表示测试总结引言在编程中，括号匹配是一个常见问题，尤其是在
阅读更多...
使用C++实现链表元素的反转

使用C++实现链表元素的反转

《使用C++实现链表元素的反转》反转链表是链表操作中一个经典的问题,也是面试中常见的考题,本文将从思路到实现一步步地讲解如何实现链表的反转,帮助初学者理解这一操作,我们将使用C++代码演示具体实现,同... 目录问题定义思路分析代码实现带头节点的链表代码讲解其他实现方式时间和空间复杂度分析总结问题定义给定
阅读更多...
C++初始化数组的几种常见方法(简单易懂)

C++初始化数组的几种常见方法(简单易懂)

《C++初始化数组的几种常见方法(简单易懂)》本文介绍了C++中数组的初始化方法,包括一维数组和二维数组的初始化,以及用new动态初始化数组,在C++11及以上版本中,还提供了使用std::array... 目录1、初始化一维数组1.1、使用列表初始化（推荐方式）1.2、初始化部分列表1.3、使用std::
阅读更多...
C++ Primer 多维数组的使用

C++ Primer 多维数组的使用

《C++Primer多维数组的使用》本文主要介绍了多维数组在C++语言中的定义、初始化、下标引用以及使用范围for语句处理多维数组的方法,具有一定的参考价值,感兴趣的可以了解一下... 目录多维数组多维数组的初始化多维数组的下标引用使用范围for语句处理多维数组指针和多维数组多维数组严格来说,C++语言没
阅读更多...
在不同系统间迁移Python程序的方法与教程

在不同系统间迁移Python程序的方法与教程

《在不同系统间迁移Python程序的方法与教程》本文介绍了几种将Windows上编写的Python程序迁移到Linux服务器上的方法,包括使用虚拟环境和依赖冻结、容器化技术（如Docker）、使用An... 目录使用虚拟环境和依赖冻结1. 创建虚拟环境2. 冻结依赖使用容器化技术（如 docker）1. 创
阅读更多...
c++中std::placeholders的使用方法

c++中std::placeholders的使用方法

《c++中std::placeholders的使用方法》std::placeholders是C++标准库中的一个工具,用于在函数对象绑定时创建占位符,本文就来详细的介绍一下,具有一定的参考价值,感兴... 目录1. 基本概念2. 使用场景3. 示例示例 1：部分参数绑定示例 2：参数重排序4. 注意事项5.
阅读更多...
使用C++将处理后的信号保存为PNG和TIFF格式

使用C++将处理后的信号保存为PNG和TIFF格式

《使用C++将处理后的信号保存为PNG和TIFF格式》在信号处理领域,我们常常需要将处理结果以图像的形式保存下来,方便后续分析和展示,C++提供了多种库来处理图像数据,本文将介绍如何使用stb_ima... 目录1. PNG格式保存使用stb_imagephp_write库1.1 安装和包含库1.2 代码解
阅读更多...
C++实现封装的顺序表的操作与实践

C++实现封装的顺序表的操作与实践

《C++实现封装的顺序表的操作与实践》在程序设计中,顺序表是一种常见的线性数据结构,通常用于存储具有固定顺序的元素,与链表不同,顺序表中的元素是连续存储的,因此访问速度较快,但插入和删除操作的效率可能... 目录一、顺序表的基本概念二、顺序表类的设计1. 顺序表类的成员变量2. 构造函数和析构函数三、顺序表
阅读更多...
使用C++实现单链表的操作与实践

使用C++实现单链表的操作与实践

《使用C++实现单链表的操作与实践》在程序设计中,链表是一种常见的数据结构,特别是在动态数据管理、频繁插入和删除元素的场景中,链表相比于数组,具有更高的灵活性和高效性,尤其是在需要频繁修改数据结构的应... 目录一、单链表的基本概念二、单链表类的设计1. 节点的定义2. 链表的类定义三、单链表的操作实现四、
阅读更多...
Go语言利用泛型封装常见的Map操作

Go语言利用泛型封装常见的Map操作

《Go语言利用泛型封装常见的Map操作》Go语言在1.18版本中引入了泛型,这是Go语言发展的一个重要里程碑,它极大地增强了语言的表达能力和灵活性,本文将通过泛型实现封装常见的Map操作,感... 目录什么是泛型泛型解决了什么问题Go泛型基于泛型的常见Map操作代码合集总结什么是泛型泛型是一种编程范式，允
阅读更多...

Copyright China编程 23002807@qq.com

沪ICP备2023033072号-2