su: move restricted_shell into su.c (the only user)
[oweals/busybox.git] / loginutils / su.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  *  Mini su implementation for busybox
4  *
5  *  Licensed under the GPL v2 or later, see the file LICENSE in this tarball.
6  */
7
8 #include "libbb.h"
9 #include <syslog.h>
10
11 #if ENABLE_FEATURE_SU_CHECKS_SHELLS
12 /* Return 1 if SHELL is a restricted shell (one not returned by
13    getusershell), else 0, meaning it is a standard shell.  */
14 static int restricted_shell(const char *shell)
15 {
16         char *line;
17
18         /*setusershell(); - getusershell does it itself*/
19         while ((line = getusershell()) != NULL) {
20                 if (/* *line != '#' && */ strcmp(line, shell) == 0)
21                         return 0;
22         }
23         endusershell();
24         return 1;
25 }
26 #endif
27
28 #define SU_OPT_mp (3)
29 #define SU_OPT_l (4)
30
31 int su_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
32 int su_main(int argc UNUSED_PARAM, char **argv)
33 {
34         unsigned flags;
35         char *opt_shell = NULL;
36         char *opt_command = NULL;
37         const char *opt_username = "root";
38         struct passwd *pw;
39         uid_t cur_uid = getuid();
40         const char *tty;
41         char *old_user;
42
43         flags = getopt32(argv, "mplc:s:", &opt_command, &opt_shell);
44         //argc -= optind;
45         argv += optind;
46
47         if (argv[0] && LONE_DASH(argv[0])) {
48                 flags |= SU_OPT_l;
49                 argv++;
50         }
51
52         /* get user if specified */
53         if (argv[0]) {
54                 opt_username = argv[0];
55                 argv++;
56         }
57
58         if (ENABLE_FEATURE_SU_SYSLOG) {
59                 /* The utmp entry (via getlogin) is probably the best way to identify
60                  * the user, especially if someone su's from a su-shell.
61                  * But getlogin can fail -- usually due to lack of utmp entry.
62                  * in this case resort to getpwuid.  */
63                 const char *user;
64 #if ENABLE_FEATURE_UTMP
65                 char user_buf[64];
66                 user = user_buf;
67                 if (getlogin_r(user_buf, sizeof(user_buf)) != 0)
68 #endif
69                 {
70                         pw = getpwuid(cur_uid);
71                         user = pw ? pw->pw_name : "";
72                 }
73                 old_user = xstrdup(user);
74                 tty = xmalloc_ttyname(2);
75                 if (!tty) {
76                         tty = "none";
77                 }
78                 openlog(applet_name, 0, LOG_AUTH);
79         }
80
81         pw = xgetpwnam(opt_username);
82
83         /* Make sure pw->pw_shell is non-NULL.  It may be NULL when NEW_USER
84            is a username that is retrieved via NIS (YP), but that doesn't have
85            a default shell listed.  */
86         if (!pw->pw_shell || !pw->pw_shell[0])
87                 pw->pw_shell = (char *)DEFAULT_SHELL;
88
89         if ((cur_uid == 0) || correct_password(pw)) {
90                 if (ENABLE_FEATURE_SU_SYSLOG)
91                         syslog(LOG_NOTICE, "%c %s %s:%s",
92                                 '+', tty, old_user, opt_username);
93         } else {
94                 if (ENABLE_FEATURE_SU_SYSLOG)
95                         syslog(LOG_NOTICE, "%c %s %s:%s",
96                                 '-', tty, old_user, opt_username);
97                 bb_error_msg_and_die("incorrect password");
98         }
99
100         if (ENABLE_FEATURE_CLEAN_UP && ENABLE_FEATURE_SU_SYSLOG) {
101                 closelog();
102                 free(old_user);
103         }
104
105         if (!opt_shell && (flags & SU_OPT_mp))
106                 opt_shell = getenv("SHELL");
107
108 #if ENABLE_FEATURE_SU_CHECKS_SHELLS
109         if (opt_shell && cur_uid != 0 && restricted_shell(pw->pw_shell)) {
110                 /* The user being su'd to has a nonstandard shell, and so is
111                    probably a uucp account or has restricted access.  Don't
112                    compromise the account by allowing access with a standard
113                    shell.  */
114                 bb_error_msg("using restricted shell");
115                 opt_shell = NULL;
116         }
117 #endif
118         if (!opt_shell)
119                 opt_shell = pw->pw_shell;
120
121         change_identity(pw);
122         setup_environment(opt_shell,
123                         ((flags & SU_OPT_l) / SU_OPT_l * SETUP_ENV_CLEARENV)
124                         + (!(flags & SU_OPT_mp) * SETUP_ENV_CHANGEENV),
125                         pw);
126         IF_SELINUX(set_current_security_context(NULL);)
127
128         /* Never returns */
129         run_shell(opt_shell, flags & SU_OPT_l, opt_command, (const char**)argv);
130
131         /* return EXIT_FAILURE; - not reached */
132 }