tee.c revision 0d8b53344490faab75b127936edb53a2f1e0dc01
/*
* CDDL HEADER START
*
* The contents of this file are subject to the terms of the
* Common Development and Distribution License, Version 1.0 only
* (the "License"). You may not use this file except in compliance
* with the License.
*
* You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
* or http://www.opensolaris.org/os/licensing.
* See the License for the specific language governing permissions
* and limitations under the License.
*
* When distributing Covered Code, include this CDDL HEADER in each
* file and include the License file at usr/src/OPENSOLARIS.LICENSE.
* If applicable, add the following below this CDDL HEADER, with the
* fields enclosed by brackets "[]" replaced with your own identifying
* information: Portions Copyright [yyyy] [name of copyright owner]
*
* CDDL HEADER END
*/
/*
* Copyright 2005 Sun Microsystems, Inc. All rights reserved.
* Use is subject to license terms.
*/
/* Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T */
/* All Rights Reserved */
#pragma ident "%Z%%M% %I% %E% SMI"
/*
* tee - pipe fitting
*/
#include <stdlib.h>
#include <limits.h>
#include <signal.h>
#include <unistd.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <locale.h>
#define min(a, b) ((a) > (b) ? (b) : (a))
#define MAXFILES 20
static void stash(int);
static int ofiles = 0;
static int ispipe = 0; /* output goes to pipe or special file */
static int openf[MAXFILES] = { 1 };
static char in[PIPE_BUF];
static const char *usage = "usage: tee [-ai] [file...]\n";
int
main(int argc, char **argv)
{
int w;
int c;
int aflag = 0;
int errorcode = 0;
struct stat buf;
(void) setlocale(LC_ALL, "");
#if !defined(TEXT_DOMAIN) /* Should be defined by cc -D */
#define TEXT_DOMAIN "SYS_TEST" /* Use this only if it weren't. */
#endif
(void) textdomain(TEXT_DOMAIN);
while ((c = getopt(argc, argv, "ai")) != EOF) {
switch (c) {
case 'a':
aflag++;
break;
case 'i':
(void) signal(SIGINT, SIG_IGN);
break;
case '?':
(void) fprintf(stderr, gettext(usage));
exit(1);
}
}
argc -= optind;
argv = &argv[optind];
(void) fstat(1, &buf);
if (S_ISFIFO(buf.st_mode) || S_ISCHR(buf.st_mode))
ispipe++;
openf[ofiles++] = 1;
while (argc-- > 0 && ofiles < MAXFILES) {
openf[ofiles] = open(argv[0],
O_WRONLY|O_CREAT|(aflag ? O_APPEND:O_TRUNC), 0666);
if (openf[ofiles] < 0) {
(void) fprintf(stderr, "tee: ");
perror(argv[0]);
errorcode++;
} else {
if (fstat(openf[ofiles], &buf) >= 0) {
if (S_ISCHR(buf.st_mode))
ispipe++;
ofiles++;
} else {
(void) fprintf(stderr, "tee: ");
perror(argv[0]);
errorcode++;
}
}
argv++;
}
if (argc >= 0 && ofiles >= MAXFILES) {
argv--;
(void) fprintf(stderr, gettext("tee: too many input files; "
"ignoring file(s) listed after %s\n"), argv[0]);
errorcode++;
}
while ((w = read(0, in, PIPE_BUF)) > 0)
stash(w);
if (w < 0) {
(void) fprintf(stderr, gettext("tee: read error on input\n"));
exit(1);
}
return (errorcode);
}
static void
stash(int nbytes)
{
register int k, i, chunk, nb;
chunk = ispipe ? PIPE_BUF : nbytes;
for (i = 0; i < nbytes; i += chunk) {
nb = min(chunk, nbytes - i);
for (k = 0; k < ofiles; k++)
(void) write(openf[k], in+i, nb);
}
}