7f5335204c17f0a74ec9030e461da2994c286734
[dragonfly.git] / usr.sbin / chown / chown.c
1 /*
2  * Copyright (c) 1988, 1993, 1994
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, 1994 The Regents of the University of California.  All rights reserved.
34  * @(#)chown.c  8.8 (Berkeley) 4/4/94
35  * $FreeBSD: src/usr.sbin/chown/chown.c,v 1.15.2.3 2002/08/07 21:24:33 schweikh Exp $
36  * $DragonFly: src/usr.sbin/chown/chown.c,v 1.7 2004/11/28 16:51:50 joerg Exp $
37  */
38
39 #include <sys/param.h>
40 #include <sys/stat.h>
41
42 #include <ctype.h>
43 #include <dirent.h>
44 #include <err.h>
45 #include <errno.h>
46 #include <fts.h>
47 #include <grp.h>
48 #include <pwd.h>
49 #include <stdio.h>
50 #include <stdlib.h>
51 #include <string.h>
52 #include <unistd.h>
53
54 static void     a_gid(char *);
55 static void     a_uid(char *);
56 static void     chownerr(const char *);
57 static uid_t    id(const char *, const char *);
58 static void     usage(void);
59
60 static uid_t     uid;
61 static gid_t     gid;
62 static int       Rflag, ischown, fflag, vflag;
63 static char      *gname, *myname;
64
65 int
66 main(int argc, char **argv)
67 {
68         FTS *ftsp;
69         FTSENT *p;
70         int Hflag, Lflag, Pflag, ch, fts_options, hflag, rval;
71         char *cp;
72
73         myname = (cp = strrchr(*argv, '/')) ? cp + 1 : *argv;
74         ischown = myname[2] == 'o';
75
76         Hflag = Lflag = Pflag = hflag = vflag = 0;
77         while ((ch = getopt(argc, argv, "HLPRfhv")) != -1)
78                 switch (ch) {
79                 case 'H':
80                         Hflag = 1;
81                         Lflag = Pflag = 0;
82                         break;
83                 case 'L':
84                         Lflag = 1;
85                         Hflag = Pflag = 0;
86                         break;
87                 case 'P':
88                         Pflag = 1;
89                         Hflag = Lflag = 0;
90                         break;
91                 case 'R':
92                         Rflag = 1;
93                         break;
94                 case 'f':
95                         fflag = 1;
96                         break;
97                 case 'h':
98                         hflag = 1;
99                         break;
100                 case 'v':
101                         vflag++;
102                         break;
103                 default:
104                         usage();
105                 }
106         argv += optind;
107         argc -= optind;
108
109         if (argc < 2)
110                 usage();
111
112         fts_options = FTS_PHYSICAL;
113         if (Rflag) {
114                 if (hflag && (Lflag || Hflag))
115                         errx(1, "the -R and -h options may not be specified together");
116                 if (Hflag)
117                         fts_options |= FTS_COMFOLLOW;
118                 if (Lflag) {
119                         fts_options &= ~FTS_PHYSICAL;
120                         fts_options |= FTS_LOGICAL;
121                 }
122         }
123
124         uid = gid = -1;
125         if (ischown) {
126                 if ((cp = strchr(*argv, ':')) != NULL) {
127                         *cp++ = '\0';
128                         a_gid(cp);
129                 }
130 #ifdef SUPPORT_DOT
131                 else if ((cp = strchr(*argv, '.')) != NULL) {
132                         warnx("separation of user and group with a period is deprecated");
133                         *cp++ = '\0';
134                         a_gid(cp);
135                 }
136 #endif
137                 a_uid(*argv);
138         } else
139                 a_gid(*argv);
140
141         if ((ftsp = fts_open(++argv, fts_options, 0)) == NULL)
142                 err(1, NULL);
143
144         for (rval = 0; (p = fts_read(ftsp)) != NULL;) {
145                 switch (p->fts_info) {
146                 case FTS_D:                     /* Change it at FTS_DP. */
147                         if (!Rflag)
148                                 fts_set(ftsp, p, FTS_SKIP);
149                         continue;
150                 case FTS_DNR:                   /* Warn, chown, continue. */
151                         warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
152                         rval = 1;
153                         break;
154                 case FTS_ERR:                   /* Warn, continue. */
155                 case FTS_NS:
156                         warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
157                         rval = 1;
158                         continue;
159                 case FTS_SL:                    /* Ignore. */
160                 case FTS_SLNONE:
161                         /*
162                          * The only symlinks that end up here are ones that
163                          * don't point to anything and ones that we found
164                          * doing a physical walk.
165                          */
166                         if (hflag)
167                                 break;
168                         else
169                                 continue;
170                 default:
171                         break;
172                 }
173                 if ((uid == (uid_t)(-1) || uid == p->fts_statp->st_uid) &&
174                     (gid == (gid_t)(-1) || gid == p->fts_statp->st_gid))
175                         continue;
176                 if (hflag) {
177                         if (lchown(p->fts_accpath, uid, gid) && !fflag) {
178                                 chownerr(p->fts_path);
179                                 rval = 1;
180                         } else {
181                                 if (vflag)
182                                         (void)printf("%s\n", p->fts_accpath);
183                         }
184                 } else {
185                         if (chown(p->fts_accpath, uid, gid) && !fflag) {
186                                 chownerr(p->fts_path);
187                                 rval = 1;
188                         } else {
189                                 if (vflag) {
190                                         (void)printf("%s", p->fts_accpath);
191                                          if (vflag > 1) {
192                                                 if (ischown) {
193                                                         printf(": %d:%d -> %d:%d",
194                                                                 (int)p->fts_statp->st_uid,
195                                                                 (int)p->fts_statp->st_gid,
196                                                                 (uid == (uid_t)-1) ?
197                                                                 (int)p->fts_statp->st_uid :
198                                                                 (int)uid,
199                                                                 (gid == (gid_t)-1) ?
200                                                                 (int) p->fts_statp->st_gid :
201                                                                 (int)gid);
202                                                 } else {
203                                                         printf(": %d -> %d",
204                                                                 (int)p->fts_statp->st_gid,
205                                                                 (gid == (gid_t)-1) ?
206                                                                 (int)p->fts_statp->st_gid :
207                                                                 (int)gid);
208                                                  }
209                                          }
210                                          printf("\n");
211                                 }
212                         }
213                 }
214         }
215         if (errno)
216                 err(1, "fts_read");
217         exit(rval);
218 }
219
220 static void
221 a_gid(char *s)
222 {
223         struct group *gr;
224
225         if (*s == '\0')                 /* Argument was "uid[:.]". */
226                 return;
227         gname = s;
228         gid = ((gr = getgrnam(s)) == NULL) ? id(s, "group") : gr->gr_gid;
229 }
230
231 static void
232 a_uid(char *s)
233 {
234         struct passwd *pw;
235
236         if (*s == '\0')                 /* Argument was "[:.]gid". */
237                 return;
238         uid = ((pw = getpwnam(s)) == NULL) ? id(s, "user") : pw->pw_uid;
239 }
240
241 static uid_t
242 id(const char *name, const char *type)
243 {
244         u_long val;
245         char *ep;
246
247         /*
248          * XXX
249          * We know that uid_t's and gid_t's are unsigned longs.
250          */
251         errno = 0;
252         val = strtoul(name, &ep, 10);
253         if (errno)
254                 err(1, "%s", name);
255         if (*ep != '\0')
256                 errx(1, "%s: illegal %s name", name, type);
257         return (val);
258 }
259
260 static void
261 chownerr(const char *file)
262 {
263         static uid_t euid = -1;
264         static int ngroups = -1;
265         gid_t groups[NGROUPS];
266
267         /* Check for chown without being root. */
268         if (errno != EPERM ||
269             (uid != (uid_t)(-1) && euid == (uid_t)(-1) && (euid = geteuid()) != 0))
270                 err(1, "%s", file);
271
272         /* Check group membership; kernel just returns EPERM. */
273         if (gid != (gid_t)(-1) && ngroups == -1 &&
274             euid == (uid_t)(-1) && (euid = geteuid()) != 0) {
275                 ngroups = getgroups(NGROUPS, groups);
276                 while (--ngroups >= 0 && gid != groups[ngroups]);
277                 if (ngroups < 0)
278                         errx(1, "you are not a member of group %s", gname);
279         }
280         warn("%s", file);
281 }
282
283 static void
284 usage(void)
285 {
286         (void)fprintf(stderr, "%s\n%s\n%s\n",
287             "usage: chown [-R [-H | -L | -P]] [-f] [-h] [-v] owner[:group] file ...",
288             "       chown [-R [-H | -L | -P]] [-f] [-h] [-v] :group file ...",
289             "       chgrp [-R [-H | -L | -P]] [-f] [-h] [-v] group file ...");
290         exit(1);
291 }