registry.cpp

   1 #include "nssm.h"\r
   2 \r
   3 int create_parameters(char *service_name, char *exe, char *flags, char *dir) {\r
   4   /* Get registry */\r
   5   char registry[MAX_PATH];\r
   6   if (_snprintf(registry, sizeof(registry), NSSM_REGISTRY, service_name) < 0) {\r
   7     eventprintf(EVENTLOG_ERROR_TYPE, "Out of memory for NSSM_REGISTRY in create_parameters()!");\r
   8     return 1;\r
   9   }\r
  10 \r
  11   /* Try to open the registry */\r
  12   HKEY key;\r
  13   if (RegCreateKeyEx(HKEY_LOCAL_MACHINE, registry, 0, 0, REG_OPTION_NON_VOLATILE, KEY_WRITE, 0, &key, 0) != ERROR_SUCCESS) {\r
  14     eventprintf(EVENTLOG_ERROR_TYPE, "Can't open service registry settings!", NSSM_REGISTRY);\r
  15     return 2;\r
  16   }\r
  17 \r
  18   /* Try to create the parameters */\r
  19   if (RegSetValueEx(key, NSSM_REG_EXE, 0, REG_SZ, (const unsigned char *) exe, strlen(exe) + 1) != ERROR_SUCCESS) {\r
  20     eventprintf(EVENTLOG_ERROR_TYPE, "Can't add registry value %s: %s", NSSM_REG_EXE, error_string(GetLastError()));\r
  21     RegDeleteKey(HKEY_LOCAL_MACHINE, NSSM_REGISTRY);\r
  22     RegCloseKey(key);\r
  23     return 3;\r
  24   }\r
  25   if (RegSetValueEx(key, NSSM_REG_FLAGS, 0, REG_SZ, (const unsigned char *) flags, strlen(flags) + 1) != ERROR_SUCCESS) {\r
  26     eventprintf(EVENTLOG_ERROR_TYPE, "Can't add registry value %s: %s", NSSM_REG_FLAGS, error_string(GetLastError()));\r
  27     RegDeleteKey(HKEY_LOCAL_MACHINE, NSSM_REGISTRY);\r
  28     RegCloseKey(key);\r
  29     return 4;\r
  30   }\r
  31   if (RegSetValueEx(key, NSSM_REG_DIR, 0, REG_SZ, (const unsigned char *) dir, strlen(dir) + 1) != ERROR_SUCCESS) {\r
  32     eventprintf(EVENTLOG_ERROR_TYPE, "Can't add registry value %s: %s", NSSM_REG_DIR, error_string(GetLastError()));\r
  33     RegDeleteKey(HKEY_LOCAL_MACHINE, NSSM_REGISTRY);\r
  34     RegCloseKey(key);\r
  35     return 5;\r
  36   }\r
  37 \r
  38   /* Close registry */\r
  39   RegCloseKey(key);\r
  40 \r
  41   return 0;\r
  42 }\r
  43 \r
  44 int create_exit_action(char *service_name, const char *action_string) {\r
  45   /* Get registry */\r
  46   char registry[MAX_PATH];\r
  47   if (_snprintf(registry, sizeof(registry), NSSM_REGISTRY "\\%s", service_name, NSSM_REG_EXIT) < 0) {\r
  48     eventprintf(EVENTLOG_ERROR_TYPE, "Out of memory for NSSM_REG_EXIT in create_exit_action()!");\r
  49     return 1;\r
  50   }\r
  51 \r
  52   /* Try to open the registry */\r
  53   HKEY key;\r
  54   unsigned long disposition;\r
  55   if (RegCreateKeyEx(HKEY_LOCAL_MACHINE, registry, 0, 0, REG_OPTION_NON_VOLATILE, KEY_WRITE, 0, &key, &disposition) != ERROR_SUCCESS) {\r
  56     eventprintf(EVENTLOG_ERROR_TYPE, "Can't open service exit action registry settings!");\r
  57     return 2;\r
  58   }\r
  59 \r
  60   /* Do nothing if the key already existed */\r
  61   if (disposition == REG_OPENED_EXISTING_KEY) {\r
  62     RegCloseKey(key);\r
  63     return 0;\r
  64   }\r
  65 \r
  66   /* Create the default value */\r
  67   if (RegSetValueEx(key, 0, 0, REG_SZ, (const unsigned char *) action_string, strlen(action_string) + 1) != ERROR_SUCCESS) {\r
  68     eventprintf(EVENTLOG_ERROR_TYPE, "Can't add default registry value %s: %s", NSSM_REG_EXIT, error_string(GetLastError()));\r
  69     RegCloseKey(key);\r
  70     return 3;\r
  71   }\r
  72 \r
  73   /* Close registry */\r
  74   RegCloseKey(key);\r
  75 \r
  76   return 0;\r
  77 }\r
  78 \r
  79 int get_parameters(char *service_name, char *exe, int exelen, char *flags, int flagslen, char *dir, int dirlen) {\r
  80   /* Get registry */\r
  81   char registry[MAX_PATH];\r
  82   if (_snprintf(registry, sizeof(registry), NSSM_REGISTRY, service_name) < 0) {\r
  83     eventprintf(EVENTLOG_ERROR_TYPE, "Out of memory for NSSM_REGISTRY in get_parameters()!");\r
  84     return 1;\r
  85   }\r
  86 \r
  87   /* Try to open the registry */\r
  88   HKEY key;\r
  89   if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, registry, 0, KEY_READ, &key) != ERROR_SUCCESS) {\r
  90     eventprintf(EVENTLOG_ERROR_TYPE, "Can't open service registry settings!", NSSM_REGISTRY);\r
  91     return 2;\r
  92   }\r
  93 \r
  94   unsigned long type = REG_SZ;\r
  95 \r
  96   /* Try to get executable file - MUST succeed */\r
  97   if (RegQueryValueEx(key, NSSM_REG_EXE, 0, &type, (unsigned char *) exe, (unsigned long *) &exelen) != ERROR_SUCCESS) {\r
  98     eventprintf(EVENTLOG_ERROR_TYPE, "Can't get application path (registry value %s): %s", NSSM_REG_EXE, error_string(GetLastError()));\r
  99     RegCloseKey(key);\r
 100     return 3;\r
 101   }\r
 102 \r
 103   /* Try to get flags - may fail */\r
 104   if (RegQueryValueEx(key, NSSM_REG_FLAGS, 0, &type, (unsigned char *) flags, (unsigned long *) &flagslen) != ERROR_SUCCESS) {\r
 105     eventprintf(EVENTLOG_WARNING_TYPE, "Can't get application flags (registry value %s): %s", NSSM_REG_FLAGS, error_string(GetLastError()));\r
 106     RegCloseKey(key);\r
 107     return 4;\r
 108   }\r
 109 \r
 110   /* Try to get startup directory - may fail */\r
 111   if (RegQueryValueEx(key, NSSM_REG_DIR, 0, &type, (unsigned char *) dir, (unsigned long *) &dirlen) != ERROR_SUCCESS) {\r
 112     eventprintf(EVENTLOG_WARNING_TYPE, "Can't get application startup directory (registry value %s): %s", NSSM_REG_DIR, error_string(GetLastError()));\r
 113     RegCloseKey(key);\r
 114     return 5;\r
 115   }\r
 116 \r
 117   /* Close registry */\r
 118   RegCloseKey(key);\r
 119 \r
 120   return 0;\r
 121 }\r
 122 \r
 123 int get_exit_action(char *service_name, unsigned long *ret, unsigned char *action) {\r
 124   /* Get registry */\r
 125   char registry[MAX_PATH];\r
 126   if (_snprintf(registry, sizeof(registry), NSSM_REGISTRY "\\%s", service_name, NSSM_REG_EXIT) < 0) {\r
 127     eventprintf(EVENTLOG_ERROR_TYPE, "Out of memory for NSSM_REG_EXIT in get_exit_action()!");\r
 128     return 1;\r
 129   }\r
 130 \r
 131   /* Try to open the registry */\r
 132   HKEY key;\r
 133   long error = RegOpenKeyEx(HKEY_LOCAL_MACHINE, registry, 0, KEY_READ, &key);\r
 134   if (error != ERROR_SUCCESS && error != ERROR_FILE_NOT_FOUND) {\r
 135     eventprintf(EVENTLOG_ERROR_TYPE, "Can't open registry %s!", registry);\r
 136     return 2;\r
 137   }\r
 138 \r
 139   unsigned long type = REG_SZ;\r
 140   unsigned long action_len = ACTION_LEN;\r
 141 \r
 142   char code[64];\r
 143   if (! ret) code[0] = '\0';\r
 144   else if (_snprintf(code, sizeof(code), "%lu", *ret) < 0) {\r
 145     RegCloseKey(key);\r
 146     return get_exit_action(service_name, 0, action);\r
 147   }\r
 148   if (RegQueryValueEx(key, code, 0, &type, action, &action_len) != ERROR_SUCCESS) {\r
 149     RegCloseKey(key);\r
 150     /* Try again with * as the key if an exit code was defined */\r
 151     if (ret) return get_exit_action(service_name, 0, action);\r
 152     return 0;\r
 153   }\r
 154 \r
 155   /* Close registry */\r
 156   RegCloseKey(key);\r
 157 \r
 158   return 0;\r
 159 }\r