命令行安装为Windows Service

 前阵子在Win2003 x64安装OpenLDAP碰到问题, 32平台可以安装为服务, 到64位就不出错了.

安装为了windows service图个方便, 一下又找不到什么好的工具, 只好自己动手写个程序把一些命令行安装为后台程序.

 

BTW，安装VS2005/2008的时候记得安装支持x64的编译组件. VCKBASE的例子有点不好使, 用MSDN的例子就没问题。

sample.h

 

// The following are message definitions. // // Values are 32 bit values laid out as follows: // // 3 3 2 2 2 2 2 2 2 2 2 2 1 1 1 1 1 1 1 1 1 1 // 1 0 9 8 7 6 5 4 3 2 1 0 9 8 7 6 5 4 3 2 1 0 9 8 7 6 5 4 3 2 1 0 // +---+-+-+-----------------------+-------------------------------+ // |Sev|C|R| Facility | Code | // +---+-+-+-----------------------+-------------------------------+ // // where // // Sev - is the severity code // // 00 - Success // 01 - Informational // 10 - Warning // 11 - Error // // C - is the Customer code flag // // R - is a reserved bit // // Facility - is the facility code // // Code - is the facility's status code // // // Define the facility codes // #define FACILITY_SYSTEM 0x0 #define FACILITY_STUBS 0x3 #define FACILITY_RUNTIME 0x2 #define FACILITY_IO_ERROR_CODE 0x4 // // Define the severity codes // #define STATUS_SEVERITY_WARNING 0x2 #define STATUS_SEVERITY_SUCCESS 0x0 #define STATUS_SEVERITY_INFORMATIONAL 0x1 #define STATUS_SEVERITY_ERROR 0x3 // // MessageId: SVC_ERROR // // MessageText: // // An error has occurred (%2). // #define SVC_ERROR ((DWORD)0xC0020001L) // A message file must end with a period on its own line // followed by a blank line.

 

 

