nssm.cpp

   1 #include "nssm.h"\r
   2 \r
   3 extern unsigned long tls_index;\r
   4 \r
   5 /* String function */\r
   6 int str_equiv(const char *a, const char *b) {\r
   7   int i;\r
   8   for (i = 0; ; i++) {\r
   9     if (tolower(b[i]) != tolower(a[i])) return 0;\r
  10     if (! a[i]) return 1;\r
  11   }\r
  12 }\r
  13 \r
  14 /* How to use me correctly */\r
  15 int usage(int ret) {\r
  16   fprintf(stderr, "NSSM: The non-sucking service manager\n");\r
  17   fprintf(stderr, "Version %s, %s\n", NSSM_VERSION, NSSM_DATE);\r
  18   fprintf(stderr, "Usage: nssm <option> [args]\n\n");\r
  19   fprintf(stderr, "To show service installation GUI:\n\n");\r
  20   fprintf(stderr, "        nssm install [<servicename>]\n\n");\r
  21   fprintf(stderr, "To install a service without confirmation:\n\n");\r
  22   fprintf(stderr, "        nssm install <servicename> <app> [<args>]\n\n");\r
  23   fprintf(stderr, "To show service removal GUI:\n\n");\r
  24   fprintf(stderr, "        nssm remove [<servicename>]\n\n");\r
  25   fprintf(stderr, "To remove a service without confirmation:\n\n");\r
  26   fprintf(stderr, "        nssm remove <servicename> confirm\n");\r
  27   return(ret);\r
  28 }\r
  29 \r
  30 int main(int argc, char **argv) {\r
  31   /* Require an argument since users may try to run nssm directly */\r
  32   if (argc == 1) exit(usage(1));\r
  33 \r
  34   /* Valid commands are install or remove */\r
  35   if (str_equiv(argv[1], "install")) {\r
  36     exit(pre_install_service(argc - 2, argv + 2));\r
  37   }\r
  38   if (str_equiv(argv[1], "remove")) {\r
  39     exit(pre_remove_service(argc - 2, argv + 2));\r
  40   }\r
  41   /* Undocumented: "run" is used to actually do service stuff */\r
  42   if (! str_equiv(argv[1], NSSM_RUN)) exit(usage(2));\r
  43 \r
  44   /* Thread local storage for error message buffer */\r
  45   tls_index = TlsAlloc();\r
  46 \r
  47   /* Register messages */\r
  48   create_messages();\r
  49 \r
  50   /* Start service magic */\r
  51   SERVICE_TABLE_ENTRY table[] = { { NSSM, service_main }, { 0, 0 } };\r
  52   if (! StartServiceCtrlDispatcher(table)) {\r
  53     log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_DISPATCHER_FAILED, error_string(GetLastError()), 0);\r
  54     return 100;\r
  55   }\r
  56 \r
  57   /* And nothing more to do */\r
  58   return 0;\r
  59 }\r