coreutils/tee.c

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