Home | History | Annotate | Line # | Download | only in script
script.c revision 1.1.1.2 
      1 /*
      2  * Copyright (c) 1980, 1992, 1993
      3  *	The Regents of the University of California.  All rights reserved.
      4  *
      5  * Redistribution and use in source and binary forms, with or without
      6  * modification, are permitted provided that the following conditions
      7  * are met:
      8  * 1. Redistributions of source code must retain the above copyright
      9  *    notice, this list of conditions and the following disclaimer.
     10  * 2. Redistributions in binary form must reproduce the above copyright
     11  *    notice, this list of conditions and the following disclaimer in the
     12  *    documentation and/or other materials provided with the distribution.
     13  * 3. All advertising materials mentioning features or use of this software
     14  *    must display the following acknowledgement:
     15  *	This product includes software developed by the University of
     16  *	California, Berkeley and its contributors.
     17  * 4. Neither the name of the University nor the names of its contributors
     18  *    may be used to endorse or promote products derived from this software
     19  *    without specific prior written permission.
     20  *
     21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     31  * SUCH DAMAGE.
     32  */
     33 
     34 #ifndef lint
     35 static char copyright[] =
     36 "@(#) Copyright (c) 1980, 1992, 1993\n\
     37 	The Regents of the University of California.  All rights reserved.\n";
     38 #endif /* not lint */
     39 
     40 #ifndef lint
     41 static char sccsid[] = "@(#)script.c	8.1 (Berkeley) 6/6/93";
     42 #endif /* not lint */
     43 
     44 #include <sys/types.h>
     45 #include <sys/wait.h>
     46 #include <sys/stat.h>
     47 #include <sys/ioctl.h>
     48 #include <sys/time.h>
     49 
     50 #include <errno.h>
     51 #include <fcntl.h>
     52 #include <paths.h>
     53 #include <signal.h>
     54 #include <stdio.h>
     55 #include <stdlib.h>
     56 #include <string.h>
     57 #include <termios.h>
     58 #include <tzfile.h>
     59 #include <unistd.h>
     60 
     61 FILE	*fscript;
     62 int	master, slave;
     63 int	child, subchild;
     64 int	outcc;
     65 char	*fname;
     66 
     67 struct	termios tt;
     68 
     69 __dead	void done __P((void));
     70 	void dooutput __P((void));
     71 	void doshell __P((void));
     72 	void err __P((const char *, ...));
     73 	void fail __P((void));
     74 	void finish __P((int));
     75 	void scriptflush __P((int));
     76 
     77 int
     78 main(argc, argv)
     79 	int argc;
     80 	char *argv[];
     81 {
     82 	register int cc;
     83 	struct termios rtt;
     84 	struct winsize win;
     85 	int aflg, ch;
     86 	char ibuf[BUFSIZ];
     87 
     88 	aflg = 0;
     89 	while ((ch = getopt(argc, argv, "a")) != EOF)
     90 		switch(ch) {
     91 		case 'a':
     92 			aflg = 1;
     93 			break;
     94 		case '?':
     95 		default:
     96 			(void)fprintf(stderr, "usage: script [-a] [file]\n");
     97 			exit(1);
     98 		}
     99 	argc -= optind;
    100 	argv += optind;
    101 
    102 	if (argc > 0)
    103 		fname = argv[0];
    104 	else
    105 		fname = "typescript";
    106 
    107 	if ((fscript = fopen(fname, aflg ? "a" : "w")) == NULL)
    108 		err("%s: %s", fname, strerror(errno));
    109 
    110 	(void)tcgetattr(STDIN_FILENO, &tt);
    111 	(void)ioctl(STDIN_FILENO, TIOCGWINSZ, &win);
    112 	if (openpty(&master, &slave, NULL, &tt, &win) == -1)
    113 		err("openpty: %s", strerror(errno));
    114 
    115 	(void)printf("Script started, output file is %s\n", fname);
    116 	rtt = tt;
    117 	cfmakeraw(&rtt);
    118 	rtt.c_lflag &= ~ECHO;
    119 	(void)tcsetattr(STDIN_FILENO, TCSAFLUSH, &rtt);
    120 
    121 	(void)signal(SIGCHLD, finish);
    122 	child = fork();
    123 	if (child < 0) {
    124 		perror("fork");
    125 		fail();
    126 	}
    127 	if (child == 0) {
    128 		subchild = child = fork();
    129 		if (child < 0) {
    130 			perror("fork");
    131 			fail();
    132 		}
    133 		if (child)
    134 			dooutput();
    135 		else
    136 			doshell();
    137 	}
    138 
    139 	(void)fclose(fscript);
    140 	while ((cc = read(STDIN_FILENO, ibuf, BUFSIZ)) > 0)
    141 		(void)write(master, ibuf, cc);
    142 	done();
    143 }
    144 
    145 void
    146 finish(signo)
    147 	int signo;
    148 {
    149 	register int die, pid;
    150 	union wait status;
    151 
    152 	die = 0;
    153 	while ((pid = wait3((int *)&status, WNOHANG, 0)) > 0)
    154 		if (pid == child)
    155 			die = 1;
    156 
    157 	if (die)
    158 		done();
    159 }
    160 
    161 void
    162 dooutput()
    163 {
    164 	struct itimerval value;
    165 	register int cc;
    166 	time_t tvec;
    167 	char obuf[BUFSIZ];
    168 
    169 	(void)close(STDIN_FILENO);
    170 	tvec = time(NULL);
    171 	(void)fprintf(fscript, "Script started on %s", ctime(&tvec));
    172 
    173 	(void)signal(SIGALRM, scriptflush);
    174 	value.it_interval.tv_sec = SECSPERMIN / 2;
    175 	value.it_interval.tv_usec = 0;
    176 	value.it_value = value.it_interval;
    177 	(void)setitimer(ITIMER_REAL, &value, NULL);
    178 	for (;;) {
    179 		cc = read(master, obuf, sizeof (obuf));
    180 		if (cc <= 0)
    181 			break;
    182 		(void)write(1, obuf, cc);
    183 		(void)fwrite(obuf, 1, cc, fscript);
    184 		outcc += cc;
    185 	}
    186 	done();
    187 }
    188 
    189 void
    190 scriptflush(signo)
    191 	int signo;
    192 {
    193 	if (outcc) {
    194 		(void)fflush(fscript);
    195 		outcc = 0;
    196 	}
    197 }
    198 
    199 void
    200 doshell()
    201 {
    202 	char *shell;
    203 
    204 	shell = getenv("SHELL");
    205 	if (shell == NULL)
    206 		shell = _PATH_BSHELL;
    207 
    208 	(void)close(master);
    209 	(void)fclose(fscript);
    210 	login_tty(slave);
    211 	execl(shell, "sh", "-i", NULL);
    212 	perror(shell);
    213 	fail();
    214 }
    215 
    216 void
    217 fail()
    218 {
    219 
    220 	(void)kill(0, SIGTERM);
    221 	done();
    222 }
    223 
    224 void
    225 done()
    226 {
    227 	time_t tvec;
    228 
    229 	if (subchild) {
    230 		tvec = time(NULL);
    231 		(void)fprintf(fscript,"\nScript done on %s", ctime(&tvec));
    232 		(void)fclose(fscript);
    233 		(void)close(master);
    234 	} else {
    235 		(void)tcsetattr(STDIN_FILENO, TCSAFLUSH, &tt);
    236 		(void)printf("Script done, output file is %s\n", fname);
    237 	}
    238 	exit(0);
    239 }
    240 
    241 #if __STDC__
    242 #include <stdarg.h>
    243 #else
    244 #include <varargs.h>
    245 #endif
    246 
    247 void
    248 #if __STDC__
    249 err(const char *fmt, ...)
    250 #else
    251 err(fmt, va_alist)
    252 	char *fmt;
    253 	va_dcl
    254 #endif
    255 {
    256 	va_list ap;
    257 #if __STDC__
    258 	va_start(ap, fmt);
    259 #else
    260 	va_start(ap);
    261 #endif
    262 	(void)fprintf(stderr, "script: ");
    263 	(void)vfprintf(stderr, fmt, ap);
    264 	va_end(ap);
    265 	(void)fprintf(stderr, "\n");
    266 	exit(1);
    267 	/* NOTREACHED */
    268 }
    269