Imported Upstream version 1.8.7
[debian/sudo] / plugins / sudoers / sudoreplay.c
index e40fcefbfd4071af9841f441dbed530b992e9a66..8cd9488bd1990f3a9e1e7ba79b5eaa800191a6f7 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2009-2011 Todd C. Miller <Todd.Miller@courtesan.com>
+ * Copyright (c) 2009-2013 Todd C. Miller <Todd.Miller@courtesan.com>
  *
  * Permission to use, copy, modify, and distribute this software for any
  * purpose with or without fee is hereby granted, provided that the above
@@ -17,7 +17,7 @@
 #include <config.h>
 
 #include <sys/types.h>
-#include <sys/param.h>
+#include <sys/uio.h>
 #ifdef HAVE_SYS_SYSMACROS_H
 # include <sys/sysmacros.h>
 #endif
@@ -52,7 +52,7 @@
 #if TIME_WITH_SYS_TIME
 # include <time.h>
 #endif
-#ifndef HAVE_TIMESPEC
+#ifndef HAVE_STRUCT_TIMESPEC
 # include "compat/timespec.h"
 #endif
 #include <ctype.h>
 #ifdef HAVE_ZLIB_H
 # include <zlib.h>
 #endif
-#ifdef HAVE_SETLOCALE
-# include <locale.h>
-#endif
 #include <signal.h>
+#ifdef HAVE_STDBOOL_H
+# include <stdbool.h>
+#else
+# include "compat/stdbool.h"
+#endif /* HAVE_STDBOOL_H */
 
 #include <pathnames.h>
 
 #include "missing.h"
 #include "alloc.h"
 #include "error.h"
+#include "gettext.h"
+#include "logging.h"
+#include "sudo_plugin.h"
+#include "sudo_conf.h"
+#include "sudo_debug.h"
 
 #ifndef LINE_MAX
 # define LINE_MAX 2048
