sbase/tee.c

54 lines
956 B
C
Raw Normal View History

2011-05-22 21:36:34 -04:00
/* See LICENSE file for copyright and license details. */
2015-01-22 16:08:25 -05:00
#include <signal.h>
2011-05-22 21:36:34 -04:00
#include <stdio.h>
2011-05-22 21:36:34 -04:00
#include "util.h"
2013-06-14 14:20:47 -04:00
static void
usage(void)
{
eprintf("usage: %s [-ai] [file ...]\n", argv0);
2013-06-14 14:20:47 -04:00
}
2011-05-22 21:36:34 -04:00
int
main(int argc, char *argv[])
{
FILE **fps = NULL;
size_t i, n, nfps;
2015-01-22 16:08:25 -05:00
int aflag = 0, iflag = 0;
2013-06-14 14:20:47 -04:00
char buf[BUFSIZ];
2011-05-22 21:36:34 -04:00
2013-06-14 14:20:47 -04:00
ARGBEGIN {
case 'a':
aflag = 1;
2013-06-14 14:20:47 -04:00
break;
2015-01-22 16:08:25 -05:00
case 'i':
iflag = 1;
break;
2013-06-14 14:20:47 -04:00
default:
usage();
} ARGEND;
2015-01-22 16:08:25 -05:00
if (iflag && signal(SIGINT, SIG_IGN) == SIG_ERR)
eprintf("signal:");
nfps = argc + 1;
fps = ecalloc(nfps, sizeof(*fps));
for (i = 0; i < argc; i++)
if (!(fps[i] = fopen(argv[i], aflag ? "a" : "w")))
eprintf("fopen %s:", argv[i]);
fps[i] = stdout;
2011-05-22 21:36:34 -04:00
while ((n = fread(buf, 1, sizeof(buf), stdin))) {
for (i = 0; i < nfps; i++) {
if (fwrite(buf, 1, n, fps[i]) == n)
continue;
eprintf("fwrite %s:", (i != argc) ? argv[i] : "<stdout>");
2013-06-14 14:20:47 -04:00
}
}
if (ferror(stdin))
eprintf("fread <stdin>:");
2013-06-14 14:20:47 -04:00
2014-10-02 18:46:04 -04:00
return 0;
2011-05-22 21:36:34 -04:00
}