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(value, sizeof(value), "%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(value, sizeof(value), "%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(value, sizeof(value), "%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(value, sizeof(value), "%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 HANDLE append_to_file(char *path, unsigned long sharing, SECURITY_ATTRIBUTES *attributes, unsigned long disposition, unsigned long flags) {\r
  54   HANDLE ret;\r
  55 \r
  56   /* Try to append to the file first. */\r
  57   ret = CreateFile(path, FILE_APPEND_DATA, sharing, attributes, disposition, flags, 0);\r
  58   if (ret) {\r
  59     SetEndOfFile(ret);\r
  60     return ret;\r
  61   }\r
  62 \r
  63   unsigned long error = GetLastError();\r
  64   if (error != ERROR_FILE_NOT_FOUND) {\r
  65     log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, path, error_string(error));\r
  66     return (HANDLE) 0;\r
  67   }\r
  68 \r
  69   /* It didn't exist.  Create it. */\r
  70   return CreateFile(path, FILE_WRITE_DATA, sharing, attributes, disposition, flags, 0);\r
  71 }\r
  72 \r
  73 int get_output_handles(HKEY key, STARTUPINFO *si) {\r
  74   char path[MAX_PATH];\r
  75   char stdout_path[MAX_PATH];\r
  76   unsigned long sharing, disposition, flags;\r
  77   bool set_flags = false;\r
  78 \r
  79   /* Standard security attributes allowing inheritance. */\r
  80   SECURITY_ATTRIBUTES attributes;\r
  81   ZeroMemory(&attributes, sizeof(attributes));\r
  82   attributes.bInheritHandle = true;\r
  83 \r
  84   /* stdin */\r
  85   if (get_createfile_parameters(key, NSSM_REG_STDIN, path, &sharing, FILE_SHARE_WRITE, &disposition, OPEN_EXISTING, &flags, FILE_ATTRIBUTE_NORMAL)) return 1;\r
  86   if (path[0]) {\r
  87     si->hStdInput = CreateFile(path, FILE_READ_DATA, sharing, &attributes, disposition, flags, 0);\r
  88     if (! si->hStdInput) {\r
  89       log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, path, error_string(GetLastError()));\r
  90       return 2;\r
  91     }\r
  92     set_flags = true;\r
  93   }\r
  94 \r
  95   /* stdout */\r
  96   if (get_createfile_parameters(key, NSSM_REG_STDOUT, path, &sharing, FILE_SHARE_READ | FILE_SHARE_WRITE, &disposition, OPEN_ALWAYS, &flags, FILE_ATTRIBUTE_NORMAL)) return 3;\r
  97   if (path[0]) {\r
  98     /* Remember path for comparison with stderr. */\r
  99     if (_snprintf(stdout_path, sizeof(stdout_path), "%s", path) < 0) {\r
 100       log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, "stdout_path", "get_output_handles", 0);\r
 101       return 4;\r
 102     }\r
 103 \r
 104     si->hStdOutput = append_to_file(path, sharing, &attributes, disposition, flags);\r
 105     if (! si->hStdOutput) return 5;\r
 106     set_flags = true;\r
 107   }\r
 108   else ZeroMemory(stdout_path, sizeof(stdout_path));\r
 109 \r
 110   /* stderr */\r
 111   if (get_createfile_parameters(key, NSSM_REG_STDERR, path, &sharing, FILE_SHARE_READ | FILE_SHARE_WRITE, &disposition, OPEN_ALWAYS, &flags, FILE_ATTRIBUTE_NORMAL)) return 6;\r
 112   if (path[0]) {\r
 113     /* Same as stdin? */\r
 114     if (str_equiv(path, stdout_path)) {\r
 115       /* Two handles to the same file will create a race. */\r
 116       if (! DuplicateHandle(GetCurrentProcess(), si->hStdOutput, GetCurrentProcess(), &si->hStdError, 0, true, DUPLICATE_SAME_ACCESS)) {\r
 117         log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_DUPLICATEHANDLE_FAILED, NSSM_REG_STDOUT, error_string(GetLastError()));\r
 118         return 7;\r
 119       }\r
 120     }\r
 121     else {\r
 122       si->hStdError = append_to_file(path, sharing, &attributes, disposition, flags);\r
 123       if (! si->hStdError) {\r
 124         log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, path, error_string(GetLastError()));\r
 125         return 8;\r
 126       }\r
 127       SetEndOfFile(si->hStdError);\r
 128     }\r
 129     set_flags = true;\r
 130   }\r
 131 \r
 132   if (! set_flags) return 0;\r
 133 \r
 134   /*\r
 135     We need to set the startup_info flags to make the new handles\r
 136     inheritable by the new process.\r
 137   */\r
 138   si->dwFlags |= STARTF_USESTDHANDLES;\r
 139 \r
 140   return 0;\r
 141 }\r
 142 \r
 143 void close_output_handles(STARTUPINFO *si) {\r
 144   if (si->hStdInput) CloseHandle(si->hStdInput);\r
 145   if (si->hStdOutput) CloseHandle(si->hStdOutput);\r
 146   if (si->hStdError) CloseHandle(si->hStdError);\r
 147 }\r