Return to tee.c CVS log

Up to [local] / src / usr.bin / tee

Annotation of src/usr.bin/tee/tee.c, Revision 1.10

1.10    ! deraadt     1: /*     $OpenBSD: tee.c,v 1.9 2015/10/07 14:34:34 deraadt 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: #include <sys/types.h>
                     34: #include <sys/stat.h>
                     35: #include <signal.h>
                     36: #include <errno.h>
                     37: #include <fcntl.h>
                     38: #include <unistd.h>
                     39: #include <stdio.h>
                     40: #include <stdlib.h>
                     41: #include <string.h>
                     42: #include <locale.h>
                     43: #include <err.h>
                     44:
1.8       tedu       45: struct list {
                     46:        struct list *next;
1.1       deraadt    47:        int fd;
                     48:        char *name;
1.8       tedu       49: };
                     50: struct list *head;
1.1       deraadt    51:
1.8       tedu       52: static void
                     53: add(int fd, char *name)
                     54: {
                     55:        struct list *p;
                     56:
                     57:        if ((p = malloc(sizeof(*p))) == NULL)
                     58:                err(1, NULL);
                     59:        p->fd = fd;
                     60:        p->name = name;
                     61:        p->next = head;
                     62:        head = p;
                     63: }
1.1       deraadt    64:
                     65: int
1.6       deraadt    66: main(int argc, char *argv[])
1.1       deraadt    67: {
1.8       tedu       68:        struct list *p;
                     69:        int fd;
                     70:        ssize_t n, rval, wval;
1.3       mpech      71:        char *bp;
1.1       deraadt    72:        int append, ch, exitval;
1.8       tedu       73:        char buf[8192];
1.1       deraadt    74:
                     75:        setlocale(LC_ALL, "");
                     76:
1.10    ! deraadt    77:        if (pledge("stdio wpath cpath", NULL) == -1)
        !            78:                err(1, "pledge");
1.9       deraadt    79:
1.1       deraadt    80:        append = 0;
1.8       tedu       81:        while ((ch = getopt(argc, argv, "ai")) != -1) {
                     82:                switch(ch) {
1.1       deraadt    83:                case 'a':
                     84:                        append = 1;
                     85:                        break;
                     86:                case 'i':
                     87:                        (void)signal(SIGINT, SIG_IGN);
                     88:                        break;
                     89:                case '?':
                     90:                default:
                     91:                        (void)fprintf(stderr, "usage: tee [-ai] [file ...]\n");
                     92:                        exit(1);
                     93:                }
1.8       tedu       94:        }
1.1       deraadt    95:        argv += optind;
                     96:        argc -= optind;
                     97:
                     98:        add(STDOUT_FILENO, "stdout");
                     99:
1.8       tedu      100:        exitval = 0;
                    101:        while (*argv) {
                    102:                if ((fd = open(*argv, O_WRONLY | O_CREAT |
                    103:                    (append ? O_APPEND : O_TRUNC), DEFFILEMODE)) == -1) {
1.1       deraadt   104:                        warn("%s", *argv);
                    105:                        exitval = 1;
                    106:                } else
                    107:                        add(fd, *argv);
1.8       tedu      108:                argv++;
                    109:        }
1.9       deraadt   110:
1.10    ! deraadt   111:        if (pledge("stdio", NULL) == -1)
        !           112:                err(1, "pledge");
1.1       deraadt   113:
1.8       tedu      114:        while ((rval = read(STDIN_FILENO, buf, sizeof(buf))) > 0) {
1.1       deraadt   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:                }
1.8       tedu      127:        }
                    128:        if (rval == -1) {
1.1       deraadt   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: }