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 check_admin(char *action) {\r
  31   /* Lifted from MSDN examples */\r
  32   PSID AdministratorsGroup;\r
  33   SID_IDENTIFIER_AUTHORITY NtAuthority = SECURITY_NT_AUTHORITY;\r
  34   BOOL ok = AllocateAndInitializeSid(&NtAuthority, 2, SECURITY_BUILTIN_DOMAIN_RID, DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, &AdministratorsGroup);\r
  35   if (ok) {\r
  36     if (! CheckTokenMembership(0, AdministratorsGroup, &ok)) ok = 0;\r
  37     FreeSid(AdministratorsGroup);\r
  38 \r
  39     if (ok) return 0;\r
  40 \r
  41     fprintf(stderr, "Administator access is needed to %s a service.\n", action);\r
  42     return 1;\r
  43   }\r
  44 \r
  45   /* Can't tell if we are admin or not; later operations may fail */\r
  46   return 0;\r
  47 }\r
  48 \r
  49 int main(int argc, char **argv) {\r
  50   /* Require an argument since users may try to run nssm directly */\r
  51   if (argc == 1) exit(usage(1));\r
  52 \r
  53   /* Elevate */\r
  54   if (str_equiv(argv[1], "install") || str_equiv(argv[1], "remove")) {\r
  55     if (check_admin(argv[1])) exit(100);\r
  56   }\r
  57 \r
  58   /* Valid commands are install or remove */\r
  59   if (str_equiv(argv[1], "install")) {\r
  60     exit(pre_install_service(argc - 2, argv + 2));\r
  61   }\r
  62   if (str_equiv(argv[1], "remove")) {\r
  63     exit(pre_remove_service(argc - 2, argv + 2));\r
  64   }\r
  65   /* Undocumented: "run" is used to actually do service stuff */\r
  66   if (! str_equiv(argv[1], NSSM_RUN)) exit(usage(2));\r
  67 \r
  68   /* Thread local storage for error message buffer */\r
  69   tls_index = TlsAlloc();\r
  70 \r
  71   /* Register messages */\r
  72   create_messages();\r
  73 \r
  74   /* Start service magic */\r
  75   SERVICE_TABLE_ENTRY table[] = { { NSSM, service_main }, { 0, 0 } };\r
  76   if (! StartServiceCtrlDispatcher(table)) {\r
  77     log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_DISPATCHER_FAILED, error_string(GetLastError()), 0);\r
  78     return 100;\r
  79   }\r
  80 \r
  81   /* And nothing more to do */\r
  82   return 0;\r
  83 }\r