service.cpp

   1 #include "nssm.h"\r
   2 \r
   3 SERVICE_STATUS service_status;\r
   4 SERVICE_STATUS_HANDLE service_handle;\r
   5 HANDLE wait_handle;\r
   6 HANDLE pid;\r
   7 char exe[MAX_PATH];\r
   8 char flags[MAX_PATH];\r
   9 char dir[MAX_PATH];\r
  10 \r
  11 /* Connect to the service manager */\r
  12 SC_HANDLE open_service_manager() {\r
  13   SC_HANDLE ret = OpenSCManager(0, SERVICES_ACTIVE_DATABASE, SC_MANAGER_ALL_ACCESS);\r
  14   if (! ret) {\r
  15     eventprintf(EVENTLOG_ERROR_TYPE, "Unable to connect to service manager!\nPerhaps you need to be an administrator...");\r
  16     return 0;\r
  17   }\r
  18 \r
  19   return ret;\r
  20 }\r
  21 \r
  22 /* Install the service */\r
  23 int install_service(char *name) {\r
  24 #ifdef GUI\r
  25   /* Show the dialogue box */\r
  26   return nssm_gui(IDD_INSTALL, name);\r
  27 #else\r
  28   fprintf(stderr, "Unimplemented\n");\r
  29   return 1;\r
  30 #endif\r
  31 }\r
  32 \r
  33 /* Remove the service */\r
  34 int remove_service(char *name) {\r
  35 #ifdef GUI\r
  36   return nssm_gui(IDD_REMOVE, name);\r
  37 #else\r
  38   fprintf(stderr, "Unimplemented\n");\r
  39   return 1;\r
  40 #endif\r
  41 }\r
  42 \r
  43 /* Service initialisation */\r
  44 void WINAPI service_main(unsigned long argc, char **argv) {\r
  45   /* Initialise status */\r
  46   ZeroMemory(&service_status, sizeof(service_status));\r
  47   service_status.dwServiceType = SERVICE_WIN32_OWN_PROCESS | SERVICE_INTERACTIVE_PROCESS;\r
  48   service_status.dwCurrentState = SERVICE_RUNNING;\r
  49   service_status.dwControlsAccepted = SERVICE_ACCEPT_SHUTDOWN | SERVICE_ACCEPT_STOP;\r
  50   service_status.dwWin32ExitCode = NO_ERROR;\r
  51   service_status.dwServiceSpecificExitCode = 0;\r
  52   service_status.dwCheckPoint = 0;\r
  53   service_status.dwWaitHint = 1000;\r
  54 \r
  55   /* Signal we AREN'T running the server */\r
  56   pid = 0;\r
  57 \r
  58   /* Get startup parameters */\r
  59   int ret = get_parameters(argv[0], exe, sizeof(exe), flags, sizeof(flags), dir, sizeof(dir));\r
  60   if (ret) {\r
  61     eventprintf(EVENTLOG_ERROR_TYPE, "service_main(): Can't get startup parameters: error %d", ret);\r
  62     return;\r
  63   }\r
  64 \r
  65   /* Register control handler */\r
  66   service_handle = RegisterServiceCtrlHandlerEx(NSSM, service_control_handler, 0);\r
  67   if (! service_handle) {\r
  68     eventprintf(EVENTLOG_ERROR_TYPE, "service_main(): RegisterServiceCtrlHandlerEx() failed: %s", error_string(GetLastError()));\r
  69     return;\r
  70   }\r
  71 \r
  72   monitor_service();\r
  73 }\r
  74 \r
  75 int monitor_service() {\r
  76   /* Set service status to started */\r
  77   int ret = start_service();\r
  78   if (ret) {\r
  79     eventprintf(EVENTLOG_ERROR_TYPE, "Can't start service: error code %d", ret);\r
  80     return ret;\r
  81   }\r
  82   eventprintf(EVENTLOG_INFORMATION_TYPE, "Started process %s %s in %s", exe, flags, dir);\r
  83 \r
  84   /* Monitor service service */\r
  85   if (! RegisterWaitForSingleObject(&wait_handle, pid, end_service, 0, INFINITE, WT_EXECUTEONLYONCE | WT_EXECUTELONGFUNCTION)) {\r
  86     eventprintf(EVENTLOG_WARNING_TYPE, "RegisterWaitForSingleObject() returned %s - service may claim to be still running when %s exits ", error_string(GetLastError()), exe);\r
  87   }\r
  88 \r
  89   return 0;\r
  90 }\r
  91 \r
  92 /* Service control handler */\r
  93 unsigned long WINAPI service_control_handler(unsigned long control, unsigned long event, void *data, void *context) {\r
  94   switch (control) {\r
  95     case SERVICE_CONTROL_SHUTDOWN:\r
  96     case SERVICE_CONTROL_STOP:\r
  97       stop_service(0);\r
  98       return NO_ERROR;\r
  99   }\r
 100 \r
 101   /* Unknown control */\r
 102   return ERROR_CALL_NOT_IMPLEMENTED;\r
 103 }\r
 104 \r
 105 /* Start the service */\r
 106 int start_service() {\r
 107   if (pid) return 0;\r
 108 \r
 109   /* Allocate a STARTUPINFO structure for a new process */\r
 110   STARTUPINFO si;\r
 111   ZeroMemory(&si, sizeof(si));\r
 112   si.cb = sizeof(si);\r
 113 \r
 114   /* Allocate a PROCESSINFO structure for the process */\r
 115   PROCESS_INFORMATION pi;\r
 116   ZeroMemory(&pi, sizeof(pi));\r
 117 \r
 118   /* Launch executable with arguments */\r
 119   char cmd[MAX_PATH];\r
 120   if (_snprintf(cmd, sizeof(cmd), "%s %s", exe, flags) < 0) {\r
 121     eventprintf(EVENTLOG_ERROR_TYPE, "Error constructing command line");\r
 122     return stop_service(2);\r
 123   }\r
 124   if (! CreateProcess(0, cmd, 0, 0, 0, 0, 0, dir, &si, &pi)) {\r
 125     eventprintf(EVENTLOG_ERROR_TYPE, "Can't launch %s.  CreateProcess() returned %s", exe, error_string(GetLastError()));\r
 126     return stop_service(3);\r
 127   }\r
 128   pid = pi.hProcess;\r
 129 \r
 130   /* Signal successful start */\r
 131   service_status.dwCurrentState = SERVICE_RUNNING;\r
 132   SetServiceStatus(service_handle, &service_status);\r
 133 \r
 134   return 0;\r
 135 }\r
 136 \r
 137 /* Stop the service */\r
 138 int stop_service(unsigned long exitcode) {\r
 139   /* Signal we are stopping */\r
 140   service_status.dwCurrentState = SERVICE_STOP_PENDING;\r
 141   SetServiceStatus(service_handle, &service_status);\r
 142 \r
 143   /* Nothing to do if server isn't running */\r
 144   if (pid) {\r
 145     /* Shut down server */\r
 146     TerminateProcess(pid, 0);\r
 147     pid = 0;\r
 148   }\r
 149 \r
 150   /* Signal we stopped */\r
 151   service_status.dwCurrentState = SERVICE_STOPPED;\r
 152   if (exitcode) {\r
 153     service_status.dwWin32ExitCode = ERROR_SERVICE_SPECIFIC_ERROR;\r
 154     service_status.dwServiceSpecificExitCode = exitcode;\r
 155   }\r
 156   else {\r
 157     service_status.dwWin32ExitCode = NO_ERROR;\r
 158     service_status.dwServiceSpecificExitCode = 0;\r
 159   }\r
 160   SetServiceStatus(service_handle, &service_status);\r
 161 \r
 162   return exitcode;\r
 163 }\r
 164 \r
 165 /* Callback function triggered when the server exits */\r
 166 void CALLBACK end_service(void *arg, unsigned char why) {\r
 167   /* Check exit code */\r
 168   unsigned long ret = 0;\r
 169   GetExitCodeProcess(pid, &ret);\r
 170 \r
 171   /* Force an error code if none given, so system can restart this service */\r
 172   /*if (! ret) {\r
 173     eventprintf(EVENTLOG_INFORMATION_TYPE, "Process exited with return code 0 - overriding with return code 111 so the service manager will notice the failure");\r
 174     ret = 111;\r
 175   }\r
 176   else */eventprintf(EVENTLOG_INFORMATION_TYPE, "Process %s exited with return code %u", exe, ret);\r
 177 \r
 178   /* Try to restart the service or return failure code to service manager */\r
 179   pid = 0;\r
 180   while (monitor_service()) {\r
 181     eventprintf(EVENTLOG_INFORMATION_TYPE, "Failed to restart %s - sleeping ", exe, ret);\r
 182     Sleep(30000);\r
 183   }\r
 184 }\r