loginutils/vlock.c

   1 /* vi: set sw=4 ts=4: */
   2
   3 /*
   4  * vlock implementation for busybox
   5  *
   6  * Copyright (C) 2000 by spoon <spoon@ix.netcom.com>
   7  * Written by spoon <spon@ix.netcom.com>
   8  *
   9  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  10  */
  11
  12 /* Shoutz to Michael K. Johnson <johnsonm@redhat.com>, author of the
  13  * original vlock.  I snagged a bunch of his code to write this
  14  * minimalistic vlock.
  15  */
  16 /* Fixed by Erik Andersen to do passwords the tinylogin way...
  17  * It now works with md5, sha1, etc passwords. */
  18
  19 #include "busybox.h"
  20 #include <sys/vt.h>
  21
  22 static struct passwd *pw;
  23 static struct vt_mode ovtm;
  24 static struct termios oterm;
  25 static int vfd;
  26 static unsigned long o_lock_all;
  27
  28 static void release_vt(int signo)
  29 {
  30         if (!o_lock_all)
  31                 ioctl(vfd, VT_RELDISP, 1);
  32         else
  33                 ioctl(vfd, VT_RELDISP, 0);
  34 }
  35
  36 static void acquire_vt(int signo)
  37 {
  38         ioctl(vfd, VT_RELDISP, VT_ACKACQ);
  39 }
  40
  41 static void restore_terminal(void)
  42 {
  43         ioctl(vfd, VT_SETMODE, &ovtm);
  44         tcsetattr(STDIN_FILENO, TCSANOW, &oterm);
  45 }
  46
  47 int vlock_main(int argc, char **argv)
  48 {
  49         sigset_t sig;
  50         struct sigaction sa;
  51         struct vt_mode vtm;
  52         struct termios term;
  53
  54         if (argc > 2) {
  55                 bb_show_usage();
  56         }
  57
  58         o_lock_all = getopt32(argc, argv, "a");
  59
  60         if((pw = getpwuid(getuid())) == NULL) {
  61                 bb_error_msg_and_die("unknown uid %d", getuid());
  62         }
  63
  64         vfd = xopen(CURRENT_TTY, O_RDWR);
  65
  66         if (ioctl(vfd, VT_GETMODE, &vtm) < 0) {
  67                 bb_perror_msg_and_die("VT_GETMODE");
  68         }
  69
  70         /* mask a bunch of signals */
  71         sigprocmask(SIG_SETMASK, NULL, &sig);
  72         sigdelset(&sig, SIGUSR1);
  73         sigdelset(&sig, SIGUSR2);
  74         sigaddset(&sig, SIGTSTP);
  75         sigaddset(&sig, SIGTTIN);
  76         sigaddset(&sig, SIGTTOU);
  77         sigaddset(&sig, SIGHUP);
  78         sigaddset(&sig, SIGCHLD);
  79         sigaddset(&sig, SIGQUIT);
  80         sigaddset(&sig, SIGINT);
  81
  82         sigemptyset(&(sa.sa_mask));
  83         sa.sa_flags = SA_RESTART;
  84         sa.sa_handler = release_vt;
  85         sigaction(SIGUSR1, &sa, NULL);
  86         sa.sa_handler = acquire_vt;
  87         sigaction(SIGUSR2, &sa, NULL);
  88
  89         /* need to handle some signals so that we don't get killed by them */
  90         sa.sa_handler = SIG_IGN;
  91         sigaction(SIGHUP, &sa, NULL);
  92         sigaction(SIGQUIT, &sa, NULL);
  93         sigaction(SIGINT, &sa, NULL);
  94         sigaction(SIGTSTP, &sa, NULL);
  95
  96         ovtm = vtm;
  97         vtm.mode = VT_PROCESS;
  98         vtm.relsig = SIGUSR1;
  99         vtm.acqsig = SIGUSR2;
 100         ioctl(vfd, VT_SETMODE, &vtm);
 101
 102         tcgetattr(STDIN_FILENO, &oterm);
 103         term = oterm;
 104         term.c_iflag &= ~BRKINT;
 105         term.c_iflag |= IGNBRK;
 106         term.c_lflag &= ~ISIG;
 107         term.c_lflag &= ~(ECHO | ECHOCTL);
 108         tcsetattr(STDIN_FILENO, TCSANOW, &term);
 109
 110         do {
 111                 printf("Virtual Console%s locked by %s.\n", (o_lock_all) ? "s" : "", pw->pw_name);
 112                 if (correct_password(pw)) {
 113                         break;
 114                 }
 115                 bb_do_delay(FAIL_DELAY);
 116                 puts("Password incorrect");
 117         } while (1);
 118         restore_terminal();
 119         return 0;
 120 }