shell_cmds-216.40.4.tar.gz

[apple/shell_cmds.git] / script / script.c

diff --git a/script/script.c b/script/script.c
index 32d80f3d98fd2b5cca23be593d0ac4f76302b8b0..90e895ef0129f311c1f95a66dc8e071a9dbbe003 100644 (file)
--- a/script/script.c
+++ b/script/script.c
@@ -1,4 +1,5 @@
 /*
 /*

+ * Copyright (c) 2010, 2012  David E. O'Brien

  * Copyright (c) 1980, 1992, 1993
  *     The Regents of the University of California.  All rights reserved.
  *
  * Copyright (c) 1980, 1992, 1993
  *     The Regents of the University of California.  All rights reserved.
  *


@@ -10,10 +11,6 @@
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.

- * 3. All advertising materials mentioning features or use of this software
- *    must display the following acknowledgement:
- *     This product includes software developed by the University of
- *     California, Berkeley and its contributors.

  * 4. Neither the name of the University nor the names of its contributors
  *    may be used to endorse or promote products derived from this software
  *    without specific prior written permission.
  * 4. Neither the name of the University nor the names of its contributors
  *    may be used to endorse or promote products derived from this software
  *    without specific prior written permission.


@@ -31,32 +28,39 @@
  * SUCH DAMAGE.
  */
 
  * SUCH DAMAGE.
  */
 

-#include <sys/cdefs.h>
-
-__FBSDID("$FreeBSD: src/usr.bin/script/script.c,v 1.24 2004/02/15 17:30:13 cperciva Exp $");
-
+#include <sys/param.h>
+__FBSDID("$FreeBSD$");

 #ifndef lint
 static const char copyright[] =
 "@(#) Copyright (c) 1980, 1992, 1993\n\
        The Regents of the University of California.  All rights reserved.\n";
 #endif
 #ifndef lint
 static const char copyright[] =
 "@(#) Copyright (c) 1980, 1992, 1993\n\
        The Regents of the University of California.  All rights reserved.\n";
 #endif

-

 #ifndef lint
 static const char sccsid[] = "@(#)script.c     8.1 (Berkeley) 6/6/93";
 #endif
 
 #ifndef lint
 static const char sccsid[] = "@(#)script.c     8.1 (Berkeley) 6/6/93";
 #endif
 

-#include <sys/types.h>

 #include <sys/wait.h>
 #include <sys/stat.h>
 #include <sys/ioctl.h>
 #include <sys/time.h>
 #include <sys/wait.h>
 #include <sys/stat.h>
 #include <sys/ioctl.h>
 #include <sys/time.h>

+#include <sys/uio.h>
+#ifdef __APPLE__
+#include <libkern/OSByteOrder.h>
+#else
+#include <sys/endian.h>
+#endif
+#ifdef ENABLE_FILEMON
+#include <dev/filemon/filemon.h>
+#endif /* ENABLE_FILEMON */

 
 #include <err.h>
 #include <errno.h>
 #include <fcntl.h>
 
 #include <err.h>
 #include <errno.h>
 #include <fcntl.h>

-#ifndef __APPLE__
+#ifdef __APPLE__
+#include <util.h>
+#else

 #include <libutil.h>
 #include <libutil.h>

-#endif /* !__APPLE__ */
+#endif

 #include <paths.h>
 #include <signal.h>
 #include <stdio.h>
 #include <paths.h>
 #include <signal.h>
 #include <stdio.h>


@@ -64,23 +68,37 @@ static const char sccsid[] = "@(#)script.c  8.1 (Berkeley) 6/6/93";
 #include <string.h>
 #include <termios.h>
 #include <unistd.h>
 #include <string.h>
 #include <termios.h>
 #include <unistd.h>

-#ifdef __APPLE__
-#include <util.h>
-#endif /* __APPLE__ */

 
 

-FILE   *fscript;
-int    master, slave;
-int    child;
-const char *fname;
-int    qflg, ttyflg;
+#define DEF_BUF 65536

 
 