// ServiceTest3.cpp : 定义控制台应用程序的入口点。 // #include "stdafx.h" #include <windows.h> #include <tchar.h> #include <strsafe.h> #include "sample.h" #pragma comment(lib, "advapi32.lib") //#define SVCNAME TEXT("OpenLDAP-test") SERVICE_STATUS gSvcStatus; SERVICE_STATUS_HANDLE gSvcStatusHandle; HANDLE ghSvcStopEvent = NULL; VOID SvcInstall(void); VOID WINAPI SvcCtrlHandler( DWORD ); VOID WINAPI SvcMain( DWORD, LPTSTR * ); VOID ReportSvcStatus( DWORD, DWORD, DWORD ); VOID SvcInit( DWORD, LPTSTR * ); VOID SvcReportEvent( LPTSTR ); VOID CloseProcess(); VOID LoadConfig(); //Current working diretory TCHAR szCurrentDir[MAX_PATH]; //The command line configured in ws.conf #define MAX_COMMAND_LINE 1024 TCHAR szCommandLine[MAX_COMMAND_LINE]; //Service Name TCHAR szServiceName[MAX_PATH]; //Process PROCESS_INFORMATION processInfo; STARTUPINFO siStartInfo; // // Purpose: // Entry point for the process // // Parameters: // None // // Return value: // None // void __cdecl _tmain(int argc, TCHAR *argv[]) { // If command-line parameter is "install", install the service. // Otherwise, the service is probably being started by the SCM. LoadConfig(); if( lstrcmpi( argv[1], TEXT("install")) == 0 ) { SvcInstall(); return; } // TO_DO: Add any additional services for the process to this table. SERVICE_TABLE_ENTRY DispatchTable[] = { { szServiceName, (LPSERVICE_MAIN_FUNCTION) SvcMain }, { NULL, NULL } }; // This call returns when the service has stopped. // The process should simply terminate when the call returns. if (!StartServiceCtrlDispatcher( DispatchTable )) { SvcReportEvent(TEXT("StartServiceCtrlDispatcher")); } } VOID LoadConfig() { TCHAR szExeFilePath[MAX_PATH]; DWORD size = ::GetModuleFileName(NULL, szExeFilePath, MAX_PATH); szExeFilePath[size] = '/0'; for (int i=(size-1); i >=0; --i) { if (szExeFilePath[i] == '//' || szExeFilePath[i] == '/') { szExeFilePath[i] = '/0'; break; } } ::lstrcpy(szCurrentDir, szExeFilePath); //Retrieve ws.conf content. TCHAR szConfFilePath[MAX_PATH]; ::lstrcpy(szConfFilePath, szCurrentDir); ::lstrcat(szConfFilePath, TEXT("//ws.conf")); ::GetPrivateProfileString(TEXT("CONF"), TEXT("COMMAND_LINE"), NULL, szCommandLine, MAX_COMMAND_LINE, szConfFilePath); ::GetPrivateProfileString(TEXT("CONF"), TEXT("SERVICE_NAME"), NULL, szServiceName, MAX_PATH, szConfFilePath); } // // Purpose: // Installs a service in the SCM database // // Parameters: // None // // Return value: // None // VOID SvcInstall() { SC_HANDLE schSCManager; SC_HANDLE schService; TCHAR szPath[MAX_PATH]; if( !GetModuleFileName( NULL, szPath, MAX_PATH ) ) { printf("Cannot install service (%d)/n", GetLastError()); return; } // Get a handle to the SCM database. schSCManager = OpenSCManager( NULL, // local computer NULL, // ServicesActive database SC_MANAGER_ALL_ACCESS); // full access rights if (NULL == schSCManager) { printf("OpenSCManager failed (%d)/n", GetLastError()); return; } // Create the service schService = CreateService( schSCManager, // SCM database szServiceName, // name of service szServiceName, // service name to display SERVICE_ALL_ACCESS, // desired access SERVICE_WIN32_OWN_PROCESS, // service type SERVICE_DEMAND_START, // start type SERVICE_ERROR_NORMAL, // error control type szPath, // path to service's binary NULL, // no load ordering group NULL, // no tag identifier NULL, // no dependencies NULL, // LocalSystem account NULL); // no password if (schService == NULL) { printf("CreateService failed (%d)/n", GetLastError()); CloseServiceHandle(schSCManager); return; } else printf("Service installed successfully/n"); CloseServiceHandle(schService); CloseServiceHandle(schSCManager); } // // Purpose: // Entry point for the service // // Parameters: // dwArgc - Number of arguments in the lpszArgv array // lpszArgv - Array of strings. The first string is the name of // the service and subsequent strings are passed by the process // that called the StartService function to start the service. // // Return value: // None. // VOID WINAPI SvcMain( DWORD dwArgc, LPTSTR *lpszArgv ) { // Register the handler function for the service gSvcStatusHandle = RegisterServiceCtrlHandler( szServiceName, SvcCtrlHandler); if( !gSvcStatusHandle ) { SvcReportEvent(TEXT("RegisterServiceCtrlHandler")); return; } // These SERVICE_STATUS members remain as set here gSvcStatus.dwServiceType = SERVICE_WIN32_OWN_PROCESS; gSvcStatus.dwServiceSpecificExitCode = 0; // Report initial status to the SCM ReportSvcStatus( SERVICE_START_PENDING, NO_ERROR, 3000 ); // Perform service-specific initialization and work. SvcInit( dwArgc, lpszArgv ); } // // Purpose: // The service code // // Parameters: // dwArgc - Number of arguments in the lpszArgv array // lpszArgv - Array of strings. The first string is the name of // the service and subsequent strings are passed by the process // that called the StartService function to start the service. // // Return value: // None // VOID SvcInit( DWORD dwArgc, LPTSTR *lpszArgv) { /* TCHAR szExeFilePath[MAX_PATH]; DWORD size = ::GetModuleFileName(NULL, szExeFilePath, MAX_PATH); szExeFilePath[size] = '/0'; for (int i=(size-1); i >=0; --i) { if (szExeFilePath[i] == '//' || szExeFilePath[i] == '/') { szExeFilePath[i] = '/0'; break; } } ::lstrcpy(szCurrentDir, szExeFilePath); //Retrieve ws.conf content. TCHAR szConfFilePath[MAX_PATH]; ::lstrcpy(szConfFilePath, szCurrentDir); ::lstrcat(szConfFilePath, TEXT("//ws.conf")); ::GetPrivateProfileString(TEXT("CONF"), TEXT("COMMAND_LINE"), NULL, szCommandLine, MAX_COMMAND_LINE, szConfFilePath); ::GetPrivateProfileString(TEXT("CONF"), TEXT("SERVICE_NAME"), NULL, szServiceName, MAX_PATH, szConfFilePath); */ /// ::ZeroMemory(&siStartInfo, sizeof(siStartInfo)); siStartInfo.cb = sizeof(siStartInfo); BOOL processRet = CreateProcess( NULL, _tcsdup(szCommandLine), NULL, // process security attributes NULL, // primary thread security attributes 0, // handles are inherited 0, // creation flags NULL, // use parent's environment szCurrentDir, // use parent's current directory &siStartInfo, // STARTUPINFO pointer &processInfo); // receives PROCESS_INFORMATION if (!processRet) { MessageBox(NULL, TEXT("Failed to create process to execute the command line."), TEXT("Failed to create process"), MB_OK); return; } // TO_DO: Declare and set any required variables. // Be sure to periodically call ReportSvcStatus() with // SERVICE_START_PENDING. If initialization fails, call // ReportSvcStatus with SERVICE_STOPPED. // Create an event. The control handler function, SvcCtrlHandler, // signals this event when it receives the stop control code. ghSvcStopEvent = CreateEvent( NULL, // default security attributes TRUE, // manual reset event FALSE, // not signaled NULL); // no name if ( ghSvcStopEvent == NULL) { ReportSvcStatus( SERVICE_STOPPED, NO_ERROR, 0 ); return; } // Report running status when initialization is complete. ReportSvcStatus( SERVICE_RUNNING, NO_ERROR, 0 ); // TO_DO: Perform work until service stops. while(1) { // Check whether to stop the service. WaitForSingleObject(ghSvcStopEvent, INFINITE); ReportSvcStatus( SERVICE_STOPPED, NO_ERROR, 0 ); return; } } // // Purpose: // Sets the current service status and reports it to the SCM. // // Parameters: // dwCurrentState - The current state (see SERVICE_STATUS) // dwWin32ExitCode - The system error code // dwWaitHint - Estimated time for pending operation, // in milliseconds // // Return value: // None // VOID ReportSvcStatus( DWORD dwCurrentState, DWORD dwWin32ExitCode, DWORD dwWaitHint) { static DWORD dwCheckPoint = 1; // Fill in the SERVICE_STATUS structure. gSvcStatus.dwCurrentState = dwCurrentState; gSvcStatus.dwWin32ExitCode = dwWin32ExitCode; gSvcStatus.dwWaitHint = dwWaitHint; if (dwCurrentState == SERVICE_START_PENDING) gSvcStatus.dwControlsAccepted = 0; else gSvcStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP; if ( (dwCurrentState == SERVICE_RUNNING) || (dwCurrentState == SERVICE_STOPPED) ) gSvcStatus.dwCheckPoint = 0; else gSvcStatus.dwCheckPoint = dwCheckPoint++; // Report the status of the service to the SCM. SetServiceStatus( gSvcStatusHandle, &gSvcStatus ); } // // Purpose: // Called by SCM whenever a control code is sent to the service // using the ControlService function. // // Parameters: // dwCtrl - control code // // Return value: // None // VOID WINAPI SvcCtrlHandler( DWORD dwCtrl ) { // Handle the requested control code. switch(dwCtrl) { case SERVICE_CONTROL_STOP: ReportSvcStatus(SERVICE_STOP_PENDING, NO_ERROR, 0); // Signal the service to stop. CloseProcess(); //SetEvent(ghSvcStopEvent); ReportSvcStatus(gSvcStatus.dwCurrentState, NO_ERROR, 0); return; case SERVICE_CONTROL_INTERROGATE: break; default: break; } } VOID CloseProcess() { //FIXME, function TerminateXXX will lead to memory leak sometimes //Check whether we should post message to thread first. if (processInfo.hProcess != NULL) { ::TerminateProcess(processInfo.hProcess, 0); ::CloseHandle(processInfo.hProcess); } if (processInfo.hThread != NULL) { ::TerminateThread(processInfo.hThread, 0); ::CloseHandle(processInfo.hThread); } SetEvent(ghSvcStopEvent); } // // Purpose: // Logs messages to the event log // // Parameters: // szFunction - name of function that failed // // Return value: // None // // Remarks: // The service must have an entry in the Application event log. // VOID SvcReportEvent(LPTSTR szFunction) { HANDLE hEventSource; LPCTSTR lpszStrings[2]; TCHAR Buffer[80]; hEventSource = RegisterEventSource(NULL, szServiceName); if( NULL != hEventSource ) { StringCchPrintf(Buffer, 80, TEXT("%s failed with %d"), szFunction, GetLastError()); lpszStrings[0] = szServiceName; lpszStrings[1] = Buffer; ReportEvent(hEventSource, // event log handle EVENTLOG_ERROR_TYPE, // event type 0, // event category SVC_ERROR, // event identifier NULL, // no security identifier 2, // size of lpszStrings array 0, // no binary data lpszStrings, // array of strings NULL); // no binary data DeregisterEventSource(hEventSource); } }

 

选择64位平台解决方案编译下, 出来了,把ws.conf

[CONF]
COMMAND_LINE= ""D:/Program Files/openldap-2.4/libexec/slapd.exe" -d 1 -f "D:/Program Files/openldap-2.4/libexec/slapd.conf""
SERVICE_NAME=OpenLDAP-test2

 

ServerTest3.exe install