Small size optimization from Aaron Lehmann
[oweals/busybox.git] / tee.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini tee implementation for busybox
4  *
5  * Copyright (C) 1999,2000,2001 by Lineo, inc.
6  * Written by Matt Kraai <kraai@alumni.carnegiemellon.edu>
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2 of the License, or
11  * (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
21  *
22  */
23
24 #include "busybox.h"
25 #include <getopt.h>
26 #include <stdio.h>
27
28 int
29 tee_main(int argc, char **argv)
30 {
31         char *mode = "w";
32         int c, i, status = 0, nfiles = 0;
33         FILE **files;
34
35         while ((c = getopt(argc, argv, "a")) != EOF) {
36                 switch (c) {
37                 case 'a': 
38                         mode = "a";
39                         break;
40                 default:
41                         show_usage();
42                 }
43         }
44
45         files = (FILE **)xmalloc(sizeof(FILE *) * (argc - optind + 1));
46         files[nfiles++] = stdout;
47         while (optind < argc) {
48                 if ((files[nfiles++] = fopen(argv[optind++], mode)) == NULL) {
49                         nfiles--;
50                         perror_msg("%s", argv[optind-1]);
51                         status = 1;
52                 }
53         }
54
55         while ((c = getchar()) != EOF)
56                 for (i = 0; i < nfiles; i++)
57                         putc(c, files[i]);
58
59         return status;
60 }
61
62 /*
63 Local Variables:
64 c-file-style: "linux"
65 c-basic-offset: 4
66 tab-width: 4
67 End:
68 */