Allow the optional setting of a user, primary group, or grouplist
[dragonfly.git] / usr.sbin / chroot / chroot.c
1 /*
2  * Copyright (c) 1988, 1993
3  *      The Regents of the University of California.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *      This product includes software developed by the University of
16  *      California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  *
33  * @(#) Copyright (c) 1988, 1993 The Regents of the University of California.  All rights reserved.
34  * @(#)chroot.c 8.1 (Berkeley) 6/9/93
35  * $FreeBSD: src/usr.sbin/chroot/chroot.c,v 1.4.2.1 2002/03/15 22:54:59 mikeh Exp $
36  * $DragonFly: src/usr.sbin/chroot/chroot.c,v 1.4 2003/08/24 15:58:55 drhodus Exp $
37  */
38
39 #include <sys/types.h>
40
41 #include <ctype.h>
42 #include <err.h>
43 #include <grp.h>
44 #include <limits.h>
45 #include <paths.h>
46 #include <pwd.h>
47 #include <stdio.h>
48 #include <stdlib.h>
49 #include <string.h>
50 #include <unistd.h>
51
52 static void usage(void);
53
54 char    *user;          /* user to switch to before running program */
55 char    *group;         /* group to switch to ... */
56 char    *grouplist;     /* group list to switch to ... */
57
58 int
59 main(argc, argv)
60         int argc;
61         char *argv[];
62 {
63         struct group    *gp;
64         struct passwd   *pw;
65         char            *endp, *p;
66         const char      *shell;
67         gid_t           gid, gidlist[NGROUPS_MAX];
68         uid_t           uid;
69         int             ch, gids;
70
71         gid = 0;
72         uid = 0;
73         while ((ch = getopt(argc, argv, "G:g:u:")) != -1) {
74                 switch(ch) {
75                 case 'u':
76                         user = optarg;
77                         if (*user == '\0')
78                                 usage();
79                         break;
80                 case 'g':
81                         group = optarg;
82                         if (*group == '\0')
83                                 usage();
84                         break;
85                 case 'G':
86                         grouplist = optarg;
87                         if (*grouplist == '\0')
88                                 usage();
89                         break;
90                 case '?':
91                 default:
92                         usage();
93                 }
94         }
95         argc -= optind;
96         argv += optind;
97
98         if (argc < 1)
99                 usage();
100
101         if (group != NULL) {
102                 if (isdigit((unsigned char)*group)) {
103                         gid = (gid_t)strtoul(group, &endp, 0);
104                         if (*endp != '\0')
105                                 goto getgroup;
106                 } else {
107  getgroup:
108                         if ((gp = getgrnam(group)) != NULL)
109                                 gid = gp->gr_gid;
110                         else
111                                 errx(1, "no such group `%s'", group);
112                 }
113         }
114
115         for (gids = 0;
116             (p = strsep(&grouplist, ",")) != NULL && gids < NGROUPS_MAX; ) {
117                 if (*p == '\0')
118                         continue;
119
120                 if (isdigit((unsigned char)*p)) {
121                         gidlist[gids] = (gid_t)strtoul(p, &endp, 0);
122                         if (*endp != '\0')
123                                 goto getglist;
124                 } else {
125  getglist:
126                         if ((gp = getgrnam(p)) != NULL)
127                                 gidlist[gids] = gp->gr_gid;
128                         else
129                                 errx(1, "no such group `%s'", p);
130                 }
131                 gids++;
132         }
133         if (p != NULL && gids == NGROUPS_MAX)
134                 errx(1, "too many supplementary groups provided");
135
136         if (user != NULL) {
137                 if (isdigit((unsigned char)*user)) {
138                         uid = (uid_t)strtoul(user, &endp, 0);
139                         if (*endp != '\0')
140                                 goto getuser;
141                 } else {
142  getuser:
143                         if ((pw = getpwnam(user)) != NULL)
144                                 uid = pw->pw_uid;
145                         else
146                                 errx(1, "no such user `%s'", user);
147                 }
148         }
149
150         if (chdir(argv[0]) == -1 || chroot(".") == -1)
151                 err(1, "%s", argv[0]);
152
153         if (gids && setgroups(gids, gidlist) == -1)
154                 err(1, "setgroups");
155         if (group && setgid(gid) == -1)
156                 err(1, "setgid");
157         if (user && setuid(uid) == -1)
158                 err(1, "setuid");
159
160         if (argv[1]) {
161                 execvp(argv[1], &argv[1]);
162                 err(1, "%s", argv[1]);
163         }
164
165         if (!(shell = getenv("SHELL")))
166                 shell = _PATH_BSHELL;
167         execlp(shell, shell, "-i", (char *)NULL);
168         err(1, "%s", shell);
169         /* NOTREACHED */
170 }
171
172 static void
173 usage()
174 {
175         (void)fprintf(stderr, "usage: chroot [-g group] [-G group,group,...] "
176             "[-u user] newroot [command]\n");
177         exit(1);
178 }
179