attempt to regularize atoi mess.
[oweals/busybox.git] / loginutils / sulogin.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini sulogin implementation for busybox
4  *
5  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
6  */
7
8 #include <syslog.h>
9
10 #include "busybox.h"
11
12 static const char * const forbid[] = {
13         "ENV",
14         "BASH_ENV",
15         "HOME",
16         "IFS",
17         "PATH",
18         "SHELL",
19         "LD_LIBRARY_PATH",
20         "LD_PRELOAD",
21         "LD_TRACE_LOADED_OBJECTS",
22         "LD_BIND_NOW",
23         "LD_AOUT_LIBRARY_PATH",
24         "LD_AOUT_PRELOAD",
25         "LD_NOWARN",
26         "LD_KEEPDIR",
27         (char *) 0
28 };
29
30
31 static void catchalarm(int ATTRIBUTE_UNUSED junk)
32 {
33         exit(EXIT_FAILURE);
34 }
35
36
37 int sulogin_main(int argc, char **argv)
38 {
39         char *cp;
40         int timeout = 0;
41         char *timeout_arg;
42         const char * const *p;
43         struct passwd *pwd;
44         struct spwd *spwd;
45
46         logmode = LOGMODE_BOTH;
47         openlog(applet_name, 0, LOG_AUTH);
48
49         if (getopt32(argc, argv, "t:", &timeout_arg)) {
50                 timeout = xatoi_u(timeout_arg);
51         }
52
53         if (argv[optind]) {
54                 close(0);
55                 close(1);
56                 dup(xopen(argv[optind], O_RDWR));
57                 close(2);
58                 dup(0);
59         }
60
61         if (!isatty(0) || !isatty(1) || !isatty(2)) {
62                 logmode = LOGMODE_SYSLOG;
63                 bb_error_msg_and_die("not a tty");
64         }
65
66         /* Clear out anything dangerous from the environment */
67         for (p = forbid; *p; p++)
68                 unsetenv(*p);
69
70         signal(SIGALRM, catchalarm);
71
72         if (!(pwd = getpwuid(0))) {
73                 goto auth_error;
74         }
75
76         if (ENABLE_FEATURE_SHADOWPASSWDS) {
77                 if (!(spwd = getspnam(pwd->pw_name))) {
78                         goto auth_error;
79                 }
80                 pwd->pw_passwd = spwd->sp_pwdp;
81         }
82
83         while (1) {
84                 /* cp points to a static buffer that is zeroed every time */
85                 cp = bb_askpass(timeout,
86                                 "Give root password for system maintenance\n"
87                                 "(or type Control-D for normal startup):");
88
89                 if (!cp || !*cp) {
90                         bb_info_msg("Normal startup");
91                         return 0;
92                 }
93                 if (strcmp(pw_encrypt(cp, pwd->pw_passwd), pwd->pw_passwd) == 0) {
94                         break;
95                 }
96                 bb_do_delay(FAIL_DELAY);
97                 bb_error_msg("login incorrect");
98         }
99         memset(cp, 0, strlen(cp));
100         signal(SIGALRM, SIG_DFL);
101
102         bb_info_msg("System Maintenance Mode");
103
104         USE_SELINUX(renew_current_security_context());
105
106         run_shell(pwd->pw_shell, 1, 0, 0);
107         /* never returns */
108
109 auth_error:
110         bb_error_msg_and_die("no password entry for `root'");
111 }