[BACK]Return to tee.c CVS log [TXT][DIR] Up to [local] / src / usr.bin / tee

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

1.11    ! schwarze    1: /*     $OpenBSD: tee.c,v 1.10 2015/10/09 01:37:09 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>
1.11    ! schwarze   35:
        !            36: #include <err.h>
1.1       deraadt    37: #include <errno.h>
                     38: #include <fcntl.h>
1.11    ! schwarze   39: #include <signal.h>
1.1       deraadt    40: #include <stdio.h>
                     41: #include <stdlib.h>
                     42: #include <string.h>
1.11    ! schwarze   43: #include <unistd.h>
1.1       deraadt    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:
1.10      deraadt    75:        if (pledge("stdio wpath cpath", NULL) == -1)
                     76:                err(1, "pledge");
1.9       deraadt    77:
1.1       deraadt    78:        append = 0;
1.8       tedu       79:        while ((ch = getopt(argc, argv, "ai")) != -1) {
                     80:                switch(ch) {
1.1       deraadt    81:                case 'a':
                     82:                        append = 1;
                     83:                        break;
                     84:                case 'i':
                     85:                        (void)signal(SIGINT, SIG_IGN);
                     86:                        break;
                     87:                default:
                     88:                        (void)fprintf(stderr, "usage: tee [-ai] [file ...]\n");
1.11    ! schwarze   89:                        return 1;
1.1       deraadt    90:                }
1.8       tedu       91:        }
1.1       deraadt    92:        argv += optind;
                     93:        argc -= optind;
                     94:
                     95:        add(STDOUT_FILENO, "stdout");
                     96:
1.8       tedu       97:        exitval = 0;
                     98:        while (*argv) {
                     99:                if ((fd = open(*argv, O_WRONLY | O_CREAT |
                    100:                    (append ? O_APPEND : O_TRUNC), DEFFILEMODE)) == -1) {
1.1       deraadt   101:                        warn("%s", *argv);
                    102:                        exitval = 1;
                    103:                } else
                    104:                        add(fd, *argv);
1.8       tedu      105:                argv++;
                    106:        }
1.9       deraadt   107:
1.10      deraadt   108:        if (pledge("stdio", NULL) == -1)
                    109:                err(1, "pledge");
1.1       deraadt   110:
1.8       tedu      111:        while ((rval = read(STDIN_FILENO, buf, sizeof(buf))) > 0) {
1.1       deraadt   112:                for (p = head; p; p = p->next) {
                    113:                        n = rval;
                    114:                        bp = buf;
                    115:                        do {
                    116:                                if ((wval = write(p->fd, bp, n)) == -1) {
                    117:                                        warn("%s", p->name);
                    118:                                        exitval = 1;
                    119:                                        break;
                    120:                                }
                    121:                                bp += wval;
                    122:                        } while (n -= wval);
                    123:                }
1.8       tedu      124:        }
                    125:        if (rval == -1) {
1.1       deraadt   126:                warn("read");
                    127:                exitval = 1;
                    128:        }
                    129:
                    130:        for (p = head; p; p = p->next) {
                    131:                if (close(p->fd) == -1) {
                    132:                        warn("%s", p->name);
                    133:                        exitval = 1;
                    134:                }
                    135:        }
                    136:
1.11    ! schwarze  137:        return exitval;
1.1       deraadt   138: }