usr/src/cmd/script/script.c

   1 /*
   2  * Copyright 2009 Sun Microsystems, Inc.  All rights reserved.
   3  * Use is subject to license terms.
   4  */
   5
   6 /*      Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T */
   7 /*        All Rights Reserved   */
   8
   9
  10 /*
  11  * Copyright (c) 1980 Regents of the University of California.
  12  * All rights reserved.  The Berkeley software License Agreement
  13  * specifies the terms and conditions for redistribution.
  14  */
  15
  16 /*      Portions Copyright(c) 1988, Sun Microsystems, Inc.      */
  17 /*      All Rights Reserved.                                    */
  18
  19 /*
  20  * script: Produce a record of a terminal session.
  21  */
  22 #include <stdio.h>
  23 #include <stdlib.h>
  24 #include <unistd.h>
  25 #include <signal.h>
  26 #include <fcntl.h>
  27 #include <locale.h>
  28 #include <time.h>
  29 #include <sys/stropts.h>
  30 #include <sys/types.h>
  31 #include <sys/stat.h>
  32 #include <sys/termios.h>
  33 #include <sys/file.h>
  34 #include <errno.h>
  35
  36 int     grantpt();
  37 int     unlockpt();
  38 char    *ptsname();
  39 void    doinput() __NORETURN;
  40 void    dooutput();
  41 void    doshell();
  42 void    fixtty();
  43 void    fail();
  44 void    done() __NORETURN;
  45 void    getmaster();
  46 void    getslave();
  47
  48 char    *shell;
  49 FILE    *fscript;
  50 int     master;                 /* file descriptor for master pseudo-tty */
  51 int     slave;                  /* file descriptor for slave pseudo-tty */
  52 int     child;
  53 int     subchild;
  54 char    *fname = "typescript";
  55 void    sigwinch();
  56 void    finish();
  57
  58 struct  termios b;
  59 struct  winsize size;
  60 int     lb;
  61 int     l;
  62 char    *mptname = "/dev/ptmx"; /* master pseudo-tty device */
  63
  64 int     aflg;
  65
  66 int
  67 main(int argc, char *argv[])
  68 {
  69         uid_t ruidt;
  70         gid_t gidt;
  71
  72         (void) setlocale(LC_ALL, "");
  73 #if !defined(TEXT_DOMAIN)
  74 #define TEXT_DOMAIN     "SYS_TEST"
  75 #endif
  76         (void) textdomain(TEXT_DOMAIN);
  77
  78         shell = getenv("SHELL");
  79         if (shell == NULL)
  80                 shell = "/bin/sh";
  81         argc--, argv++;
  82         while (argc > 0 && argv[0][0] == '-') {
  83                 switch (argv[0][1]) {
  84
  85                 case 'a':
  86                         aflg++;
  87                         break;
  88
  89                 default:
  90                         fprintf(stderr,
  91                             gettext("usage: script [ -a ] [ typescript ]\n"));
  92                         exit(1);
  93                 }
  94                 argc--, argv++;
  95         }
  96         if (argc > 0)
  97                 fname = argv[0];
  98         ruidt = getuid();
  99         gidt = getgid();
 100         if ((fscript = fopen(fname, aflg ? "a" : "w")) == NULL) {
 101                 perror(fname);
 102                 fail();
 103         }
 104         setbuf(fscript, NULL);
 105         chown(fname, ruidt, gidt);
 106         getmaster();
 107         printf(gettext("Script started, file is %s\n"), fname);
 108         fixtty();
 109
 110         (void) signal(SIGCHLD, finish);
 111         child = fork();
 112         if (child < 0) {
 113                 perror("fork");
 114                 fail();
 115         }
 116         if (child == 0) {
 117                 subchild = child = fork();
 118                 if (child < 0) {
 119                         perror("fork");
 120                         fail();
 121                 }
 122                 if (child)
 123                         dooutput();
 124                 else
 125                         doshell();
 126         }
 127         doinput();
 128         /* NOTREACHED */
 129         return (0);
 130 }
 131
 132 void
 133 doinput()
 134 {
 135         char ibuf[BUFSIZ];
 136         int cc;
 137
 138         (void) fclose(fscript);
 139         sigset(SIGWINCH, sigwinch);
 140
 141         while ((cc = read(0, ibuf, BUFSIZ)) != 0) {
 142                 if (cc == -1) {
 143                         if (errno == EINTR) {   /* SIGWINCH probably */
 144                                 continue;
 145                         } else {
 146                                 break;
 147                         }
 148                 }
 149                 (void) write(master, ibuf, cc);
 150         }
 151         done();
 152 }
 153
 154 void
 155 sigwinch()
 156 {
 157         struct winsize ws;
 158
 159         if (ioctl(0, TIOCGWINSZ, &ws) == 0)
 160                 (void) ioctl(master, TIOCSWINSZ, &ws);
 161 }
 162
 163 #include <sys/wait.h>
 164
 165 void
 166 finish()
 167 {
 168         int status;
 169         register int pid;
 170         register int die = 0;
 171
 172         while ((pid = wait(&status)) > 0)
 173                 if (pid == child)
 174                         die = 1;
 175
 176         if (die)
 177                 done();
 178 }
 179
 180 void
 181 dooutput()
 182 {
 183         time_t tvec;
 184         char obuf[BUFSIZ];
 185         char tbuf[BUFSIZ];
 186         int cc;
 187
 188         (void) close(0);
 189         tvec = time((time_t *)0);
 190         strftime(tbuf, BUFSIZ, "%c", localtime(&tvec));
 191         fprintf(fscript, gettext("Script started on %s\n"), tbuf);
 192         for (;;) {
 193                 cc = read(master, obuf, sizeof (obuf));
 194                 if (cc <= 0)
 195                         break;
 196                 (void) write(1, obuf, cc);
 197                 (void) fwrite(obuf, 1, cc, fscript);
 198         }
 199         done();
 200 }
 201
 202 void
 203 doshell()
 204 {
 205
 206         setpgrp();      /* relinquish control terminal */
 207         getslave();
 208         (void) close(master);
 209         (void) fclose(fscript);
 210         (void) dup2(slave, 0);
 211         (void) dup2(slave, 1);
 212         (void) dup2(slave, 2);
 213         (void) close(slave);
 214         execl(shell, shell, "-i", (char *)0);
 215         perror(shell);
 216         fail();
 217 }
 218
 219 void
 220 fixtty()
 221 {
 222         struct termios sbuf;
 223
 224         sbuf = b;
 225         sbuf.c_iflag &= ~(INLCR|IGNCR|ICRNL|IUCLC|IXON);
 226         sbuf.c_oflag &= ~OPOST;
 227         sbuf.c_lflag &= ~(ICANON|ISIG|ECHO);
 228         sbuf.c_cc[VMIN] = 1;
 229         sbuf.c_cc[VTIME] = 0;
 230         (void) ioctl(0, TCSETSF, (char *)&sbuf);
 231 }
 232
 233 void
 234 fail()
 235 {
 236
 237         (void) kill(0, SIGTERM);
 238         done();
 239 }
 240
 241 void
 242 done()
 243 {
 244         time_t tvec;
 245         char tbuf[BUFSIZ];
 246
 247         if (subchild) {
 248                 tvec = time((time_t *)0);
 249                 strftime(tbuf, BUFSIZ, "%c", localtime(&tvec));
 250                 fprintf(fscript, gettext("\nscript done on %s\n"), tbuf);
 251                 (void) fclose(fscript);
 252                 (void) close(master);
 253         } else {
 254                 (void) ioctl(0, TCSETSW, (char *)&b);
 255                 printf(gettext("Script done, file is %s\n"), fname);
 256         }
 257         exit(0);
 258 }
 259
 260 void
 261 getmaster()
 262 {
 263         struct stat stb;
 264
 265         if ((master = open(mptname, O_RDWR)) >= 0) { /* a pseudo-tty is free */
 266                 (void) ioctl(0, TCGETS, (char *)&b);
 267                 (void) ioctl(0, TIOCGWINSZ, (char *)&size);
 268                 return;
 269         } else {                                /* out of pseudo-tty's */
 270                 perror(mptname);
 271                 fprintf(stderr, gettext("Out of pseudo-tty's\n"));
 272                 fail();
 273         }
 274 }
 275
 276 void
 277 getslave()
 278 {
 279         char *slavename;        /* name of slave pseudo-tty */
 280
 281         grantpt(master);                /* change permissions of slave */
 282         unlockpt(master);                       /* unlock slave */
 283         slavename = ptsname(master);            /* get name of slave */
 284         slave = open(slavename, O_RDWR);        /* open slave */
 285         if (slave < 0) {                        /* error on opening slave */
 286                 perror(slavename);
 287                 fail();
 288         }
 289         ioctl(slave, I_PUSH, "ptem");   /* push pt hw emulation module */
 290         ioctl(slave, I_PUSH, "ldterm");         /* push line discipline */
 291
 292         (void) ioctl(slave, TCSETSF, (char *)&b);
 293         (void) ioctl(slave, TIOCSWINSZ, (char *)&size);
 294 }