发一个自己写的服务程序

看看自己的博客，只能感叹自己确实太懒了，发个自己写的windows 服务程序。

这个程序有以下功能:

1.把自己注册为windows服务;

2.根据配置文件启动多个其他可执行程序，相当于可以吧其他可执行文件改为windows服务;

3.根据windows对服务的操作停止自己启动的可执行程序;

4.接收可执行程序通过标准输出输出的信息，并记录到log文件中;

多了不说，直接贴源码。

 

 

/** /brief 寄主服务主程序 /author whyjld */ // --------------------------------------------------------------------------- #pragma hdrstop #include <tchar.h> #include <windows.h> #include <time.h> #include <cstdio> #include <vector> #include <fstream> #include <string> #include <boost/lexical_cast.hpp> #include <boost/format.hpp> // --------------------------------------------------------------------------- #pragma argsused // --------------------------------------------------------------------------- /** /brief 服务管理的命令 */ struct ServiceCommandParam { /** /brief 初始化命令参数 /param name 命令名称 /param cl 命令行 */ ServiceCommandParam(const std::string& name, const std::string& cl) : Name(name), CommandLine(cl), Process(NULL), Pipe(NULL) { } /** /brief 命令名称 */ std::string Name; /** brief 命令行 */ std::string CommandLine; /** /brief 命令进程句柄 */ HANDLE Process; /** /brief 命令管道 */ HANDLE Pipe; }; // --------------------------------------------------------------------------- /** /brief 服务状态 详细信息请参阅MSDN */ SERVICE_STATUS g_ServiceStatus = { // 这个重要,服务类型,关系到启动类型和能否禁用, // 文件系统服务和引导服务都不能被禁用而且权限很高, // 可以在引导系统的时候自动启动. SERVICE_WIN32_OWN_PROCESS | SERVICE_INTERACTIVE_PROCESS, SERVICE_STOPPED, SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_PAUSE_CONTINUE, 0, 0, 0, 0 }; /** /brief 服务名称 */ TCHAR g_ServiceName[] = TEXT("Viewer Service"); /** /brief 服务状态句柄 */ SERVICE_STATUS_HANDLE g_Status = NULL; /** /brief Log文件名 */ std::string g_LogFileName; /** /brief Log文件 */ std::ofstream g_LogFile; /** /brief 配置文件名 */ std::string g_CfgFileName; /** /brief 命令列表 */ std::vector<ServiceCommandParam> g_Commands; // --------------------------------------------------------------------------- /** /brief 输出纪录信息 /param log 纪录内容 */ void Log(const std::string& log) { time_t ltime; time(&ltime); tm* now = localtime(&ltime); g_LogFile << "[" << (boost::format("%04d-%02d-%02d %02d:%02d:%02d") % (now->tm_year + 1900) % now->tm_mon % now->tm_mday % now->tm_hour % now->tm_min % now->tm_sec) << "]" << log << std::endl; } // --------------------------------------------------------------------------- /** /brief 输出错误信息 /param msg 提示信息 */ void ShowLastError(const std::string& msg) { LPVOID lpMsgBuf; LPVOID lpDisplayBuf; DWORD dw = GetLastError(); FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM, NULL, dw, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR) & lpMsgBuf, 0, NULL); lpDisplayBuf = (LPVOID)LocalAlloc(LMEM_ZEROINIT, (lstrlen((LPCTSTR)lpMsgBuf) + 40)*sizeof(TCHAR)); std::sprintf((char*)lpDisplayBuf, "发生错误 %d: %s", dw, (char*)lpMsgBuf); Log(msg + ":" + (const char*) lpMsgBuf + "(" + boost::lexical_cast<std::string>(dw) + ")"); LocalFree(lpMsgBuf); LocalFree(lpDisplayBuf); } // --------------------------------------------------------------------------- /** /brief 把自己添加到服务管理控制器中 /param name 程序全路径名 */ void RegisteService(const TCHAR* name) { SC_HANDLE SCMgr = OpenSCManager(NULL, NULL, SC_MANAGER_ALL_ACCESS); if (NULL != SCMgr) { SCMgr = CreateService(SCMgr, g_ServiceName, g_ServiceName, SERVICE_START, SERVICE_WIN32_OWN_PROCESS, SERVICE_AUTO_START, SERVICE_ERROR_IGNORE, name, NULL, NULL, NULL, NULL, NULL); } if (NULL == SCMgr) { ShowLastError(TEXT("无法注册服务")); } } // --------------------------------------------------------------------------- /** /brief 关闭所有命令 */ void TerminateCommands(void) { for(std::vector<ServiceCommandParam>::iterator i = g_Commands.begin();i != g_Commands.end();++i) { if(i->Process != NULL) { if (TerminateProcess(i->Process, 0)) { i->Process = NULL; } } } } // --------------------------------------------------------------------------- /** /brief 控制句柄 /param request 要求的动作 */ void WINAPI ControlHandler(DWORD request) { if (NULL == g_Status) { return; } switch(request) { case SERVICE_CONTROL_SHUTDOWN: TerminateCommands(); g_ServiceStatus.dwCurrentState = SERVICE_STOPPED; SetServiceStatus(g_Status, &g_ServiceStatus); break; case SERVICE_CONTROL_STOP: TerminateCommands(); g_ServiceStatus.dwCurrentState = SERVICE_STOPPED; SetServiceStatus(g_Status, &g_ServiceStatus); break; } } // --------------------------------------------------------------------------- /** /brief 服务命令 /param cmd命令 /return 是否装入成功 */ bool ServiceCommand(ServiceCommandParam& cmd) { SECURITY_ATTRIBUTES sa; HANDLE hRead, hWrite; sa.nLength = sizeof(SECURITY_ATTRIBUTES); sa.lpSecurityDescriptor = NULL; sa.bInheritHandle = TRUE; if (!CreatePipe(&hRead, &hWrite, &sa, 0)) { ShowLastError(TEXT("打开命令行管道失败")); return false; } STARTUPINFO si; PROCESS_INFORMATION pi; si.cb = sizeof(STARTUPINFO); GetStartupInfo(&si); si.hStdError = hWrite; si.hStdOutput = hWrite; si.wShowWindow = SW_HIDE; si.dwFlags = STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES; if (!CreateProcessA(NULL, (char*)cmd.CommandLine.c_str(), NULL, NULL, TRUE, NULL, NULL, NULL, &si, π)) { ShowLastError(TEXT("无法装入命令")); return false; } CloseHandle(hWrite); cmd.Process = pi.hProcess; cmd.Pipe = hRead; return true; } // --------------------------------------------------------------------------- /** /brief 纪录命令输出 */ void LogCommandMessage(void) { char buffer[4096]; DWORD readed; DWORD running; do { running = 0; for(std::vector<ServiceCommandParam>::iterator i = g_Commands.begin();i != g_Commands.end();++i) { if(NULL != i->Pipe) { ++running; if(FAILED(ReadFile(i->Pipe, buffer, 4095, &readed, NULL))) { i->Pipe = NULL; Log((boost::format("[%s]已经关闭") % i->Name).str()); } else { if(0 != readed) { buffer[readed] = 0; Log((boost::format("[%s]%s(%d)") % i->Name % buffer % readed).str()); } } } else { Log((boost::format("[%s]初始化失败") % i->Name).str()); } } Sleep(200); } while(running > 0); return ; } // --------------------------------------------------------------------------- /** /brief 装入服务程序 /param filename 配置文件名称 /return 是否装入成功 */ bool LoadConfig(const std::string& filename) { std::ifstream cfgFile(filename.c_str()); if (!cfgFile) { Log(std::string("无法打开配置文件/"") + filename + "/""); return false; } std::string line; g_Commands.clear(); do { std::getline(cfgFile, line, '/n'); std::string::size_type pos = line.find("="); if (std::string::npos != pos) { ServiceCommandParam sc(line.substr(0, pos), line.substr(pos + 1, line.length())); if (0 < sc.CommandLine.length()) { Log(std::string("装入命令[") + sc.Name + "=" + sc.CommandLine + "]"); g_Commands.push_back(sc); } else { Log(std::string("无效配置数据:") + sc.Name + ";" + sc.CommandLine); } } else { Log(std::string("无效配置行:") + line); } } while (!cfgFile.eof()); return true; } // --------------------------------------------------------------------------- /** /brief 服务主程序 /param argc /param argv */ void ServiceMain(int argc, char* argv) { g_Status = RegisterServiceCtrlHandler(g_ServiceName, ControlHandler); g_ServiceStatus.dwCurrentState = SERVICE_START_PENDING; g_ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_SHUTDOWN; SetServiceStatus(g_Status, &g_ServiceStatus); g_LogFile.open(g_LogFileName.c_str(), std::ios::out | std::ios::app); if (!g_LogFile) { Log("无法打开纪录文件，服务启动失败"); g_ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN; g_ServiceStatus.dwCurrentState = SERVICE_STOPPED; SetServiceStatus(g_Status, &g_ServiceStatus); return; } Log("打开纪录文件"); if (!LoadConfig(g_CfgFileName)) { Log("无法读取配置文件文件/"" + g_CfgFileName + "/"，服务启动失败"); g_ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN; g_ServiceStatus.dwCurrentState = SERVICE_STOPPED; SetServiceStatus(g_Status, &g_ServiceStatus); return; } Log("读取命令" + boost::lexical_cast<std::string>(g_Commands.size())); bool allFailed = false; for(std::vector<ServiceCommandParam>::iterator i = g_Commands.begin();i != g_Commands.end();++i) { allFailed |= (!ServiceCommand(*i)); } Log("启动命令"); if(allFailed) { Log("未启动任何命令，服务启动失败"); g_ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN; g_ServiceStatus.dwCurrentState = SERVICE_STOPPED; SetServiceStatus(g_Status, &g_ServiceStatus); return; } Log("命令已启动"); g_ServiceStatus.dwWin32ExitCode = NO_ERROR; g_ServiceStatus.dwCheckPoint = 0; g_ServiceStatus.dwWaitHint = 0; g_ServiceStatus.dwCurrentState = SERVICE_RUNNING; g_ServiceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN; SetServiceStatus(g_Status, &g_ServiceStatus); Log("纪录信息"); LogCommandMessage(); g_LogFile.close(); g_ServiceStatus.dwCurrentState = SERVICE_STOPPED; SetServiceStatus(g_Status, &g_ServiceStatus); } // --------------------------------------------------------------------------- int _tmain(int argc, _TCHAR* argv[]) { std::string name = argv[0]; if(name.find(".exe") != name.length() - 4) { return 1; } name = name.substr(0, name.length() - 4); g_CfgFileName = name + ".cfg"; { time_t ltime; time(&ltime); tm* now = localtime(&ltime); size_t p = name.length() - 1; for(;p > 0;--p) { if('//' == name.c_str()[p]) { name = name.substr(0, p + 1); break; } } if(0 == p) { name = "c://"; } g_LogFileName = name + (boost::format("%04d%02d%02d") % (now->tm_year + 1900) % now->tm_mon % now->tm_mday).str() + ".log"; } SERVICE_TABLE_ENTRY ServiceTable[2]; ServiceTable[0].lpServiceName = g_ServiceName; ServiceTable[0].lpServiceProc = (LPSERVICE_MAIN_FUNCTION)ServiceMain; ServiceTable[1].lpServiceName = NULL; ServiceTable[1].lpServiceProc = NULL; // 启动服务的控制分派机线程 if (StartServiceCtrlDispatcher(ServiceTable) == 0) { RegisteService(argv[0]); } return 0; } // ---------------------------------------------------------------------------  

配置文件是一个文本文件，格式是

名称=命令

名称随意，只要不重复就好，命令就是要启动的可执行程序，可以任意带参数