-struct termios tt;
+struct stamp {
+       uint64_t scr_len;       /* amount of data */
+       uint64_t scr_sec;       /* time it arrived in seconds... */
+       uint32_t scr_usec;      /* ...and microseconds */
+       uint32_t scr_direction; /* 'i', 'o', etc (also indicates endianness) */
+};

 
 

-void   done(int) __dead2;
-void   dooutput(void);
-void   doshell(char **);
-void   fail(void);
-void   finish(void);
+static FILE *fscript;
+static int master, slave;
+static int child;
+static const char *fname;
+static char *fmfname;
+#ifdef ENABLE_FILEMON
+static int fflg, qflg, ttyflg;
+#else /* !ENABLE_FILEMON */
+static int qflg, ttyflg;
+#endif /* ENABLE_FILEMON */
+static int usesleep, rawout;
+
+static struct termios tt;
+
+static void done(int) __dead2;
+static void doshell(char **);
+static void fail(void);
+static void finish(void);
+static void record(FILE *, char *, size_t, int);
+static void consume(FILE *, off_t, char *, int);
+static void playback(FILE *) __dead2;

 static void usage(void);
 
 int
 static void usage(void);
 
 int


@@ -89,26 +107,58 @@ main(int argc, char *argv[])
        int cc;
        struct termios rtt, stt;
        struct winsize win;
        int cc;
        struct termios rtt, stt;
        struct winsize win;

-       int aflg, kflg, ch, n;

        struct timeval tv, *tvp;
        time_t tvec, start;
        char obuf[BUFSIZ];
        char ibuf[BUFSIZ];
        fd_set rfd;
        struct timeval tv, *tvp;
        time_t tvec, start;
        char obuf[BUFSIZ];
        char ibuf[BUFSIZ];
        fd_set rfd;

-       int flushtime = 30;
+       int aflg, Fflg, kflg, pflg, ch, k, n;
+       int flushtime, readstdin;
+#ifdef ENABLE_FILEMON
+       int fm_fd, fm_log;
+#endif /* ENABLE_FILEMON */
+
+       aflg = Fflg = kflg = pflg = 0;
+       usesleep = 1;
+       rawout = 0;
+       flushtime = 30;
+#ifdef ENABLE_FILEMON
+       fm_fd = -1;     /* Shut up stupid "may be used uninitialized" GCC
+                          warning. (not needed w/clang) */
+#endif /* ENABLE_FILEMON */

 
 

-       aflg = kflg = 0;
-       while ((ch = getopt(argc, argv, "aqkt:")) != -1)
+#ifdef ENABLE_FILEMON
+       while ((ch = getopt(argc, argv, "adFfkpqrt:")) != -1)
+#else /* !ENABLE_FILEMON */
+       while ((ch = getopt(argc, argv, "adFkpqrt:")) != -1)
+#endif /* ENABLE_FILEMON */

                switch(ch) {
                case 'a':
                        aflg = 1;
                        break;
                switch(ch) {
                case 'a':
                        aflg = 1;
                        break;

-               case 'q':
-                       qflg = 1;
+               case 'd':
+                       usesleep = 0;

                        break;
                        break;

+               case 'F':
+                       Fflg = 1;
+                       break;
+#ifdef ENABLE_FILEMON
+               case 'f':
+                       fflg = 1;
+                       break;
+#endif /* ENABLE_FILEMON */

                case 'k':
                        kflg = 1;
                        break;
                case 'k':
                        kflg = 1;
                        break;

+               case 'p':
+                       pflg = 1;
+                       break;
+               case 'q':
+                       qflg = 1;
+                       break;
+               case 'r':
+                       rawout = 1;
+                       break;

                case 't':
                        flushtime = atoi(optarg);
                        if (flushtime < 0)
                case 't':
                        flushtime = atoi(optarg);
                        if (flushtime < 0)


@@ -128,10 +178,32 @@ main(int argc, char *argv[])
        } else
                fname = "typescript";
 
        } else
                fname = "typescript";
 

