Rotate output files.

[nssm.git] / io.cpp

diff --git a/io.cpp b/io.cpp
index 5331188..30deb72 100644 (file)
--- a/io.cpp
+++ b/io.cpp
@@ -81,7 +81,77 @@ HANDLE append_to_file(TCHAR *path, unsigned long sharing, SECURITY_ATTRIBUTES *a
   return CreateFile(path, FILE_WRITE_DATA, sharing, attributes, disposition, flags, 0);\r
 }\r
 \r
-int get_output_handles(HKEY key, STARTUPINFO *si) {\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
   TCHAR path[MAX_PATH];\r
   TCHAR stdout_path[MAX_PATH];\r
   unsigned long sharing, disposition, flags;\r
@@ -112,6 +182,7 @@ int get_output_handles(HKEY key, STARTUPINFO *si) {
       return 4;\r
     }\r
 \r
+    if (service->rotate_files) rotate_file(service->name, path, service->rotate_seconds, service->rotate_bytes_low, service->rotate_bytes_high);\r
     si->hStdOutput = append_to_file(path, sharing, &attributes, disposition, flags);\r
     if (! si->hStdOutput) return 5;\r
     set_flags = true;\r
@@ -130,6 +201,7 @@ int get_output_handles(HKEY key, STARTUPINFO *si) {
       }\r
     }\r
     else {\r
+      if (service->rotate_files) rotate_file(service->name, path, service->rotate_seconds, service->rotate_bytes_low, service->rotate_bytes_high);\r
       si->hStdError = append_to_file(path, sharing, &attributes, disposition, flags);\r
       if (! si->hStdError) {\r
         log_event(EVENTLOG_ERROR_TYPE, NSSM_EVENT_CREATEFILE_FAILED, path, error_string(GetLastError()));\r