userapps/opensource/busybox/coreutils/tee.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * tee implementation for busybox
   4  *
   5  * Copyright (C) 2003  Manuel Novoa III  <mjn3@codepoet.org>
   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 /* BB_AUDIT SUSv3 compliant */
  24 /* http://www.opengroup.org/onlinepubs/007904975/utilities/tee.html */
  25
  26 #include <stdio.h>
  27 #include <stdlib.h>
  28 #include <signal.h>
  29 #include <unistd.h>
  30 #include "busybox.h"
  31
  32 int tee_main(int argc, char **argv)
  33 {
  34         const char *mode = "w\0a";
  35         FILE **files;
  36         FILE **p;
  37         char **filenames;
  38         int flags;
  39         int retval = EXIT_SUCCESS;
  40 #ifdef CONFIG_FEATURE_TEE_USE_BLOCK_IO
  41         ssize_t c;
  42         RESERVE_CONFIG_BUFFER(buf, BUFSIZ);
  43 #else
  44         int c;
  45 #endif
  46
  47         flags = bb_getopt_ulflags(argc, argv, "ia");    /* 'a' must be 2nd */
  48
  49         mode += (flags & 2);    /* Since 'a' is the 2nd option... */
  50
  51         if (flags & 1) {
  52                 signal(SIGINT, SIG_IGN);        /* TODO - switch to sigaction.*/
  53         }
  54
  55         /* gnu tee ignores SIGPIPE in case one of the output files is a pipe
  56          * that doesn't consume all its input.  Good idea... */
  57         signal(SIGPIPE, SIG_IGN);               /* TODO - switch to sigaction.*/
  58
  59         /* Allocate an array of FILE *'s, with one extra for a sentinal. */
  60         p = files = (FILE **)xmalloc(sizeof(FILE *) * (argc - optind + 2));
  61         *p = stdout;
  62         argv += optind - 1;
  63         filenames = argv - 1;
  64         *filenames = (char *) bb_msg_standard_input;    /* for later */
  65         goto GOT_NEW_FILE;
  66
  67         do {
  68                 if ((*p = bb_wfopen(*argv, mode)) == NULL) {
  69                         retval = EXIT_FAILURE;
  70                         continue;
  71                 }
  72                 filenames[(int)(p - files)] = *argv;
  73         GOT_NEW_FILE:
  74                 setbuf(*p, NULL);       /* tee must not buffer output. */
  75                 ++p;
  76         } while (*++argv);
  77
  78         *p = NULL;                              /* Store the sentinal value. */
  79
  80 #ifdef CONFIG_FEATURE_TEE_USE_BLOCK_IO
  81         while ((c = safe_read(STDIN_FILENO, buf, BUFSIZ)) > 0) {
  82                 for (p=files ; *p ; p++) {
  83                         fwrite(buf, 1, c, *p);
  84                 }
  85         }
  86
  87         if (c < 0) {                    /* Make sure read errors are signaled. */
  88                 retval = EXIT_FAILURE;
  89         }
  90
  91 #ifdef CONFIG_FEATURE_CLEAN_UP
  92         RELEASE_CONFIG_BUFFER(buf);
  93 #endif
  94
  95 #else
  96         setvbuf(stdout, NULL, _IONBF, 0);
  97         while ((c = getchar()) != EOF) {
  98                 for (p=files ; *p ; p++) {
  99                         putc(c, *p);
 100                 }
 101         }
 102 #endif
 103
 104         /* Now we need to check for i/o errors on stdin and the various
 105          * output files.  Since we know that the first entry in the output
 106          * file table is stdout, we can save one "if ferror" test by
 107          * setting the first entry to stdin and checking stdout error
 108          * status with bb_fflush_stdout_and_exit()... although fflush()ing
 109          * is unnecessary here. */
 110
 111         p = files;
 112         *p = stdin;
 113         do {            /* Now check for (input and) output errors. */
 114                 /* Checking ferror should be sufficient, but we may want to fclose.
 115                  * If we do, remember not to close stdin! */
 116                 bb_xferror(*p, filenames[(int)(p - files)]);
 117         } while (*++p);
 118
 119         bb_fflush_stdout_and_exit(retval);
 120 }