-       if ((fscript = fopen(fname, aflg ? "a" : "w")) == NULL)
+       if ((fscript = fopen(fname, pflg ? "r" : aflg ? "a" : "w")) == NULL)

                err(1, "%s", fname);
 
                err(1, "%s", fname);
 

-       if (ttyflg = isatty(STDIN_FILENO)) {
+#ifdef ENABLE_FILEMON
+       if (fflg) {
+               asprintf(&fmfname, "%s.filemon", fname);
+               if (!fmfname)
+                       err(1, "%s.filemon", fname);
+               if ((fm_fd = open("/dev/filemon", O_RDWR)) == -1)
+                       err(1, "open(\"/dev/filemon\", O_RDWR)");
+               if ((fm_log = open(fmfname, O_WRONLY | O_CREAT | O_TRUNC,
+                   S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH)) == -1)
+                       err(1, "open(%s)", fmfname);
+               if (ioctl(fm_fd, FILEMON_SET_FD, &fm_log) < 0)
+                       err(1, "Cannot set filemon log file descriptor");
+
+               /* Set up these two fd's to close on exec. */
+               (void)fcntl(fm_fd, F_SETFD, FD_CLOEXEC);
+               (void)fcntl(fm_log, F_SETFD, FD_CLOEXEC);
+       }
+#endif /* ENABLE_FILEMON */
+
+       if (pflg)
+               playback(fscript);
+
+       if ((ttyflg = isatty(STDIN_FILENO)) != 0) {

                if (tcgetattr(STDIN_FILENO, &tt) == -1)
                        err(1, "tcgetattr");
                if (ioctl(STDIN_FILENO, TIOCGWINSZ, &win) == -1)
                if (tcgetattr(STDIN_FILENO, &tt) == -1)
                        err(1, "tcgetattr");
                if (ioctl(STDIN_FILENO, TIOCGWINSZ, &win) == -1)


@@ -143,11 +215,30 @@ main(int argc, char *argv[])
                        err(1, "openpty");
        }
 
                        err(1, "openpty");
        }
 

+       if (rawout)
+               record(fscript, NULL, 0, 's');
+

        if (!qflg) {
                tvec = time(NULL);
                (void)printf("Script started, output file is %s\n", fname);
        if (!qflg) {
                tvec = time(NULL);
                (void)printf("Script started, output file is %s\n", fname);

-               (void)fprintf(fscript, "Script started on %s", ctime(&tvec));
+               if (!rawout) {
+                       (void)fprintf(fscript, "Script started on %s",
+                           ctime(&tvec));
+                       if (argv[0]) {
+                               fprintf(fscript, "command: ");
+                               for (k = 0 ; argv[k] ; ++k)
+                                       fprintf(fscript, "%s%s", k ? " " : "",
+                                               argv[k]);
+                               fprintf(fscript, "\n");
+                       }
+               }

                fflush(fscript);
                fflush(fscript);

+#ifdef ENABLE_FILEMON
+               if (fflg) {
+                       (void)printf("Filemon started, output file is %s\n",
+                           fmfname);
+               }
+#endif /* ENABLE_FILEMON */

        }
        if (ttyflg) {
                rtt = tt;
        }
        if (ttyflg) {
                rtt = tt;


@@ -163,23 +254,31 @@ main(int argc, char *argv[])
        }
        if (child == 0)
                doshell(argv);
        }
        if (child == 0)
                doshell(argv);

-#ifdef __APPLE__
-       (void)close(slave);
-#endif /* __APPLE__ */
+       close(slave);

 
 

-       if (flushtime > 0)
-               tvp = &tv;
-       else
-               tvp = NULL;
+#ifdef ENABLE_FILEMON
+       if (fflg && ioctl(fm_fd, FILEMON_SET_PID, &child) < 0)
+               err(1, "Cannot set filemon PID");
+#endif /* ENABLE_FILEMON */

 
 