@@ -132,6 +139,8 @@ struct log_info {
     char *tty;
     char *cmd;
     time_t tstamp;
+    int rows;
+    int cols;
 };
 
 /*
@@ -189,8 +198,7 @@ extern time_t get_date(char *);
 extern char *get_timestr(time_t, int);
 extern int term_raw(int, int);
 extern int term_restore(int, int);
-extern void zero_bytes(volatile void *, size_t);
-void cleanup(int);
+extern void get_ttysize(int *rowp, int *colp);
 
 static int list_sessions(int, char **, const char *, const char *, const char *);
 static int parse_expr(struct search_node **, char **);
@@ -198,8 +206,13 @@ static void check_input(int, double *);
 static void delay(double);
 static void help(void) __attribute__((__noreturn__));
 static void usage(int);
-static void *open_io_fd(char *pathbuf, int len, const char *suffix);
+static int open_io_fd(char *pathbuf, int len, const char *suffix, union io_fd *fdp);
 static int parse_timing(const char *buf, const char *decimal, int *idx, double *seconds, size_t *nbytes);
+static struct log_info *parse_logfile(char *logfile);
+static void free_log_info(struct log_info *li);
+static size_t atomic_writev(int fd, struct iovec *iov, int iovcnt);
+static void sudoreplay_handler(int);
+static void sudoreplay_cleanup(void);
 
 #ifdef HAVE_REGCOMP
 # define REGEX_T       regex_t
@@ -219,29 +232,46 @@ static int parse_timing(const char *buf, const char *decimal, int *idx, double *
     (s)[5] == '/' && \
     isalnum((unsigned char)(s)[6]) && isalnum((unsigned char)(s)[7]) && \
     (s)[8] == '/' && (s)[9] == 'l' && (s)[10] == 'o' && (s)[11] == 'g' && \
-    (s)[9] == '\0')
+    (s)[12] == '\0')
+
+__dso_public int main(int argc, char *argv[]);
 
 int
 main(int argc, char *argv[])
 {
-    int ch, idx, plen, nready, interactive = 0, listonly = 0;
-    const char *id, *user = NULL, *pattern = NULL, *tty = NULL, *decimal = ".";
+    int ch, idx, plen, exitcode = 0, rows = 0, cols = 0;
+    bool interactive = false, listonly = false, need_nlcr = false;
+    const char *decimal, *id, *user = NULL, *pattern = NULL, *tty = NULL;
     char path[PATH_MAX], buf[LINE_MAX], *cp, *ep;
     double seconds, to_wait, speed = 1.0, max_wait = 0;
-    FILE *lfile;
-    fd_set *fdsw;
     sigaction_t sa;
-    size_t len, nbytes, nread, off;
-    ssize_t nwritten;
+    size_t len, nbytes, nread;
+    struct log_info *li;
+    struct iovec *iov = NULL;
+    int iovcnt = 0, iovmax = 0;
+    debug_decl(main, SUDO_DEBUG_MAIN)
+
+#if defined(SUDO_DEVEL) && defined(__OpenBSD__)
+    {
+       extern char *malloc_options;
+       malloc_options = "AFGJPR";
+    }  
+#endif
 
 #if !defined(HAVE_GETPROGNAME) && !defined(HAVE___PROGNAME)
     setprogname(argc > 0 ? argv[0] : "sudoreplay");
 #endif
 
-#ifdef HAVE_SETLOCALE
-    setlocale(LC_ALL, "");
+    sudoers_setlocale(SUDOERS_LOCALE_USER, NULL);
     decimal = localeconv()->decimal_point;
-#endif
+    bindtextdomain("sudoers", LOCALEDIR); /* XXX - should have sudoreplay domain */
+    textdomain("sudoers");
+
+    /* Register fatal/fatalx callback. */
+    fatal_callback_register(sudoreplay_cleanup);
+
+    /* Read sudo.conf. */
+    sudo_conf_read(NULL);
 
     while ((ch = getopt(argc, argv, "d:f:hlm:s:V")) != -1) {
        switch(ch) {
@@ -259,30 +289,30 @@ main(int argc, char *argv[])
                else if (strcmp(cp, "ttyout") == 0)
                    SET(replay_filter, 1 << IOFD_TTYOUT);
                else
-                   errorx(1, "invalid filter option: %s", optarg);
+                   fatalx(_("invalid filter option: %s"), optarg);
            }
            break;
        case 'h':
            help();
            /* NOTREACHED */
        case 'l':
-           listonly = 1;
+           listonly = true;
            break;
        case 'm':
            errno = 0;
            max_wait = strtod(optarg, &ep);
            if (*ep != '\0' || errno != 0)
-               errorx(1, "invalid max wait: %s", optarg);
+               fatalx(_("invalid max wait: %s"), optarg);
            break;
        case 's':
            errno = 0;
            speed = strtod(optarg, &ep);
            if (*ep != '\0' || errno != 0)
-               errorx(1, "invalid speed factor: %s", optarg);
+               fatalx(_("invalid speed factor: %s"), optarg);
            break;
        case 'V':
-           (void) printf("%s version %s\n", getprogname(), PACKAGE_VERSION);
-           exit(0);
+           (void) printf(_("%s version %s\n"), getprogname(), PACKAGE_VERSION);
+           goto done;
        default:
            usage(1);
            /* NOTREACHED */
@@ -292,8 +322,10 @@ main(int argc, char *argv[])
     argc -= optind;
     argv += optind;
 
-    if (listonly)
-       exit(list_sessions(argc, argv, pattern, user, tty));
+    if (listonly) {
+       exitcode = list_sessions(argc, argv, pattern, user, tty);
+       goto done;
+    }
 
     if (argc != 1)
        usage(1);
@@ -304,13 +336,13 @@ main(int argc, char *argv[])
        plen = snprintf(path, sizeof(path), "%s/%.2s/%.2s/%.2s/timing",
            session_dir, id, &id[2], &id[4]);
        if (plen <= 0 || plen >= sizeof(path))
-           errorx(1, "%s/%.2s/%.2s/%.2s/%.2s/timing: %s", session_dir,
+           fatalx(_("%s/%.2s/%.2s/%.2s/timing: %s"), session_dir,
                id, &id[2], &id[4], strerror(ENAMETOOLONG));
     } else {
        plen = snprintf(path, sizeof(path), "%s/%s/timing",
            session_dir, id);
        if (plen <= 0 || plen >= sizeof(path))
-           errorx(1, "%s/%s/timing: %s", session_dir,
+           fatalx(_("%s/%s/timing: %s"), session_dir,
                id, strerror(ENAMETOOLONG));
     }
     plen -= 7;
@@ -318,35 +350,36 @@ main(int argc, char *argv[])
     /* Open files for replay, applying replay filter for the -f flag. */
     for (idx = 0; idx < IOFD_MAX; idx++) {
        if (ISSET(replay_filter, 1 << idx) || idx == IOFD_TIMING) {
-           io_fds[idx].v = open_io_fd(path, plen, io_fnames[idx]);
-           if (io_fds[idx].v == NULL)
-               error(1, "unable to open %s", path);
+           if (open_io_fd(path, plen, io_fnames[idx], &io_fds[idx]) == -1)
+               fatal(_("unable to open %s"), path);
        }
     }
 
-    /* Read log file. */
+    /* Parse log file. */
     path[plen] = '\0';
     strlcat(path, "/log", sizeof(path));
-    lfile = fopen(path, "r");
-    if (lfile == NULL)
-       error(1, "unable to open %s", path);
-    cp = NULL;
-    len = 0;
-    /* Pull out command (third line). */
-    if (getline(&cp, &len, lfile) == -1 ||
-       getline(&cp, &len, lfile) == -1 ||
-       getline(&cp, &len, lfile) == -1) {
-       errorx(1, "invalid log file %s", path);
+    if ((li = parse_logfile(path)) == NULL)
+       exit(1);
+    printf(_("Replaying sudo session: %s\n"), li->cmd);
+
+    /* Make sure the terminal is large enough. */
+    get_ttysize(&rows, &cols);
+    if (li->rows != 0 && li->cols != 0) {
+       if (li->rows > rows) {
+           printf(_("Warning: your terminal is too small to properly replay the log.\n"));
+           printf(_("Log geometry is %d x %d, your terminal's geometry is %d x %d."), li->rows, li->cols, rows, cols);
+       }
     }
-    printf("Replaying sudo session: %s", cp);
-    free(cp);
-    fclose(lfile);
+
+    /* Done with parsed log file. */
+    free_log_info(li);
+    li = NULL;
 
     fflush(stdout);
-    zero_bytes(&sa, sizeof(sa));
+    memset(&sa, 0, sizeof(sa));
     sigemptyset(&sa.sa_mask);
     sa.sa_flags = SA_RESETHAND;
-    sa.sa_handler = cleanup;
+    sa.sa_handler = sudoreplay_handler;
     (void) sigaction(SIGINT, &sa, NULL);
     (void) sigaction(SIGKILL, &sa, NULL);
     (void) sigaction(SIGTERM, &sa, NULL);
@@ -364,10 +397,10 @@ main(int argc, char *argv[])
        if (ch != -1)
            (void) fcntl(STDIN_FILENO, F_SETFL, ch | O_NONBLOCK);
        if (!term_raw(STDIN_FILENO, 1))
-           error(1, "cannot set tty to raw mode");
+           fatal(_("unable to set tty to raw mode"));
+       iovmax = 32;
+       iov = ecalloc(iovmax, sizeof(*iov));
     }
-    fdsw = (fd_set *)emalloc2(howmany(STDOUT_FILENO + 1, NFDBITS),
-       sizeof(fd_mask));
 
     /*
      * Timing file consists of line of the format: "%f %d\n"
@@ -377,8 +410,10 @@ main(int argc, char *argv[])
 #else
     while (fgets(buf, sizeof(buf), io_fds[IOFD_TIMING].f) != NULL) {
 #endif
+       char last_char = '\0';
+
        if (!parse_timing(buf, decimal, &idx, &seconds, &nbytes))
-           errorx(1, "invalid timing file line: %s", buf);
+           fatalx(_("invalid timing file line: %s"), buf);
 
        if (interactive)
            check_input(STDIN_FILENO, &speed);
@@ -393,6 +428,10 @@ main(int argc, char *argv[])
        if (io_fds[idx].v == NULL)
            continue;
 
+       /* Check whether we need to convert newline to CR LF pairs. */
+       if (interactive) 
+           need_nlcr = (idx == IOFD_STDOUT || idx == IOFD_STDERR);
+
        /* All output is sent to stdout. */
        while (nbytes != 0) {
            if (nbytes > sizeof(buf))
@@ -405,29 +444,65 @@ main(int argc, char *argv[])
            nread = fread(buf, 1, len, io_fds[idx].f);
 #endif
            nbytes -= nread;
-           off = 0;
-           do {
-               /* no stdio, must be unbuffered */
-               nwritten = write(STDOUT_FILENO, buf + off, nread - off);
-               if (nwritten == -1) {
-                   if (errno == EINTR)
-                       continue;
-                   if (errno == EAGAIN) {
-                       FD_SET(STDOUT_FILENO, fdsw);
-                       do {
-                           nready = select(STDOUT_FILENO + 1, NULL, fdsw, NULL, NULL);
-                       } while (nready == -1 && errno == EINTR);
-                       if (nready == 1)
-                           continue;
+
+           /* Convert newline to carriage return + linefeed if needed. */
+           if (need_nlcr) {
+               size_t remainder = nread;
+               size_t linelen;
+               iovcnt = 0;
+               cp = buf;
+               ep = cp - 1;
+               /* Handle a "\r\n" pair that spans a buffer. */
+               if (last_char == '\r' && buf[0] == '\n') {
+                   ep++;
+                   remainder--;
+               }
+               while ((ep = memchr(ep + 1, '\n', remainder)) != NULL) {
+                   /* Is there already a carriage return? */
+                   if (cp != ep && ep[-1] == '\r') {
+                       remainder = (size_t)(&buf[nread - 1] - ep);
+                       continue;
                    }
-                   error(1, "writing to standard output");
+
+                   /* Store the line in iov followed by \r\n pair. */
+                   if (iovcnt + 3 > iovmax) {
+                       iovmax <<= 1;
+                       iov = erealloc3(iov, iovmax, sizeof(*iov));
+                   }
+                   linelen = (size_t)(ep - cp) + 1;
+                   iov[iovcnt].iov_base = cp;
+                   iov[iovcnt].iov_len = linelen - 1; /* not including \n */
+                   iovcnt++;
+                   iov[iovcnt].iov_base = "\r\n";
+                   iov[iovcnt].iov_len = 2;
+                   iovcnt++;
+                   cp = ep + 1;
+                   remainder -= linelen;
+               }
+               if (cp - buf != nread) {
+                   /*
+                    * Partial line without a linefeed or multiple lines
+                    * with \r\n pairs.
+                    */
+                   iov[iovcnt].iov_base = cp;
+                   iov[iovcnt].iov_len = nread - (cp - buf);
+                   iovcnt++;
                }
-               off += nwritten;
-           } while (nread > off);
+               last_char = buf[nread - 1]; /* stash last char of old buffer */
+           } else {
+               /* No conversion needed. */
+               iov[0].iov_base = buf;
+               iov[0].iov_len = nread;
+               iovcnt = 1;
+           }
+           if (atomic_writev(STDOUT_FILENO, iov, iovcnt) == -1)
+               fatal(_("writing to standard output"));
        }
     }
     term_restore(STDIN_FILENO, 1);
-    exit(0);
+done:
+    sudo_debug_exit_int(__func__, __FILE__, __LINE__, sudo_debug_subsys, exitcode);
+    exit(exitcode);
 }
 
 static void
@@ -449,21 +524,89 @@ delay(double secs)
       memcpy(&ts, &rts, sizeof(ts));
       rval = nanosleep(&ts, &rts);
     } while (rval == -1 && errno == EINTR);
-    if (rval == -1)
-       error(1, "nanosleep: tv_sec %ld, tv_nsec %ld", ts.tv_sec, ts.tv_nsec);
+    if (rval == -1) {
+       fatal_nodebug("nanosleep: tv_sec %lld, tv_nsec %ld",
+           (long long)ts.tv_sec, (long)ts.tv_nsec);
+    }
 }
 
