diff options
Diffstat (limited to 'util/sbase/tee.c')
-rw-r--r-- | util/sbase/tee.c | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/util/sbase/tee.c b/util/sbase/tee.c new file mode 100644 index 00000000..eac106ca --- /dev/null +++ b/util/sbase/tee.c @@ -0,0 +1,60 @@ +/* See LICENSE file for copyright and license details. */ +#include <fcntl.h> +#include <signal.h> +#include <unistd.h> + +#include "util.h" + +static void +usage(void) +{ + eprintf("usage: %s [-ai] [file ...]\n", argv0); +} + +int +main(int argc, char *argv[]) +{ + int *fds = NULL; + size_t i, nfds; + ssize_t n; + int ret = 0, aflag = O_TRUNC, iflag = 0; + char buf[BUFSIZ]; + + ARGBEGIN { + case 'a': + aflag = O_APPEND; + break; + case 'i': + iflag = 1; + break; + default: + usage(); + } ARGEND + + if (iflag && signal(SIGINT, SIG_IGN) == SIG_ERR) + eprintf("signal:"); + nfds = argc + 1; + fds = ecalloc(nfds, sizeof(*fds)); + + for (i = 0; i < argc; i++) { + if ((fds[i] = open(argv[i], O_WRONLY|O_CREAT|aflag, 0666)) < 0) { + weprintf("open %s:", argv[i]); + ret = 1; + } + } + fds[i] = 1; + + while ((n = read(0, buf, sizeof(buf))) > 0) { + for (i = 0; i < nfds; i++) { + if (fds[i] >= 0 && writeall(fds[i], buf, n) < 0) { + weprintf("write %s:", (i != argc) ? argv[i] : "<stdout>"); + fds[i] = -1; + ret = 1; + } + } + } + if (n < 0) + eprintf("read <stdin>:"); + + return ret; +} |