Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 1 | /* vi: set sw=4 ts=4: */ |
| 2 | /* |
| 3 | * vlock implementation for busybox |
| 4 | * |
| 5 | * Copyright (C) 2000 by spoon <spoon@ix.netcom.com> |
| 6 | * Written by spoon <spon@ix.netcom.com> |
| 7 | * |
Denys Vlasenko | 0ef64bd | 2010-08-16 20:14:46 +0200 | [diff] [blame] | 8 | * Licensed under GPLv2 or later, see file LICENSE in this source tree. |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 9 | */ |
| 10 | |
| 11 | /* Shoutz to Michael K. Johnson <johnsonm@redhat.com>, author of the |
| 12 | * original vlock. I snagged a bunch of his code to write this |
| 13 | * minimalistic vlock. |
| 14 | */ |
| 15 | /* Fixed by Erik Andersen to do passwords the tinylogin way... |
| 16 | * It now works with md5, sha1, etc passwords. */ |
| 17 | |
Pere Orga | 6a3e01d | 2011-04-01 22:56:30 +0200 | [diff] [blame] | 18 | //usage:#define vlock_trivial_usage |
| 19 | //usage: "[-a]" |
| 20 | //usage:#define vlock_full_usage "\n\n" |
| 21 | //usage: "Lock a virtual terminal. A password is required to unlock.\n" |
Pere Orga | 6a3e01d | 2011-04-01 22:56:30 +0200 | [diff] [blame] | 22 | //usage: "\n -a Lock all VTs" |
| 23 | |
Denis Vlasenko | 4cf1d08 | 2008-03-12 23:13:50 +0000 | [diff] [blame] | 24 | #include "libbb.h" |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 25 | |
Jeremie Koenig | 68fca4c | 2010-07-29 04:29:52 +0200 | [diff] [blame] | 26 | #ifdef __linux__ |
| 27 | #include <sys/vt.h> |
| 28 | |
Denis Vlasenko | a60f84e | 2008-07-05 09:18:54 +0000 | [diff] [blame] | 29 | static void release_vt(int signo UNUSED_PARAM) |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 30 | { |
Denis Vlasenko | 4cf1d08 | 2008-03-12 23:13:50 +0000 | [diff] [blame] | 31 | /* If -a, param is 0, which means: |
| 32 | * "no, kernel, we don't allow console switch away from us!" */ |
| 33 | ioctl(STDIN_FILENO, VT_RELDISP, (unsigned long) !option_mask32); |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 34 | } |
| 35 | |
Denis Vlasenko | a60f84e | 2008-07-05 09:18:54 +0000 | [diff] [blame] | 36 | static void acquire_vt(int signo UNUSED_PARAM) |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 37 | { |
Denis Vlasenko | 4cf1d08 | 2008-03-12 23:13:50 +0000 | [diff] [blame] | 38 | /* ACK to kernel that switch to console is successful */ |
| 39 | ioctl(STDIN_FILENO, VT_RELDISP, VT_ACKACQ); |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 40 | } |
Jeremie Koenig | 68fca4c | 2010-07-29 04:29:52 +0200 | [diff] [blame] | 41 | #endif |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 42 | |
Denis Vlasenko | 9b49a5e | 2007-10-11 10:05:36 +0000 | [diff] [blame] | 43 | int vlock_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE; |
Denis Vlasenko | a60f84e | 2008-07-05 09:18:54 +0000 | [diff] [blame] | 44 | int vlock_main(int argc UNUSED_PARAM, char **argv) |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 45 | { |
Jeremie Koenig | 68fca4c | 2010-07-29 04:29:52 +0200 | [diff] [blame] | 46 | #ifdef __linux__ |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 47 | struct vt_mode vtm; |
Jeremie Koenig | 68fca4c | 2010-07-29 04:29:52 +0200 | [diff] [blame] | 48 | struct vt_mode ovtm; |
| 49 | #endif |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 50 | struct termios term; |
Denis Vlasenko | 33b900f | 2007-09-27 10:12:02 +0000 | [diff] [blame] | 51 | struct termios oterm; |
Denis Vlasenko | 33b900f | 2007-09-27 10:12:02 +0000 | [diff] [blame] | 52 | struct passwd *pw; |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 53 | |
Denis Vlasenko | 0c68a87 | 2008-12-02 22:56:59 +0000 | [diff] [blame] | 54 | pw = xgetpwuid(getuid()); |
Denis Vlasenko | 4cf1d08 | 2008-03-12 23:13:50 +0000 | [diff] [blame] | 55 | opt_complementary = "=0"; /* no params! */ |
| 56 | getopt32(argv, "a"); |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 57 | |
Denis Vlasenko | 4cf1d08 | 2008-03-12 23:13:50 +0000 | [diff] [blame] | 58 | /* Ignore some signals so that we don't get killed by them */ |
| 59 | bb_signals(0 |
| 60 | + (1 << SIGTSTP) |
| 61 | + (1 << SIGTTIN) |
| 62 | + (1 << SIGTTOU) |
| 63 | + (1 << SIGHUP ) |
| 64 | + (1 << SIGCHLD) /* paranoia :) */ |
| 65 | + (1 << SIGQUIT) |
| 66 | + (1 << SIGINT ) |
| 67 | , SIG_IGN); |
Rob Landley | 68eb9dd | 2005-12-19 02:50:10 +0000 | [diff] [blame] | 68 | |
Jeremie Koenig | 68fca4c | 2010-07-29 04:29:52 +0200 | [diff] [blame] | 69 | #ifdef __linux__ |
Denis Vlasenko | 4cf1d08 | 2008-03-12 23:13:50 +0000 | [diff] [blame] | 70 | /* We will use SIGUSRx for console switch control: */ |
| 71 | /* 1: set handlers */ |
Denis Vlasenko | 8e2cfec | 2008-03-12 23:19:35 +0000 | [diff] [blame] | 72 | signal_SA_RESTART_empty_mask(SIGUSR1, release_vt); |
| 73 | signal_SA_RESTART_empty_mask(SIGUSR2, acquire_vt); |
Denis Vlasenko | 4cf1d08 | 2008-03-12 23:13:50 +0000 | [diff] [blame] | 74 | /* 2: unmask them */ |
Denis Vlasenko | 8e2cfec | 2008-03-12 23:19:35 +0000 | [diff] [blame] | 75 | sig_unblock(SIGUSR1); |
| 76 | sig_unblock(SIGUSR2); |
Jeremie Koenig | 68fca4c | 2010-07-29 04:29:52 +0200 | [diff] [blame] | 77 | #endif |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 78 | |
Denis Vlasenko | 4cf1d08 | 2008-03-12 23:13:50 +0000 | [diff] [blame] | 79 | /* Revert stdin/out to our controlling tty |
| 80 | * (or die if we have none) */ |
| 81 | xmove_fd(xopen(CURRENT_TTY, O_RDWR), STDIN_FILENO); |
| 82 | xdup2(STDIN_FILENO, STDOUT_FILENO); |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 83 | |
Jeremie Koenig | 68fca4c | 2010-07-29 04:29:52 +0200 | [diff] [blame] | 84 | #ifdef __linux__ |
Denis Vlasenko | 4cf1d08 | 2008-03-12 23:13:50 +0000 | [diff] [blame] | 85 | xioctl(STDIN_FILENO, VT_GETMODE, &vtm); |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 86 | ovtm = vtm; |
Denis Vlasenko | 4cf1d08 | 2008-03-12 23:13:50 +0000 | [diff] [blame] | 87 | /* "console switches are controlled by us, not kernel!" */ |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 88 | vtm.mode = VT_PROCESS; |
| 89 | vtm.relsig = SIGUSR1; |
| 90 | vtm.acqsig = SIGUSR2; |
Denis Vlasenko | 4cf1d08 | 2008-03-12 23:13:50 +0000 | [diff] [blame] | 91 | ioctl(STDIN_FILENO, VT_SETMODE, &vtm); |
Jeremie Koenig | 68fca4c | 2010-07-29 04:29:52 +0200 | [diff] [blame] | 92 | #endif |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 93 | |
| 94 | tcgetattr(STDIN_FILENO, &oterm); |
| 95 | term = oterm; |
| 96 | term.c_iflag &= ~BRKINT; |
| 97 | term.c_iflag |= IGNBRK; |
| 98 | term.c_lflag &= ~ISIG; |
| 99 | term.c_lflag &= ~(ECHO | ECHOCTL); |
Denis Vlasenko | 202ac50 | 2008-11-05 13:20:58 +0000 | [diff] [blame] | 100 | tcsetattr_stdin_TCSANOW(&term); |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 101 | |
Denys Vlasenko | 940c720 | 2011-03-02 04:07:14 +0100 | [diff] [blame] | 102 | while (1) { |
Denis Vlasenko | 33b900f | 2007-09-27 10:12:02 +0000 | [diff] [blame] | 103 | printf("Virtual console%s locked by %s.\n", |
Denys Vlasenko | 940c720 | 2011-03-02 04:07:14 +0100 | [diff] [blame] | 104 | /* "s" if -a, else "": */ "s" + !option_mask32, |
| 105 | pw->pw_name |
| 106 | ); |
maxwen | 27116ba | 2015-08-14 21:41:28 +0200 | [diff] [blame] | 107 | if (ask_and_check_password(pw) > 0) { |
Rob Landley | 68eb9dd | 2005-12-19 02:50:10 +0000 | [diff] [blame] | 108 | break; |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 109 | } |
Denys Vlasenko | 7d4e7a2 | 2011-03-08 21:07:05 +0100 | [diff] [blame] | 110 | bb_do_delay(LOGIN_FAIL_DELAY); |
| 111 | puts("Incorrect password"); |
Denys Vlasenko | 940c720 | 2011-03-02 04:07:14 +0100 | [diff] [blame] | 112 | } |
Denis Vlasenko | 33b900f | 2007-09-27 10:12:02 +0000 | [diff] [blame] | 113 | |
Jeremie Koenig | 68fca4c | 2010-07-29 04:29:52 +0200 | [diff] [blame] | 114 | #ifdef __linux__ |
Denis Vlasenko | 4cf1d08 | 2008-03-12 23:13:50 +0000 | [diff] [blame] | 115 | ioctl(STDIN_FILENO, VT_SETMODE, &ovtm); |
Jeremie Koenig | 68fca4c | 2010-07-29 04:29:52 +0200 | [diff] [blame] | 116 | #endif |
Denis Vlasenko | 202ac50 | 2008-11-05 13:20:58 +0000 | [diff] [blame] | 117 | tcsetattr_stdin_TCSANOW(&oterm); |
Bernhard Reutner-Fischer | 636a1f8 | 2008-05-19 09:29:47 +0000 | [diff] [blame] | 118 | fflush_stdout_and_exit(EXIT_SUCCESS); |
Eric Andersen | 27f64e1 | 2002-06-23 04:24:25 +0000 | [diff] [blame] | 119 | } |