1/*
2 * Copyright (c) 1988, 1993
3 *	The Regents of the University of California.  All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 * 4. Neither the name of the University nor the names of its contributors
14 *    may be used to endorse or promote products derived from this software
15 *    without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 */
29
30#ifndef lint
31static const char copyright[] =
32"@(#) Copyright (c) 1988, 1993\n\
33	The Regents of the University of California.  All rights reserved.\n";
34#endif /* not lint */
35
36#ifndef lint
37#if 0
38static char sccsid[] = "@(#)tee.c	8.1 (Berkeley) 6/6/93";
39#endif
40static const char rcsid[] =
41  "$FreeBSD$";
42#endif /* not lint */
43
44#include <sys/types.h>
45#include <sys/stat.h>
46#include <err.h>
47#include <fcntl.h>
48#include <signal.h>
49#include <stdio.h>
50#include <stdlib.h>
51#include <string.h>
52#include <unistd.h>
53
54typedef struct _list {
55	struct _list *next;
56	int fd;
57	const char *name;
58} LIST;
59LIST *head;
60
61void add(int, const char *);
62static void usage(void);
63
64int
65main(int argc, char *argv[])
66{
67	LIST *p;
68	int n, fd, rval, wval;
69	char *bp;
70	int append, ch, exitval;
71	char *buf;
72#define	BSIZE (8 * 1024)
73
74	append = 0;
75	while ((ch = getopt(argc, argv, "ai")) != -1)
76		switch((char)ch) {
77		case 'a':
78			append = 1;
79			break;
80		case 'i':
81			(void)signal(SIGINT, SIG_IGN);
82			break;
83		case '?':
84		default:
85			usage();
86		}
87	argv += optind;
88	argc -= optind;
89
90	if ((buf = malloc(BSIZE)) == NULL)
91		err(1, "malloc");
92
93	add(STDOUT_FILENO, "stdout");
94
95	for (exitval = 0; *argv; ++argv)
96		if ((fd = open(*argv, append ? O_WRONLY|O_CREAT|O_APPEND :
97		    O_WRONLY|O_CREAT|O_TRUNC, DEFFILEMODE)) < 0) {
98			warn("%s", *argv);
99			exitval = 1;
100		} else
101			add(fd, *argv);
102
103	while ((rval = read(STDIN_FILENO, buf, BSIZE)) > 0)
104		for (p = head; p; p = p->next) {
105			n = rval;
106			bp = buf;
107			do {
108				if ((wval = write(p->fd, bp, n)) == -1) {
109					warn("%s", p->name);
110					exitval = 1;
111					break;
112				}
113				bp += wval;
114			} while (n -= wval);
115		}
116	if (rval < 0)
117		err(1, "read");
118	exit(exitval);
119}
120
121static void
122usage(void)
123{
124	(void)fprintf(stderr, "usage: tee [-ai] [file ...]\n");
125	exit(1);
126}
127
128void
129add(int fd, const char *name)
130{
131	LIST *p;
132
133	if ((p = malloc(sizeof(LIST))) == NULL)
134		err(1, "malloc");
135	p->fd = fd;
136	p->name = name;
137	p->next = head;
138	head = p;
139}
140