src/tee.c

   1 /* tee - read from standard input and write to standard output and files.
   2    Copyright (C) 1985, 1990, 1991 Free Software Foundation, Inc.
   3
   4    This program is free software; you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation; either version 2, or (at your option)
   7    any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program; if not, write to the Free Software
  16    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  17
  18 /* Mike Parker, Richard M. Stallman, and David MacKenzie */
  19
  20 #ifdef HAVE_CONFIG_H
  21 #if defined (CONFIG_BROKETS)
  22 /* We use <config.h> instead of "config.h" so that a compilation
  23    using -I. -I$srcdir will use ./config.h rather than $srcdir/config.h
  24    (which it would do because it found this file in $srcdir).  */
  25 #include <config.h>
  26 #else
  27 #include "config.h"
  28 #endif
  29 #endif
  30
  31 #include <stdio.h>
  32 #include <sys/types.h>
  33 #include <signal.h>
  34 #include <getopt.h>
  35
  36 #include "system.h"
  37 #include "version.h"
  38
  39 char *xmalloc ();
  40 void error ();
  41 int safe_write ();
  42
  43 static int tee ();
  44
  45 /* If nonzero, append to output files rather than truncating them. */
  46 static int append;
  47
  48 /* If nonzero, ignore interrupts. */
  49 static int ignore_interrupts;
  50
  51 /* The name that this program was run with. */
  52 char *program_name;
  53
  54 /* If non-zero, display usage information and exit.  */
  55 static int show_help;
  56
  57 /* If non-zero, print the version on standard output and exit.  */
  58 static int show_version;
  59
  60 static struct option const long_options[] =
  61 {
  62   {"append", no_argument, NULL, 'a'},
  63   {"help", no_argument, &show_help, 1},
  64   {"ignore-interrupts", no_argument, NULL, 'i'},
  65   {"version", no_argument, &show_version, 1},
  66   {NULL, 0, NULL, 0}
  67 };
  68
  69 static void
  70 usage (status)
  71      int status;
  72 {
  73   if (status != 0)
  74     fprintf (stderr, "Try `%s --help' for more information.\n",
  75              program_name);
  76   else
  77     {
  78       printf ("Usage: %s [OPTION]... [FILE]...\n", program_name);
  79       printf ("\
  80 \n\
  81   -a, --append              append to the given FILEs, do not overwrite\n\
  82   -i, --ignore-interrupts   ignore interrupt signals\n\
  83       --help                display this help and exit\n\
  84       --version             output version information and exit\n\
  85 ");
  86     }
  87   exit (status);
  88 }
  89
  90 void
  91 main (argc, argv)
  92      int argc;
  93      char **argv;
  94 {
  95   int errs;
  96   int optc;
  97
  98   program_name = argv[0];
  99   append = 0;
 100   ignore_interrupts = 0;
 101
 102   while ((optc = getopt_long (argc, argv, "ai", long_options, (int *) 0))
 103          != EOF)
 104     {
 105       switch (optc)
 106         {
 107         case 0:
 108           break;
 109
 110         case 'a':
 111           append = 1;
 112           break;
 113
 114         case 'i':
 115           ignore_interrupts = 1;
 116           break;
 117
 118         default:
 119           usage (1);
 120         }
 121     }
 122
 123   if (show_version)
 124     {
 125       printf ("%s\n", version_string);
 126       exit (0);
 127     }
 128
 129   if (show_help)
 130     usage (0);
 131
 132   if (ignore_interrupts)
 133     {
 134 #ifdef _POSIX_VERSION
 135       struct sigaction sigact;
 136
 137       sigact.sa_handler = SIG_IGN;
 138       sigemptyset (&sigact.sa_mask);
 139       sigact.sa_flags = 0;
 140       sigaction (SIGINT, &sigact, NULL);
 141 #else                           /* !_POSIX_VERSION */
 142       signal (SIGINT, SIG_IGN);
 143 #endif                          /* _POSIX_VERSION */
 144     }
 145
 146   errs = tee (argc - optind, &argv[optind]);
 147   if (close (0) != 0)
 148     error (1, errno, "standard input");
 149   if (close (1) != 0)
 150     error (1, errno, "standard output");
 151   exit (errs);
 152 }
 153
 154 /* Copy the standard input into each of the NFILES files in FILES
 155    and into the standard output.
 156    Return 0 if successful, 1 if any errors occur. */
 157
 158 static int
 159 tee (nfiles, files)
 160      int nfiles;
 161      char **files;
 162 {
 163   int *descriptors;
 164   char buffer[BUFSIZ];
 165   register int bytes_read, i, ret = 0, mode;
 166
 167   descriptors = (int *) xmalloc ((nfiles + 1) * sizeof (int));
 168
 169   mode = O_WRONLY | O_CREAT;
 170   if (append)
 171     mode |= O_APPEND;
 172   else
 173     mode |= O_TRUNC;
 174
 175   /* Move all the names `up' one in the argv array to make room for
 176      the entry for standard output.  This writes into argv[argc].  */
 177   for (i = nfiles; i >= 1; i--)
 178     files[i] = files[i - 1];
 179
 180   /* In the array of NFILES + 1 descriptors, make
 181      the first one correspond to standard output.   */
 182   descriptors[0] = 1;
 183   files[0] = "standard output";
 184
 185   for (i = 1; i <= nfiles; i++)
 186     {
 187       descriptors[i] = open (files[i], mode, 0666);
 188       if (descriptors[i] == -1)
 189         {
 190           error (0, errno, "%s", files[i]);
 191           ret = 1;
 192         }
 193     }
 194
 195   while (1)
 196     {
 197       bytes_read = read (0, buffer, sizeof buffer);
 198 #ifdef EINTR
 199       if (bytes_read < 0 && errno == EINTR)
 200         continue;
 201 #endif
 202       if (bytes_read <= 0)
 203         break;
 204
 205       /* Write to all NFILES + 1 descriptors.
 206          Standard output is the first one.  */
 207       for (i = 0; i <= nfiles; i++)
 208         {
 209           if (descriptors[i] != -1
 210               && safe_write (descriptors[i], buffer, bytes_read))
 211             {
 212               error (0, errno, "%s", files[i]);
 213               /* Don't close stdout.  That's done in main.  */
 214               if (descriptors[i] != 1)
 215                 close (descriptors[i]);
 216               descriptors[i] = -1;
 217               ret = 1;
 218             }
 219         }
 220     }
 221
 222   if (bytes_read == -1)
 223     {
 224       error (0, errno, "read error");
 225       ret = 1;
 226     }
 227
 228   /* Close the files, but not standard output.  */
 229   for (i = 1; i <= nfiles; i++)
 230     if (descriptors[i] != -1 && close (descriptors[i]) != 0)
 231       {
 232         error (0, errno, "%s", files[i]);
 233         ret = 1;
 234       }
 235
 236   free (descriptors);
 237
 238   return ret;
 239 }