sbase/tee.c

/* See LICENSE file for copyright and license details. */
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include "util.h"

static void
usage(void)
{
	eprintf("usage: %s [-a] [file...]\n", argv0);
}

int
main(int argc, char *argv[])
{
	bool aflag = false;
	char buf[BUFSIZ];
	int i, nfps;
	size_t n;
	FILE **fps = NULL;

	ARGBEGIN {
	case 'a':
		aflag = true;
		break;
	default:
		usage();
	} ARGEND;

	nfps = argc + 1;
	if(!(fps = calloc(nfps, sizeof *fps)))
		eprintf("calloc:");

	for(i = 0; argc > 0; argc--, argv++, i++)
		if(!(fps[i] = fopen(*argv, aflag ? "a" : "w")))
			eprintf("fopen %s:", *argv);
	fps[i] = stdout;

	while((n = fread(buf, 1, sizeof buf, stdin)) > 0) {
		for(i = 0; i < nfps; i++) {
			if(fwrite(buf, 1, n, fps[i]) != n)
				eprintf("%s: write error:", buf);
		}
	}
	if(ferror(stdin))
		eprintf("<stdin>: read error:");
	free(fps);

	return EXIT_SUCCESS;
}