git://git.iain.cx/iain / nssm.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | inline | side by side
remove dead breaks and returns
[nssm.git] / registry.cpp
diff --git a/registry.cpp b/registry.cpp
index 5953f8a..258b96a 100644 (file)
--- a/registry.cpp
+++ b/registry.cpp
@@ -2,6 +2,46 @@
 \r
 extern const TCHAR *exit_action_strings[];\r
 \r
+static int service_registry_path(const TCHAR *service_name, bool parameters, const TCHAR *sub, TCHAR *buffer, unsigned long buflen) {\r
+  int ret;\r
+\r
+  if (parameters) {\r
+    if (sub) ret = _sntprintf_s(buffer, buflen, _TRUNCATE, NSSM_REGISTRY _T("\\") NSSM_REG_PARAMETERS _T("\\%s"), service_name, sub);\r
+    else ret = _sntprintf_s(buffer, buflen, _TRUNCATE, NSSM_REGISTRY _T("\\") NSSM_REG_PARAMETERS, service_name);\r
+  }\r
+  else ret = _sntprintf_s(buffer, buflen, _TRUNCATE, NSSM_REGISTRY, service_name);\r
+\r
+  return ret;\r
+}\r
+\r
+static long open_registry_key(const TCHAR *registry, REGSAM sam, HKEY *key, bool must_exist) {\r
+  long error;\r
+\r
+  if (sam & KEY_SET_VALUE) {\r
+    error = RegCreateKeyEx(HKEY_LOCAL_MACHINE, registry, 0, 0, REG_OPTION_NON_VOLATILE, sam, 0, key, 0);\r
+    if (error != ERROR_SUCCESS) {\r
+      *key = 0;\r
+      log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OPENKEY_FAILED, registry, error_string(GetLastError()), 0);\r
+      return error;\r
+    }\r
+  }\r
+  else {\r
+    error = RegOpenKeyEx(HKEY_LOCAL_MACHINE, registry, 0, sam, key);\r
+    if (error != ERROR_SUCCESS) {\r
+      *key = 0;\r
+      if (error != ERROR_FILE_NOT_FOUND || must_exist) log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OPENKEY_FAILED, registry, error_string(GetLastError()), 0);\r
+    }\r
+  }\r
+\r
+  return error;\r
+}\r
+\r
+static HKEY open_registry_key(const TCHAR *registry, REGSAM sam, bool must_exist) {\r
+  HKEY key;\r
+  long error = open_registry_key(registry, sam, &key, must_exist);\r
+  return key;\r
+}\r
+\r
 int create_messages() {\r
   HKEY key;\r
 \r
@@ -17,7 +57,7 @@ int create_messages() {
   }\r
 \r
   /* Get path of this program */\r
-  const TCHAR *path = nssm_imagepath();\r
+  const TCHAR *path = nssm_unquoted_imagepath();\r
 \r
   /* Try to register the module but don't worry so much on failure */\r
   RegSetValueEx(key, _T("EventMessageFile"), 0, REG_SZ, (const unsigned char *) path, (unsigned long) (_tcslen(path) +  1) * sizeof(TCHAR));\r
@@ -32,19 +72,23 @@ int create_parameters(nssm_service_t *service, bool editing) {
   HKEY key = open_registry(service->name, KEY_WRITE);\r
   if (! key) return 1;\r
 \r
+  /* Remember parameters in case we need to delete them. */\r
+  TCHAR registry[KEY_LENGTH];\r
+  int ret = service_registry_path(service->name, true, 0, registry, _countof(registry));\r
+\r
   /* Try to create the parameters */\r
   if (set_expand_string(key, NSSM_REG_EXE, service->exe)) {\r
-    RegDeleteKey(HKEY_LOCAL_MACHINE, NSSM_REGISTRY);\r
+    if (ret > 0) RegDeleteKey(HKEY_LOCAL_MACHINE, registry);\r
     RegCloseKey(key);\r
     return 2;\r
   }\r
   if (set_expand_string(key, NSSM_REG_FLAGS, service->flags)) {\r
-    RegDeleteKey(HKEY_LOCAL_MACHINE, NSSM_REGISTRY);\r
+    if (ret > 0) RegDeleteKey(HKEY_LOCAL_MACHINE, registry);\r
     RegCloseKey(key);\r
     return 3;\r
   }\r
   if (set_expand_string(key, NSSM_REG_DIR, service->dir)) {\r
-    RegDeleteKey(HKEY_LOCAL_MACHINE, NSSM_REGISTRY);\r
+    if (ret > 0) RegDeleteKey(HKEY_LOCAL_MACHINE, registry);\r
     RegCloseKey(key);\r
     return 4;\r
   }\r
@@ -152,7 +196,7 @@ int create_parameters(nssm_service_t *service, bool editing) {
 int create_exit_action(TCHAR *service_name, const TCHAR *action_string, bool editing) {\r
   /* Get registry */\r
   TCHAR registry[KEY_LENGTH];\r
-  if (_sntprintf_s(registry, _countof(registry), _TRUNCATE, NSSM_REGISTRY _T("\\%s"), service_name, NSSM_REG_EXIT) < 0) {\r
+  if (service_registry_path(service_name, true, NSSM_REG_EXIT, registry, _countof(registry)) < 0) {\r
     log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, _T("NSSM_REG_EXIT"), _T("create_exit_action()"), 0);\r
     return 1;\r
   }\r
@@ -309,12 +353,10 @@ int set_string(HKEY key, TCHAR *value, TCHAR *string, bool expand) {
 \r
 int set_string(HKEY key, TCHAR *value, TCHAR *string) {\r
   return set_string(key, value, string, false);\r
-  return 1;\r
 }\r
 \r
 int set_expand_string(HKEY key, TCHAR *value, TCHAR *string) {\r
   return set_string(key, value, string, true);\r
-  return 1;\r
 }\r
 \r
 /*\r
@@ -456,39 +498,6 @@ void override_milliseconds(TCHAR *service_name, HKEY key, TCHAR *value, unsigned
   if (! ok) *buffer = default_value;\r
 }\r
 \r
-static int service_registry_path(const TCHAR *service_name, bool parameters, const TCHAR *sub, TCHAR *buffer, unsigned long buflen) {\r
-  int ret;\r
-\r
-  if (parameters) {\r
-    if (sub) ret = _sntprintf_s(buffer, buflen, _TRUNCATE, NSSM_REGISTRY _T("\\") NSSM_REG_PARAMETERS _T("\\%s"), service_name, sub);\r
-    else ret = _sntprintf_s(buffer, buflen, _TRUNCATE, NSSM_REGISTRY _T("\\") NSSM_REG_PARAMETERS, service_name);\r
-  }\r
-  else ret = _sntprintf_s(buffer, buflen, _TRUNCATE, NSSM_REGISTRY, service_name);\r
-\r
-  return ret;\r
-}\r
-\r
-static HKEY open_registry_key(const TCHAR *registry, REGSAM sam, bool must_exist) {\r
-  HKEY key;\r
-\r
-  if (sam & KEY_SET_VALUE) {\r
-    if (RegCreateKeyEx(HKEY_LOCAL_MACHINE, registry, 0, 0, REG_OPTION_NON_VOLATILE, sam, 0, &key, 0) != ERROR_SUCCESS) {\r
-      log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OPENKEY_FAILED, registry, error_string(GetLastError()), 0);\r
-      return 0;\r
-    }\r
-  }\r
-  else {\r
-    long error = RegOpenKeyEx(HKEY_LOCAL_MACHINE, registry, 0, sam, &key);\r
-    if (error != ERROR_SUCCESS) {\r
-      if (error == ERROR_FILE_NOT_FOUND && ! must_exist) return 0;\r
-      log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OPENKEY_FAILED, registry, error_string(GetLastError()), 0);\r
-      return 0;\r
-    }\r
-  }\r
-\r
-  return key;\r
-}\r
-\r
 HKEY open_service_registry(const TCHAR *service_name, REGSAM sam, bool must_exist) {\r
   /* Get registry */\r
   TCHAR registry[KEY_LENGTH];\r
@@ -500,7 +509,7 @@ HKEY open_service_registry(const TCHAR *service_name, REGSAM sam, bool must_exis
   return open_registry_key(registry, sam, must_exist);\r
 }\r
 \r
-HKEY open_registry(const TCHAR *service_name, const TCHAR *sub, REGSAM sam, bool must_exist) {\r
+long open_registry(const TCHAR *service_name, const TCHAR *sub, REGSAM sam, HKEY *key, bool must_exist) {\r
   /* Get registry */\r
   TCHAR registry[KEY_LENGTH];\r
   if (service_registry_path(service_name, true, sub, registry, _countof(registry)) < 0) {\r
@@ -508,7 +517,13 @@ HKEY open_registry(const TCHAR *service_name, const TCHAR *sub, REGSAM sam, bool
     return 0;\r
   }\r
 \r
-  return open_registry_key(registry, sam, must_exist);\r
+  return open_registry_key(registry, sam, key, must_exist);\r
+}\r
+\r
+HKEY open_registry(const TCHAR *service_name, const TCHAR *sub, REGSAM sam, bool must_exist) {\r
+  HKEY key;\r
+  long error = open_registry(service_name, sub, sam, &key, must_exist);\r
+  return key;\r
 }\r
 \r
 HKEY open_registry(const TCHAR *service_name, const TCHAR *sub, REGSAM sam) {\r
@@ -805,8 +820,15 @@ int get_hook(const TCHAR *service_name, const TCHAR *hook_event, const TCHAR *ho
     log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, _T("hook registry"), _T("get_hook()"), 0);\r
     return 1;\r
   }\r
-  HKEY key = open_registry(service_name, registry, KEY_READ, false);\r
-  if (! key) return 1;\r
+  HKEY key;\r
+  long error = open_registry(service_name, registry, KEY_READ, &key, false);\r
+  if (! key) {\r
+    if (error == ERROR_FILE_NOT_FOUND) {\r
+      ZeroMemory(buffer, buflen);\r
+      return 0;\r
+    }\r
+    return 1;\r
+  }\r
 \r
   int ret = expand_parameter(key, (TCHAR *) hook_action, buffer, buflen, true, false);\r
 \r
The Non-Sucking Service Manager