granicus.if.org Git - apache/blob - support/rotatelogs.c

   1 /* ====================================================================
   2  * The Apache Software License, Version 1.1
   3  *
   4  * Copyright (c) 2000-2001 The Apache Software Foundation.  All rights
   5  * reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  *
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  *
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in
  16  *    the documentation and/or other materials provided with the
  17  *    distribution.
  18  *
  19  * 3. The end-user documentation included with the redistribution,
  20  *    if any, must include the following acknowledgment:
  21  *       "This product includes software developed by the
  22  *        Apache Software Foundation (http://www.apache.org/)."
  23  *    Alternately, this acknowledgment may appear in the software itself,
  24  *    if and wherever such third-party acknowledgments normally appear.
  25  *
  26  * 4. The names "Apache" and "Apache Software Foundation" must
  27  *    not be used to endorse or promote products derived from this
  28  *    software without prior written permission. For written
  29  *    permission, please contact apache@apache.org.
  30  *
  31  * 5. Products derived from this software may not be called "Apache",
  32  *    nor may "Apache" appear in their name, without prior written
  33  *    permission of the Apache Software Foundation.
  34  *
  35  * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
  36  * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  37  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  38  * DISCLAIMED.  IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
  39  * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  40  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  41  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  42  * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  43  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  44  * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  45  * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  46  * SUCH DAMAGE.
  47  * ====================================================================
  48  *
  49  * This software consists of voluntary contributions made by many
  50  * individuals on behalf of the Apache Software Foundation.  For more
  51  * information on the Apache Software Foundation, please see
  52  * <http://www.apache.org/>.
  53  */
  54
  55 /*
  56  * Simple program to rotate Apache logs without having to kill the server.
  57  *
  58  * Contributed by Ben Laurie <ben@algroup.co.uk>
  59  *
  60  * 12 Mar 1996
  61  */
  62
  63
  64 #include "apr.h"
  65 #include <stdlib.h>
  66 #include <time.h>
  67 #include <errno.h>
  68 #if APR_HAVE_STDIO_H
  69 #include <stdio.h>
  70 #endif
  71 #if APR_HAVE_UNISTD_H
  72 #include <unistd.h>
  73 #endif
  74 #if APR_HAVE_IO_H
  75 #include <io.h>
  76 #endif
  77 #if APR_HAVE_FCNTL_H
  78 #include <fcntl.h>
  79 #endif
  80
  81 #define BUFSIZE         65536
  82 #define ERRMSGSZ        82
  83
  84 #ifndef MAX_PATH
  85 #define MAX_PATH        1024
  86 #endif
  87
  88 int main (int argc, char *argv[])
  89 {
  90     char buf[BUFSIZE], buf2[MAX_PATH], errbuf[ERRMSGSZ];
  91     time_t tLogEnd = 0, tRotation;
  92     int nLogFD = -1, nLogFDprev = -1, nMessCount = 0, nRead, nWrite;
  93     char *szLogRoot;
  94
  95     if (argc != 3) {
  96         fprintf(stderr,
  97                 "%s <logfile> <rotation time in seconds>\n\n",
  98                 argv[0]);
  99 #ifdef OS2
 100         fprintf(stderr,
 101                 "Add this:\n\nTransferLog \"|%s.exe /some/where 86400\"\n\n",
 102                 argv[0]);
 103 #else
 104         fprintf(stderr,
 105                 "Add this:\n\nTransferLog \"|%s /some/where 86400\"\n\n",
 106                 argv[0]);
 107 #endif
 108         fprintf(stderr,
 109                 "to httpd.conf. The generated name will be /some/where.nnnn "
 110                 "where nnnn is the\nsystem time at which the log nominally "
 111                 "starts (N.B. this time will always be a\nmultiple of the "
 112                 "rotation time, so you can synchronize cron scripts with it).\n"
 113                 "At the end of each rotation time a new log is started.\n");
 114         exit(1);
 115     }
 116
 117     szLogRoot = argv[1];
 118     tRotation = atoi(argv[2]);
 119     if (tRotation <= 0) {
 120         fprintf(stderr, "Rotation time must be > 0\n");
 121         exit(6);
 122     }
 123
 124     for (;;) {
 125         nRead = read(0, buf, sizeof buf);
 126         if (nRead == 0)
 127             exit(3);
 128         if (nRead < 0)
 129             if (errno != EINTR)
 130                 exit(4);
 131         if (nLogFD >= 0 && (time(NULL) >= tLogEnd || nRead < 0)) {
 132             nLogFDprev = nLogFD;
 133             nLogFD = -1;
 134         }
 135         if (nLogFD < 0) {
 136             time_t tLogStart = (time(NULL) / tRotation) * tRotation;
 137             sprintf(buf2, "%s.%010d", szLogRoot, (int) tLogStart);
 138             tLogEnd = tLogStart + tRotation;
 139             nLogFD = open(buf2, O_WRONLY | O_CREAT | O_APPEND, 0666);
 140             if (nLogFD < 0) {
 141                 /* Uh-oh. Failed to open the new log file. Try to clear
 142                  * the previous log file, note the lost log entries,
 143                  * and keep on truckin'. */
 144                 if (nLogFDprev == -1) {
 145                     perror(buf2);
 146                     exit(2);
 147                 }
 148                 else {
 149                     nLogFD = nLogFDprev;
 150                     sprintf(errbuf,
 151                             "Resetting log file due to error opening "
 152                             "new log file. %10d messages lost.\n",
 153                             nMessCount);
 154                     nWrite = strlen(errbuf);
 155 #ifdef WIN32
 156                     chsize(nLogFD, 0);
 157 #else
 158                     ftruncate(nLogFD, 0);
 159 #endif
 160                     write(nLogFD, errbuf, nWrite);
 161                 }
 162             }
 163             else {
 164                 close(nLogFDprev);
 165             }
 166             nMessCount = 0;
 167         }
 168         do {
 169             nWrite = write(nLogFD, buf, nRead);
 170         } while (nWrite < 0 && errno == EINTR);
 171         if (nWrite != nRead) {
 172             nMessCount++;
 173             sprintf(errbuf,
 174                     "Error writing to log file. "
 175                     "%10d messages lost.\n",
 176                     nMessCount);
 177             nWrite = strlen(errbuf);
 178 #ifdef WIN32
 179             chsize(nLogFD, 0);
 180 #else
 181             ftruncate(nLogFD, 0);
 182 #endif
 183             write (nLogFD, errbuf, nWrite);
 184         }
 185         else {
 186             nMessCount++;
 187         }
 188     }
 189     /* Of course we never, but prevent compiler warnings */
 190     return 0;
 191 }