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 get_parameters(char *service_name, char *exe, int exelen, char *flags, int flagslen, char *dir, int dirlen) {\r
  45   /* Get registry */\r
  46   char registry[MAX_PATH];\r
  47   if (_snprintf(registry, sizeof(registry), NSSM_REGISTRY, service_name) < 0) {\r
  48     eventprintf(EVENTLOG_ERROR_TYPE, "Out of memory for NSSM_REGISTRY in get_parameters()!");\r
  49     return 1;\r
  50   }\r
  51 \r
  52   /* Try to open the registry */\r
  53   HKEY key;\r
  54   if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, registry, 0, KEY_READ, &key) != ERROR_SUCCESS) {\r
  55     eventprintf(EVENTLOG_ERROR_TYPE, "Can't open service registry settings!", NSSM_REGISTRY);\r
  56     return 2;\r
  57   }\r
  58 \r
  59   unsigned long type = REG_SZ;\r
  60 \r
  61   /* Try to get executable file - MUST succeed */\r
  62   if (RegQueryValueEx(key, NSSM_REG_EXE, 0, &type, (unsigned char *) exe, (unsigned long *) &exelen) != ERROR_SUCCESS) {\r
  63     eventprintf(EVENTLOG_ERROR_TYPE, "Can't get application path (registry value %s): %s", NSSM_REG_EXE, error_string(GetLastError()));\r
  64     RegCloseKey(key);\r
  65     return 3;\r
  66   }\r
  67 \r
  68   /* Try to get flags - may fail */\r
  69   if (RegQueryValueEx(key, NSSM_REG_FLAGS, 0, &type, (unsigned char *) flags, (unsigned long *) &flagslen) != ERROR_SUCCESS) {\r
  70     eventprintf(EVENTLOG_WARNING_TYPE, "Can't get application flags (registry value %s): %s", NSSM_REG_FLAGS, error_string(GetLastError()));\r
  71     RegCloseKey(key);\r
  72     return 4;\r
  73   }\r
  74 \r
  75   /* Try to get startup directory - may fail */\r
  76   if (RegQueryValueEx(key, NSSM_REG_DIR, 0, &type, (unsigned char *) dir, (unsigned long *) &dirlen) != ERROR_SUCCESS) {\r
  77     eventprintf(EVENTLOG_WARNING_TYPE, "Can't get application startup directory (registry value %s): %s", NSSM_REG_DIR, error_string(GetLastError()));\r
  78     RegCloseKey(key);\r
  79     return 5;\r
  80   }\r
  81 \r
  82   /* Close registry */\r
  83   RegCloseKey(key);\r
  84 \r
  85   return 0;\r
  86 }\r