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