發一個自己寫的服務程序

看看自己的博客，只能感嘆自己確實太懶了，發個自己寫的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; } // ---------------------------------------------------------------------------  

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

名稱=命令

名稱隨意，只要不重複就好，命令就是要啓動的可執行程序，可以任意帶參數