sbase/tee.c

61 lines
1.0 KiB
C
Raw Normal View History

2011-05-23 01:36:34 +00:00
/* See LICENSE file for copyright and license details. */
#include <fcntl.h>
2015-01-22 21:08:25 +00:00
#include <signal.h>
#include <unistd.h>
2011-05-23 01:36:34 +00:00
#include "util.h"
2013-06-14 18:20:47 +00:00
static void
usage(void)
{
eprintf("usage: %s [-ai] [file ...]\n", argv0);
2013-06-14 18:20:47 +00:00
}
2011-05-23 01:36:34 +00:00
int
main(int argc, char *argv[])
{
int *fds = NULL;
size_t i, nfds;
ssize_t n;
int ret = 0, aflag = O_TRUNC, iflag = 0;
2013-06-14 18:20:47 +00:00
char buf[BUFSIZ];
2011-05-23 01:36:34 +00:00
2013-06-14 18:20:47 +00:00
ARGBEGIN {
case 'a':
aflag = O_APPEND;
2013-06-14 18:20:47 +00:00
break;
2015-01-22 21:08:25 +00:00
case 'i':
iflag = 1;
break;
2013-06-14 18:20:47 +00:00
default:
usage();
} ARGEND
2013-06-14 18:20:47 +00:00
2015-01-22 21:08:25 +00:00
if (iflag && signal(SIGINT, SIG_IGN) == SIG_ERR)
eprintf("signal:");
nfds = argc + 1;
fds = ecalloc(nfds, sizeof(*fds));
2016-06-23 18:37:23 +00:00
for (i = 0; i < argc; i++) {
if ((fds[i] = open(argv[i], O_WRONLY|O_CREAT|aflag, 0666)) < 0) {
weprintf("open %s:", argv[i]);
2016-06-23 18:37:23 +00:00
ret = 1;
}
}
fds[i] = 1;
2011-05-23 01:36:34 +00:00
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;
2016-06-23 18:37:23 +00:00
ret = 1;
}
2013-06-14 18:20:47 +00:00
}
}
if (n < 0)
eprintf("read <stdin>:");
return ret;
2011-05-23 01:36:34 +00:00
}