-       start = time(0);
-       FD_ZERO(&rfd);
+       start = tvec = time(0);
+       readstdin = 1;

        for (;;) {
        for (;;) {

+               FD_ZERO(&rfd);

                FD_SET(master, &rfd);
                FD_SET(master, &rfd);

-               FD_SET(STDIN_FILENO, &rfd);
-               if (flushtime > 0) {
-                       tv.tv_sec = flushtime;
+               if (readstdin)
+                       FD_SET(STDIN_FILENO, &rfd);
+               if (!readstdin && ttyflg) {
+                       tv.tv_sec = 1;
+                       tv.tv_usec = 0;
+                       tvp = &tv;
+                       readstdin = 1;
+               } else if (flushtime > 0) {
+                       tv.tv_sec = flushtime - (tvec - start);

                        tv.tv_usec = 0;
                        tv.tv_usec = 0;

+                       tvp = &tv;
+               } else {
+                       tvp = NULL;

                }
                n = select(master + 1, &rfd, 0, 0, tvp);
                if (n < 0 && errno != EINTR)
                }
                n = select(master + 1, &rfd, 0, 0, tvp);
                if (n < 0 && errno != EINTR)


@@ -188,9 +287,16 @@ main(int argc, char *argv[])
                        cc = read(STDIN_FILENO, ibuf, BUFSIZ);
                        if (cc < 0)
                                break;
                        cc = read(STDIN_FILENO, ibuf, BUFSIZ);
                        if (cc < 0)
                                break;

-                       if (cc == 0)
-                               (void)write(master, ibuf, 0);
+                       if (cc == 0) {
+                               if (tcgetattr(master, &stt) == 0 &&
+                                   (stt.c_lflag & ICANON) != 0) {
+                                       (void)write(master, &stt.c_cc[VEOF], 1);
+                               }
+                               readstdin = 0;
+                       }

                        if (cc > 0) {
                        if (cc > 0) {

+                               if (rawout)
+                                       record(fscript, ibuf, cc, 'i');

                                (void)write(master, ibuf, cc);
                                if (kflg && tcgetattr(master, &stt) >= 0 &&
                                    ((stt.c_lflag & ECHO) == 0)) {
                                (void)write(master, ibuf, cc);
                                if (kflg && tcgetattr(master, &stt) >= 0 &&
                                    ((stt.c_lflag & ECHO) == 0)) {


@@ -203,13 +309,18 @@ main(int argc, char *argv[])
                        if (cc <= 0)
                                break;
                        (void)write(STDOUT_FILENO, obuf, cc);
                        if (cc <= 0)
                                break;
                        (void)write(STDOUT_FILENO, obuf, cc);

-                       (void)fwrite(obuf, 1, cc, fscript);
+                       if (rawout)
+                               record(fscript, obuf, cc, 'o');
+                       else
+                               (void)fwrite(obuf, 1, cc, fscript);

                }
                tvec = time(0);
                if (tvec - start >= flushtime) {
                        fflush(fscript);
                        start = tvec;
                }
                }
                tvec = time(0);
                if (tvec - start >= flushtime) {
                        fflush(fscript);
                        start = tvec;
                }

+               if (Fflg)
+                       fflush(fscript);

        }
        finish();
        done(0);
        }
        finish();
        done(0);


@@ -219,33 +330,31 @@ static void
 usage(void)
 {
        (void)fprintf(stderr,
 usage(void)
 {
        (void)fprintf(stderr,

-           "usage: script [-akq] [-t time] [file [command ...]]\n");
+#ifdef ENABLE_FILEMON
+           "usage: script [-adfkpqr] [-t time] [file [command ...]]\n");
+#else /* !ENABLE_FILEMON */
+           "usage: script [-adkpqr] [-t time] [file [command ...]]\n");
+#endif /* ENABLE_FILEMON */

        exit(1);
 }
 
        exit(1);
 }
 

-void
+static void

 finish(void)
 {
 finish(void)
 {

-       pid_t pid;
-       int die, e, status;
-
-       die = e = 0;
-       while ((pid = wait3(&status, WNOHANG, 0)) > 0)
-               if (pid == child) {
-                       die = 1;
-                       if (WIFEXITED(status))
-                               e = WEXITSTATUS(status);
-                       else if (WIFSIGNALED(status))
-                               e = WTERMSIG(status);
-                       else /* can't happen */
-                               e = 1;
-               }
+       int e, status;

 
 

-       if (die)
+       if (waitpid(child, &status, 0) == child) {
+               if (WIFEXITED(status))
+                       e = WEXITSTATUS(status);
+               else if (WIFSIGNALED(status))
+                       e = WTERMSIG(status);
+               else /* can't happen */
+                       e = 1;

                done(e);
                done(e);

+       }

 }
 
 }
 

-void
+static void

 doshell(char **av)
 {
        const char *shell;
 doshell(char **av)
 {
        const char *shell;


@@ -256,7 +365,9 @@ doshell(char **av)
 
        (void)close(master);
        (void)fclose(fscript);
 
        (void)close(master);
        (void)fclose(fscript);

+       free(fmfname);

        login_tty(slave);
        login_tty(slave);

+       setenv("SCRIPT", fname, 1);

        if (av[0]) {
                execvp(av[0], av);
                warn("%s", av[0]);
        if (av[0]) {
                execvp(av[0], av);
                warn("%s", av[0]);


@@ -267,14 +378,14 @@ doshell(char **av)
        fail();
 }
 
        fail();
 }
 

-void
+static void

 fail(void)
 {
        (void)kill(0, SIGTERM);
        done(1);
 }
 
 fail(void)
 {
        (void)kill(0, SIGTERM);
        done(1);
 }
 

-void
+static void

 done(int eno)
 {
        time_t tvec;
 done(int eno)
 {
        time_t tvec;


@@ -282,11 +393,154 @@ done(int eno)
        if (ttyflg)
                (void)tcsetattr(STDIN_FILENO, TCSAFLUSH, &tt);
        tvec = time(NULL);
        if (ttyflg)
                (void)tcsetattr(STDIN_FILENO, TCSAFLUSH, &tt);
        tvec = time(NULL);

+       if (rawout)
+               record(fscript, NULL, 0, 'e');

        if (!qflg) {
        if (!qflg) {

-               (void)fprintf(fscript,"\nScript done on %s", ctime(&tvec));
+               if (!rawout)
+                       (void)fprintf(fscript,"\nScript done on %s",
+                           ctime(&tvec));

                (void)printf("\nScript done, output file is %s\n", fname);
                (void)printf("\nScript done, output file is %s\n", fname);

+#ifdef ENABLE_FILEMON
+               if (fflg) {
+                       (void)printf("Filemon done, output file is %s\n",
+                           fmfname);
+               }
+#endif /* ENABLE_FILEMON */

        }
        (void)fclose(fscript);
        (void)close(master);
        exit(eno);
 }
        }
        (void)fclose(fscript);
        (void)close(master);
        exit(eno);
 }

+
+static void
+record(FILE *fp, char *buf, size_t cc, int direction)
+{
+       struct iovec iov[2];
+       struct stamp stamp;
+       struct timeval tv;
+
+       (void)gettimeofday(&tv, NULL);
+       stamp.scr_len = cc;
+       stamp.scr_sec = tv.tv_sec;
+       stamp.scr_usec = tv.tv_usec;
+       stamp.scr_direction = direction;
+       iov[0].iov_len = sizeof(stamp);
+       iov[0].iov_base = &stamp;
+       iov[1].iov_len = cc;
+       iov[1].iov_base = buf;
+       if (writev(fileno(fp), &iov[0], 2) == -1)
+               err(1, "writev");
+}
+
+static void
+consume(FILE *fp, off_t len, char *buf, int reg)
+{
+       size_t l;
+
+       if (reg) {
+               if (fseeko(fp, len, SEEK_CUR) == -1)
+                       err(1, NULL);
+       }
+       else {
+               while (len > 0) {
+                       l = MIN(DEF_BUF, len);
+                       if (fread(buf, sizeof(char), l, fp) != l)
+                               err(1, "cannot read buffer");
+                       len -= l;
+               }
+       }
+}
+
+#ifdef __APPLE__
+#define bswap32 OSSwapInt32
+#define bswap64 OSSwapInt64
+#endif /* __APPLE__ */
+#define swapstamp(stamp) do { \
+       if (stamp.scr_direction > 0xff) { \
+               stamp.scr_len = bswap64(stamp.scr_len); \
+               stamp.scr_sec = bswap64(stamp.scr_sec); \
+               stamp.scr_usec = bswap32(stamp.scr_usec); \
+               stamp.scr_direction = bswap32(stamp.scr_direction); \
+       } \
+} while (0/*CONSTCOND*/)
+
+static void
+playback(FILE *fp)
+{
+       struct timespec tsi, tso;
+       struct stamp stamp;
+       struct stat pst;
+       char buf[DEF_BUF];
+       off_t nread, save_len;
+       size_t l;
+       time_t tclock;
+       int reg;
+
+       if (fstat(fileno(fp), &pst) == -1)
+               err(1, "fstat failed");
+
+       reg = S_ISREG(pst.st_mode);
+
+       for (nread = 0; !reg || nread < pst.st_size; nread += save_len) {
+               if (fread(&stamp, sizeof(stamp), 1, fp) != 1) {
+                       if (reg)
+                               err(1, "reading playback header");
+                       else
+                               break;
+               }
+               swapstamp(stamp);
+               save_len = sizeof(stamp);
+
+               if (reg && stamp.scr_len >
+                   (uint64_t)(pst.st_size - save_len) - nread)
+                       errx(1, "invalid stamp");
+
+               save_len += stamp.scr_len;
+               tclock = stamp.scr_sec;
+               tso.tv_sec = stamp.scr_sec;
+               tso.tv_nsec = stamp.scr_usec * 1000;
+
+               switch (stamp.scr_direction) {
+               case 's':
+                       if (!qflg)
+                           (void)printf("Script started on %s",
+                               ctime(&tclock));
+                       tsi = tso;
+                       (void)consume(fp, stamp.scr_len, buf, reg);
+                       break;
+               case 'e':
+                       if (!qflg)
+                               (void)printf("\nScript done on %s",
+                                   ctime(&tclock));
+                       (void)consume(fp, stamp.scr_len, buf, reg);
+                       break;
+               case 'i':
+                       /* throw input away */
+                       (void)consume(fp, stamp.scr_len, buf, reg);
+                       break;
+               case 'o':
+                       tsi.tv_sec = tso.tv_sec - tsi.tv_sec;
+                       tsi.tv_nsec = tso.tv_nsec - tsi.tv_nsec;
+                       if (tsi.tv_nsec < 0) {
+                               tsi.tv_sec -= 1;
+                               tsi.tv_nsec += 1000000000;
+                       }
+                       if (usesleep)
+                               (void)nanosleep(&tsi, NULL);
+                       tsi = tso;
+                       while (stamp.scr_len > 0) {
+                               l = MIN(DEF_BUF, stamp.scr_len);
+                               if (fread(buf, sizeof(char), l, fp) != l)
+                                       err(1, "cannot read buffer");
+
+                               (void)write(STDOUT_FILENO, buf, l);
+                               stamp.scr_len -= l;
+                       }
+                       break;
+               default:
+                       errx(1, "invalid direction");
+               }
+       }
+       (void)fclose(fp);
+       exit(0);
+}