-static void *
-open_io_fd(char *path, int len, const char *suffix)
+static int
+open_io_fd(char *path, int len, const char *suffix, union io_fd *fdp)
 {
+    debug_decl(open_io_fd, SUDO_DEBUG_UTIL)
+
     path[len] = '\0';
     strlcat(path, suffix, PATH_MAX);
 
 #ifdef HAVE_ZLIB_H
-    return gzopen(path, "r");
+    fdp->g = gzopen(path, "r");
 #else
-    return fopen(path, "r");
+    fdp->f = fopen(path, "r");
 #endif
+    debug_return_int(fdp->v ? 0 : -1);
+}
+
+/*
+ * Call writev(), restarting as needed and handling EAGAIN since
+ * fd may be in non-blocking mode.
+ */
+static size_t
+atomic_writev(int fd, struct iovec *iov, int iovcnt)
+{
+    ssize_t n, nwritten = 0;
+    size_t count, remainder, nbytes = 0;
+    int i;
+    debug_decl(atomic_writev, SUDO_DEBUG_UTIL)
+
+    for (i = 0; i < iovcnt; i++)
+       nbytes += iov[i].iov_len;
+
+    for (;;) {
+       n = writev(STDOUT_FILENO, iov, iovcnt);
+       if (n > 0) {
+           nwritten += n;
+           remainder = nbytes - nwritten;
+           if (remainder == 0)
+               break;
+           /* short writev, adjust iov and do the rest. */
+           count = 0;
+           i = iovcnt;
+           while (i--) {
+               count += iov[i].iov_len;
+               if (count == remainder) {
+                   iov += i;
+                   iovcnt -= i;
+                   break;
+               }
+               if (count > remainder) {
+                   size_t off = (count - remainder);
+                   /* XXX - side effect prevents iov from being const */
+                   iov[i].iov_base = (char *)iov[i].iov_base + off;
+                   iov[i].iov_len -= off;
+                   iov += i;
+                   iovcnt -= i;
+                   break;
+               }
+           }
+           continue;
+       }
+       if (n == 0 || errno == EAGAIN) {
+           int nready;
+           fd_set fdsw;
+           FD_ZERO(&fdsw);
+           FD_SET(STDOUT_FILENO, &fdsw);
+           do {
+               nready = select(STDOUT_FILENO + 1, NULL, &fdsw, NULL, NULL);
+           } while (nready == -1 && errno == EINTR);
+           if (nready == 1)
+               continue;
+       }
+       if (errno == EINTR)
+           continue;
+       nwritten = -1;
+       break;
+    }
+    debug_return_size_t(nwritten);
 }
 
 /*
@@ -474,6 +617,7 @@ parse_expr(struct search_node **headp, char *argv[])
 {
     struct search_node *sn, *newsn;
     char or = 0, not = 0, type, **av;
+    debug_decl(parse_expr, SUDO_DEBUG_UTIL)
 
     sn = *headp;
     for (av = argv; *av; av++) {
@@ -494,7 +638,7 @@ parse_expr(struct search_node **headp, char *argv[])
            continue;
        case 'c': /* command */
            if (av[0][1] == '\0')
