Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
John Beppu | 059f152 | 1999-12-10 05:27:16 +0000 | [diff] [blame] | 2 | /* |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 3 | * tee implementation for busybox |
John Beppu | 059f152 | 1999-12-10 05:27:16 +0000 | [diff] [blame] | 4 | * |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 5 | * Copyright (C) 2003 Manuel Novoa III <mjn3@codepoet.org> |
John Beppu | 059f152 | 1999-12-10 05:27:16 +0000 | [diff] [blame] | 6 | * |
"Robert P. J. Day" | 801ab14 | 2006-07-12 07:56:04 +0000 | [diff] [blame] | 7 | * Licensed under GPLv2 or later, see file LICENSE in this tarball for details. |
John Beppu | 059f152 | 1999-12-10 05:27:16 +0000 | [diff] [blame] | 8 | */ |
| 9 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 10 | /* BB_AUDIT SUSv3 compliant */ |
| 11 | /* http://www.opengroup.org/onlinepubs/007904975/utilities/tee.html */ |
| 12 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 13 | #include "busybox.h" |
Bernhard Reutner-Fischer | 73561cc | 2006-08-28 23:31:54 +0000 | [diff] [blame] | 14 | #include <signal.h> |
John Beppu | a3e0d79 | 1999-12-10 07:41:03 +0000 | [diff] [blame] | 15 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 16 | int tee_main(int argc, char **argv) |
John Beppu | 059f152 | 1999-12-10 05:27:16 +0000 | [diff] [blame] | 17 | { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 18 | const char *mode = "w\0a"; |
Matt Kraai | 1638488 | 2000-08-28 03:53:27 +0000 | [diff] [blame] | 19 | FILE **files; |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 20 | FILE **p; |
| 21 | char **filenames; |
| 22 | int flags; |
| 23 | int retval = EXIT_SUCCESS; |
| 24 | #ifdef CONFIG_FEATURE_TEE_USE_BLOCK_IO |
Manuel Novoa III | d709743 | 2004-05-26 15:21:19 +0000 | [diff] [blame] | 25 | ssize_t c; |
"Vladimir N. Oleynik" | 6f347ef | 2005-10-15 10:23:55 +0000 | [diff] [blame] | 26 | # define buf bb_common_bufsiz1 |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 27 | #else |
| 28 | int c; |
| 29 | #endif |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 30 | |
Denis Vlasenko | 67b23e6 | 2006-10-03 21:00:06 +0000 | [diff] [blame^] | 31 | flags = getopt32(argc, argv, "ia"); /* 'a' must be 2nd */ |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 32 | |
| 33 | mode += (flags & 2); /* Since 'a' is the 2nd option... */ |
| 34 | |
| 35 | if (flags & 1) { |
| 36 | signal(SIGINT, SIG_IGN); /* TODO - switch to sigaction.*/ |
| 37 | } |
| 38 | |
| 39 | /* gnu tee ignores SIGPIPE in case one of the output files is a pipe |
| 40 | * that doesn't consume all its input. Good idea... */ |
Bernhard Reutner-Fischer | 73561cc | 2006-08-28 23:31:54 +0000 | [diff] [blame] | 41 | signal(SIGPIPE, SIG_IGN); /* TODO - switch to sigaction.*/ |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 42 | |
| 43 | /* Allocate an array of FILE *'s, with one extra for a sentinal. */ |
| 44 | p = files = (FILE **)xmalloc(sizeof(FILE *) * (argc - optind + 2)); |
| 45 | *p = stdout; |
| 46 | argv += optind - 1; |
| 47 | filenames = argv - 1; |
| 48 | *filenames = (char *) bb_msg_standard_input; /* for later */ |
| 49 | goto GOT_NEW_FILE; |
| 50 | |
| 51 | do { |
| 52 | if ((*p = bb_wfopen(*argv, mode)) == NULL) { |
| 53 | retval = EXIT_FAILURE; |
| 54 | continue; |
| 55 | } |
| 56 | filenames[(int)(p - files)] = *argv; |
| 57 | GOT_NEW_FILE: |
| 58 | setbuf(*p, NULL); /* tee must not buffer output. */ |
| 59 | ++p; |
| 60 | } while (*++argv); |
| 61 | |
| 62 | *p = NULL; /* Store the sentinal value. */ |
| 63 | |
| 64 | #ifdef CONFIG_FEATURE_TEE_USE_BLOCK_IO |
Manuel Novoa III | d709743 | 2004-05-26 15:21:19 +0000 | [diff] [blame] | 65 | while ((c = safe_read(STDIN_FILENO, buf, BUFSIZ)) > 0) { |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 66 | for (p=files ; *p ; p++) { |
| 67 | fwrite(buf, 1, c, *p); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 68 | } |
John Beppu | 059f152 | 1999-12-10 05:27:16 +0000 | [diff] [blame] | 69 | } |
John Beppu | 059f152 | 1999-12-10 05:27:16 +0000 | [diff] [blame] | 70 | |
Manuel Novoa III | d709743 | 2004-05-26 15:21:19 +0000 | [diff] [blame] | 71 | if (c < 0) { /* Make sure read errors are signaled. */ |
| 72 | retval = EXIT_FAILURE; |
| 73 | } |
| 74 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 75 | #else |
Eric Andersen | 637d226 | 2003-10-22 10:18:24 +0000 | [diff] [blame] | 76 | setvbuf(stdout, NULL, _IONBF, 0); |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 77 | while ((c = getchar()) != EOF) { |
| 78 | for (p=files ; *p ; p++) { |
| 79 | putc(c, *p); |
Erik Andersen | e49d5ec | 2000-02-08 19:58:47 +0000 | [diff] [blame] | 80 | } |
Eric Andersen | 2cb5507 | 1999-12-10 08:25:07 +0000 | [diff] [blame] | 81 | } |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 82 | #endif |
John Beppu | 059f152 | 1999-12-10 05:27:16 +0000 | [diff] [blame] | 83 | |
Eric Andersen | c7bda1c | 2004-03-15 08:29:22 +0000 | [diff] [blame] | 84 | /* Now we need to check for i/o errors on stdin and the various |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 85 | * output files. Since we know that the first entry in the output |
| 86 | * file table is stdout, we can save one "if ferror" test by |
| 87 | * setting the first entry to stdin and checking stdout error |
| 88 | * status with bb_fflush_stdout_and_exit()... although fflush()ing |
| 89 | * is unnecessary here. */ |
John Beppu | 059f152 | 1999-12-10 05:27:16 +0000 | [diff] [blame] | 90 | |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 91 | p = files; |
| 92 | *p = stdin; |
| 93 | do { /* Now check for (input and) output errors. */ |
| 94 | /* Checking ferror should be sufficient, but we may want to fclose. |
| 95 | * If we do, remember not to close stdin! */ |
Rob Landley | d921b2e | 2006-08-03 15:41:12 +0000 | [diff] [blame] | 96 | xferror(*p, filenames[(int)(p - files)]); |
Manuel Novoa III | cad5364 | 2003-03-19 09:13:01 +0000 | [diff] [blame] | 97 | } while (*++p); |
| 98 | |
| 99 | bb_fflush_stdout_and_exit(retval); |
John Beppu | 059f152 | 1999-12-10 05:27:16 +0000 | [diff] [blame] | 100 | } |