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