-               errorx(1, "ambiguous expression \"%s\"", *av);
+               fatalx(_("ambiguous expression \"%s\""), *av);
            if (strncmp(*av, "cwd", strlen(*av)) == 0)
                type = ST_CWD;
            else if (strncmp(*av, "command", strlen(*av)) == 0)
@@ -519,7 +663,7 @@ parse_expr(struct search_node **headp, char *argv[])
            break;
        case 't': /* tty or to date */
            if (av[0][1] == '\0')
-               errorx(1, "ambiguous expression \"%s\"", *av);
+               fatalx(_("ambiguous expression \"%s\""), *av);
            if (strncmp(*av, "todate", strlen(*av)) == 0)
                type = ST_TODATE;
            else if (strncmp(*av, "tty", strlen(*av)) == 0)
@@ -536,7 +680,7 @@ parse_expr(struct search_node **headp, char *argv[])
            if (av[0][1] != '\0')
                goto bad;
            if (stack_top + 1 == STACK_NODE_SIZE) {
-               errorx(1, "too many parenthesized expressions, max %d",
+               fatalx(_("too many parenthesized expressions, max %d"),
                    STACK_NODE_SIZE);
            }
            node_stack[stack_top++] = sn;
@@ -547,37 +691,37 @@ parse_expr(struct search_node **headp, char *argv[])
                goto bad;
            /* pop */
            if (--stack_top < 0)
-               errorx(1, "unmatched ')' in expression");
+               fatalx(_("unmatched ')' in expression"));
            if (node_stack[stack_top])
                sn->next = node_stack[stack_top]->next;
-           return av - argv + 1;
+           debug_return_int(av - argv + 1);
        bad:
        default:
-           errorx(1, "unknown search term \"%s\"", *av);
+           fatalx(_("unknown search term \"%s\""), *av);
            /* NOTREACHED */
        }
 
        /* Allocate new search node */
