xref: /freebsd/usr.bin/script/script.c (revision 8ceac8e13dccbe4e177c8f2f443b87b7d2e3edb3)
19b50d902SRodney W. Grimes /*
28a16b7a1SPedro F. Giffuni  * SPDX-License-Identifier: BSD-3-Clause
38a16b7a1SPedro F. Giffuni  *
46cff4e07SDavid E. O'Brien  * Copyright (c) 2010, 2012  David E. O'Brien
59b50d902SRodney W. Grimes  * Copyright (c) 1980, 1992, 1993
69b50d902SRodney W. Grimes  *	The Regents of the University of California.  All rights reserved.
79b50d902SRodney W. Grimes  *
89b50d902SRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
99b50d902SRodney W. Grimes  * modification, are permitted provided that the following conditions
109b50d902SRodney W. Grimes  * are met:
119b50d902SRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
129b50d902SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
139b50d902SRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
149b50d902SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
159b50d902SRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
16fbbd9655SWarner Losh  * 3. Neither the name of the University nor the names of its contributors
179b50d902SRodney W. Grimes  *    may be used to endorse or promote products derived from this software
189b50d902SRodney W. Grimes  *    without specific prior written permission.
199b50d902SRodney W. Grimes  *
209b50d902SRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
219b50d902SRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
229b50d902SRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
239b50d902SRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
249b50d902SRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
259b50d902SRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
269b50d902SRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
279b50d902SRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
289b50d902SRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
299b50d902SRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
309b50d902SRodney W. Grimes  * SUCH DAMAGE.
319b50d902SRodney W. Grimes  */
329b50d902SRodney W. Grimes 
3313f02f28SDavid E. O'Brien #include <sys/param.h>
349b50d902SRodney W. Grimes #include <sys/wait.h>
359b50d902SRodney W. Grimes #include <sys/stat.h>
369b50d902SRodney W. Grimes #include <sys/ioctl.h>
379b50d902SRodney W. Grimes #include <sys/time.h>
38c0ba4c2eSKonstantin Belousov #include <sys/queue.h>
39df53360cSBrian Somers #include <sys/uio.h>
40df53360cSBrian Somers #include <sys/endian.h>
416cff4e07SDavid E. O'Brien #include <dev/filemon/filemon.h>
429b50d902SRodney W. Grimes 
434459abe3SXavier Beaudouin #include <assert.h>
44236d2f55SPhilippe Charnier #include <err.h>
45e8eb82a8SPeter Wemm #include <errno.h>
469b50d902SRodney W. Grimes #include <fcntl.h>
47236d2f55SPhilippe Charnier #include <libutil.h>
489b50d902SRodney W. Grimes #include <paths.h>
499b50d902SRodney W. Grimes #include <signal.h>
509b50d902SRodney W. Grimes #include <stdio.h>
519b50d902SRodney W. Grimes #include <stdlib.h>
529b50d902SRodney W. Grimes #include <string.h>
539b50d902SRodney W. Grimes #include <termios.h>
549b50d902SRodney W. Grimes #include <unistd.h>
559b50d902SRodney W. Grimes 
56df53360cSBrian Somers #define DEF_BUF 65536
57df53360cSBrian Somers 
58df53360cSBrian Somers struct stamp {
59df53360cSBrian Somers 	uint64_t scr_len;	/* amount of data */
60df53360cSBrian Somers 	uint64_t scr_sec;	/* time it arrived in seconds... */
61df53360cSBrian Somers 	uint32_t scr_usec;	/* ...and microseconds */
62df53360cSBrian Somers 	uint32_t scr_direction; /* 'i', 'o', etc (also indicates endianness) */
63df53360cSBrian Somers };
64df53360cSBrian Somers 
65c0ba4c2eSKonstantin Belousov struct buf_elm {
66c0ba4c2eSKonstantin Belousov 	TAILQ_ENTRY(buf_elm) link;
673276866fSDag-Erling Smørgrav 	size_t rpos;
683276866fSDag-Erling Smørgrav 	size_t len;
69c0ba4c2eSKonstantin Belousov 	char ibuf[];
70c0ba4c2eSKonstantin Belousov };
71c0ba4c2eSKonstantin Belousov 
72e8efeec6SEd Schouten static FILE *fscript;
73e8efeec6SEd Schouten static int master, slave;
74e8efeec6SEd Schouten static int child;
75e8efeec6SEd Schouten static const char *fname;
766cff4e07SDavid E. O'Brien static char *fmfname;
776cff4e07SDavid E. O'Brien static int fflg, qflg, ttyflg;
7866d93438SBryan Drewery static int usesleep, rawout, showexit;
79c0ba4c2eSKonstantin Belousov static TAILQ_HEAD(, buf_elm) obuf_list = TAILQ_HEAD_INITIALIZER(obuf_list);
80*8ceac8e1SKyle Evans static volatile sig_atomic_t doresize;
819b50d902SRodney W. Grimes 
82e8efeec6SEd Schouten static struct termios tt;
839b50d902SRodney W. Grimes 
846c4afed5SSimon J. Gerraty #ifndef TSTAMP_FMT
856c4afed5SSimon J. Gerraty /* useful for tool and human reading */
8631fde973SSimon J. Gerraty # define TSTAMP_FMT "%n@ %s [%Y-%m-%d %T]%n"
876c4afed5SSimon J. Gerraty #endif
886c4afed5SSimon J. Gerraty static const char *tstamp_fmt = TSTAMP_FMT;
896c4afed5SSimon J. Gerraty static int tflg;
906c4afed5SSimon J. Gerraty 
91e8efeec6SEd Schouten static void done(int) __dead2;
92e8efeec6SEd Schouten static void doshell(char **);
93e8efeec6SEd Schouten static void finish(void);
94df53360cSBrian Somers static void record(FILE *, char *, size_t, int);
95df53360cSBrian Somers static void consume(FILE *, off_t, char *, int);
96df53360cSBrian Somers static void playback(FILE *) __dead2;
97cccdaf50SAlfonso Gregory static void usage(void) __dead2;
98*8ceac8e1SKyle Evans static void resizeit(int);
999b50d902SRodney W. Grimes 
1009b50d902SRodney W. Grimes int
main(int argc,char * argv[])101f4ac32deSDavid Malone main(int argc, char *argv[])
1029b50d902SRodney W. Grimes {
103e8eb82a8SPeter Wemm 	struct termios rtt, stt;
1049b50d902SRodney W. Grimes 	struct winsize win;
105*8ceac8e1SKyle Evans 	struct timespec tv, *tvp;
106e8eb82a8SPeter Wemm 	time_t tvec, start;
107e8eb82a8SPeter Wemm 	char obuf[BUFSIZ];
1089b50d902SRodney W. Grimes 	char ibuf[BUFSIZ];
109*8ceac8e1SKyle Evans 	sigset_t *pselmask, selmask;
110c0ba4c2eSKonstantin Belousov 	fd_set rfd, wfd;
111c0ba4c2eSKonstantin Belousov 	struct buf_elm *be;
1123276866fSDag-Erling Smørgrav 	ssize_t cc;
113*8ceac8e1SKyle Evans 	int aflg, Fflg, kflg, pflg, wflg, ch, k, n, fcm;
1146cff4e07SDavid E. O'Brien 	int flushtime, readstdin;
1156cff4e07SDavid E. O'Brien 	int fm_fd, fm_log;
1169b50d902SRodney W. Grimes 
117*8ceac8e1SKyle Evans 	aflg = Fflg = kflg = pflg = wflg = 0;
118*8ceac8e1SKyle Evans 	doresize = 0;
119df53360cSBrian Somers 	usesleep = 1;
120df53360cSBrian Somers 	rawout = 0;
1216cff4e07SDavid E. O'Brien 	flushtime = 30;
1224459abe3SXavier Beaudouin 	fm_fd = -1;
12366d93438SBryan Drewery 	showexit = 0;
124df53360cSBrian Somers 
125*8ceac8e1SKyle Evans 	/*
126*8ceac8e1SKyle Evans 	 * For normal operation, we'll leave pselmask == NULL so that pselect(2)
127*8ceac8e1SKyle Evans 	 * leaves the signal mask alone.  If -w is specified, we'll restore the
128*8ceac8e1SKyle Evans 	 * process signal mask upon entry with SIGWINCH unblocked so that we can
129*8ceac8e1SKyle Evans 	 * forward resize events properly.
130*8ceac8e1SKyle Evans 	 */
131*8ceac8e1SKyle Evans 	sigemptyset(&selmask);
132*8ceac8e1SKyle Evans 	pselmask = NULL;
133*8ceac8e1SKyle Evans 
134*8ceac8e1SKyle Evans 	while ((ch = getopt(argc, argv, "adeFfkpqrT:t:w")) != -1)
1359b50d902SRodney W. Grimes 		switch (ch) {
1369b50d902SRodney W. Grimes 		case 'a':
1379b50d902SRodney W. Grimes 			aflg = 1;
1389b50d902SRodney W. Grimes 			break;
139df53360cSBrian Somers 		case 'd':
140df53360cSBrian Somers 			usesleep = 0;
14151afb8dfSPeter Wemm 			break;
1424459abe3SXavier Beaudouin 		case 'e':
1434459abe3SXavier Beaudouin 			/* Default behavior, accepted for linux compat. */
144f9177b6cSWarner Losh 			break;
1458d105abcSTom Rhodes 		case 'F':
1468d105abcSTom Rhodes 			Fflg = 1;
1478d105abcSTom Rhodes 			break;
1486cff4e07SDavid E. O'Brien 		case 'f':
1496cff4e07SDavid E. O'Brien 			fflg = 1;
1506cff4e07SDavid E. O'Brien 			break;
151e8eb82a8SPeter Wemm 		case 'k':
152e8eb82a8SPeter Wemm 			kflg = 1;
153e8eb82a8SPeter Wemm 			break;
154df53360cSBrian Somers 		case 'p':
155df53360cSBrian Somers 			pflg = 1;
156df53360cSBrian Somers 			break;
157df53360cSBrian Somers 		case 'q':
158df53360cSBrian Somers 			qflg = 1;
159df53360cSBrian Somers 			break;
160df53360cSBrian Somers 		case 'r':
161df53360cSBrian Somers 			rawout = 1;
162df53360cSBrian Somers 			break;
163e8eb82a8SPeter Wemm 		case 't':
164e8eb82a8SPeter Wemm 			flushtime = atoi(optarg);
165e8eb82a8SPeter Wemm 			if (flushtime < 0)
166e8eb82a8SPeter Wemm 				err(1, "invalid flush time %d", flushtime);
167e8eb82a8SPeter Wemm 			break;
1686c4afed5SSimon J. Gerraty 		case 'T':
1696c4afed5SSimon J. Gerraty 			tflg = pflg = 1;
1706c4afed5SSimon J. Gerraty 			if (strchr(optarg, '%'))
1716c4afed5SSimon J. Gerraty 				tstamp_fmt = optarg;
1726c4afed5SSimon J. Gerraty 			break;
173*8ceac8e1SKyle Evans 		case 'w':
174*8ceac8e1SKyle Evans 			wflg = 1;
175*8ceac8e1SKyle Evans 			break;
1769b50d902SRodney W. Grimes 		case '?':
1779b50d902SRodney W. Grimes 		default:
178236d2f55SPhilippe Charnier 			usage();
1799b50d902SRodney W. Grimes 		}
1809b50d902SRodney W. Grimes 	argc -= optind;
1819b50d902SRodney W. Grimes 	argv += optind;
1829b50d902SRodney W. Grimes 
18351afb8dfSPeter Wemm 	if (argc > 0) {
1849b50d902SRodney W. Grimes 		fname = argv[0];
18551afb8dfSPeter Wemm 		argv++;
18651afb8dfSPeter Wemm 		argc--;
18751afb8dfSPeter Wemm 	} else
1889b50d902SRodney W. Grimes 		fname = "typescript";
1899b50d902SRodney W. Grimes 
190df53360cSBrian Somers 	if ((fscript = fopen(fname, pflg ? "r" : aflg ? "a" : "w")) == NULL)
191236d2f55SPhilippe Charnier 		err(1, "%s", fname);
1929b50d902SRodney W. Grimes 
1936cff4e07SDavid E. O'Brien 	if (fflg) {
1946cff4e07SDavid E. O'Brien 		asprintf(&fmfname, "%s.filemon", fname);
1956cff4e07SDavid E. O'Brien 		if (!fmfname)
1966cff4e07SDavid E. O'Brien 			err(1, "%s.filemon", fname);
19726e736b2SBaptiste Daroussin 		if ((fm_fd = open("/dev/filemon", O_RDWR | O_CLOEXEC)) == -1)
1986cff4e07SDavid E. O'Brien 			err(1, "open(\"/dev/filemon\", O_RDWR)");
19926e736b2SBaptiste Daroussin 		if ((fm_log = open(fmfname,
20026e736b2SBaptiste Daroussin 		    O_WRONLY | O_CREAT | O_TRUNC | O_CLOEXEC,
2016cff4e07SDavid E. O'Brien 		    S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH)) == -1)
2026cff4e07SDavid E. O'Brien 			err(1, "open(%s)", fmfname);
2036cff4e07SDavid E. O'Brien 		if (ioctl(fm_fd, FILEMON_SET_FD, &fm_log) < 0)
2046cff4e07SDavid E. O'Brien 			err(1, "Cannot set filemon log file descriptor");
2056cff4e07SDavid E. O'Brien 	}
2066cff4e07SDavid E. O'Brien 
207df53360cSBrian Somers 	if (pflg)
208df53360cSBrian Somers 		playback(fscript);
209df53360cSBrian Somers 
2108bc30d2aSMark Johnston 	if (tcgetattr(STDIN_FILENO, &tt) == -1 ||
2118bc30d2aSMark Johnston 	    ioctl(STDIN_FILENO, TIOCGWINSZ, &win) == -1) {
2128bc30d2aSMark Johnston 		if (errno != ENOTTY) /* For debugger. */
2138bc30d2aSMark Johnston 			err(1, "tcgetattr/ioctl");
214e292a0e3SColin Percival 		if (openpty(&master, &slave, NULL, NULL, NULL) == -1)
215e292a0e3SColin Percival 			err(1, "openpty");
2168bc30d2aSMark Johnston 	} else {
2178bc30d2aSMark Johnston 		if (openpty(&master, &slave, NULL, &tt, &win) == -1)
2188bc30d2aSMark Johnston 			err(1, "openpty");
2198bc30d2aSMark Johnston 		ttyflg = 1;
220e292a0e3SColin Percival 	}
221c0ba4c2eSKonstantin Belousov 	fcm = fcntl(master, F_GETFL);
222c0ba4c2eSKonstantin Belousov 	if (fcm == -1)
223c0ba4c2eSKonstantin Belousov 		err(1, "master F_GETFL");
224c0ba4c2eSKonstantin Belousov 	fcm |= O_NONBLOCK;
225c0ba4c2eSKonstantin Belousov 	if (fcntl(master, F_SETFL, fcm) == -1)
226c0ba4c2eSKonstantin Belousov 		err(1, "master F_SETFL");
2279b50d902SRodney W. Grimes 
228df53360cSBrian Somers 	if (rawout)
229df53360cSBrian Somers 		record(fscript, NULL, 0, 's');
230df53360cSBrian Somers 
231e8eb82a8SPeter Wemm 	if (!qflg) {
232e8eb82a8SPeter Wemm 		tvec = time(NULL);
2339b50d902SRodney W. Grimes 		(void)printf("Script started, output file is %s\n", fname);
234df53360cSBrian Somers 		if (!rawout) {
235df53360cSBrian Somers 			(void)fprintf(fscript, "Script started on %s",
236df53360cSBrian Somers 			    ctime(&tvec));
237df53360cSBrian Somers 			if (argv[0]) {
23866d93438SBryan Drewery 				showexit = 1;
23966d93438SBryan Drewery 				fprintf(fscript, "Command: ");
240df53360cSBrian Somers 				for (k = 0 ; argv[k] ; ++k)
241df53360cSBrian Somers 					fprintf(fscript, "%s%s", k ? " " : "",
242df53360cSBrian Somers 						argv[k]);
243df53360cSBrian Somers 				fprintf(fscript, "\n");
244df53360cSBrian Somers 			}
245df53360cSBrian Somers 		}
246e8eb82a8SPeter Wemm 		fflush(fscript);
2476cff4e07SDavid E. O'Brien 		if (fflg) {
2486cff4e07SDavid E. O'Brien 			(void)printf("Filemon started, output file is %s\n",
2496cff4e07SDavid E. O'Brien 			    fmfname);
2506cff4e07SDavid E. O'Brien 		}
251e8eb82a8SPeter Wemm 	}
252e292a0e3SColin Percival 	if (ttyflg) {
2539b50d902SRodney W. Grimes 		rtt = tt;
2549b50d902SRodney W. Grimes 		cfmakeraw(&rtt);
2559b50d902SRodney W. Grimes 		rtt.c_lflag &= ~ECHO;
2569b50d902SRodney W. Grimes 		(void)tcsetattr(STDIN_FILENO, TCSAFLUSH, &rtt);
257e292a0e3SColin Percival 	}
2589b50d902SRodney W. Grimes 
2594459abe3SXavier Beaudouin 	assert(fflg ? fm_fd >= 0 : fm_fd < 0);
2604459abe3SXavier Beaudouin 
2619b50d902SRodney W. Grimes 	child = fork();
2629b50d902SRodney W. Grimes 	if (child < 0) {
263236d2f55SPhilippe Charnier 		warn("fork");
264b1c66970SDag-Erling Smørgrav 		done(1);
2659b50d902SRodney W. Grimes 	}
2667b245cb8SBryan Drewery 	if (child == 0) {
2677b245cb8SBryan Drewery 		if (fflg) {
2687b245cb8SBryan Drewery 			int pid;
2699b50d902SRodney W. Grimes 
2707b245cb8SBryan Drewery 			pid = getpid();
2717b245cb8SBryan Drewery 			if (ioctl(fm_fd, FILEMON_SET_PID, &pid) < 0)
2726cff4e07SDavid E. O'Brien 				err(1, "Cannot set filemon PID");
2737b245cb8SBryan Drewery 		}
2747b245cb8SBryan Drewery 
2757b245cb8SBryan Drewery 		doshell(argv);
2767b245cb8SBryan Drewery 	}
2777b245cb8SBryan Drewery 	close(slave);
2786cff4e07SDavid E. O'Brien 
279*8ceac8e1SKyle Evans 	if (wflg) {
280*8ceac8e1SKyle Evans 		struct sigaction sa = { .sa_handler = resizeit };
281*8ceac8e1SKyle Evans 		sigset_t smask;
282*8ceac8e1SKyle Evans 
283*8ceac8e1SKyle Evans 		sigaction(SIGWINCH, &sa, NULL);
284*8ceac8e1SKyle Evans 
285*8ceac8e1SKyle Evans 		sigemptyset(&smask);
286*8ceac8e1SKyle Evans 		sigaddset(&smask, SIGWINCH);
287*8ceac8e1SKyle Evans 
288*8ceac8e1SKyle Evans 		if (sigprocmask(SIG_BLOCK, &smask, &selmask) != 0)
289*8ceac8e1SKyle Evans 			err(1, "Failed to block SIGWINCH");
290*8ceac8e1SKyle Evans 
291*8ceac8e1SKyle Evans 		/* Just in case SIGWINCH was blocked before we came in. */
292*8ceac8e1SKyle Evans 		sigdelset(&selmask, SIGWINCH);
293*8ceac8e1SKyle Evans 		pselmask = &selmask;
294*8ceac8e1SKyle Evans 	}
295*8ceac8e1SKyle Evans 
29629da7547SMikolaj Golub 	start = tvec = time(0);
29729da7547SMikolaj Golub 	readstdin = 1;
298e8eb82a8SPeter Wemm 	for (;;) {
29929da7547SMikolaj Golub 		FD_ZERO(&rfd);
300c0ba4c2eSKonstantin Belousov 		FD_ZERO(&wfd);
301e8eb82a8SPeter Wemm 		FD_SET(master, &rfd);
30229da7547SMikolaj Golub 		if (readstdin)
303e8eb82a8SPeter Wemm 			FD_SET(STDIN_FILENO, &rfd);
304c0ba4c2eSKonstantin Belousov 		if (!TAILQ_EMPTY(&obuf_list))
305c0ba4c2eSKonstantin Belousov 			FD_SET(master, &wfd);
30629f4384aSMikolaj Golub 		if (!readstdin && ttyflg) {
30729f4384aSMikolaj Golub 			tv.tv_sec = 1;
308*8ceac8e1SKyle Evans 			tv.tv_nsec = 0;
30929da7547SMikolaj Golub 			tvp = &tv;
31029da7547SMikolaj Golub 			readstdin = 1;
31129f4384aSMikolaj Golub 		} else if (flushtime > 0) {
31229f4384aSMikolaj Golub 			tv.tv_sec = flushtime - (tvec - start);
313*8ceac8e1SKyle Evans 			tv.tv_nsec = 0;
31429f4384aSMikolaj Golub 			tvp = &tv;
31529da7547SMikolaj Golub 		} else {
31629da7547SMikolaj Golub 			tvp = NULL;
317e8eb82a8SPeter Wemm 		}
318*8ceac8e1SKyle Evans 		n = pselect(master + 1, &rfd, &wfd, NULL, tvp, pselmask);
319e8eb82a8SPeter Wemm 		if (n < 0 && errno != EINTR)
320e8eb82a8SPeter Wemm 			break;
321*8ceac8e1SKyle Evans 
322*8ceac8e1SKyle Evans 		if (doresize) {
323*8ceac8e1SKyle Evans 			if (ioctl(STDIN_FILENO, TIOCGWINSZ, &win) != -1)
324*8ceac8e1SKyle Evans 				ioctl(master, TIOCSWINSZ, &win);
325*8ceac8e1SKyle Evans 			doresize = 0;
326*8ceac8e1SKyle Evans 		}
327*8ceac8e1SKyle Evans 
328e8eb82a8SPeter Wemm 		if (n > 0 && FD_ISSET(STDIN_FILENO, &rfd)) {
329e8eb82a8SPeter Wemm 			cc = read(STDIN_FILENO, ibuf, BUFSIZ);
330d6a68195SColin Percival 			if (cc < 0)
331e8eb82a8SPeter Wemm 				break;
33229da7547SMikolaj Golub 			if (cc == 0) {
33329da7547SMikolaj Golub 				if (tcgetattr(master, &stt) == 0 &&
33429da7547SMikolaj Golub 				    (stt.c_lflag & ICANON) != 0) {
33529da7547SMikolaj Golub 					(void)write(master, &stt.c_cc[VEOF], 1);
33629da7547SMikolaj Golub 				}
33729da7547SMikolaj Golub 				readstdin = 0;
33829da7547SMikolaj Golub 			}
339e8eb82a8SPeter Wemm 			if (cc > 0) {
340df53360cSBrian Somers 				if (rawout)
341df53360cSBrian Somers 					record(fscript, ibuf, cc, 'i');
342c0ba4c2eSKonstantin Belousov 				be = malloc(sizeof(*be) + cc);
343c0ba4c2eSKonstantin Belousov 				be->rpos = 0;
344c0ba4c2eSKonstantin Belousov 				be->len = cc;
345c0ba4c2eSKonstantin Belousov 				memcpy(be->ibuf, ibuf, cc);
346c0ba4c2eSKonstantin Belousov 				TAILQ_INSERT_TAIL(&obuf_list, be, link);
347c0ba4c2eSKonstantin Belousov 			}
348c0ba4c2eSKonstantin Belousov 		}
349c0ba4c2eSKonstantin Belousov 		if (n > 0 && FD_ISSET(master, &wfd)) {
350c0ba4c2eSKonstantin Belousov 			while ((be = TAILQ_FIRST(&obuf_list)) != NULL) {
351c0ba4c2eSKonstantin Belousov 				cc = write(master, be->ibuf + be->rpos,
352c0ba4c2eSKonstantin Belousov 				    be->len);
353c0ba4c2eSKonstantin Belousov 				if (cc == -1) {
354c0ba4c2eSKonstantin Belousov 					if (errno == EWOULDBLOCK ||
355c0ba4c2eSKonstantin Belousov 					    errno == EINTR)
356c0ba4c2eSKonstantin Belousov 						break;
357c0ba4c2eSKonstantin Belousov 					warn("write master");
358c0ba4c2eSKonstantin Belousov 					done(1);
359c0ba4c2eSKonstantin Belousov 				}
360c0ba4c2eSKonstantin Belousov 				if (cc == 0)
361c0ba4c2eSKonstantin Belousov 					break;		/* retry later ? */
362e8eb82a8SPeter Wemm 				if (kflg && tcgetattr(master, &stt) >= 0 &&
363e8eb82a8SPeter Wemm 				    ((stt.c_lflag & ECHO) == 0)) {
364c0ba4c2eSKonstantin Belousov 					(void)fwrite(be->ibuf + be->rpos,
365c0ba4c2eSKonstantin Belousov 					    1, cc, fscript);
366c0ba4c2eSKonstantin Belousov 				}
367c0ba4c2eSKonstantin Belousov 				be->len -= cc;
368c0ba4c2eSKonstantin Belousov 				if (be->len == 0) {
369c0ba4c2eSKonstantin Belousov 					TAILQ_REMOVE(&obuf_list, be, link);
370c0ba4c2eSKonstantin Belousov 					free(be);
371c0ba4c2eSKonstantin Belousov 				} else {
372c0ba4c2eSKonstantin Belousov 					be->rpos += cc;
373e8eb82a8SPeter Wemm 				}
374e8eb82a8SPeter Wemm 			}
375e8eb82a8SPeter Wemm 		}
376e8eb82a8SPeter Wemm 		if (n > 0 && FD_ISSET(master, &rfd)) {
377e8eb82a8SPeter Wemm 			cc = read(master, obuf, sizeof(obuf));
378e8eb82a8SPeter Wemm 			if (cc <= 0)
379e8eb82a8SPeter Wemm 				break;
380e1b4d8d0SSheldon Hearn 			(void)write(STDOUT_FILENO, obuf, cc);
381df53360cSBrian Somers 			if (rawout)
382df53360cSBrian Somers 				record(fscript, obuf, cc, 'o');
383df53360cSBrian Somers 			else
384e8eb82a8SPeter Wemm 				(void)fwrite(obuf, 1, cc, fscript);
385e8eb82a8SPeter Wemm 		}
386e8eb82a8SPeter Wemm 		tvec = time(0);
387e8eb82a8SPeter Wemm 		if (tvec - start >= flushtime) {
388e8eb82a8SPeter Wemm 			fflush(fscript);
389e8eb82a8SPeter Wemm 			start = tvec;
390e8eb82a8SPeter Wemm 		}
3918d105abcSTom Rhodes 		if (Fflg)
3928d105abcSTom Rhodes 			fflush(fscript);
393e8eb82a8SPeter Wemm 	}
394b1c66970SDag-Erling Smørgrav 	finish();
395b1c66970SDag-Erling Smørgrav 	done(0);
3969b50d902SRodney W. Grimes }
3979b50d902SRodney W. Grimes 
398236d2f55SPhilippe Charnier static void
usage(void)399f4ac32deSDavid Malone usage(void)
400236d2f55SPhilippe Charnier {
401e8eb82a8SPeter Wemm 	(void)fprintf(stderr,
402*8ceac8e1SKyle Evans 	    "usage: script [-aeFfkpqrw] [-t time] [file [command ...]]\n");
403e7c13cf4SDag-Erling Smørgrav 	(void)fprintf(stderr,
4046ac087cfSDag-Erling Smørgrav 	    "       script -p [-deq] [-T fmt] [file]\n");
405236d2f55SPhilippe Charnier 	exit(1);
406236d2f55SPhilippe Charnier }
407236d2f55SPhilippe Charnier 
408e8efeec6SEd Schouten static void
finish(void)409f4ac32deSDavid Malone finish(void)
4109b50d902SRodney W. Grimes {
411e1e9ba33SEd Schouten 	int e, status;
4129b50d902SRodney W. Grimes 
413e1e9ba33SEd Schouten 	if (waitpid(child, &status, 0) == child) {
414b1c66970SDag-Erling Smørgrav 		if (WIFEXITED(status))
415b1c66970SDag-Erling Smørgrav 			e = WEXITSTATUS(status);
416b1c66970SDag-Erling Smørgrav 		else if (WIFSIGNALED(status))
417b1c66970SDag-Erling Smørgrav 			e = WTERMSIG(status);
418b1c66970SDag-Erling Smørgrav 		else /* can't happen */
419b1c66970SDag-Erling Smørgrav 			e = 1;
420b1c66970SDag-Erling Smørgrav 		done(e);
4219b50d902SRodney W. Grimes 	}
422e1e9ba33SEd Schouten }
4239b50d902SRodney W. Grimes 
424e8efeec6SEd Schouten static void
doshell(char ** av)425f4ac32deSDavid Malone doshell(char **av)
4269b50d902SRodney W. Grimes {
427b139689cSDavid Malone 	const char *shell;
4289b50d902SRodney W. Grimes 
4299b50d902SRodney W. Grimes 	shell = getenv("SHELL");
4309b50d902SRodney W. Grimes 	if (shell == NULL)
4319b50d902SRodney W. Grimes 		shell = _PATH_BSHELL;
4329b50d902SRodney W. Grimes 
4339b50d902SRodney W. Grimes 	(void)close(master);
4349b50d902SRodney W. Grimes 	(void)fclose(fscript);
4356cff4e07SDavid E. O'Brien 	free(fmfname);
4369b50d902SRodney W. Grimes 	login_tty(slave);
4379b91846cSDavid E. O'Brien 	setenv("SCRIPT", fname, 1);
438b1c66970SDag-Erling Smørgrav 	if (av[0]) {
4390e604e98SPeter Wemm 		execvp(av[0], av);
440b7ffba17SKris Kennaway 		warn("%s", av[0]);
441b1c66970SDag-Erling Smørgrav 	} else {
4427bc6d015SBrian Somers 		execl(shell, shell, "-i", (char *)NULL);
443b7ffba17SKris Kennaway 		warn("%s", shell);
444b1c66970SDag-Erling Smørgrav 	}
44595d13d1bSBryan Drewery 	exit(1);
4469b50d902SRodney W. Grimes }
4479b50d902SRodney W. Grimes 
448e8efeec6SEd Schouten static void
done(int eno)449f4ac32deSDavid Malone done(int eno)
4509b50d902SRodney W. Grimes {
4519b50d902SRodney W. Grimes 	time_t tvec;
4529b50d902SRodney W. Grimes 
453e292a0e3SColin Percival 	if (ttyflg)
454e8eb82a8SPeter Wemm 		(void)tcsetattr(STDIN_FILENO, TCSAFLUSH, &tt);
4559b50d902SRodney W. Grimes 	tvec = time(NULL);
456df53360cSBrian Somers 	if (rawout)
457df53360cSBrian Somers 		record(fscript, NULL, 0, 'e');
458e8eb82a8SPeter Wemm 	if (!qflg) {
45966d93438SBryan Drewery 		if (!rawout) {
46066d93438SBryan Drewery 			if (showexit)
46166d93438SBryan Drewery 				(void)fprintf(fscript, "\nCommand exit status:"
46266d93438SBryan Drewery 				    " %d", eno);
463df53360cSBrian Somers 			(void)fprintf(fscript, "\nScript done on %s",
464df53360cSBrian Somers 			    ctime(&tvec));
46566d93438SBryan Drewery 		}
466e8eb82a8SPeter Wemm 		(void)printf("\nScript done, output file is %s\n", fname);
4676cff4e07SDavid E. O'Brien 		if (fflg) {
4686cff4e07SDavid E. O'Brien 			(void)printf("Filemon done, output file is %s\n",
4696cff4e07SDavid E. O'Brien 			    fmfname);
4706cff4e07SDavid E. O'Brien 		}
471e8eb82a8SPeter Wemm 	}
4729b50d902SRodney W. Grimes 	(void)fclose(fscript);
4739b50d902SRodney W. Grimes 	(void)close(master);
474b1c66970SDag-Erling Smørgrav 	exit(eno);
4759b50d902SRodney W. Grimes }
476df53360cSBrian Somers 
477df53360cSBrian Somers static void
record(FILE * fp,char * buf,size_t cc,int direction)478df53360cSBrian Somers record(FILE *fp, char *buf, size_t cc, int direction)
479df53360cSBrian Somers {
480df53360cSBrian Somers 	struct iovec iov[2];
481df53360cSBrian Somers 	struct stamp stamp;
482df53360cSBrian Somers 	struct timeval tv;
483df53360cSBrian Somers 
484df53360cSBrian Somers 	(void)gettimeofday(&tv, NULL);
485df53360cSBrian Somers 	stamp.scr_len = cc;
486df53360cSBrian Somers 	stamp.scr_sec = tv.tv_sec;
487df53360cSBrian Somers 	stamp.scr_usec = tv.tv_usec;
488df53360cSBrian Somers 	stamp.scr_direction = direction;
489df53360cSBrian Somers 	iov[0].iov_len = sizeof(stamp);
490df53360cSBrian Somers 	iov[0].iov_base = &stamp;
491df53360cSBrian Somers 	iov[1].iov_len = cc;
492df53360cSBrian Somers 	iov[1].iov_base = buf;
493df53360cSBrian Somers 	if (writev(fileno(fp), &iov[0], 2) == -1)
494df53360cSBrian Somers 		err(1, "writev");
495df53360cSBrian Somers }
496df53360cSBrian Somers 
497df53360cSBrian Somers static void
consume(FILE * fp,off_t len,char * buf,int reg)498df53360cSBrian Somers consume(FILE *fp, off_t len, char *buf, int reg)
499df53360cSBrian Somers {
500df53360cSBrian Somers 	size_t l;
501df53360cSBrian Somers 
502df53360cSBrian Somers 	if (reg) {
503df53360cSBrian Somers 		if (fseeko(fp, len, SEEK_CUR) == -1)
504df53360cSBrian Somers 			err(1, NULL);
5054459abe3SXavier Beaudouin 	} else {
506df53360cSBrian Somers 		while (len > 0) {
507df53360cSBrian Somers 			l = MIN(DEF_BUF, len);
508df53360cSBrian Somers 			if (fread(buf, sizeof(char), l, fp) != l)
509df53360cSBrian Somers 				err(1, "cannot read buffer");
510df53360cSBrian Somers 			len -= l;
511df53360cSBrian Somers 		}
512df53360cSBrian Somers 	}
513df53360cSBrian Somers }
514df53360cSBrian Somers 
515df53360cSBrian Somers #define swapstamp(stamp) do { \
516df53360cSBrian Somers 	if (stamp.scr_direction > 0xff) { \
517df53360cSBrian Somers 		stamp.scr_len = bswap64(stamp.scr_len); \
518df53360cSBrian Somers 		stamp.scr_sec = bswap64(stamp.scr_sec); \
519df53360cSBrian Somers 		stamp.scr_usec = bswap32(stamp.scr_usec); \
520df53360cSBrian Somers 		stamp.scr_direction = bswap32(stamp.scr_direction); \
521df53360cSBrian Somers 	} \
522df53360cSBrian Somers } while (0/*CONSTCOND*/)
523df53360cSBrian Somers 
524df53360cSBrian Somers static void
termset(void)52569bc4fa9SMark Johnston termset(void)
52669bc4fa9SMark Johnston {
52769bc4fa9SMark Johnston 	struct termios traw;
52869bc4fa9SMark Johnston 
52969bc4fa9SMark Johnston 	if (tcgetattr(STDOUT_FILENO, &tt) == -1) {
5308bc30d2aSMark Johnston 		if (errno != ENOTTY) /* For debugger. */
5318bc30d2aSMark Johnston 			err(1, "tcgetattr");
53269bc4fa9SMark Johnston 		return;
53369bc4fa9SMark Johnston 	}
53469bc4fa9SMark Johnston 	ttyflg = 1;
53569bc4fa9SMark Johnston 	traw = tt;
53669bc4fa9SMark Johnston 	cfmakeraw(&traw);
53769bc4fa9SMark Johnston 	traw.c_lflag |= ISIG;
53869bc4fa9SMark Johnston 	(void)tcsetattr(STDOUT_FILENO, TCSANOW, &traw);
53969bc4fa9SMark Johnston }
54069bc4fa9SMark Johnston 
54169bc4fa9SMark Johnston static void
termreset(void)54269bc4fa9SMark Johnston termreset(void)
54369bc4fa9SMark Johnston {
54469bc4fa9SMark Johnston 	if (ttyflg) {
54569bc4fa9SMark Johnston 		tcsetattr(STDOUT_FILENO, TCSADRAIN, &tt);
54669bc4fa9SMark Johnston 		ttyflg = 0;
54769bc4fa9SMark Johnston 	}
54869bc4fa9SMark Johnston }
54969bc4fa9SMark Johnston 
55069bc4fa9SMark Johnston static void
playback(FILE * fp)551df53360cSBrian Somers playback(FILE *fp)
552df53360cSBrian Somers {
553df53360cSBrian Somers 	struct timespec tsi, tso;
554df53360cSBrian Somers 	struct stamp stamp;
555df53360cSBrian Somers 	struct stat pst;
556df53360cSBrian Somers 	char buf[DEF_BUF];
557df53360cSBrian Somers 	off_t nread, save_len;
558df53360cSBrian Somers 	size_t l;
559df53360cSBrian Somers 	time_t tclock;
5607b45ad3fSSimon J. Gerraty 	time_t lclock;
561df53360cSBrian Somers 	int reg;
562df53360cSBrian Somers 
563df53360cSBrian Somers 	if (fstat(fileno(fp), &pst) == -1)
564df53360cSBrian Somers 		err(1, "fstat failed");
565df53360cSBrian Somers 
566df53360cSBrian Somers 	reg = S_ISREG(pst.st_mode);
5677b45ad3fSSimon J. Gerraty 	lclock = 0;
568df53360cSBrian Somers 
569df53360cSBrian Somers 	for (nread = 0; !reg || nread < pst.st_size; nread += save_len) {
570df53360cSBrian Somers 		if (fread(&stamp, sizeof(stamp), 1, fp) != 1) {
571df53360cSBrian Somers 			if (reg)
572df53360cSBrian Somers 				err(1, "reading playback header");
573df53360cSBrian Somers 			else
574df53360cSBrian Somers 				break;
575df53360cSBrian Somers 		}
576df53360cSBrian Somers 		swapstamp(stamp);
577df53360cSBrian Somers 		save_len = sizeof(stamp);
578df53360cSBrian Somers 
579df53360cSBrian Somers 		if (reg && stamp.scr_len >
580df53360cSBrian Somers 		    (uint64_t)(pst.st_size - save_len) - nread)
581df53360cSBrian Somers 			errx(1, "invalid stamp");
582df53360cSBrian Somers 
583df53360cSBrian Somers 		save_len += stamp.scr_len;
584df53360cSBrian Somers 		tclock = stamp.scr_sec;
585df53360cSBrian Somers 		tso.tv_sec = stamp.scr_sec;
586df53360cSBrian Somers 		tso.tv_nsec = stamp.scr_usec * 1000;
587473e9fcaSDag-Erling Smørgrav 		if (nread == 0)
588473e9fcaSDag-Erling Smørgrav 			tsi = tso;
589df53360cSBrian Somers 
590df53360cSBrian Somers 		switch (stamp.scr_direction) {
591df53360cSBrian Somers 		case 's':
592df53360cSBrian Somers 			if (!qflg)
593df53360cSBrian Somers 			    (void)printf("Script started on %s",
594df53360cSBrian Somers 				ctime(&tclock));
595df53360cSBrian Somers 			tsi = tso;
596df53360cSBrian Somers 			(void)consume(fp, stamp.scr_len, buf, reg);
59769bc4fa9SMark Johnston 			termset();
59869bc4fa9SMark Johnston 			atexit(termreset);
599df53360cSBrian Somers 			break;
600df53360cSBrian Somers 		case 'e':
60169bc4fa9SMark Johnston 			termreset();
602df53360cSBrian Somers 			if (!qflg)
603df53360cSBrian Somers 				(void)printf("\nScript done on %s",
604df53360cSBrian Somers 				    ctime(&tclock));
605df53360cSBrian Somers 			(void)consume(fp, stamp.scr_len, buf, reg);
606df53360cSBrian Somers 			break;
607df53360cSBrian Somers 		case 'i':
608df53360cSBrian Somers 			/* throw input away */
609df53360cSBrian Somers 			(void)consume(fp, stamp.scr_len, buf, reg);
610df53360cSBrian Somers 			break;
611df53360cSBrian Somers 		case 'o':
6126c4afed5SSimon J. Gerraty 			if (tflg) {
6136c4afed5SSimon J. Gerraty 				if (stamp.scr_len == 0)
6146c4afed5SSimon J. Gerraty 					continue;
6157b45ad3fSSimon J. Gerraty 				if (tclock - lclock > 0) {
6166c4afed5SSimon J. Gerraty 				    l = strftime(buf, sizeof buf, tstamp_fmt,
6176c4afed5SSimon J. Gerraty 					localtime(&tclock));
6186c4afed5SSimon J. Gerraty 				    (void)write(STDOUT_FILENO, buf, l);
6197b45ad3fSSimon J. Gerraty 				}
6207b45ad3fSSimon J. Gerraty 				lclock = tclock;
6216c4afed5SSimon J. Gerraty 			} else {
622df53360cSBrian Somers 				tsi.tv_sec = tso.tv_sec - tsi.tv_sec;
623df53360cSBrian Somers 				tsi.tv_nsec = tso.tv_nsec - tsi.tv_nsec;
624df53360cSBrian Somers 				if (tsi.tv_nsec < 0) {
625df53360cSBrian Somers 					tsi.tv_sec -= 1;
626df53360cSBrian Somers 					tsi.tv_nsec += 1000000000;
627df53360cSBrian Somers 				}
628df53360cSBrian Somers 				if (usesleep)
629df53360cSBrian Somers 					(void)nanosleep(&tsi, NULL);
630df53360cSBrian Somers 				tsi = tso;
6316c4afed5SSimon J. Gerraty 			}
632df53360cSBrian Somers 			while (stamp.scr_len > 0) {
633df53360cSBrian Somers 				l = MIN(DEF_BUF, stamp.scr_len);
634df53360cSBrian Somers 				if (fread(buf, sizeof(char), l, fp) != l)
635df53360cSBrian Somers 					err(1, "cannot read buffer");
636df53360cSBrian Somers 
637df53360cSBrian Somers 				(void)write(STDOUT_FILENO, buf, l);
638df53360cSBrian Somers 				stamp.scr_len -= l;
639df53360cSBrian Somers 			}
640df53360cSBrian Somers 			break;
641df53360cSBrian Somers 		default:
642df53360cSBrian Somers 			errx(1, "invalid direction");
643df53360cSBrian Somers 		}
644df53360cSBrian Somers 	}
645df53360cSBrian Somers 	(void)fclose(fp);
646df53360cSBrian Somers 	exit(0);
647df53360cSBrian Somers }
648*8ceac8e1SKyle Evans 
649*8ceac8e1SKyle Evans static void
resizeit(int signo __unused)650*8ceac8e1SKyle Evans resizeit(int signo __unused)
651*8ceac8e1SKyle Evans {
652*8ceac8e1SKyle Evans 	doresize = 1;
653*8ceac8e1SKyle Evans }
654