summaryrefslogtreecommitdiff
path: root/util/sbase/tee.c
diff options
context:
space:
mode:
Diffstat (limited to 'util/sbase/tee.c')
-rw-r--r--util/sbase/tee.c60
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;
+}