-       newsn = emalloc(sizeof(*newsn));
-       newsn->next = NULL;
+       newsn = ecalloc(1, sizeof(*newsn));
        newsn->type = type;
        newsn->or = or;
        newsn->negated = not;
+       /* newsn->next = NULL; */
        if (type == ST_EXPR) {
            av += parse_expr(&newsn->u.expr, av + 1);
        } else {
            if (*(++av) == NULL)
-               errorx(1, "%s requires an argument", av[-1]);
+               fatalx(_("%s requires an argument"), av[-1]);
 #ifdef HAVE_REGCOMP
            if (type == ST_PATTERN) {
                if (regcomp(&newsn->u.cmdre, *av, REG_EXTENDED|REG_NOSUB) != 0)
-                   errorx(1, "invalid regex: %s", *av);
+                   fatalx(_("invalid regular expression: %s"), *av);
            } else
 #endif
            if (type == ST_TODATE || type == ST_FROMDATE) {
                newsn->u.tstamp = get_date(*av);
                if (newsn->u.tstamp == -1)
-                   errorx(1, "could not parse date \"%s\"", *av);
+                   fatalx(_("could not parse date \"%s\""), *av);
            } else {
                newsn->u.ptr = *av;
            }
@@ -590,20 +734,22 @@ parse_expr(struct search_node **headp, char *argv[])
        sn = newsn;
     }
     if (stack_top)
-       errorx(1, "unmatched '(' in expression");
+       fatalx(_("unmatched '(' in expression"));
     if (or)
-       errorx(1, "illegal trailing \"or\"");
+       fatalx(_("illegal trailing \"or\""));
     if (not)
-       errorx(1, "illegal trailing \"!\"");
+       fatalx(_("illegal trailing \"!\""));
 
-    return av - argv;
+    debug_return_int(av - argv);
 }
 
-static int
+static bool
 match_expr(struct search_node *head, struct log_info *log)
 {
     struct search_node *sn;
-    int matched = 1, rc;
+    bool matched = true;
+    int rc;
+    debug_decl(match_expr, SUDO_DEBUG_UTIL)
 
     for (sn = head; sn; sn = sn->next) {
        /* If we have no match, skip ahead to the next OR entry. */
@@ -635,7 +781,7 @@ match_expr(struct search_node *head, struct log_info *log)
            if (rc && rc != REG_NOMATCH) {
                char buf[BUFSIZ];
                regerror(rc, &sn->u.cmdre, buf, sizeof(buf));
-               errorx(1, "%s", buf);
+               fatalx("%s", buf);
            }
            matched = rc == REG_NOMATCH ? 0 : 1;
 #else
@@ -652,22 +798,22 @@ match_expr(struct search_node *head, struct log_info *log)
        if (sn->negated)
            matched = !matched;
     }
-    return matched;
+    debug_return_bool(matched);
 }
 
