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\n");\r
  17   fprintf(stderr, "To show service installation GUI:\n\n");\r
  18   fprintf(stderr, "        nssm install [<servicename>]\n\n");\r
  19   fprintf(stderr, "To install a service without confirmation:\n\n");\r
  20   fprintf(stderr, "        nssm install <servicename> <app> [<args>]\n\n");\r
  21   fprintf(stderr, "To show service removal GUI:\n\n");\r
  22   fprintf(stderr, "        nssm remove [<servicename>]\n\n");\r
  23   fprintf(stderr, "To remove a service without confirmation:\n\n");\r
  24   fprintf(stderr, "        nssm remove <servicename> confirm\n");\r
  25   return(ret);\r
  26 }\r
  27 \r
  28 int main(int argc, char **argv) {\r
  29   /* Require an argument since users may try to run nssm directly */\r
  30   if (argc == 1) exit(usage(1));\r
  31 \r
  32   /* Valid commands are install or remove */\r
  33   if (str_equiv(argv[1], "install")) {\r
  34     exit(pre_install_service(argc - 2, argv + 2));\r
  35   }\r
  36   if (str_equiv(argv[1], "remove")) {\r
  37     exit(pre_remove_service(argc - 2, argv + 2));\r
  38   }\r
  39   /* Undocumented: "run" is used to actually do service stuff */\r
  40   if (! str_equiv(argv[1], NSSM_RUN)) exit(usage(2));\r
  41 \r
  42   /* Register messages */\r
  43   create_messages();\r
  44 \r
  45   /* Start service magic */\r
  46   SERVICE_TABLE_ENTRY table[] = { { NSSM, service_main }, { 0, 0 } };\r
  47   if (! StartServiceCtrlDispatcher(table)) {\r
  48     char *message = error_string(GetLastError());\r
  49     eventprintf(EVENTLOG_ERROR_TYPE, NSSM_MESSAGE_DEFAULT, "StartServiceCtrlDispatcher() failed: %s", message);\r
  50     if (message) LocalFree(message);\r
  51     return 100;\r
  52   }\r
  53 \r
  54   /* And nothing more to do */\r
  55   return 0;\r
  56 }\r