--- /dev/null
+/* Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You 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(<, 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_APPEND | APR_WRITE | APR_CREATE | APR_LARGEFILE,
+ 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;
+}
Code Review / bottlenecks.git / blobdiff