-static int
-list_session(char *logfile, REGEX_T *re, const char *user, const char *tty)
+static struct log_info *
+parse_logfile(char *logfile)
 {
     FILE *fp;
-    char *buf = NULL, *cmd = NULL, *cwd = NULL, idbuf[7], *idstr, *cp;
-    struct log_info li;
+    char *buf = NULL, *cp, *ep;
     size_t bufsize = 0, cwdsize = 0, cmdsize = 0;
-    int rval = -1;
+    struct log_info *li = NULL;
+    debug_decl(list_session, SUDO_DEBUG_UTIL)
 
     fp = fopen(logfile, "r");
     if (fp == NULL) {
-       warning("unable to open %s", logfile);
-       goto done;
+       warning(_("unable to open %s"), logfile);
+       goto bad;
     }
 
     /*
@@ -676,56 +822,111 @@ list_session(char *logfile, REGEX_T *re, const char *user, const char *tty)
      *  2) cwd
      *  3) command with args
      */
+    li = ecalloc(1, sizeof(*li));
     if (getline(&buf, &bufsize, fp) == -1 ||
-       getline(&cwd, &cwdsize, fp) == -1 ||
-       getline(&cmd, &cmdsize, fp) == -1) {
-       goto done;
+       getline(&li->cwd, &cwdsize, fp) == -1 ||
+       getline(&li->cmd, &cmdsize, fp) == -1) {
+       goto bad;
     }
 
-    /* crack the log line: timestamp:user:runas_user:runas_group:tty */
+    /* Strip the newline from the cwd and command. */
+    li->cwd[strcspn(li->cwd, "\n")] = '\0';
+    li->cmd[strcspn(li->cmd, "\n")] = '\0';
+
+    /*
+     * Crack the log line (rows and cols not present in old versions).
+     * timestamp:user:runas_user:runas_group:tty:rows:cols
+     */
     buf[strcspn(buf, "\n")] = '\0';
-    if ((li.tstamp = atoi(buf)) == 0)
-       goto done;
 
-    if ((cp = strchr(buf, ':')) == NULL)
-       goto done;
-    *cp++ = '\0';
-    li.user = cp;
+    /* timestamp */
+    if ((ep = strchr(buf, ':')) == NULL)
+       goto bad;
+    if ((li->tstamp = atoi(buf)) == 0)
+       goto bad;
 
-    if ((cp = strchr(cp, ':')) == NULL)
-       goto done;
-    *cp++ = '\0';
-    li.runas_user = cp;
+    /* user */
+    cp = ep + 1;
+    if ((ep = strchr(cp, ':')) == NULL)
+       goto bad;
+    li->user = estrndup(cp, (size_t)(ep - cp));
 
-    if ((cp = strchr(cp, ':')) == NULL)
-       goto done;
-    *cp++ = '\0';
-    li.runas_group = cp;
+    /* runas user */
+    cp = ep + 1;
+    if ((ep = strchr(cp, ':')) == NULL)
+       goto bad;
+    li->runas_user = estrndup(cp, (size_t)(ep - cp));
 
-    if ((cp = strchr(cp, ':')) == NULL)
-       goto done;
-    *cp++ = '\0';
-    li.tty = cp;
+    /* runas group */
+    cp = ep + 1;
+    if ((ep = strchr(cp, ':')) == NULL)
+       goto bad;
+    if (cp != ep)
+       li->runas_group = estrndup(cp, (size_t)(ep - cp));
+
+    /* tty, followed by optional rows + columns */
+    cp = ep + 1;
+    if ((ep = strchr(cp, ':')) == NULL) {
+       li->tty = estrdup(cp);
+    } else {
+       li->tty = estrndup(cp, (size_t)(ep - cp));
+       cp = ep + 1;
+       li->rows = atoi(cp);
+       if ((ep = strchr(cp, ':')) != NULL) {
+           cp = ep + 1;
+           li->cols = atoi(cp);
+       }
+    }
+    fclose(fp);
+    efree(buf);
+    debug_return_ptr(li);
 
-    cwd[strcspn(cwd, "\n")] = '\0';
-    li.cwd = cwd;
+bad:
+    if (fp != NULL)
+       fclose(fp);
+    efree(buf);
+    free_log_info(li);
+    debug_return_ptr(NULL);
+}
+
+static void
+free_log_info(struct log_info *li)
+{
+    if (li != NULL) {
+       efree(li->cwd);
+       efree(li->user);
+       efree(li->runas_user);
+       efree(li->runas_group);
+       efree(li->tty);
+       efree(li->cmd);
+       efree(li);
+    }
+}
 
-    cmd[strcspn(cmd, "\n")] = '\0';
-    li.cmd = cmd;
+static int
+list_session(char *logfile, REGEX_T *re, const char *user, const char *tty)
+{
+    char idbuf[7], *idstr, *cp;
+    struct log_info *li;
+    int rval = -1;
+    debug_decl(list_session, SUDO_DEBUG_UTIL)
+
+    if ((li = parse_logfile(logfile)) == NULL)
+       goto done;
 
     /* Match on search expression if there is one. */
-    if (search_expr && !match_expr(search_expr, &li))
+    if (search_expr && !match_expr(search_expr, li))
        goto done;
 
     /* Convert from /var/log/sudo-sessions/00/00/01/log to 000001 */
     cp = logfile + strlen(session_dir) + 1;
     if (IS_IDLOG(cp)) {
-       idbuf[0] = cp[7];
-       idbuf[1] = cp[6];
-       idbuf[2] = cp[4];
-       idbuf[3] = cp[3];
-       idbuf[4] = cp[1];
-       idbuf[5] = cp[0];
+       idbuf[0] = cp[0];
+       idbuf[1] = cp[1];
+       idbuf[2] = cp[3];
+       idbuf[3] = cp[4];
+       idbuf[4] = cp[6];
+       idbuf[5] = cp[7];
        idbuf[6] = '\0';
        idstr = idbuf;
     } else {
@@ -733,53 +934,95 @@ list_session(char *logfile, REGEX_T *re, const char *user, const char *tty)
        cp[strlen(cp) - 4] = '\0';
        idstr = cp;
     }
+    /* XXX - print rows + cols? */
     printf("%s : %s : TTY=%s ; CWD=%s ; USER=%s ; ",
-       get_timestr(li.tstamp, 1), li.user, li.tty, li.cwd, li.runas_user);
-    if (*li.runas_group)
-       printf("GROUP=%s ; ", li.runas_group);
-    printf("TSID=%s ; COMMAND=%s\n", idstr, li.cmd);
+       get_timestr(li->tstamp, 1), li->user, li->tty, li->cwd, li->runas_user);
+    if (li->runas_group)
+       printf("GROUP=%s ; ", li->runas_group);
+    printf("TSID=%s ; COMMAND=%s\n", idstr, li->cmd);
 
     rval = 0;
 
 done:
