io.cpp

   1 #include "nssm.h"\r
   2 \r
   3 /* Get path, share mode, creation disposition and flags for a stream. */\r
   4 int get_createfile_parameters(HKEY key, char *prefix, char *path, unsigned long *sharing, unsigned long default_sharing, unsigned long *disposition, unsigned long default_disposition, unsigned long *flags, unsigned long default_flags) {\r
   5   char value[NSSM_STDIO_LENGTH];\r
   6 \r
   7   /* Path. */\r
   8   if (_snprintf_s(value, sizeof(value), _TRUNCATE, "%s", prefix) < 0) {\r
   9     log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, prefix, "get_createfile_parameters()", 0);\r
  10     return 1;\r
  11   }\r
  12   switch (expand_parameter(key, value, path, MAX_PATH, true, false)) {\r
  13     case 0: if (! path[0]) return 0; break; /* OK. */\r
  14     default: return 2; /* Error. */\r
  15   }\r
  16 \r
  17   /* ShareMode. */\r
  18   if (_snprintf_s(value, sizeof(value), _TRUNCATE, "%s%s", prefix, NSSM_REG_STDIO_SHARING) < 0) {\r
  19     log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, NSSM_REG_STDIO_SHARING, "get_createfile_parameters()", 0);\r
  20     return 3;\r
  21   }\r
  22   switch (get_number(key, value, sharing, false)) {\r
  23     case 0: *sharing = default_sharing; break; /* Missing. */\r
  24     case 1: break; /* Found. */\r
  25     case -2: return 4; break; /* Error. */\r
  26   }\r
  27 \r
  28   /* CreationDisposition. */\r
  29   if (_snprintf_s(value, sizeof(value), _TRUNCATE, "%s%s", prefix, NSSM_REG_STDIO_DISPOSITION) < 0) {\r
  30     log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, NSSM_REG_STDIO_DISPOSITION, "get_createfile_parameters()", 0);\r
  31     return 5;\r
  32   }\r
  33   switch (get_number(key, value, disposition, false)) {\r
  34     case 0: *disposition = default_disposition; break; /* Missing. */\r
  35     case 1: break; /* Found. */\r
  36     case -2: return 6; break; /* Error. */\r
  37   }\r
  38 \r
  39   /* Flags. */\r
  40   if (_snprintf_s(value, sizeof(value), _TRUNCATE, "%s%s", prefix, NSSM_REG_STDIO_FLAGS) < 0) {\r
  41     log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, NSSM_REG_STDIO_FLAGS, "get_createfile_parameters()", 0);\r
  42     return 7;\r
  43   }\r
  44   switch (get_number(key, value, flags, false)) {\r
  45     case 0: *flags = default_flags; break; /* Missing. */\r
  46     case 1: break; /* Found. */\r
  47     case -2: return 8; break; /* Error. */\r
  48   }\r
  49 \r
  50   return 0;\r
  51 }\r
  52 \r
  53 int set_createfile_parameter(HKEY key, char *prefix, char *suffix, unsigned long number) {\r
  54   char value[NSSM_STDIO_LENGTH];\r
  55 \r
  56   if (_snprintf_s(value, sizeof(value), _TRUNCATE, "%s%s", prefix, suffix) < 0) {\r
  57     log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, suffix, "set_createfile_parameter()", 0);\r
  58     return 1;\r
  59   }\r
  60 \r
  61   return set_number(key, value, number);\r
  62 }\r
  63 \r
  64 HANDLE append_to_file(char *path, unsigned long sharing, SECURITY_ATTRIBUTES *attributes, unsigned long disposition, unsigned long flags) {\r
  65   HANDLE ret;\r
  66 \r
  67   /* Try to append to the file first. */\r
  68   ret = CreateFile(path, FILE_APPEND_DATA, sharing, attributes, disposition, flags, 0);\r
  69   if (ret) {\r
  70     SetEndOfFile(ret);\r
  71     return ret;\r
  72   }\r
  73 \r
  74   unsigned long error = GetLastError();\r
  75   if (error != ERROR_FILE_NOT_FOUND) {\r
  76     log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, path, error_string(error));\r
  77     return (HANDLE) 0;\r
  78   }\r
  79 \r
  80   /* It didn't exist.  Create it. */\r
  81   return CreateFile(path, FILE_WRITE_DATA, sharing, attributes, disposition, flags, 0);\r
  82 }\r
  83 \r
  84 int get_output_handles(HKEY key, STARTUPINFO *si) {\r
  85   char path[MAX_PATH];\r
  86   char stdout_path[MAX_PATH];\r
  87   unsigned long sharing, disposition, flags;\r
  88   bool set_flags = false;\r
  89 \r
  90   /* Standard security attributes allowing inheritance. */\r
  91   SECURITY_ATTRIBUTES attributes;\r
  92   ZeroMemory(&attributes, sizeof(attributes));\r
  93   attributes.bInheritHandle = true;\r
  94 \r
  95   /* stdin */\r
  96   if (get_createfile_parameters(key, NSSM_REG_STDIN, path, &sharing, NSSM_STDIN_SHARING, &disposition, NSSM_STDIN_DISPOSITION, &flags, NSSM_STDIN_FLAGS)) return 1;\r
  97   if (path[0]) {\r
  98     si->hStdInput = CreateFile(path, FILE_READ_DATA, sharing, &attributes, disposition, flags, 0);\r
  99     if (! si->hStdInput) {\r
 100       log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, path, error_string(GetLastError()));\r
 101       return 2;\r
 102     }\r
 103     set_flags = true;\r
 104   }\r
 105 \r
 106   /* stdout */\r
 107   if (get_createfile_parameters(key, NSSM_REG_STDOUT, path, &sharing, NSSM_STDOUT_SHARING, &disposition, NSSM_STDOUT_DISPOSITION, &flags, NSSM_STDOUT_FLAGS)) return 3;\r
 108   if (path[0]) {\r
 109     /* Remember path for comparison with stderr. */\r
 110     if (_snprintf_s(stdout_path, sizeof(stdout_path), _TRUNCATE, "%s", path) < 0) {\r
 111       log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, "stdout_path", "get_output_handles", 0);\r
 112       return 4;\r
 113     }\r
 114 \r
 115     si->hStdOutput = append_to_file(path, sharing, &attributes, disposition, flags);\r
 116     if (! si->hStdOutput) return 5;\r
 117     set_flags = true;\r
 118   }\r
 119   else ZeroMemory(stdout_path, sizeof(stdout_path));\r
 120 \r
 121   /* stderr */\r
 122   if (get_createfile_parameters(key, NSSM_REG_STDERR, path, &sharing, NSSM_STDERR_SHARING, &disposition, NSSM_STDERR_DISPOSITION, &flags, NSSM_STDERR_FLAGS)) return 6;\r
 123   if (path[0]) {\r
 124     /* Same as stdin? */\r
 125     if (str_equiv(path, stdout_path)) {\r
 126       /* Two handles to the same file will create a race. */\r
 127       if (! DuplicateHandle(GetCurrentProcess(), si->hStdOutput, GetCurrentProcess(), &si->hStdError, 0, true, DUPLICATE_SAME_ACCESS)) {\r
 128         log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_DUPLICATEHANDLE_FAILED, NSSM_REG_STDOUT, error_string(GetLastError()));\r
 129         return 7;\r
 130       }\r
 131     }\r
 132     else {\r
 133       si->hStdError = append_to_file(path, sharing, &attributes, disposition, flags);\r
 134       if (! si->hStdError) {\r
 135         log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, path, error_string(GetLastError()));\r
 136         return 8;\r
 137       }\r
 138       SetEndOfFile(si->hStdError);\r
 139     }\r
 140     set_flags = true;\r
 141   }\r
 142 \r
 143   if (! set_flags) return 0;\r
 144 \r
 145   /*\r
 146     We need to set the startup_info flags to make the new handles\r
 147     inheritable by the new process.\r
 148   */\r
 149   si->dwFlags |= STARTF_USESTDHANDLES;\r
 150 \r
 151   return 0;\r
 152 }\r
 153 \r
 154 void close_output_handles(STARTUPINFO *si) {\r
 155   if (si->hStdInput) CloseHandle(si->hStdInput);\r
 156   if (si->hStdOutput) CloseHandle(si->hStdOutput);\r
 157   if (si->hStdError) CloseHandle(si->hStdError);\r
 158 }\r