support/rotatelogs.c - httpd - Git at Google

 /* Copyright 1999-2004 The Apache Software Foundation
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *     http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */

 /*
  * Simple program to rotate Apache logs without having to kill the server.
  *
  * Contributed by Ben Laurie <ben algroup.co.uk>
  *
  * 12 Mar 1996
  *
  * Ported to APR by Mladen Turk <mturk mappingsoft.com>
  *
  * 23 Sep 2001
  *
  * -l option added 2004-06-11
  *
  * -l causes the use of local time rather than GMT as the base for the
  * interval.  NB: Using -l in an environment which changes the GMT offset
  * (such as for BST or DST) can lead to unpredictable results!
  *
  */


 #include "apr.h"
 #include "apr_lib.h"
 #include "apr_strings.h"
 #include "apr_errno.h"
 #include "apr_file_io.h"
 #include "apr_file_info.h"
 #include "apr_general.h"
 #include "apr_time.h"

 #if APR_HAVE_STDLIB_H
 #include <stdlib.h>
 #endif
 #if APR_HAVE_STRING_H
 #include <string.h>
 #endif
 #if APR_HAVE_STRINGS_H
 #include <strings.h>
 #endif

 #define BUFSIZE         65536
 #define ERRMSGSZ        82

 #ifndef MAX_PATH
 #define MAX_PATH        1024
 #endif

 int main (int argc, const char * const argv[])
 {
     char buf[BUFSIZE], buf2[MAX_PATH], errbuf[ERRMSGSZ];
     int tLogEnd = 0, tRotation = 0, utc_offset = 0;
     unsigned int sRotation = 0;
     int nMessCount = 0;
     apr_size_t nRead, nWrite;
     int use_strftime = 0;
     int use_localtime = 0;
     int now = 0;
     const char *szLogRoot;
     apr_file_t *f_stdin, *nLogFD = NULL, *nLogFDprev = NULL;
     apr_pool_t *pool;
     char *ptr = NULL;
     int argBase = 0;
     int argFile = 1;
     int argIntv = 2;
     int argOffset = 3;

     apr_app_initialize(&argc, &argv, NULL);
     atexit(apr_terminate);

     apr_pool_create(&pool, NULL);
     if ((argc > 2) && (strcmp(argv[1], "-l") == 0)) {
         argBase++;
         argFile += argBase;
         argIntv += argBase;
         argOffset += argBase;
         use_localtime = 1;
     }
     if (argc < (argBase + 3) || argc > (argBase + 4)) {
         fprintf(stderr,
                 "Usage: %s [-l] <logfile> <rotation time in seconds> "
                 "[offset minutes from UTC] or <rotation size in megabytes>\n\n",
                 argv[0]);
 #ifdef OS2
         fprintf(stderr,
                 "Add this:\n\nTransferLog \"|%s.exe /some/where 86400\"\n\n",
                 argv[0]);
 #else
         fprintf(stderr,
                 "Add this:\n\nTransferLog \"|%s /some/where 86400\"\n\n",
                 argv[0]);
         fprintf(stderr,
                 "or \n\nTransferLog \"|%s /some/where 5M\"\n\n", argv[0]);
 #endif
         fprintf(stderr,
                 "to httpd.conf. The generated name will be /some/where.nnnn "
                 "where nnnn is the\nsystem time at which the log nominally "
                 "starts (N.B. if using a rotation time,\nthe time will always "
                 "be a multiple of the rotation time, so you can synchronize\n"
                 "cron scripts with it). At the end of each rotation time or "
                 "when the file size\nis reached a new log is started.\n");
         exit(1);
     }

     szLogRoot = argv[argFile];

     ptr = strchr(argv[argIntv], 'M');
     if (ptr) {
         if (*(ptr+1) == '\0') {
             sRotation = atoi(argv[argIntv]) * 1048576;
         }
         if (sRotation == 0) {
             fprintf(stderr, "Invalid rotation size parameter\n");
             exit(1);
         }
     }
     else {
         if (argc >= (argBase + 4)) {
             utc_offset = atoi(argv[argOffset]) * 60;
         }
         tRotation = atoi(argv[argIntv]);
         if (tRotation <= 0) {
             fprintf(stderr, "Rotation time must be > 0\n");
             exit(6);
         }
     }

     use_strftime = (strchr(szLogRoot, '%') != NULL);
     if (apr_file_open_stdin(&f_stdin, pool) != APR_SUCCESS) {
         fprintf(stderr, "Unable to open stdin\n");
         exit(1);
     }

     for (;;) {
         nRead = sizeof(buf);
         if (apr_file_read(f_stdin, buf, &nRead) != APR_SUCCESS) {
             exit(3);
         }
         if (tRotation) {
             /*
              * Check for our UTC offset every time through the loop, since
              * it might change if there's a switch between standard and
              * daylight savings time.
              */
             if (use_localtime) {
                 apr_time_exp_t lt;
                 apr_time_exp_lt(&lt, apr_time_now());
                 utc_offset = lt.tm_gmtoff;
             }
             now = (int)(apr_time_now() / APR_USEC_PER_SEC) + utc_offset;
             if (nLogFD != NULL && now >= tLogEnd) {
                 nLogFDprev = nLogFD;
                 nLogFD = NULL;
             }
         }
         else if (sRotation) {
             apr_finfo_t finfo;
             apr_off_t current_size = -1;

             if ((nLogFD != NULL) &&
                 (apr_file_info_get(&finfo, APR_FINFO_SIZE, nLogFD) == APR_SUCCESS)) {
                 current_size = finfo.size;
             }

             if (current_size > sRotation) {
                 nLogFDprev = nLogFD;
                 nLogFD = NULL;
             }
         }
         else {
             fprintf(stderr, "No rotation time or size specified\n");
             exit(2);
         }

         if (nLogFD == NULL) {
             int tLogStart;

             if (tRotation) {
                 tLogStart = (now / tRotation) * tRotation;
             }
             else {
                 tLogStart = (int)apr_time_sec(apr_time_now());
             }

             if (use_strftime) {
                 apr_time_t tNow = apr_time_from_sec(tLogStart);
                 apr_time_exp_t e;
                 apr_size_t rs;

                 apr_time_exp_gmt(&e, tNow);
                 apr_strftime(buf2, &rs, sizeof(buf2), szLogRoot, &e);
             }
             else {
                 sprintf(buf2, "%s.%010d", szLogRoot, tLogStart);
             }
             tLogEnd = tLogStart + tRotation;
             apr_file_open(&nLogFD, buf2, APR_READ | APR_WRITE | APR_CREATE | APR_APPEND,
                           APR_OS_DEFAULT, pool);
             if (nLogFD == NULL) {
                 /* Uh-oh. Failed to open the new log file. Try to clear
                  * the previous log file, note the lost log entries,
                  * and keep on truckin'. */
                 if (nLogFDprev == NULL) {
                     fprintf(stderr, "1 Previous file handle doesn't exists %s\n", buf2);
                     exit(2);
                 }
                 else {
                     nLogFD = nLogFDprev;
                     sprintf(errbuf,
                             "Resetting log file due to error opening "
                             "new log file. %10d messages lost.\n",
                             nMessCount);
                     nWrite = strlen(errbuf);
                     apr_file_trunc(nLogFD, 0);
                     if (apr_file_write(nLogFD, errbuf, &nWrite) != APR_SUCCESS) {
                         fprintf(stderr, "Error writing to the file %s\n", buf2);
                         exit(2);
                     }
                 }
             }
             else if (nLogFDprev) {
                 apr_file_close(nLogFDprev);
             }
             nMessCount = 0;
         }
         nWrite = nRead;
         apr_file_write(nLogFD, buf, &nWrite);
         if (nWrite != nRead) {
             nMessCount++;
             sprintf(errbuf,
                     "Error writing to log file. "
                     "%10d messages lost.\n",
                     nMessCount);
             nWrite = strlen(errbuf);
             apr_file_trunc(nLogFD, 0);
             if (apr_file_write(nLogFD, errbuf, &nWrite) != APR_SUCCESS) {
                 fprintf(stderr, "Error writing to the file %s\n", buf2);
                 exit(2);
             }
         }
         else {
             nMessCount++;
         }
     }
     /* Of course we never, but prevent compiler warnings */
     return 0;
 }
	/* Copyright 1999-2004 The Apache Software Foundation
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	/*
	* Simple program to rotate Apache logs without having to kill the server.
	*
	* Contributed by Ben Laurie <ben algroup.co.uk>
	*
	* 12 Mar 1996
	*
	* Ported to APR by Mladen Turk <mturk mappingsoft.com>
	*
	* 23 Sep 2001
	*
	* -l option added 2004-06-11
	*
	* -l causes the use of local time rather than GMT as the base for the
	* interval. NB: Using -l in an environment which changes the GMT offset
	* (such as for BST or DST) can lead to unpredictable results!
	*
	*/


	#include "apr.h"
	#include "apr_lib.h"
	#include "apr_strings.h"
	#include "apr_errno.h"
	#include "apr_file_io.h"
	#include "apr_file_info.h"
	#include "apr_general.h"
	#include "apr_time.h"

	#if APR_HAVE_STDLIB_H
	#include <stdlib.h>
	#endif
	#if APR_HAVE_STRING_H
	#include <string.h>
	#endif
	#if APR_HAVE_STRINGS_H
	#include <strings.h>
	#endif

	#define BUFSIZE 65536
	#define ERRMSGSZ 82

	#ifndef MAX_PATH
	#define MAX_PATH 1024
	#endif

	int main (int argc, const char * const argv[])
	{
	char buf[BUFSIZE], buf2[MAX_PATH], errbuf[ERRMSGSZ];
	int tLogEnd = 0, tRotation = 0, utc_offset = 0;
	unsigned int sRotation = 0;
	int nMessCount = 0;
	apr_size_t nRead, nWrite;
	int use_strftime = 0;
	int use_localtime = 0;
	int now = 0;
	const char *szLogRoot;
	apr_file_t f_stdin, nLogFD = NULL, *nLogFDprev = NULL;
	apr_pool_t *pool;
	char *ptr = NULL;
	int argBase = 0;
	int argFile = 1;
	int argIntv = 2;
	int argOffset = 3;

	apr_app_initialize(&argc, &argv, NULL);
	atexit(apr_terminate);

	apr_pool_create(&pool, NULL);
	if ((argc > 2) && (strcmp(argv[1], "-l") == 0)) {
	argBase++;
	argFile += argBase;
	argIntv += argBase;
	argOffset += argBase;
	use_localtime = 1;
	}
	if (argc < (argBase + 3) \|\| argc > (argBase + 4)) {
	fprintf(stderr,
	"Usage: %s [-l] <logfile> <rotation time in seconds> "
	"[offset minutes from UTC] or <rotation size in megabytes>\n\n",
	argv[0]);
	#ifdef OS2
	fprintf(stderr,
	"Add this:\n\nTransferLog \"\|%s.exe /some/where 86400\"\n\n",
	argv[0]);
	#else
	fprintf(stderr,
	"Add this:\n\nTransferLog \"\|%s /some/where 86400\"\n\n",
	argv[0]);
	fprintf(stderr,
	"or \n\nTransferLog \"\|%s /some/where 5M\"\n\n", argv[0]);
	#endif
	fprintf(stderr,
	"to httpd.conf. The generated name will be /some/where.nnnn "
	"where nnnn is the\nsystem time at which the log nominally "
	"starts (N.B. if using a rotation time,\nthe time will always "
	"be a multiple of the rotation time, so you can synchronize\n"
	"cron scripts with it). At the end of each rotation time or "
	"when the file size\nis reached a new log is started.\n");
	exit(1);
	}

	szLogRoot = argv[argFile];

	ptr = strchr(argv[argIntv], 'M');
	if (ptr) {
	if (*(ptr+1) == '\0') {
	sRotation = atoi(argv[argIntv]) * 1048576;
	}
	if (sRotation == 0) {
	fprintf(stderr, "Invalid rotation size parameter\n");
	exit(1);
	}
	}
	else {
	if (argc >= (argBase + 4)) {
	utc_offset = atoi(argv[argOffset]) * 60;
	}
	tRotation = atoi(argv[argIntv]);
	if (tRotation <= 0) {
	fprintf(stderr, "Rotation time must be > 0\n");
	exit(6);
	}
	}

	use_strftime = (strchr(szLogRoot, '%') != NULL);
	if (apr_file_open_stdin(&f_stdin, pool) != APR_SUCCESS) {
	fprintf(stderr, "Unable to open stdin\n");
	exit(1);
	}

	for (;;) {
	nRead = sizeof(buf);
	if (apr_file_read(f_stdin, buf, &nRead) != APR_SUCCESS) {
	exit(3);
	}
	if (tRotation) {
	/*
	* Check for our UTC offset every time through the loop, since
	* it might change if there's a switch between standard and
	* daylight savings time.
	*/
	if (use_localtime) {
	apr_time_exp_t lt;
	apr_time_exp_lt(&lt, apr_time_now());
	utc_offset = lt.tm_gmtoff;
	}
	now = (int)(apr_time_now() / APR_USEC_PER_SEC) + utc_offset;
	if (nLogFD != NULL && now >= tLogEnd) {
	nLogFDprev = nLogFD;
	nLogFD = NULL;
	}
	}
	else if (sRotation) {
	apr_finfo_t finfo;
	apr_off_t current_size = -1;

	if ((nLogFD != NULL) &&
	(apr_file_info_get(&finfo, APR_FINFO_SIZE, nLogFD) == APR_SUCCESS)) {
	current_size = finfo.size;
	}

	if (current_size > sRotation) {
	nLogFDprev = nLogFD;
	nLogFD = NULL;
	}
	}
	else {
	fprintf(stderr, "No rotation time or size specified\n");
	exit(2);
	}

	if (nLogFD == NULL) {
	int tLogStart;

	if (tRotation) {
	tLogStart = (now / tRotation) * tRotation;
	}
	else {
	tLogStart = (int)apr_time_sec(apr_time_now());
	}

	if (use_strftime) {
	apr_time_t tNow = apr_time_from_sec(tLogStart);
	apr_time_exp_t e;
	apr_size_t rs;

	apr_time_exp_gmt(&e, tNow);
	apr_strftime(buf2, &rs, sizeof(buf2), szLogRoot, &e);
	}
	else {
	sprintf(buf2, "%s.%010d", szLogRoot, tLogStart);
	}
	tLogEnd = tLogStart + tRotation;
	apr_file_open(&nLogFD, buf2, APR_READ \| APR_WRITE \| APR_CREATE \| APR_APPEND,
	APR_OS_DEFAULT, pool);
	if (nLogFD == NULL) {
	/* Uh-oh. Failed to open the new log file. Try to clear
	* the previous log file, note the lost log entries,
	* and keep on truckin'. */
	if (nLogFDprev == NULL) {
	fprintf(stderr, "1 Previous file handle doesn't exists %s\n", buf2);
	exit(2);
	}
	else {
	nLogFD = nLogFDprev;
	sprintf(errbuf,
	"Resetting log file due to error opening "
	"new log file. %10d messages lost.\n",
	nMessCount);
	nWrite = strlen(errbuf);
	apr_file_trunc(nLogFD, 0);
	if (apr_file_write(nLogFD, errbuf, &nWrite) != APR_SUCCESS) {
	fprintf(stderr, "Error writing to the file %s\n", buf2);
	exit(2);
	}
	}
	}
	else if (nLogFDprev) {
	apr_file_close(nLogFDprev);
	}
	nMessCount = 0;
	}
	nWrite = nRead;
	apr_file_write(nLogFD, buf, &nWrite);
	if (nWrite != nRead) {
	nMessCount++;
	sprintf(errbuf,
	"Error writing to log file. "
	"%10d messages lost.\n",
	nMessCount);
	nWrite = strlen(errbuf);
	apr_file_trunc(nLogFD, 0);
	if (apr_file_write(nLogFD, errbuf, &nWrite) != APR_SUCCESS) {
	fprintf(stderr, "Error writing to the file %s\n", buf2);
	exit(2);
	}
	}
	else {
	nMessCount++;
	}
	}
	/* Of course we never, but prevent compiler warnings */
	return 0;
	}