Init

bin/tee.c

@@ -0,0 +1,119 @@
+#ifndef lint
+static	char sccsid[] = "@(#)tee.c 1.1 94/10/31 SMI"; /* from S5R2 1.3 */
+#endif
+
+/*
+ * tee-- pipe fitting
+ */
+
+#include <signal.h>
+#include <stdio.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <errno.h>
+#include <fcntl.h>
+int *openf;
+char **namef;
+int n = 0;
+int t = 0;
+int errflg;
+int aflag;
+
+#define	TEEBUFSIZ	8192
+
+char in[TEEBUFSIZ];
+
+extern errno;
+long	lseek();
+char	*malloc();
+
+main(argc,argv)
+char **argv;
+{
+	int dtablesize;
+	int register w;
+	extern int optind;
+	int c;
+	struct stat buf;
+	while ((c = getopt(argc, argv, "ai")) != EOF)
+		switch(c) {
+			case 'a':
+				aflag++;
+				break;
+			case 'i':
+				signal(SIGINT, SIG_IGN);
+				break;
+			case '?':
+				errflg++;
+		}
+	if (errflg) {
+		fprintf(stderr, "usage: tee [ -i ] [ -a ] [file ] ...\n");
+		exit(2);
+	}
+	argc -= optind;
+	argv = &argv[optind];
+	if (argc > 0 && strcmp(argv[0], "-") == 0) {
+		/*
+		 * An undocumented feature of the old version was that
+		 * "tee -" disabled interrupts like "tee -i".
+		 */
+		signal(SIGINT, SIG_IGN);
+		argc--;
+		argv++;
+	}
+	dtablesize = getdtablesize();
+	if ((openf = (int *)malloc(dtablesize * sizeof (int))) == NULL
+	    || (namef = (char **)malloc(dtablesize * sizeof (char *))) == NULL) {
+		perror("tee");
+		exit(1);
+	}
+	openf[n] = 1;
+	namef[n] = "standard output";
+	n++;
+	fstat(1,&buf);
+	t = (buf.st_mode&S_IFMT)==S_IFCHR;
+	if(lseek(1,0L,1)==-1&&errno==ESPIPE)
+		t++;
+	while(argc-->0) {
+		openf[n] = open(argv[0],(aflag?O_WRONLY|O_CREAT|O_APPEND:
+			O_WRONLY|O_CREAT|O_TRUNC), 0666);
+		namef[n] = argv[0];
+		if(openf[n] < 0) {
+			fprintf(stderr, "tee: ");
+			perror(argv[0]);
+		} else {
+			if(fstat(openf[n],&buf) < 0) {
+				fprintf(stderr, "tee: cannot stat ");
+				perror(argv[0]);
+			} else {
+				n++;
+				if((buf.st_mode&S_IFMT)==S_IFCHR)
+					t++;
+			}
+		}
+		argv++;
+	}
+	while ((w = read(0, in, TEEBUFSIZ)) > 0)
+		stash(w);
+	if(w < 0) {
+		perror("tee: read error on input");
+		exit(1);
+	}
+	exit(0);
+	/* NOTREACHED */
+}
+
+stash(p)
+{
+	int k;
+	int i;
+	int d;
+	d = t ? 16 : p;
+	for(i=0; i<p; i+=d)
+		for(k=0;k<n;k++)
+			if(write(openf[k], in+i, d<p-i?d:p-i) < 0) {
+				fprintf(stderr, "tee: ");
+				perror(namef[k]);
+				exit(1);
+			}
+}