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