usr.sbin/clog/clog.c

   1 /*-
   2  * Copyright (c) 2001
   3  *     Jeff Wheelhouse (jdw@wwwi.com)
   4  *
   5  * This code was originally developed by Jeff Wheelhouse (jdw@wwwi.com).
   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  * 1. Redistribution of source code must retail the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY JEFF WHEELHOUSE ``AS IS'' AND ANY EXPRESS OR
  17  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN
  19  * NO EVENT SHALL JEFF WHEELHOUSE BE LIABLE FOR ANY DIRECT, INDIRECT,
  20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING BUT NOT
  21  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA,
  22  * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  23  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  24  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
  25  * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26  *
  27  * $Id: clog.c,v 1.3 2001/10/02 18:51:26 jdw Exp $
  28  * $DragonFly: src/usr.sbin/clog/clog.c,v 1.4 2007/06/26 23:30:05 josepht Exp $
  29  */
  30
  31
  32 #include <assert.h>
  33 #include <errno.h>
  34 #include <fcntl.h>
  35 #include <poll.h>
  36 #include <signal.h>
  37 #include <stdio.h>
  38 #include <stdlib.h>
  39 #include <string.h>
  40 #include <unistd.h>
  41
  42 #include <sys/mman.h>
  43 #include <sys/sched.h>
  44 #include <sys/stat.h>
  45 #include <sys/uio.h>
  46
  47
  48 #include "clog.h"
  49
  50
  51 /*
  52  *  The BUFFER_SIZE value is just used to allocate a buffer full of NULLs
  53  *  so that a new logfile can be extended to its full size.
  54  *
  55  *  Compiling with -pedantic complains when the buffer array is declared
  56  *  if I declare this as a const instead of a #define.
  57  */
  58 #define BUFFER_SIZE 16384
  59
  60 void init_log(const char *lname, size_t size);
  61 void read_log(const char *lname, int optf);
  62 void usage(void);
  63
  64 const char *pname;
  65
  66 int
  67 main(int argc, char **argv)
  68 {
  69         int ch;
  70         int init = 0;
  71         int size = 0;
  72         int optf = 0;
  73
  74         pname = argv[0];
  75
  76         while ((ch = getopt(argc, argv, "fis:")) != -1)
  77                 switch(ch) {
  78                 case 'i':
  79                         init = 1;
  80                         break;
  81                 case 's':
  82                         size = atol(optarg);
  83                         if (size==0) usage();
  84                         break;
  85                 case 'f':
  86                         optf = 1;
  87                 }
  88
  89         if ((size>0)&&(init==0)) {
  90                 fprintf(stderr,"%s: WARNING: -s argument ignored without -i.\n",pname);
  91                 size = 0;
  92         }
  93         if (argv[optind]==NULL) {
  94                 fprintf(stderr,"%s: ERROR: log_file argument must be specified.\n",pname);
  95                 usage();
  96         }
  97         if ((init==1)&&(size==0)) {
  98                 fprintf(stderr,"%s: ERROR: -i argument requires -s.\n",pname);
  99                 usage();
 100         }
 101         if ((init==1)&&(optf==1)) {
 102                 fprintf(stderr,"%s: ERROR: flags -f and -i are incompatible.\n",pname);
 103                 usage();
 104         }
 105
 106         if (init==1) init_log(argv[optind],size);
 107         /* if (optf==1) follow_log(artv[optind]); */
 108         read_log(argv[optind],optf);
 109
 110         return 0;
 111 }
 112
 113
 114 void
 115 usage(void)
 116 {
 117   fprintf(stderr,"usage: %s [-i -s log_size] [ -f ] log_file\n",pname);
 118   exit(1);
 119 }
 120
 121
 122 void
 123 read_log(const char *lname, int optf)
 124 {
 125         int fd;
 126         struct stat sb;
 127         struct clog_footer *pcf;
 128         char *pbuffer;
 129         struct iovec iov[2];
 130         int iovcnt = 0;
 131         uint32_t start = 0;
 132         uint32_t next;
 133         struct pollfd pfd;
 134
 135         pfd.fd = -1;
 136
 137         fd = open(lname,O_RDONLY);
 138         if (fd==-1) {
 139                 fprintf(stderr,"%s: ERROR: could not open %s (%s)\n",pname,lname,strerror(errno));
 140                 exit(11);
 141         }
 142
 143         if (fstat(fd,&sb)==-1) {
 144                 fprintf(stderr,"%s: ERROR: could not stat %s (%s)\n",pname,lname,strerror(errno));
 145                 exit(13);
 146         }
 147         pbuffer = mmap(NULL,sb.st_size,PROT_READ,MAP_SHARED,fd,0);
 148         if (pbuffer==NULL) {
 149                 fprintf(stderr,"%s: ERROR: could not mmap %s body (%s)\n",pname,lname,strerror(errno));
 150                 exit(14);
 151         }
 152         pcf = (struct clog_footer*)(pbuffer + sb.st_size - sizeof(struct clog_footer));
 153
 154         if (pcf->cf_wrap==1) start = pcf->cf_next + 1;
 155         while(1) {
 156                 while(pcf->cf_lock==1) sched_yield();
 157                 next = pcf->cf_next;
 158                 iovcnt = 0;
 159                 if (start>next) {
 160                         iov[iovcnt].iov_base = pbuffer + start;
 161                         iov[iovcnt++].iov_len = pcf->cf_max - start;
 162                         start = 0;
 163                 }
 164                 iov[iovcnt].iov_base = pbuffer + start;
 165                 iov[iovcnt++].iov_len = next - start;
 166                 if (writev(1,iov,iovcnt)==-1) {
 167                         fprintf(stderr,"%s: ERROR: could not write output (%s)\n",pname,strerror(errno));
 168                         exit(15);
 169                 }
 170                 start = next;
 171                 if (optf==0) break;
 172                 if (poll(&pfd,1,50)==-1) {
 173                         fprintf(stderr,"%s: ERROR: could not poll (%s)\n",pname,strerror(errno));
 174                         exit(16);
 175                 }
 176         }
 177
 178         munmap(pbuffer,sb.st_size);
 179         close(fd);
 180
 181         exit(0);
 182 }
 183
 184
 185 void
 186 init_log(const char *lname, size_t size)
 187 {
 188         int fd;
 189         size_t fill = size;
 190         char buffer[BUFFER_SIZE];
 191         struct clog_footer cf;
 192
 193         memcpy(&cf.cf_magic,MAGIC_CONST,4);
 194         cf.cf_max = size - sizeof(struct clog_footer);
 195
 196         memset(buffer,0,BUFFER_SIZE);
 197
 198         fd = open(lname,O_RDWR|O_CREAT,0666);
 199         if (fd==-1) {
 200                 fprintf(stderr,"%s: ERROR: could not open %s (%s)\n",pname,lname,strerror(errno));
 201                 exit(2);
 202         }
 203         if (ftruncate(fd,(off_t)0)==-1) {
 204                 fprintf(stderr,"%s: ERROR: could not truncate %s (%s)\n",pname,lname,strerror(errno));
 205                 exit(3);
 206         }
 207
 208         while(fill>BUFFER_SIZE) {
 209                 if (write(fd,buffer,BUFFER_SIZE)==-1){
 210                         fprintf(stderr,"%s: ERROR: could not write %s (%s)\n",pname,lname,strerror(errno));
 211                         exit(4);
 212                 }
 213                 fill -= BUFFER_SIZE;
 214         }
 215         assert(fill<=BUFFER_SIZE);
 216         if (fill>0) {
 217                 if (write(fd,buffer,fill)==-1) {
 218                         fprintf(stderr,"%s: ERROR: could not write %s (%s)\n",pname,lname,strerror(errno));
 219                         exit(5);
 220                 }
 221         }
 222         if (lseek(fd,-(off_t)(sizeof(struct clog_footer)),SEEK_END)==-1) {
 223                 fprintf(stderr,"%s: ERROR: could not seek in %s (%s)\n",pname,lname,strerror(errno));
 224                 exit(6);
 225         }
 226         if (write(fd,&cf,sizeof(cf))==-1) {
 227                 fprintf(stderr,"%s: ERROR: could not write magic in %s (%s)\n",pname,lname,strerror(errno));
 228                 exit(7);
 229         }
 230         close(fd);
 231         exit(0);
 232 }
 233
 234