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 #include <stdio.h>
  21 #include <sys/types.h>
  22 #include <signal.h>
  23 #include <getopt.h>
  24 #include "system.h"
  25
  26 char *xmalloc ();
  27 int tee ();
  28 void error ();
  29 void xwrite ();
  30
  31 /* If nonzero, append to output files rather than truncating them. */
  32 static int append;
  33
  34 /* If nonzero, ignore interrupts. */
  35 static int ignore_interrupts;
  36
  37 /* The name that this program was run with. */
  38 char *program_name;
  39
  40 static struct option long_options[] =
  41 {
  42   {"append", 0, NULL, 'a'},
  43   {"ignore-interrupts", 0, NULL, 'i'},
  44   {NULL, 0, NULL, 0}
  45 };
  46
  47 void
  48 main (argc, argv)
  49      int argc;
  50      char **argv;
  51 {
  52   int errs;
  53   int optc;
  54
  55   program_name = argv[0];
  56   append = 0;
  57   ignore_interrupts = 0;
  58
  59   while ((optc = getopt_long (argc, argv, "ai", long_options, (int *) 0))
  60          != EOF)
  61     {
  62       switch (optc)
  63         {
  64         case 'a':
  65           append = 1;
  66           break;
  67         case 'i':
  68           ignore_interrupts = 1;
  69           break;
  70         default:
  71           fprintf (stderr, "\
  72 Usage: %s [-ai] [--append] [--ignore-interrupts] [file...]\n",
  73                    program_name);
  74           exit (1);
  75         }
  76     }
  77
  78   if (ignore_interrupts)
  79 #ifdef _POSIX_VERSION
  80     {
  81       struct sigaction sigact;
  82
  83       sigact.sa_handler = SIG_IGN;
  84       sigemptyset (&sigact.sa_mask);
  85       sigact.sa_flags = 0;
  86       sigaction (SIGINT, &sigact, NULL);
  87     }
  88 #else                           /* !_POSIX_VERSION */
  89     signal (SIGINT, SIG_IGN);
  90 #endif                          /* _POSIX_VERSION */
  91
  92   errs = tee (argc - optind, &argv[optind]);
  93   if (close (0) == -1)
  94     error (1, errno, "standard input");
  95   if (close (1) == -1)
  96     error (1, errno, "standard output");
  97   exit (errs);
  98 }
  99
 100 /* Copy the standard input into each of the NFILES files in FILES
 101    and into the standard output.
 102    Return 0 if successful, 1 if any errors occur. */
 103
 104 int
 105 tee (nfiles, files)
 106      int nfiles;
 107      char **files;
 108 {
 109   int *descriptors;
 110   char buffer[BUFSIZ];
 111   register int bytes_read, i, ret = 0, mode;
 112
 113   if (nfiles)
 114     descriptors = (int *) xmalloc (nfiles * sizeof (int));
 115   mode = O_WRONLY | O_CREAT;
 116   if (append)
 117     mode |= O_APPEND;
 118   else
 119     mode |= O_TRUNC;
 120
 121   for (i = 0; i < nfiles; i++)
 122     {
 123       descriptors[i] = open (files[i], mode, 0666);
 124       if (descriptors[i] == -1)
 125         {
 126           error (0, errno, "%s", files[i]);
 127           ret = 1;
 128         }
 129     }
 130
 131   while ((bytes_read = read (0, buffer, sizeof buffer)) > 0)
 132     {
 133       xwrite (1, buffer, bytes_read);
 134       for (i = 0; i < nfiles; i++)
 135         if (descriptors[i] != -1)
 136           xwrite (descriptors[i], buffer, bytes_read);
 137     }
 138   if (bytes_read == -1)
 139     {
 140       error (0, errno, "read error");
 141       ret = 1;
 142     }
 143
 144   for (i = 0; i < nfiles; i++)
 145     if (descriptors[i] != -1 && close (descriptors[i]) == -1)
 146       {
 147         error (0, errno, "%s", files[i]);
 148         ret = 1;
 149       }
 150
 151   if (nfiles)
 152     free (descriptors);
 153
 154   return ret;
 155 }