sbase/tee.c

53 lines
979 B
C
Raw Normal View History

2011-05-23 01:36:34 +00:00
/* See LICENSE file for copyright and license details. */
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
2011-05-24 00:13:34 +00:00
#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 [-a] [file...]\n", argv0);
}
2011-05-23 01:36:34 +00:00
int
main(int argc, char *argv[])
{
bool aflag = false;
2013-06-14 18:20:47 +00:00
char buf[BUFSIZ];
2011-05-23 01:36:34 +00:00
int i, nfps = 1;
size_t n;
FILE **fps = NULL;
2011-05-23 01:36:34 +00:00
2013-06-14 18:20:47 +00:00
ARGBEGIN {
case 'a':
aflag = true;
break;
default:
usage();
} ARGEND;
2011-05-23 01:36:34 +00:00
if(!(fps = malloc(sizeof *fps)))
eprintf("malloc:");
fps[nfps-1] = stdout;
2013-06-14 18:20:47 +00:00
for(; argc > 0; argc--, argv++) {
2011-05-23 01:36:34 +00:00
if(!(fps = realloc(fps, ++nfps * sizeof *fps)))
eprintf("realloc:");
2013-06-14 18:20:47 +00:00
if(!(fps[nfps-1] = fopen(argv[0], aflag ? "a" : "w")))
eprintf("fopen %s:", argv[0]);
2011-05-23 01:36:34 +00:00
}
2013-06-14 18:20:47 +00:00
while((n = fread(buf, 1, sizeof buf, stdin)) > 0) {
for(i = 0; i < nfps; i++) {
2011-05-23 01:36:34 +00:00
if(fwrite(buf, 1, n, fps[i]) != n)
eprintf("%s: write error:", buf);
2013-06-14 18:20:47 +00:00
}
}
2011-05-23 01:36:34 +00:00
if(ferror(stdin))
eprintf("<stdin>: read error:");
free(fps);
2013-06-14 18:20:47 +00:00
return EXIT_SUCCESS;
2011-05-23 01:36:34 +00:00
}