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