-    fclose(fp);
-    return rval;
+    free_log_info(li);
+    debug_return_int(rval);
+}
+
+static int
+session_compare(const void *v1, const void *v2)
+{
+    const char *s1 = *(const char **)v1;
+    const char *s2 = *(const char **)v2;
+    return strcmp(s1, s2);
 }
 
+/* XXX - always returns 0, calls fatal() on failure */
 static int
 find_sessions(const char *dir, REGEX_T *re, const char *user, const char *tty)
 {
     DIR *d;
     struct dirent *dp;
     struct stat sb;
-    size_t sdlen;
-    int len;
-    char pathbuf[PATH_MAX];
+    size_t sdlen, sessions_len = 0, sessions_size = 36*36;
+    int i, len;
+    char pathbuf[PATH_MAX], **sessions = NULL;
+#ifdef HAVE_STRUCT_DIRENT_D_TYPE
+    bool checked_type = true;
+#else
+    const bool checked_type = false;
+#endif
+    debug_decl(find_sessions, SUDO_DEBUG_UTIL)
 
     d = opendir(dir);
     if (d == NULL)
-       error(1, "unable to open %s", dir);
+       fatal(_("unable to open %s"), dir);
 
     /* XXX - would be faster to chdir and use relative names */
     sdlen = strlcpy(pathbuf, dir, sizeof(pathbuf));
     if (sdlen + 1 >= sizeof(pathbuf)) {
        errno = ENAMETOOLONG;
-       error(1, "%s/", dir);
+       fatal("%s/", dir);
     }
     pathbuf[sdlen++] = '/';
     pathbuf[sdlen] = '\0';
+
+    /* Store potential session dirs for sorting. */
+    sessions = emalloc2(sessions_size, sizeof(char *));
     while ((dp = readdir(d)) != NULL) {
        /* Skip "." and ".." */
        if (dp->d_name[0] == '.' && (dp->d_name[1] == '\0' ||
            (dp->d_name[1] == '.' && dp->d_name[2] == '\0')))
            continue;
+#ifdef HAVE_STRUCT_DIRENT_D_TYPE
+       if (checked_type) {
+           if (dp->d_type != DT_DIR) {
+               /* Not all file systems support d_type. */
+               if (dp->d_type != DT_UNKNOWN)
+                   continue;
+               checked_type = false;
+           }
+       }
+#endif
 
+       /* Add name to session list. */
+       if (sessions_len + 1 > sessions_size) {
+           sessions_size <<= 1;
+           sessions = erealloc3(sessions, sessions_size, sizeof(char *));
+       }
+       sessions[sessions_len++] = estrdup(dp->d_name);
+    }
+    closedir(d);
+
+    /* Sort and list the sessions. */
+    qsort(sessions, sessions_len, sizeof(char *), session_compare);
+    for (i = 0; i < sessions_len; i++) {
        len = snprintf(&pathbuf[sdlen], sizeof(pathbuf) - sdlen,
-           "%s/log", dp->d_name);
+           "%s/log", sessions[i]);
        if (len <= 0 || len >= sizeof(pathbuf) - sdlen) {
            errno = ENAMETOOLONG;
-           error(1, "%s/%s/log", dir, dp->d_name);
+           fatal("%s/%s/log", dir, sessions[i]);
        }
+       efree(sessions[i]);
 
        /* Check for dir with a log file. */
        if (lstat(pathbuf, &sb) == 0 && S_ISREG(sb.st_mode)) {
@@ -787,20 +1030,22 @@ find_sessions(const char *dir, REGEX_T *re, const char *user, const char *tty)
        } else {
            /* Strip off "/log" and recurse if a dir. */
            pathbuf[sdlen + len - 4] = '\0';
-           if (lstat(pathbuf, &sb) == 0 && S_ISDIR(sb.st_mode))
+           if (checked_type || (lstat(pathbuf, &sb) == 0 && S_ISDIR(sb.st_mode)))
                find_sessions(pathbuf, re, user, tty);
        }
     }
