From e1550b15c5975d9554a175033a35d89645be2b76 Mon Sep 17 00:00:00 2001 From: Iain Patterson Date: Tue, 30 Jul 2013 22:39:40 +0100 Subject: [PATCH] Fixed missing io.cpp and io.h. The files were inadvertently not pushed along with the stream redirection code. Thanks Runner Mei. --- io.cpp | 147 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ io.h | 9 ++++ 2 files changed, 156 insertions(+) create mode 100644 io.cpp create mode 100644 io.h diff --git a/io.cpp b/io.cpp new file mode 100644 index 0000000..d692a4d --- /dev/null +++ b/io.cpp @@ -0,0 +1,147 @@ +#include "nssm.h" + +/* Get path, share mode, creation disposition and flags for a stream. */ +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) { + char value[NSSM_STDIO_LENGTH]; + + /* Path. */ + if (_snprintf(value, sizeof(value), "%s", prefix) < 0) { + log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, prefix, "get_createfile_parameters()", 0); + return 1; + } + switch (expand_parameter(key, value, path, MAX_PATH, true, false)) { + case 0: if (! path[0]) return 0; break; /* OK. */ + default: return 2; /* Error. */ + } + + /* ShareMode. */ + if (_snprintf(value, sizeof(value), "%s%s", prefix, NSSM_REG_STDIO_SHARING) < 0) { + log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, NSSM_REG_STDIO_SHARING, "get_createfile_parameters()", 0); + return 3; + } + switch (get_number(key, value, sharing, false)) { + case 0: *sharing = default_sharing; break; /* Missing. */ + case 1: break; /* Found. */ + case -2: return 4; break; /* Error. */ + } + + /* CreationDisposition. */ + if (_snprintf(value, sizeof(value), "%s%s", prefix, NSSM_REG_STDIO_DISPOSITION) < 0) { + log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, NSSM_REG_STDIO_DISPOSITION, "get_createfile_parameters()", 0); + return 5; + } + switch (get_number(key, value, disposition, false)) { + case 0: *disposition = default_disposition; break; /* Missing. */ + case 1: break; /* Found. */ + case -2: return 6; break; /* Error. */ + } + + /* Flags. */ + if (_snprintf(value, sizeof(value), "%s%s", prefix, NSSM_REG_STDIO_FLAGS) < 0) { + log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, NSSM_REG_STDIO_FLAGS, "get_createfile_parameters()", 0); + return 7; + } + switch (get_number(key, value, flags, false)) { + case 0: *flags = default_flags; break; /* Missing. */ + case 1: break; /* Found. */ + case -2: return 8; break; /* Error. */ + } + + return 0; +} + +HANDLE append_to_file(char *path, unsigned long sharing, SECURITY_ATTRIBUTES *attributes, unsigned long disposition, unsigned long flags) { + HANDLE ret; + + /* Try to append to the file first. */ + ret = CreateFile(path, FILE_APPEND_DATA, sharing, attributes, disposition, flags, 0); + if (ret) { + SetEndOfFile(ret); + return ret; + } + + unsigned long error = GetLastError(); + if (error != ERROR_FILE_NOT_FOUND) { + log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, path, error_string(error)); + return (HANDLE) 0; + } + + /* It didn't exist. Create it. */ + return CreateFile(path, FILE_WRITE_DATA, sharing, attributes, disposition, flags, 0); +} + +int get_output_handles(HKEY key, STARTUPINFO *si) { + char path[MAX_PATH]; + char stdout_path[MAX_PATH]; + unsigned long sharing, disposition, flags; + bool set_flags = false; + + /* Standard security attributes allowing inheritance. */ + SECURITY_ATTRIBUTES attributes; + ZeroMemory(&attributes, sizeof(attributes)); + attributes.bInheritHandle = true; + + /* stdin */ + if (get_createfile_parameters(key, NSSM_REG_STDIN, path, &sharing, FILE_SHARE_WRITE, &disposition, OPEN_EXISTING, &flags, FILE_ATTRIBUTE_NORMAL)) return 1; + if (path[0]) { + si->hStdInput = CreateFile(path, FILE_READ_DATA, sharing, &attributes, disposition, flags, 0); + if (! si->hStdInput) { + log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, path, error_string(GetLastError())); + return 2; + } + set_flags = true; + } + + /* stdout */ + 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; + if (path[0]) { + /* Remember path for comparison with stderr. */ + if (_snprintf(stdout_path, sizeof(stdout_path), "%s", path) < 0) { + log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, "stdout_path", "get_output_handles", 0); + return 4; + } + + si->hStdOutput = append_to_file(path, sharing, &attributes, disposition, flags); + if (! si->hStdOutput) return 5; + set_flags = true; + } + else ZeroMemory(stdout_path, sizeof(stdout_path)); + + /* stderr */ + 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; + if (path[0]) { + /* Same as stdin? */ + if (str_equiv(path, stdout_path)) { + /* Two handles to the same file will create a race. */ + if (! DuplicateHandle(GetCurrentProcess(), si->hStdOutput, GetCurrentProcess(), &si->hStdError, 0, true, DUPLICATE_SAME_ACCESS)) { + log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_DUPLICATEHANDLE_FAILED, NSSM_REG_STDOUT, error_string(GetLastError())); + return 7; + } + } + else { + si->hStdError = append_to_file(path, sharing, &attributes, disposition, flags); + if (! si->hStdError) { + log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, path, error_string(GetLastError())); + return 8; + } + SetEndOfFile(si->hStdError); + } + set_flags = true; + } + + if (! set_flags) return 0; + + /* + We need to set the startup_info flags to make the new handles + inheritable by the new process. + */ + si->dwFlags |= STARTF_USESTDHANDLES; + + return 0; +} + +void close_output_handles(STARTUPINFO *si) { + if (si->hStdInput) CloseHandle(si->hStdInput); + if (si->hStdOutput) CloseHandle(si->hStdOutput); + if (si->hStdError) CloseHandle(si->hStdError); +} diff --git a/io.h b/io.h new file mode 100644 index 0000000..f8fb256 --- /dev/null +++ b/io.h @@ -0,0 +1,9 @@ +#ifndef IO_H +#define IO_H + +int get_createfile_parameters(HKEY, char *, char *, unsigned long *, unsigned long, unsigned long *, unsigned long, unsigned long *, unsigned long); +HANDLE append_to_file(char *, unsigned long, SECURITY_ATTRIBUTES *, unsigned long, unsigned long); +int get_output_handles(HKEY, STARTUPINFO *); +void close_output_handles(STARTUPINFO *); + +#endif -- 2.7.4