Update FSF postal mail address.
[platform/upstream/coreutils.git] / src / tee.c
1 /* tee - read from standard input and write to standard output and files.
2    Copyright (C) 85,1990-2005 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 Foundation,
16    Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.  */
17
18 /* Mike Parker, Richard M. Stallman, and David MacKenzie */
19
20 #include <config.h>
21 #include <stdio.h>
22 #include <sys/types.h>
23 #include <signal.h>
24 #include <getopt.h>
25
26 #include "system.h"
27 #include "error.h"
28 #include "stdio-safer.h"
29
30 /* The official name of this program (e.g., no `g' prefix).  */
31 #define PROGRAM_NAME "tee"
32
33 #define AUTHORS "Mike Parker", "Richard M. Stallman", "David MacKenzie"
34
35 static bool tee (int nfiles, const char **files);
36
37 /* If true, append to output files rather than truncating them. */
38 static bool append;
39
40 /* If true, ignore interrupts. */
41 static bool ignore_interrupts;
42
43 /* The name that this program was run with. */
44 char *program_name;
45
46 static struct option const long_options[] =
47 {
48   {"append", no_argument, NULL, 'a'},
49   {"ignore-interrupts", no_argument, NULL, 'i'},
50   {GETOPT_HELP_OPTION_DECL},
51   {GETOPT_VERSION_OPTION_DECL},
52   {NULL, 0, NULL, 0}
53 };
54
55 void
56 usage (int status)
57 {
58   if (status != EXIT_SUCCESS)
59     fprintf (stderr, _("Try `%s --help' for more information.\n"),
60              program_name);
61   else
62     {
63       printf (_("Usage: %s [OPTION]... [FILE]...\n"), program_name);
64       fputs (_("\
65 Copy standard input to each FILE, and also to standard output.\n\
66 \n\
67   -a, --append              append to the given FILEs, do not overwrite\n\
68   -i, --ignore-interrupts   ignore interrupt signals\n\
69 "), stdout);
70       fputs (HELP_OPTION_DESCRIPTION, stdout);
71       fputs (VERSION_OPTION_DESCRIPTION, stdout);
72       fputs (_("\
73 \n\
74 If a FILE is -, copy again to standard output.\n\
75 "), stdout);
76       printf (_("\nReport bugs to <%s>.\n"), PACKAGE_BUGREPORT);
77     }
78   exit (status);
79 }
80
81 int
82 main (int argc, char **argv)
83 {
84   bool ok;
85   int optc;
86
87   initialize_main (&argc, &argv);
88   program_name = argv[0];
89   setlocale (LC_ALL, "");
90   bindtextdomain (PACKAGE, LOCALEDIR);
91   textdomain (PACKAGE);
92
93   atexit (close_stdout);
94
95   append = false;
96   ignore_interrupts = false;
97
98   while ((optc = getopt_long (argc, argv, "ai", long_options, NULL)) != -1)
99     {
100       switch (optc)
101         {
102         case 'a':
103           append = true;
104           break;
105
106         case 'i':
107           ignore_interrupts = true;
108           break;
109
110         case_GETOPT_HELP_CHAR;
111
112         case_GETOPT_VERSION_CHAR (PROGRAM_NAME, AUTHORS);
113
114         default:
115           usage (EXIT_FAILURE);
116         }
117     }
118
119   if (ignore_interrupts)
120     signal (SIGINT, SIG_IGN);
121
122   /* Do *not* warn if tee is given no file arguments.
123      POSIX requires that it work when given no arguments.  */
124
125   ok = tee (argc - optind, (const char **) &argv[optind]);
126   if (close (STDIN_FILENO) != 0)
127     error (EXIT_FAILURE, errno, _("standard input"));
128
129   exit (ok ? EXIT_SUCCESS : EXIT_FAILURE);
130 }
131
132 /* Copy the standard input into each of the NFILES files in FILES
133    and into the standard output.
134    Return true if successful.  */
135
136 static bool
137 tee (int nfiles, const char **files)
138 {
139   FILE **descriptors;
140   char buffer[BUFSIZ];
141   ssize_t bytes_read;
142   int i;
143   bool ok = true;
144   const char *mode_string = (append ? "a" : "w");
145
146   descriptors = xnmalloc (nfiles + 1, sizeof *descriptors);
147
148   /* Move all the names `up' one in the argv array to make room for
149      the entry for standard output.  This writes into argv[argc].  */
150   for (i = nfiles; i >= 1; i--)
151     files[i] = files[i - 1];
152
153   SET_BINARY2 (0, 1);
154
155   /* In the array of NFILES + 1 descriptors, make
156      the first one correspond to standard output.   */
157   descriptors[0] = stdout;
158   files[0] = _("standard output");
159   SETVBUF (stdout, NULL, _IONBF, 0);
160
161   for (i = 1; i <= nfiles; i++)
162     {
163       descriptors[i] = (STREQ (files[i], "-")
164                         ? stdout
165                         : fopen_safer (files[i], mode_string));
166       if (descriptors[i] == NULL)
167         {
168           error (0, errno, "%s", files[i]);
169           ok = false;
170         }
171       else
172         {
173           SETVBUF (descriptors[i], NULL, _IONBF, 0);
174           SET_BINARY (fileno (descriptors[i]));
175         }
176     }
177
178   while (1)
179     {
180       bytes_read = read (0, buffer, sizeof buffer);
181 #ifdef EINTR
182       if (bytes_read < 0 && errno == EINTR)
183         continue;
184 #endif
185       if (bytes_read <= 0)
186         break;
187
188       /* Write to all NFILES + 1 descriptors.
189          Standard output is the first one.  */
190       for (i = 0; i <= nfiles; i++)
191         if (descriptors[i]
192             && fwrite (buffer, 1, bytes_read, descriptors[i]) != bytes_read)
193           {
194             error (0, errno, "%s", files[i]);
195             descriptors[i] = NULL;
196             ok = false;
197           }
198     }
199
200   if (bytes_read == -1)
201     {
202       error (0, errno, _("read error"));
203       ok = false;
204     }
205
206   /* Close the files, but not standard output.  */
207   for (i = 1; i <= nfiles; i++)
208     if (!STREQ (files[i], "-")
209         && descriptors[i] && fclose (descriptors[i]) != 0)
210       {
211         error (0, errno, "%s", files[i]);
212         ok = false;
213       }
214
215   free (descriptors);
216
217   return ok;
218 }