merge with 1.8.1i
[platform/upstream/coreutils.git] / 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 void xwrite ();
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   if (nfiles)
168     descriptors = (int *) xmalloc (nfiles * sizeof (int));
169   mode = O_WRONLY | O_CREAT;
170   if (append)
171     mode |= O_APPEND;
172   else
173     mode |= O_TRUNC;
174
175   for (i = 0; i < nfiles; i++)
176     {
177       descriptors[i] = open (files[i], mode, 0666);
178       if (descriptors[i] == -1)
179         {
180           error (0, errno, "%s", files[i]);
181           ret = 1;
182         }
183     }
184
185   while ((bytes_read = read (0, buffer, sizeof buffer)) > 0)
186     {
187       xwrite (1, buffer, bytes_read);
188       for (i = 0; i < nfiles; i++)
189         if (descriptors[i] != -1)
190           xwrite (descriptors[i], buffer, bytes_read);
191     }
192   if (bytes_read == -1)
193     {
194       error (0, errno, "read error");
195       ret = 1;
196     }
197
198   for (i = 0; i < nfiles; i++)
199     if (descriptors[i] != -1 && close (descriptors[i]) != 0)
200       {
201         error (0, errno, "%s", files[i]);
202         ret = 1;
203       }
204
205   if (nfiles)
206     free (descriptors);
207
208   return ret;
209 }