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