#include "nssm.h"\r
\r
/* Get path, share mode, creation disposition and flags for a stream. */\r
-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
- char value[NSSM_STDIO_LENGTH];\r
+int get_createfile_parameters(HKEY key, TCHAR *prefix, TCHAR *path, unsigned long *sharing, unsigned long default_sharing, unsigned long *disposition, unsigned long default_disposition, unsigned long *flags, unsigned long default_flags) {\r
+ TCHAR value[NSSM_STDIO_LENGTH];\r
\r
/* Path. */\r
- if (_snprintf_s(value, sizeof(value), _TRUNCATE, "%s", prefix) < 0) {\r
- log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, prefix, "get_createfile_parameters()", 0);\r
+ if (_sntprintf_s(value, _countof(value), _TRUNCATE, _T("%s"), prefix) < 0) {\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, prefix, _T("get_createfile_parameters()"), 0);\r
return 1;\r
}\r
switch (expand_parameter(key, value, path, MAX_PATH, true, false)) {\r
}\r
\r
/* ShareMode. */\r
- if (_snprintf_s(value, sizeof(value), _TRUNCATE, "%s%s", prefix, NSSM_REG_STDIO_SHARING) < 0) {\r
- log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, NSSM_REG_STDIO_SHARING, "get_createfile_parameters()", 0);\r
+ if (_sntprintf_s(value, _countof(value), _TRUNCATE, _T("%s%s"), prefix, NSSM_REG_STDIO_SHARING) < 0) {\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, NSSM_REG_STDIO_SHARING, _T("get_createfile_parameters()"), 0);\r
return 3;\r
}\r
switch (get_number(key, value, sharing, false)) {\r
}\r
\r
/* CreationDisposition. */\r
- if (_snprintf_s(value, sizeof(value), _TRUNCATE, "%s%s", prefix, NSSM_REG_STDIO_DISPOSITION) < 0) {\r
- log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, NSSM_REG_STDIO_DISPOSITION, "get_createfile_parameters()", 0);\r
+ if (_sntprintf_s(value, _countof(value), _TRUNCATE, _T("%s%s"), prefix, NSSM_REG_STDIO_DISPOSITION) < 0) {\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, NSSM_REG_STDIO_DISPOSITION, _T("get_createfile_parameters()"), 0);\r
return 5;\r
}\r
switch (get_number(key, value, disposition, false)) {\r
}\r
\r
/* Flags. */\r
- if (_snprintf_s(value, sizeof(value), _TRUNCATE, "%s%s", prefix, NSSM_REG_STDIO_FLAGS) < 0) {\r
- log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, NSSM_REG_STDIO_FLAGS, "get_createfile_parameters()", 0);\r
+ if (_sntprintf_s(value, _countof(value), _TRUNCATE, _T("%s%s"), prefix, NSSM_REG_STDIO_FLAGS) < 0) {\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, NSSM_REG_STDIO_FLAGS, _T("get_createfile_parameters()"), 0);\r
return 7;\r
}\r
switch (get_number(key, value, flags, false)) {\r
return 0;\r
}\r
\r
-HANDLE append_to_file(char *path, unsigned long sharing, SECURITY_ATTRIBUTES *attributes, unsigned long disposition, unsigned long flags) {\r
+int set_createfile_parameter(HKEY key, TCHAR *prefix, TCHAR *suffix, unsigned long number) {\r
+ TCHAR value[NSSM_STDIO_LENGTH];\r
+\r
+ if (_sntprintf_s(value, _countof(value), _TRUNCATE, _T("%s%s"), prefix, suffix) < 0) {\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, suffix, _T("set_createfile_parameter()"), 0);\r
+ return 1;\r
+ }\r
+\r
+ return set_number(key, value, number);\r
+}\r
+\r
+int delete_createfile_parameter(HKEY key, TCHAR *prefix, TCHAR *suffix) {\r
+ TCHAR value[NSSM_STDIO_LENGTH];\r
+\r
+ if (_sntprintf_s(value, _countof(value), _TRUNCATE, _T("%s%s"), prefix, suffix) < 0) {\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, suffix, _T("delete_createfile_parameter()"), 0);\r
+ return 1;\r
+ }\r
+\r
+ if (RegDeleteValue(key, value)) return 0;\r
+ return 1;\r
+}\r
+\r
+HANDLE append_to_file(TCHAR *path, unsigned long sharing, SECURITY_ATTRIBUTES *attributes, unsigned long disposition, unsigned long flags) {\r
HANDLE ret;\r
\r
/* Try to append to the file first. */\r
\r
unsigned long error = GetLastError();\r
if (error != ERROR_FILE_NOT_FOUND) {\r
- log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, path, error_string(error));\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, path, error_string(error), 0);\r
return (HANDLE) 0;\r
}\r
\r
return CreateFile(path, FILE_WRITE_DATA, sharing, attributes, disposition, flags, 0);\r
}\r
\r
-int get_output_handles(HKEY key, STARTUPINFO *si) {\r
- char path[MAX_PATH];\r
- char stdout_path[MAX_PATH];\r
- unsigned long sharing, disposition, flags;\r
+void rotate_file(TCHAR *service_name, TCHAR *path, unsigned long seconds, unsigned long low, unsigned long high) {\r
+ unsigned long error;\r
+\r
+ /* Now. */\r
+ SYSTEMTIME st;\r
+ GetSystemTime(&st);\r
+\r
+ BY_HANDLE_FILE_INFORMATION info;\r
+\r
+ /* Try to open the file to check if it exists and to get attributes. */\r
+ HANDLE file = CreateFile(path, 0, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, 0);\r
+ if (file) {\r
+ /* Get file attributes. */\r
+ if (! GetFileInformationByHandle(file, &info)) {\r
+ /* Reuse current time for rotation timestamp. */\r
+ seconds = low = high = 0;\r
+ SystemTimeToFileTime(&st, &info.ftLastWriteTime);\r
+ }\r
+\r
+ CloseHandle(file);\r
+ }\r
+ else {\r
+ error = GetLastError();\r
+ if (error == ERROR_FILE_NOT_FOUND) return;\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_ROTATE_FILE_FAILED, service_name, path, _T("CreateFile()"), path, error_string(error), 0);\r
+ /* Reuse current time for rotation timestamp. */\r
+ seconds = low = high = 0;\r
+ SystemTimeToFileTime(&st, &info.ftLastWriteTime);\r
+ }\r
+\r
+ /* Check file age. */\r
+ if (seconds) {\r
+ FILETIME ft;\r
+ SystemTimeToFileTime(&st, &ft);\r
+\r
+ ULARGE_INTEGER s;\r
+ s.LowPart = ft.dwLowDateTime;\r
+ s.HighPart = ft.dwHighDateTime;\r
+ s.QuadPart -= seconds * 10000000LL;\r
+ ft.dwLowDateTime = s.LowPart;\r
+ ft.dwHighDateTime = s.HighPart;\r
+ if (CompareFileTime(&info.ftLastWriteTime, &ft) > 0) return;\r
+ }\r
+\r
+ /* Check file size. */\r
+ if (low || high) {\r
+ if (info.nFileSizeHigh < high) return;\r
+ if (info.nFileSizeHigh == high && info.nFileSizeLow < low) return;\r
+ }\r
+\r
+ /* Get new filename. */\r
+ FileTimeToSystemTime(&info.ftLastWriteTime, &st);\r
+\r
+ TCHAR buffer[MAX_PATH];\r
+ memmove(buffer, path, sizeof(buffer));\r
+ TCHAR *ext = PathFindExtension(buffer);\r
+ TCHAR extension[MAX_PATH];\r
+ _sntprintf_s(extension, _countof(extension), _TRUNCATE, _T("-%04u%02u%02uT%02u%02u%02u.%03u%s"), st.wYear, st.wMonth, st.wDay, st.wHour, st.wMinute, st.wSecond, st.wMilliseconds, ext);\r
+ *ext = _T('\0');\r
+ TCHAR rotated[MAX_PATH];\r
+ _sntprintf_s(rotated, _countof(rotated), _TRUNCATE, _T("%s%s"), buffer, extension);\r
+\r
+ /* Rotate. */\r
+ if (MoveFile(path, rotated)) return;\r
+ error = GetLastError();\r
+\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_ROTATE_FILE_FAILED, service_name, path, _T("MoveFile()"), rotated, error_string(error), 0);\r
+ return;\r
+}\r
+\r
+int get_output_handles(nssm_service_t *service, HKEY key, STARTUPINFO *si) {\r
bool set_flags = false;\r
\r
/* Standard security attributes allowing inheritance. */\r
attributes.bInheritHandle = true;\r
\r
/* stdin */\r
- if (get_createfile_parameters(key, NSSM_REG_STDIN, path, &sharing, FILE_SHARE_WRITE, &disposition, OPEN_EXISTING, &flags, FILE_ATTRIBUTE_NORMAL)) return 1;\r
- if (path[0]) {\r
- si->hStdInput = CreateFile(path, FILE_READ_DATA, sharing, &attributes, disposition, flags, 0);\r
+ if (get_createfile_parameters(key, NSSM_REG_STDIN, service->stdin_path, &service->stdin_sharing, NSSM_STDIN_SHARING, &service->stdin_disposition, NSSM_STDIN_DISPOSITION, &service->stdin_flags, NSSM_STDIN_FLAGS)) {\r
+ service->stdin_sharing = service->stdin_disposition = service->stdin_flags = 0;\r
+ ZeroMemory(service->stdin_path, _countof(service->stdin_path) * sizeof(TCHAR));\r
+ return 1;\r
+ }\r
+ if (si && service->stdin_path[0]) {\r
+ si->hStdInput = CreateFile(service->stdin_path, FILE_READ_DATA, service->stdin_sharing, &attributes, service->stdin_disposition, service->stdin_flags, 0);\r
if (! si->hStdInput) {\r
- log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, path, error_string(GetLastError()));\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, service->stdin_path, error_string(GetLastError()), 0);\r
return 2;\r
}\r
set_flags = true;\r
}\r
\r
/* stdout */\r
- 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
- if (path[0]) {\r
- /* Remember path for comparison with stderr. */\r
- if (_snprintf_s(stdout_path, sizeof(stdout_path), _TRUNCATE, "%s", path) < 0) {\r
- log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_OUT_OF_MEMORY, "stdout_path", "get_output_handles", 0);\r
- return 4;\r
- }\r
-\r
- si->hStdOutput = append_to_file(path, sharing, &attributes, disposition, flags);\r
- if (! si->hStdOutput) return 5;\r
+ if (get_createfile_parameters(key, NSSM_REG_STDOUT, service->stdout_path, &service->stdout_sharing, NSSM_STDOUT_SHARING, &service->stdout_disposition, NSSM_STDOUT_DISPOSITION, &service->stdout_flags, NSSM_STDOUT_FLAGS)) {\r
+ service->stdout_sharing = service->stdout_disposition = service->stdout_flags = 0;\r
+ ZeroMemory(service->stdout_path, _countof(service->stdout_path) * sizeof(TCHAR));\r
+ return 3;\r
+ }\r
+ if (si && service->stdout_path[0]) {\r
+ if (service->rotate_files) rotate_file(service->name, service->stdout_path, service->rotate_seconds, service->rotate_bytes_low, service->rotate_bytes_high);\r
+ si->hStdOutput = append_to_file(service->stdout_path, service->stdout_sharing, &attributes, service->stdout_disposition, service->stdout_flags);\r
+ if (! si->hStdOutput) return 4;\r
set_flags = true;\r
}\r
- else ZeroMemory(stdout_path, sizeof(stdout_path));\r
\r
/* stderr */\r
- 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
- if (path[0]) {\r
- /* Same as stdin? */\r
- if (str_equiv(path, stdout_path)) {\r
- /* Two handles to the same file will create a race. */\r
- if (! DuplicateHandle(GetCurrentProcess(), si->hStdOutput, GetCurrentProcess(), &si->hStdError, 0, true, DUPLICATE_SAME_ACCESS)) {\r
- log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_DUPLICATEHANDLE_FAILED, NSSM_REG_STDOUT, error_string(GetLastError()));\r
- return 7;\r
+ if (get_createfile_parameters(key, NSSM_REG_STDERR, service->stderr_path, &service->stdout_sharing, NSSM_STDERR_SHARING, &service->stdout_disposition, NSSM_STDERR_DISPOSITION, &service->stdout_flags, NSSM_STDERR_FLAGS)) {\r
+ service->stderr_sharing = service->stderr_disposition = service->stderr_flags = 0;\r
+ ZeroMemory(service->stderr_path, _countof(service->stderr_path) * sizeof(TCHAR));\r
+ return 5;\r
+ }\r
+ if (service->stderr_path[0]) {\r
+ /* Same as stdout? */\r
+ if (str_equiv(service->stderr_path, service->stdout_path)) {\r
+ service->stderr_sharing = service->stdout_sharing;\r
+ service->stderr_disposition = service->stdout_disposition;\r
+ service->stderr_flags = service->stdout_flags;\r
+\r
+ if (si) {\r
+ /* Two handles to the same file will create a race. */\r
+ if (! DuplicateHandle(GetCurrentProcess(), si->hStdOutput, GetCurrentProcess(), &si->hStdError, 0, true, DUPLICATE_SAME_ACCESS)) {\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_DUPLICATEHANDLE_FAILED, NSSM_REG_STDOUT, error_string(GetLastError()), 0);\r
+ return 6;\r
+ }\r
}\r
}\r
- else {\r
- si->hStdError = append_to_file(path, sharing, &attributes, disposition, flags);\r
+ else if (si) {\r
+ if (service->rotate_files) rotate_file(service->name, service->stderr_path, service->rotate_seconds, service->rotate_bytes_low, service->rotate_bytes_high);\r
+ si->hStdError = append_to_file(service->stderr_path, service->stdout_sharing, &attributes, service->stdout_disposition, service->stdout_flags);\r
if (! si->hStdError) {\r
- log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, path, error_string(GetLastError()));\r
- return 8;\r
+ log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, service->stderr_path, error_string(GetLastError()), 0);\r
+ return 7;\r
}\r
SetEndOfFile(si->hStdError);\r
}\r
We need to set the startup_info flags to make the new handles\r
inheritable by the new process.\r
*/\r
- si->dwFlags |= STARTF_USESTDHANDLES;\r
+ if (si) si->dwFlags |= STARTF_USESTDHANDLES;\r
\r
return 0;\r
}\r