[unix-history] / usr / src / cmd / tee.c

/*
 * tee-- pipe fitting
 */

#include <signal.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <errno.h>
int openf[20] = { 1 };
int n = 1;
int t = 0;
int aflag;

char in[512];

char out[512];

extern errno;
long	lseek();

main(argc,argv)
char **argv;
{
	int register r,w,p;
	struct stat buf;
	while(argc>1&&argv[1][0]=='-') {
		switch(argv[1][1]) {
		case 'a':
			aflag++;
			break;
		case 'i':
		case 0:
			signal(SIGINT, SIG_IGN);
		}
		argv++;
		argc--;
	}
	fstat(1,&buf);
	t = (buf.st_mode&S_IFMT)==S_IFCHR;
	if(lseek(1,0L,1)==-1&&errno==ESPIPE)
		t++;
	while(argc-->1) {
		if(aflag) {
			openf[n] = open(argv[1],1);
			if(openf[n] < 0)
				openf[n] = creat(argv[1],0666);
			lseek(openf[n++],0L,2);
		} else
			openf[n++] = creat(argv[1],0666);
		if(stat(argv[1],&buf)>=0) {
			if((buf.st_mode&S_IFMT)==S_IFCHR)
				t++;
		} else {
			puts("tee: cannot open ");
			puts(argv[1]);
			puts("\n");
			n--;
		}
		argv++;
	}
	r = w = 0;
	for(;;) {
		for(p=0;p<512;) {
			if(r>=w) {
				if(t>0&&p>0) break;
				w = read(0,in,512);
				r = 0;
				if(w<=0) {
					stash(p);
					return;
				}
			}
			out[p++] = in[r++];
		}
		stash(p);
	}
}

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++)
			write(openf[k], out+i, d<p-i?d:p-i);
}

puts(s)
char *s;
{
	while(*s)
		write(2,s++,1);
}
Commit	Line	Data
3b600ead TL	1	/*
	2	* tee-- pipe fitting
	3	*/
	4
	5	#include <signal.h>
	6	#include <sys/types.h>
	7	#include <sys/stat.h>
	8	#include <errno.h>
	9	int openf[20] = { 1 };
	10	int n = 1;
	11	int t = 0;
	12	int aflag;
	13
	14	char in[512];
	15
	16	char out[512];
	17
	18	extern errno;
	19	long lseek();
	20
	21	main(argc,argv)
	22	char **argv;
	23	{
	24	int register r,w,p;
	25	struct stat buf;
	26	while(argc>1&&argv[1][0]=='-') {
	27	switch(argv[1][1]) {
	28	case 'a':
	29	aflag++;
	30	break;
	31	case 'i':
	32	case 0:
	33	signal(SIGINT, SIG_IGN);
	34	}
	35	argv++;
	36	argc--;
	37	}
	38	fstat(1,&buf);
	39	t = (buf.st_mode&S_IFMT)==S_IFCHR;
	40	if(lseek(1,0L,1)==-1&&errno==ESPIPE)
	41	t++;
	42	while(argc-->1) {
	43	if(aflag) {
	44	openf[n] = open(argv[1],1);
	45	if(openf[n] < 0)
	46	openf[n] = creat(argv[1],0666);
	47	lseek(openf[n++],0L,2);
	48	} else
	49	openf[n++] = creat(argv[1],0666);
	50	if(stat(argv[1],&buf)>=0) {
	51	if((buf.st_mode&S_IFMT)==S_IFCHR)
	52	t++;
	53	} else {
	54	puts("tee: cannot open ");
	55	puts(argv[1]);
	56	puts("\n");
	57	n--;
	58	}
	59	argv++;
	60	}
	61	r = w = 0;
	62	for(;;) {
	63	for(p=0;p<512;) {
	64	if(r>=w) {
65	if(t>0&&p>0) break;
66	w = read(0,in,512);
67	r = 0;
68	if(w<=0) {
69	stash(p);
70	return;
71	}
72	}
73	out[p++] = in[r++];
74	}
75	stash(p);
76	}
77	}
78
79	stash(p)
80	{
81	int k;
82	int i;
83	int d;
84	d = t ? 16 : p;
85	for(i=0; i<p; i+=d)
86	for(k=0;k<n;k++)
87	write(openf[k], out+i, d<p-i?d:p-i);
88	}
89
90	puts(s)
91	char *s;
92	{
93	while(*s)
94	write(2,s++,1);
95	}