-    closedir(d);
+    efree(sessions);
 
-    return 0;
+    debug_return_int(0);
 }
 
+/* XXX - always returns 0, calls fatal() on failure */
 static int
 list_sessions(int argc, char **argv, const char *pattern, const char *user,
     const char *tty)
 {
     REGEX_T rebuf, *re = NULL;
+    debug_decl(list_sessions, SUDO_DEBUG_UTIL)
 
     /* Parse search expression if present */
     parse_expr(&search_expr, argv);
@@ -810,13 +1055,13 @@ list_sessions(int argc, char **argv, const char *pattern, const char *user,
     if (pattern) {
        re = &rebuf;
        if (regcomp(re, pattern, REG_EXTENDED|REG_NOSUB) != 0)
-           errorx(1, "invalid regex: %s", pattern);
+           fatalx(_("invalid regular expression: %s"), pattern);
     }
 #else
     re = (char *) pattern;
 #endif /* HAVE_REGCOMP */
 
-    return find_sessions(session_dir, re, user, tty);
+    debug_return_int(find_sessions(session_dir, re, user, tty));
 }
 
 /*
@@ -831,9 +1076,9 @@ check_input(int ttyfd, double *speed)
     struct timeval tv;
     char ch;
     ssize_t n;
+    debug_decl(check_input, SUDO_DEBUG_UTIL)
 
-    fdsr = (fd_set *)emalloc2(howmany(ttyfd + 1, NFDBITS), sizeof(fd_mask));
-
+    fdsr = ecalloc(howmany(ttyfd + 1, NFDBITS), sizeof(fd_mask));
     for (;;) {
        FD_SET(ttyfd, fdsr);
        tv.tv_sec = 0;
@@ -862,6 +1107,7 @@ check_input(int ttyfd, double *speed)
        }
     }
     free(fdsr);
+    debug_return;
 }
 
 /*
@@ -883,6 +1129,7 @@ parse_timing(buf, decimal, idx, seconds, nbytes)
     long l;
     double d, fract = 0;
     char *cp, *ep;
+    debug_decl(parse_timing, SUDO_DEBUG_UTIL)
 
     /* Parse index */
     ul = strtoul(buf, &ep, 10);
@@ -923,19 +1170,19 @@ parse_timing(buf, decimal, idx, seconds, nbytes)
        goto bad;
     *nbytes = (size_t)ul;
 
-    return 1;
+    debug_return_int(1);
 bad:
-    return 0;
+    debug_return_int(0);
 }
 
 static void
 usage(int fatal)
 {
     fprintf(fatal ? stderr : stdout,
-       "usage: %s [-h] [-d directory] [-m max_wait] [-s speed_factor] ID\n",
+       _("usage: %s [-h] [-d directory] [-m max_wait] [-s speed_factor] ID\n"),
        getprogname());
     fprintf(fatal ? stderr : stdout,
-       "usage: %s [-h] [-d directory] -l [search expression]\n",
+       _("usage: %s [-h] [-d directory] -l [search expression]\n"),
        getprogname());
     if (fatal)
        exit(1);
@@ -944,26 +1191,35 @@ usage(int fatal)
 static void
 help(void)
 {
-    (void) printf("%s - replay sudo session logs\n\n", getprogname());
+    (void) printf(_("%s - replay sudo session logs\n\n"), getprogname());
     usage(0);
-    (void) puts("\nOptions:");
-    (void) puts("  -d directory     specify directory for session logs");
-    (void) puts("  -f filter        specify which I/O type to display");
-    (void) puts("  -h               display help message and exit");
-    (void) puts("  -l [expression]  list available session IDs that match expression");
-    (void) puts("  -m max_wait      max number of seconds to wait between events");
-    (void) puts("  -s speed_factor  speed up or slow down output");
-    (void) puts("  -V               display version information and exit");
+    (void) puts(_("\nOptions:\n"
+       "  -d directory     specify directory for session logs\n"
+       "  -f filter        specify which I/O type to display\n"
+       "  -h               display help message and exit\n"
+       "  -l [expression]  list available session IDs that match expression\n"
+       "  -m max_wait      max number of seconds to wait between events\n"
+       "  -s speed_factor  speed up or slow down output\n"
+       "  -V               display version information and exit"));
     exit(0);
 }
 
 /*
- * Cleanup hook for error()/errorx()
+ * Cleanup hook for fatal()/fatalx()
   */
-void
-cleanup(int signo)
+static void
+sudoreplay_cleanup(void)
+{
+    term_restore(STDIN_FILENO, 0);
+}
+
+/*
+ * Signal handler for SIGINT, SIGKILL, SIGTERM, SIGHUP
+ * Must be installed with SA_RESETHAND enabled.
+ */
+static void
+sudoreplay_handler(int signo)
 {
     term_restore(STDIN_FILENO, 0);
-    if (signo)
-       kill(getpid(), signo);
+    kill(getpid(), signo);
 }