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