nssm.cpp

   1 #include "nssm.h"\r
   2 \r
   3 extern unsigned long tls_index;\r
   4 extern bool is_admin;\r
   5 extern imports_t imports;\r
   6 \r
   7 /* String function */\r
   8 int str_equiv(const char *a, const char *b) {\r
   9   int i;\r
  10   for (i = 0; ; i++) {\r
  11     if (tolower(b[i]) != tolower(a[i])) return 0;\r
  12     if (! a[i]) return 1;\r
  13   }\r
  14 }\r
  15 \r
  16 /* How to use me correctly */\r
  17 int usage(int ret) {\r
  18   print_message(stderr, NSSM_MESSAGE_USAGE, NSSM_VERSION, NSSM_DATE);\r
  19   return(ret);\r
  20 }\r
  21 \r
  22 void check_admin() {\r
  23   is_admin = false;\r
  24 \r
  25   /* Lifted from MSDN examples */\r
  26   PSID AdministratorsGroup;\r
  27   SID_IDENTIFIER_AUTHORITY NtAuthority = SECURITY_NT_AUTHORITY;\r
  28   if (! AllocateAndInitializeSid(&NtAuthority, 2, SECURITY_BUILTIN_DOMAIN_RID, DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, &AdministratorsGroup)) return;\r
  29   CheckTokenMembership(0, AdministratorsGroup, /*XXX*/(PBOOL) &is_admin);\r
  30   FreeSid(AdministratorsGroup);\r
  31 }\r
  32 \r
  33 int main(int argc, char **argv) {\r
  34   /* Remember if we are admin */\r
  35   check_admin();\r
  36 \r
  37   /* Elevate */\r
  38   if (argc > 1) {\r
  39     /* Valid commands are install or remove */\r
  40     if (str_equiv(argv[1], "install")) {\r
  41       if (! is_admin) {\r
  42         print_message(stderr, NSSM_MESSAGE_NOT_ADMINISTRATOR_CANNOT_INSTALL);\r
  43         exit(100);\r
  44       }\r
  45       exit(pre_install_service(argc - 2, argv + 2));\r
  46     }\r
  47     if (str_equiv(argv[1], "remove")) {\r
  48       if (! is_admin) {\r
  49         print_message(stderr, NSSM_MESSAGE_NOT_ADMINISTRATOR_CANNOT_REMOVE);\r
  50         exit(100);\r
  51       }\r
  52       exit(pre_remove_service(argc - 2, argv + 2));\r
  53     }\r
  54   }\r
  55 \r
  56   /* Thread local storage for error message buffer */\r
  57   tls_index = TlsAlloc();\r
  58 \r
  59   /* Register messages */\r
  60   if (is_admin) create_messages();\r
  61 \r
  62   /*\r
  63     Optimisation for Windows 2000:\r
  64     When we're run from the command line the StartServiceCtrlDispatcher() call\r
  65     will time out after a few seconds on Windows 2000.  On newer versions the\r
  66     call returns instantly.  Check for stdin first and only try to call the\r
  67     function if there's no input stream found.  Although it's possible that\r
  68     we're running with input redirected it's much more likely that we're\r
  69     actually running as a service.\r
  70     This will save time when running with no arguments from a command prompt.\r
  71   */\r
  72   if (_fileno(stdin) < 0) {\r
  73     /* Set up function pointers. */\r
  74     if (get_imports()) exit(111);\r
  75 \r
  76     /* Start service magic */\r
  77     SERVICE_TABLE_ENTRY table[] = { { NSSM, service_main }, { 0, 0 } };\r
  78     if (! StartServiceCtrlDispatcher(table)) {\r
  79       unsigned long error = GetLastError();\r
  80       /* User probably ran nssm with no argument */\r
  81       if (error == ERROR_FAILED_SERVICE_CONTROLLER_CONNECT) exit(usage(1));\r
  82       log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_DISPATCHER_FAILED, error_string(error), 0);\r
  83       free_imports();\r
  84       exit(100);\r
  85     }\r
  86   }\r
  87   else exit(usage(1));\r
  88 \r
  89   /* And nothing more to do */\r
  90   exit(0);\r
  91 }\r