diff options
Diffstat (limited to 'src/shared/ServiceWin32.cpp')
| -rw-r--r-- | src/shared/ServiceWin32.cpp | 41 |
1 files changed, 0 insertions, 41 deletions
diff --git a/src/shared/ServiceWin32.cpp b/src/shared/ServiceWin32.cpp index 5c68ba31954..dc374a4ee41 100644 --- a/src/shared/ServiceWin32.cpp +++ b/src/shared/ServiceWin32.cpp @@ -17,15 +17,12 @@ * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ - #ifdef WIN32 - #include "Common.h" #include "Log.h" #include <cstring> #include <windows.h> #include <winsvc.h> - #if !defined(WINADVAPI) #if !defined(_ADVAPI32_) #define WINADVAPI DECLSPEC_IMPORT @@ -33,26 +30,19 @@ #define WINADVAPI #endif #endif - extern int main(int argc, char ** argv); extern char serviceLongName[]; extern char serviceName[]; extern char serviceDescription[]; - extern int m_ServiceStatus; - SERVICE_STATUS serviceStatus; - SERVICE_STATUS_HANDLE serviceStatusHandle = 0; - typedef WINADVAPI BOOL (WINAPI *CSD_T)(SC_HANDLE, DWORD, LPCVOID); - bool WinServiceInstall() { CSD_T ChangeService_Config2; HMODULE advapi32; SC_HANDLE serviceControlManager = OpenSCManager(0, 0, SC_MANAGER_CREATE_SERVICE); - if (serviceControlManager) { char path[_MAX_PATH + 10]; @@ -83,7 +73,6 @@ bool WinServiceInstall() CloseServiceHandle(serviceControlManager); return false; } - ChangeService_Config2 = (CSD_T) GetProcAddress(advapi32, "ChangeServiceConfig2A"); if (!ChangeService_Config2) { @@ -91,14 +80,12 @@ bool WinServiceInstall() CloseServiceHandle(serviceControlManager); return false; } - SERVICE_DESCRIPTION sdBuf; sdBuf.lpDescription = serviceDescription; ChangeService_Config2( service, // handle to service SERVICE_CONFIG_DESCRIPTION, // change: description &sdBuf); // new data - SC_ACTION _action[1]; _action[0].Type = SC_ACTION_RESTART; _action[0].Delay = 10000; @@ -111,20 +98,16 @@ bool WinServiceInstall() service, // handle to service SERVICE_CONFIG_FAILURE_ACTIONS, // information level &sfa); // new data - CloseServiceHandle(service); - } } CloseServiceHandle(serviceControlManager); } return true; } - bool WinServiceUninstall() { SC_HANDLE serviceControlManager = OpenSCManager(0, 0, SC_MANAGER_CONNECT); - if (serviceControlManager) { SC_HANDLE service = OpenService(serviceControlManager, @@ -139,39 +122,32 @@ bool WinServiceUninstall() } CloseServiceHandle(service); } - CloseServiceHandle(serviceControlManager); } return true; } - void WINAPI ServiceControlHandler(DWORD controlCode) { switch (controlCode) { case SERVICE_CONTROL_INTERROGATE: break; - case SERVICE_CONTROL_SHUTDOWN: case SERVICE_CONTROL_STOP: serviceStatus.dwCurrentState = SERVICE_STOP_PENDING; SetServiceStatus(serviceStatusHandle, &serviceStatus); - m_ServiceStatus = 0; return; - case SERVICE_CONTROL_PAUSE: m_ServiceStatus = 2; serviceStatus.dwCurrentState = SERVICE_PAUSED; SetServiceStatus(serviceStatusHandle, &serviceStatus); break; - case SERVICE_CONTROL_CONTINUE: serviceStatus.dwCurrentState = SERVICE_RUNNING; SetServiceStatus(serviceStatusHandle, &serviceStatus); m_ServiceStatus = 1; break; - default: if ( controlCode >= 128 && controlCode <= 255 ) // user defined control code @@ -180,10 +156,8 @@ void WINAPI ServiceControlHandler(DWORD controlCode) // unrecognized control code break; } - SetServiceStatus(serviceStatusHandle, &serviceStatus); } - void WINAPI ServiceMain(DWORD argc, char *argv[]) { // initialise service status @@ -194,56 +168,42 @@ void WINAPI ServiceMain(DWORD argc, char *argv[]) serviceStatus.dwServiceSpecificExitCode = NO_ERROR; serviceStatus.dwCheckPoint = 0; serviceStatus.dwWaitHint = 0; - serviceStatusHandle = RegisterServiceCtrlHandler(serviceName, ServiceControlHandler); - if ( serviceStatusHandle ) { char path[_MAX_PATH + 1]; unsigned int i, last_slash = 0; - GetModuleFileName(0, path, sizeof(path)/sizeof(path[0])); - for (i = 0; i < std::strlen(path); i++) { if (path[i] == '\\') last_slash = i; } - path[last_slash] = 0; - // service is starting serviceStatus.dwCurrentState = SERVICE_START_PENDING; SetServiceStatus(serviceStatusHandle, &serviceStatus); - // do initialisation here SetCurrentDirectory(path); - // running serviceStatus.dwControlsAccepted |= (SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN); serviceStatus.dwCurrentState = SERVICE_RUNNING; SetServiceStatus( serviceStatusHandle, &serviceStatus ); - //////////////////////// // service main cycle // //////////////////////// - m_ServiceStatus = 1; argc = 1; main(argc , argv); - // service was stopped serviceStatus.dwCurrentState = SERVICE_STOP_PENDING; SetServiceStatus(serviceStatusHandle, &serviceStatus); - // do cleanup here - // service is now stopped serviceStatus.dwControlsAccepted &= ~(SERVICE_ACCEPT_STOP | SERVICE_ACCEPT_SHUTDOWN); serviceStatus.dwCurrentState = SERVICE_STOPPED; SetServiceStatus(serviceStatusHandle, &serviceStatus); } } - bool WinServiceRun() { SERVICE_TABLE_ENTRY serviceTable[] = @@ -251,7 +211,6 @@ bool WinServiceRun() { serviceName, ServiceMain }, { 0, 0 } }; - if (!StartServiceCtrlDispatcher(serviceTable)) { sLog.outError("StartService Failed. Error [%u]", ::GetLastError()); |