Windows以后台服务的形式启动程序

////////////////////////////////////////////////////////////////////// // NT Service Stub Code (For XYROOT ) ////////////////////////////////////////////////////////////////////// #include <stdio.h> #include <windows.h> #include <winbase.h> #include <winsvc.h> #include <process.h> const int nBufferSize = 500; char pServiceName[nBufferSize+1]; char pExeFile[nBufferSize+1]; char pInitFile[nBufferSize+1]; char pLogFile[nBufferSize+1]; int nProcCount = 0; PROCESS_INFORMATION* pProcInfo = 0; SERVICE_STATUS serviceStatus; SERVICE_STATUS_HANDLE hServiceStatusHandle; VOID WINAPI XYNTServiceMain( DWORD dwArgc, LPTSTR *lpszArgv ); VOID WINAPI XYNTServiceHandler( DWORD fdwControl ); CRITICAL_SECTION myCS; void WriteLog(char* pFile, char* pMsg) { ::EnterCriticalSection(&myCS); try { FILE* pLog = fopen(pFile,"a"); fprintf(pLog,pMsg); fclose(pLog); } catch(...) {} ::LeaveCriticalSection(&myCS); } ////////////////////////////////////////////////////////////////////// // // Configuration Data and Tables // SERVICE_TABLE_ENTRY DispatchTable[] = { {pServiceName, XYNTServiceMain}, {NULL, NULL} }; // helper functions BOOL StartProcess(int nIndex) { STARTUPINFO startUpInfo = { sizeof(STARTUPINFO),NULL,"",NULL,0,0,0,0,0,0,0,STARTF_USESHOWWINDOW,0,0,NULL,0,0,0}; char pItem[nBufferSize+1]; sprintf(pItem,"Process%d\0",nIndex); char pCommandLine[nBufferSize+1]; GetPrivateProfileString(pItem,"CommandLine","",pCommandLine,nBufferSize,pInitFile); char pUserInterface[nBufferSize+1]; GetPrivateProfileString(pItem,"UserInterface","N",pUserInterface,nBufferSize,pInitFile); BOOL bUserInterface = (pUserInterface[0]=='y'||pUserInterface[0]=='Y'||pUserInterface[0]=='1')?TRUE:FALSE; if(bUserInterface) { startUpInfo.wShowWindow = SW_SHOW; startUpInfo.lpDesktop = NULL; } else { startUpInfo.wShowWindow = SW_HIDE; startUpInfo.lpDesktop = ""; } char pWorkingDir[nBufferSize+1]; GetPrivateProfileString(pItem,"WorkingDir","",pWorkingDir,nBufferSize,pInitFile); if(CreateProcess(NULL,pCommandLine,NULL,NULL,TRUE,NORMAL_PRIORITY_CLASS,NULL,strlen(pWorkingDir)==0?NULL:pWorkingDir,&startUpInfo,&pProcInfo[nIndex])) { char pPause[nBufferSize+1]; GetPrivateProfileString(pItem,"PauseStart","100",pPause,nBufferSize,pInitFile); Sleep(atoi(pPause)); return TRUE; } else { long nError = GetLastError(); char pTemp[121]; sprintf(pTemp,"Failed to start program '%s', error code = %d\n", pCommandLine, nError); WriteLog(pLogFile, pTemp); return FALSE; } } void EndProcess(int nIndex) { char pItem[nBufferSize+1]; sprintf(pItem,"Process%d\0",nIndex); char pPause[nBufferSize+1]; GetPrivateProfileString(pItem,"PauseEnd","100",pPause,nBufferSize,pInitFile); int nPauseEnd = atoi(pPause); if(nIndex>=0&&nIndex<nProcCount) { if(pProcInfo[nIndex].hProcess) { if(nPauseEnd>0) { PostThreadMessage(pProcInfo[nIndex].dwThreadId,WM_QUIT,0,0); Sleep(nPauseEnd); } TerminateProcess(pProcInfo[nIndex].hProcess,0); } } } BOOL BounceProcess(char* pName, int nIndex) { SC_HANDLE schSCManager = OpenSCManager( NULL, NULL, SC_MANAGER_ALL_ACCESS); if (schSCManager==0) { long nError = GetLastError(); char pTemp[121]; sprintf(pTemp, "OpenSCManager failed, error code = %d\n", nError); WriteLog(pLogFile, pTemp); } else { SC_HANDLE schService = OpenService( schSCManager, pName, SERVICE_ALL_ACCESS); if (schService==0) { long nError = GetLastError(); char pTemp[121]; sprintf(pTemp, "OpenService failed, error code = %d\n", nError); WriteLog(pLogFile, pTemp); } else { SERVICE_STATUS status; if(nIndex>=0&&nIndex<128) { if(ControlService(schService,(nIndex|0x80),&status)) { CloseServiceHandle(schService); CloseServiceHandle(schSCManager); return TRUE; } long nError = GetLastError(); char pTemp[121]; sprintf(pTemp, "ControlService failed, error code = %d\n", nError); WriteLog(pLogFile, pTemp); } else { char pTemp[121]; sprintf(pTemp, "Invalid argument to BounceProcess: %d\n", nIndex); WriteLog(pLogFile, pTemp); } CloseServiceHandle(schService); } CloseServiceHandle(schSCManager); } return FALSE; } BOOL KillService(char* pName) { SC_HANDLE schSCManager = OpenSCManager( NULL, NULL, SC_MANAGER_ALL_ACCESS); if (schSCManager==0) { long nError = GetLastError(); char pTemp[121]; sprintf(pTemp, "OpenSCManager failed, error code = %d\n", nError); WriteLog(pLogFile, pTemp); } else { SC_HANDLE schService = OpenService( schSCManager, pName, SERVICE_ALL_ACCESS); if (schService==0) { long nError = GetLastError(); char pTemp[121]; sprintf(pTemp, "OpenService failed, error code = %d\n", nError); WriteLog(pLogFile, pTemp); } else { SERVICE_STATUS status; if(ControlService(schService,SERVICE_CONTROL_STOP,&status)) { CloseServiceHandle(schService); CloseServiceHandle(schSCManager); return TRUE; } else { long nError = GetLastError(); char pTemp[121]; sprintf(pTemp, "ControlService failed, error code = %d\n", nError); WriteLog(pLogFile, pTemp); } CloseServiceHandle(schService); } CloseServiceHandle(schSCManager); } return FALSE; } BOOL RunService(char* pName, int nArg, char** pArg) { SC_HANDLE schSCManager = OpenSCManager( NULL, NULL, SC_MANAGER_ALL_ACCESS); if (schSCManager==0) { long nError = GetLastError(); char pTemp[121]; sprintf(pTemp, "OpenSCManager failed, error code = %d\n", nError); WriteLog(pLogFile, pTemp); } else { SC_HANDLE schService = OpenService( schSCManager, pName, SERVICE_ALL_ACCESS); if (schService==0) { long nError = GetLastError(); char pTemp[121]; sprintf(pTemp, "OpenService failed, error code = %d\n", nError); WriteLog(pLogFile, pTemp); } else { if(StartService(schService,nArg,(const char**)pArg)) { CloseServiceHandle(schService); CloseServiceHandle(schSCManager); return TRUE; } else { long nError = GetLastError(); char pTemp[121]; sprintf(pTemp, "StartService failed, error code = %d\n", nError); WriteLog(pLogFile, pTemp); } CloseServiceHandle(schService); } CloseServiceHandle(schSCManager); } return FALSE; } ////////////////////////////////////////////////////////////////////// // // This routine gets used to start your service // VOID WINAPI XYNTServiceMain( DWORD dwArgc, LPTSTR *lpszArgv ) { DWORD status = 0; DWORD specificError = 0xfffffff; serviceStatus.dwServiceType = SERVICE_WIN32; serviceStatus.dwCurrentState = SERVICE_START_PENDING; serviceStatus.dwControlsAccepted = SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN | SERVICE_ACCEPT_PAUSE_CONTINUE; serviceStatus.dwWin32ExitCode = 0; serviceStatus.dwServiceSpecificExitCode = 0; serviceStatus.dwCheckPoint = 0; serviceStatus.dwWaitHint = 0; hServiceStatusHandle = RegisterServiceCtrlHandler(pServiceName, XYNTServiceHandler); if (hServiceStatusHandle==0) { long nError = GetLastError(); char pTemp[121]; sprintf(pTemp, "RegisterServiceCtrlHandler failed, error code = %d\n", nError); WriteLog(pLogFile, pTemp); return; } // Handle error condition status = GetLastError(); if (status!=NO_ERROR) { serviceStatus.dwCurrentState = SERVICE_STOPPED; serviceStatus.dwCheckPoint = 0; serviceStatus.dwWaitHint = 0; serviceStatus.dwWin32ExitCode = status; serviceStatus.dwServiceSpecificExitCode = specificError; SetServiceStatus(hServiceStatusHandle