nssm.cpp

   1 #include "nssm.h"\r
   2 \r
   3 /* String function */\r
   4 int str_equiv(const char *a, const char *b) {\r
   5   int i;\r
   6   for (i = 0; ; i++) {\r
   7     if (tolower(b[i]) != tolower(a[i])) return 0;\r
   8     if (! a[i]) return 1;\r
   9   }\r
  10 }\r
  11 \r
  12 /* How to use me correctly */\r
  13 int usage(int ret) {\r
  14   fprintf(stderr, "NSSM: The non-sucking service manager\n");\r
  15   fprintf(stderr, "Version %s, %s\n", NSSM_VERSION, NSSM_DATE);\r
  16   fprintf(stderr, "Usage: nssm option [args]\n");\r
  17   fprintf(stderr, "To install a service: nssm install [servicename]\n");\r
  18   fprintf(stderr, "To remove a service: nssm remove [servicename]\n");\r
  19   exit(ret);\r
  20 }\r
  21 \r
  22 int main(int argc, char **argv) {\r
  23   /* Require an argument since users may try to run nssm directly */\r
  24   if (argc == 1) exit(usage(1));\r
  25 \r
  26   /* Valid commands are install or remove */\r
  27   if (str_equiv(argv[1], "install")) exit(install_service(argv[2]));\r
  28   if (str_equiv(argv[1], "remove")) exit(remove_service(argv[2]));\r
  29   /* Undocumented: "run" is used to actually do service stuff */\r
  30   if (! str_equiv(argv[1], NSSM_RUN)) exit(usage(2));\r
  31 \r
  32   /* Start service magic */\r
  33   SERVICE_TABLE_ENTRY table[] = { { NSSM, service_main }, { 0, 0 } };\r
  34   if (! StartServiceCtrlDispatcher(table)) {\r
  35     char *message = error_string(GetLastError());\r
  36     eventprintf(EVENTLOG_ERROR_TYPE, "StartServiceCtrlDispatcher() failed: %s", message);\r
  37     if (message) LocalFree(message);\r
  38     return 100;\r
  39   }\r
  40 \r
  41   /* And nothing more to do */\r
  42   return 0;\r
  43 }\r