X-Git-Url: http://git.iain.cx/?a=blobdiff_plain;ds=sidebyside;f=service.cpp;h=164fbc0c2fddab5238e7f2f26baeb40b81b04ed2;hb=2a259b54caa08ef6d4dbc99f6fe7ec97837c25e8;hp=b3b0c628559eb17907539aac57a5e9292f0a6dd5;hpb=1f0b03b38f7d76814d1c7c627f64462362100223;p=nssm.git diff --git a/service.cpp b/service.cpp index b3b0c62..164fbc0 100644 --- a/service.cpp +++ b/service.cpp @@ -653,8 +653,6 @@ int get_service_description(const TCHAR *service_name, SC_HANDLE service_handle, print_message(stderr, NSSM_MESSAGE_QUERYSERVICECONFIG2_FAILED, service_name, _T("SERVICE_CONFIG_DESCRIPTION"), error_string(error)); return 4; } - - return 0; } int get_service_startup(const TCHAR *service_name, SC_HANDLE service_handle, const QUERY_SERVICE_CONFIG *qsc, unsigned long *startup) { @@ -762,7 +760,7 @@ void cleanup_nssm_service(nssm_service_t *service) { if (! service) return; if (service->username) HeapFree(GetProcessHeap(), 0, service->username); if (service->password) { - SecureZeroMemory(service->password, service->passwordlen); + SecureZeroMemory(service->password, service->passwordlen * sizeof(TCHAR)); HeapFree(GetProcessHeap(), 0, service->password); } if (service->dependencies) HeapFree(GetProcessHeap(), 0, service->dependencies); @@ -1723,15 +1721,7 @@ int start_service(nssm_service_t *service) { service->status.dwControlsAccepted = SERVICE_ACCEPT_POWEREVENT | SERVICE_ACCEPT_SHUTDOWN | SERVICE_ACCEPT_STOP; SetServiceStatus(service->status_handle, &service->status); - /* Pre-start hook. */ unsigned long control = NSSM_SERVICE_CONTROL_START; - if (nssm_hook(&hook_threads, service, NSSM_HOOK_EVENT_START, NSSM_HOOK_ACTION_PRE, &control, NSSM_SERVICE_STATUS_DEADLINE, false) == NSSM_HOOK_STATUS_ABORT) { - TCHAR code[16]; - _sntprintf_s(code, _countof(code), _TRUNCATE, _T("%lu"), NSSM_HOOK_STATUS_ABORT); - log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_PRESTART_HOOK_ABORT, NSSM_HOOK_EVENT_START, NSSM_HOOK_ACTION_PRE, service->name, code, 0); - unset_service_environment(service); - return stop_service(service, 5, true, true); - } /* Did another thread receive a stop control? */ if (service->allow_restart) { @@ -1744,6 +1734,15 @@ int start_service(nssm_service_t *service) { return stop_service(service, 4, true, true); } + /* Pre-start hook. May need I/O to have been redirected already. */ + if (nssm_hook(&hook_threads, service, NSSM_HOOK_EVENT_START, NSSM_HOOK_ACTION_PRE, &control, NSSM_SERVICE_STATUS_DEADLINE, false) == NSSM_HOOK_STATUS_ABORT) { + TCHAR code[16]; + _sntprintf_s(code, _countof(code), _TRUNCATE, _T("%lu"), NSSM_HOOK_STATUS_ABORT); + log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_PRESTART_HOOK_ABORT, NSSM_HOOK_EVENT_START, NSSM_HOOK_ACTION_PRE, service->name, code, 0); + unset_service_environment(service); + return stop_service(service, 5, true, true); + } + /* The pre-start hook will have cleaned the environment. */ set_service_environment(service); @@ -1982,7 +1981,6 @@ void CALLBACK end_service(void *arg, unsigned char why) { wait_for_hooks(service, false); free_imports(); exit(exitcode); - break; } }