git.gag.com Git - debian/sudo/blob - tgetpass.c

   1 /*
   2  * Copyright (c) 1996, 1998-2004 Todd C. Miller <Todd.Miller@courtesan.com>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  *
  16  * Sponsored in part by the Defense Advanced Research Projects
  17  * Agency (DARPA) and Air Force Research Laboratory, Air Force
  18  * Materiel Command, USAF, under agreement number F39502-99-1-0512.
  19  */
  20
  21 #ifdef __TANDEM
  22 # include <floss.h>
  23 #endif
  24
  25 #include "config.h"
  26
  27 #include <sys/types.h>
  28 #include <sys/param.h>
  29 #ifdef HAVE_SYS_BSDTYPES_H
  30 # include <sys/bsdtypes.h>
  31 #endif /* HAVE_SYS_BSDTYPES_H */
  32 #include <sys/time.h>
  33 #include <stdio.h>
  34 #ifdef STDC_HEADERS
  35 # include <stdlib.h>
  36 # include <stddef.h>
  37 #else
  38 # ifdef HAVE_STDLIB_H
  39 #  include <stdlib.h>
  40 # endif
  41 #endif /* STDC_HEADERS */
  42 #ifdef HAVE_STRING_H
  43 # if defined(HAVE_MEMORY_H) && !defined(STDC_HEADERS)
  44 #  include <memory.h>
  45 # endif
  46 # include <string.h>
  47 #else
  48 # ifdef HAVE_STRINGS_H
  49 #  include <strings.h>
  50 # endif
  51 #endif /* HAVE_STRING_H */
  52 #ifdef HAVE_UNISTD_H
  53 # include <unistd.h>
  54 #endif /* HAVE_UNISTD_H */
  55 #include <pwd.h>
  56 #include <errno.h>
  57 #include <signal.h>
  58 #include <fcntl.h>
  59 #ifdef HAVE_TERMIOS_H
  60 # include <termios.h>
  61 #else
  62 # ifdef HAVE_TERMIO_H
  63 #  include <termio.h>
  64 # else
  65 #  include <sgtty.h>
  66 #  include <sys/ioctl.h>
  67 # endif /* HAVE_TERMIO_H */
  68 #endif /* HAVE_TERMIOS_H */
  69
  70 #include "sudo.h"
  71
  72 #ifndef lint
  73 static const char rcsid[] = "$Sudo: tgetpass.c,v 1.111 2004/06/06 23:58:11 millert Exp $";
  74 #endif /* lint */
  75
  76 #ifndef TCSASOFT
  77 # define TCSASOFT       0
  78 #endif
  79 #ifndef ECHONL
  80 # define ECHONL 0
  81 #endif
  82
  83 #ifndef _POSIX_VDISABLE
  84 # ifdef VDISABLE
  85 #  define _POSIX_VDISABLE       VDISABLE
  86 # else
  87 #  define _POSIX_VDISABLE       0
  88 # endif
  89 #endif
  90
  91 /*
  92  * Abstract method of getting at the term flags.
  93  */
  94 #undef TERM
  95 #undef tflags
  96 #ifdef HAVE_TERMIOS_H
  97 # define TERM                   termios
  98 # define tflags                 c_lflag
  99 # define term_getattr(f, t)     tcgetattr(f, t)
 100 # define term_setattr(f, t)     tcsetattr(f, TCSAFLUSH|TCSASOFT, t)
 101 #else
 102 # ifdef HAVE_TERMIO_H
 103 # define TERM                   termio
 104 # define tflags                 c_lflag
 105 # define term_getattr(f, t)     ioctl(f, TCGETA, t)
 106 # define term_setattr(f, t)     ioctl(f, TCSETAF, t)
 107 # else
 108 #  define TERM                  sgttyb
 109 #  define tflags                sg_flags
 110 #  define term_getattr(f, t)    ioctl(f, TIOCGETP, t)
 111 #  define term_setattr(f, t)    ioctl(f, TIOCSETP, t)
 112 # endif /* HAVE_TERMIO_H */
 113 #endif /* HAVE_TERMIOS_H */
 114
 115 static volatile sig_atomic_t signo;
 116
 117 static void handler __P((int));
 118 static char *getln __P((int, char *, size_t));
 119
 120 /*
 121  * Like getpass(3) but with timeout and echo flags.
 122  */
 123 char *
 124 tgetpass(prompt, timeout, flags)
 125     const char *prompt;
 126     int timeout;
 127     int flags;
 128 {
 129     sigaction_t sa, savealrm, saveint, savehup, savequit, saveterm;
 130     sigaction_t savetstp, savettin, savettou;
 131     struct TERM term, oterm;
 132     char *pass;
 133     static char buf[SUDO_PASS_MAX + 1];
 134     int input, output, save_errno;
 135
 136     (void) fflush(stdout);
 137 restart:
 138     /* Open /dev/tty for reading/writing if possible else use stdin/stderr. */
 139     if (ISSET(flags, TGP_STDIN) ||
 140         (input = output = open(_PATH_TTY, O_RDWR|O_NOCTTY)) == -1) {
 141         input = STDIN_FILENO;
 142         output = STDERR_FILENO;
 143     }
 144
 145     /*
 146      * Catch signals that would otherwise cause the user to end
 147      * up with echo turned off in the shell.
 148      */
 149     sigemptyset(&sa.sa_mask);
 150     sa.sa_flags = SA_INTERRUPT; /* don't restart system calls */
 151     sa.sa_handler = handler;
 152     (void) sigaction(SIGALRM, &sa, &savealrm);
 153     (void) sigaction(SIGINT, &sa, &saveint);
 154     (void) sigaction(SIGHUP, &sa, &savehup);
 155     (void) sigaction(SIGQUIT, &sa, &savequit);
 156     (void) sigaction(SIGTERM, &sa, &saveterm);
 157     (void) sigaction(SIGTSTP, &sa, &savetstp);
 158     (void) sigaction(SIGTTIN, &sa, &savettin);
 159     (void) sigaction(SIGTTOU, &sa, &savettou);
 160
 161     /* Turn echo off/on as specified by flags.  */
 162     if (term_getattr(input, &oterm) == 0) {
 163         (void) memcpy(&term, &oterm, sizeof(term));
 164         if (!ISSET(flags, TGP_ECHO))
 165             CLR(term.tflags, (ECHO | ECHONL));
 166 #ifdef VSTATUS
 167         term.c_cc[VSTATUS] = _POSIX_VDISABLE;
 168 #endif
 169         (void) term_setattr(input, &term);
 170     } else {
 171         memset(&term, 0, sizeof(term));
 172         memset(&oterm, 0, sizeof(oterm));
 173     }
 174
 175     if (prompt)
 176         (void) write(output, prompt, strlen(prompt));
 177
 178     if (timeout > 0)
 179         alarm(timeout);
 180     pass = getln(input, buf, sizeof(buf));
 181     alarm(0);
 182     save_errno = errno;
 183
 184     if (!ISSET(term.tflags, ECHO))
 185         (void) write(output, "\n", 1);
 186
 187     /* Restore old tty settings and signals. */
 188     if (memcmp(&term, &oterm, sizeof(term)) != 0)
 189         (void) term_setattr(input, &oterm);
 190     (void) sigaction(SIGALRM, &savealrm, NULL);
 191     (void) sigaction(SIGINT, &saveint, NULL);
 192     (void) sigaction(SIGHUP, &savehup, NULL);
 193     (void) sigaction(SIGQUIT, &savequit, NULL);
 194     (void) sigaction(SIGTERM, &saveterm, NULL);
 195     (void) sigaction(SIGTSTP, &savetstp, NULL);
 196     (void) sigaction(SIGTTIN, &savettin, NULL);
 197     (void) sigaction(SIGTTOU, &savettou, NULL);
 198     if (input != STDIN_FILENO)
 199         (void) close(input);
 200
 201     /*
 202      * If we were interrupted by a signal, resend it to ourselves
 203      * now that we have restored the signal handlers.
 204      */
 205     if (signo) {
 206         kill(getpid(), signo);
 207         switch (signo) {
 208             case SIGTSTP:
 209             case SIGTTIN:
 210             case SIGTTOU:
 211                 signo = 0;
 212                 goto restart;
 213         }
 214     }
 215
 216     errno = save_errno;
 217     return(pass);
 218 }
 219
 220 static char *
 221 getln(fd, buf, bufsiz)
 222     int fd;
 223     char *buf;
 224     size_t bufsiz;
 225 {
 226     char c, *cp;
 227     ssize_t nr;
 228
 229     if (bufsiz == 0) {
 230         errno = EINVAL;
 231         return(NULL);                   /* sanity */
 232     }
 233
 234     cp = buf;
 235     nr = -1;
 236     while (--bufsiz && (nr = read(fd, &c, 1)) == 1 && c != '\n' && c != '\r')
 237         *cp++ = c;
 238     *cp = '\0';
 239     return(nr == -1 ? NULL : buf);
 240 }
 241
 242 static void
 243 handler(s)
 244     int s;
 245 {
 246     if (s != SIGALRM)
 247         signo = s;
 248 }