Merge from vendor branch NTPD:
[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.6 2004/11/28 16:38:21 liamfoy 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(int argc, char **argv)
60 {
61         struct group    *gp;
62         struct passwd   *pw;
63         char            *endp, *p;
64         const char      *shell;
65         gid_t           gid, gidlist[NGROUPS_MAX];
66         uid_t           uid;
67         int             ch, gids;
68
69         gid = 0;
70         uid = 0;
71         while ((ch = getopt(argc, argv, "G:g:u:")) != -1) {
72                 switch(ch) {
73                 case 'u':
74                         user = optarg;
75                         if (*user == '\0')
76                                 usage();
77                         break;
78                 case 'g':
79                         group = optarg;
80                         if (*group == '\0')
81                                 usage();
82                         break;
83                 case 'G':
84                         grouplist = optarg;
85                         if (*grouplist == '\0')
86                                 usage();
87                         break;
88                 default:
89                         usage();
90                 }
91         }
92         argc -= optind;
93         argv += optind;
94
95         if (argc < 1)
96                 usage();
97
98         if (group != NULL) {
99                 if (isdigit(*group)) {
100                         gid = (gid_t)strtoul(group, &endp, 0);
101                         if (*endp != '\0')
102                                 goto getgroup;
103                 } else {
104  getgroup:
105                         if ((gp = getgrnam(group)) != NULL)
106                                 gid = gp->gr_gid;
107                         else
108                                 errx(1, "no such group `%s'", group);
109                 }
110         }
111
112         for (gids = 0;
113             (p = strsep(&grouplist, ",")) != NULL && gids < NGROUPS_MAX; ) {
114                 if (*p == '\0')
115                         continue;
116
117                 if (isdigit(*p)) {
118                         gidlist[gids] = (gid_t)strtoul(p, &endp, 0);
119                         if (*endp != '\0')
120                                 goto getglist;
121                 } else {
122  getglist:
123                         if ((gp = getgrnam(p)) != NULL)
124                                 gidlist[gids] = gp->gr_gid;
125                         else
126                                 errx(1, "no such group `%s'", p);
127                 }
128                 gids++;
129         }
130         if (p != NULL && gids == NGROUPS_MAX)
131                 errx(1, "too many supplementary groups provided");
132
133         if (user != NULL) {
134                 if (isdigit(*user)) {
135                         uid = (uid_t)strtoul(user, &endp, 0);
136                         if (*endp != '\0')
137                                 goto getuser;
138                 } else {
139  getuser:
140                         if ((pw = getpwnam(user)) != NULL)
141                                 uid = pw->pw_uid;
142                         else
143                                 errx(1, "no such user `%s'", user);
144                 }
145         }
146
147         if (chdir(argv[0]) == -1 || chroot(".") == -1)
148                 err(1, "%s", argv[0]);
149
150         if (gids && setgroups(gids, gidlist) == -1)
151                 err(1, "setgroups");
152         if (group && setgid(gid) == -1)
153                 err(1, "setgid");
154         if (user && setuid(uid) == -1)
155                 err(1, "setuid");
156
157         if (argv[1]) {
158                 execvp(argv[1], &argv[1]);
159                 err(1, "%s", argv[1]);
160         }
161
162         if (!(shell = getenv("SHELL")))
163                 shell = _PATH_BSHELL;
164         execlp(shell, shell, "-i", (char *)NULL);
165         err(1, "%s", shell);
166         /* NOTREACHED */
167 }
168
169 static void
170 usage(void)
171 {
172         (void)fprintf(stderr, "usage: chroot [-g group] [-G group,group,...] "
173             "[-u user] newroot [command]\n");
174         exit(1);
175 }
176