aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--toys/posix/tee.c27
1 files changed, 11 insertions, 16 deletions
diff --git a/toys/posix/tee.c b/toys/posix/tee.c
index 43529426..88f73618 100644
--- a/toys/posix/tee.c
+++ b/toys/posix/tee.c
@@ -24,6 +24,7 @@ config TEE
GLOBALS(
void *outputs;
+ int out;
)
struct fd_list {
@@ -39,33 +40,27 @@ static void do_tee_open(int fd, char *name)
temp = xmalloc(sizeof(struct fd_list));
temp->next = TT.outputs;
- temp->fd = fd;
+ if (1 == (temp->fd = fd)) TT.out++;
TT.outputs = temp;
}
void tee_main(void)
{
+ struct fd_list *fdl;
+ int len;
+
if (FLAG(i)) xsignal(SIGINT, SIG_IGN);
- // Open output files
+ // Open output files (plus stdout if not already in output list)
loopfiles_rw(toys.optargs,
O_RDWR|O_CREAT|WARN_ONLY|(FLAG(a)?O_APPEND:O_TRUNC),
0666, do_tee_open);
+ if (!TT.out) do_tee_open(1, 0);
+ // Read data from stdin, write to each output file.
for (;;) {
- struct fd_list *fdl;
- int len, out = 0;
-
- // Read data from stdin
- len = xread(0, toybuf, sizeof(toybuf));
- if (len<1) break;
-
- // Write data to each output file, plus stdout.
- for (fdl = TT.outputs; ;fdl = fdl->next) {
- if (!fdl && out) break;
- if (len != writeall(fdl ? fdl->fd : 1, toybuf, len)) toys.exitval=1;
- if (!fdl) break;
- if (fdl->fd == 1) out++;
- }
+ if (1>(len = xread(0, toybuf, sizeof(toybuf)))) break;
+ for (fdl = TT.outputs; fdl;fdl = fdl->next)
+ if (len != writeall(fdl->fd, toybuf, len)) toys.exitval = 1;
}
}