chflags.c revision 1.9 1 /* $NetBSD: chflags.c,v 1.9 2002/07/07 11:44:03 bjh21 Exp $ */
2
3 /*
4 * Copyright (c) 1992, 1993, 1994
5 * The Regents of the University of California. All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 * 3. All advertising materials mentioning features or use of this software
16 * must display the following acknowledgement:
17 * This product includes software developed by the University of
18 * California, Berkeley and its contributors.
19 * 4. Neither the name of the University nor the names of its contributors
20 * may be used to endorse or promote products derived from this software
21 * without specific prior written permission.
22 *
23 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 * SUCH DAMAGE.
34 */
35
36 #include <sys/cdefs.h>
37 #ifndef lint
38 __COPYRIGHT("@(#) Copyright (c) 1992, 1993, 1994\n\
39 The Regents of the University of California. All rights reserved.\n");
40 #endif /* not lint */
41
42 #ifndef lint
43 #if 0
44 static char sccsid[] = "from: @(#)chflags.c 8.5 (Berkeley) 4/1/94";
45 #else
46 __RCSID("$NetBSD: chflags.c,v 1.9 2002/07/07 11:44:03 bjh21 Exp $");
47 #endif
48 #endif /* not lint */
49
50 #include <sys/types.h>
51 #include <sys/stat.h>
52
53 #include <err.h>
54 #include <errno.h>
55 #include <fts.h>
56 #include <stdio.h>
57 #include <stdlib.h>
58 #include <string.h>
59 #include <unistd.h>
60
61 #include "stat_flags.h"
62
63 int main __P((int, char **));
64 void usage __P((void));
65
66 int
67 main(argc, argv)
68 int argc;
69 char *argv[];
70 {
71 FTS *ftsp;
72 FTSENT *p;
73 u_long clear, set, newflags;
74 long val;
75 int Hflag, Lflag, Rflag, ch, fts_options, hflag, oct, rval;
76 char *flags, *ep;
77 int (*change_flags) __P((const char *, u_long));
78
79 Hflag = Lflag = Rflag = hflag = 0;
80 while ((ch = getopt(argc, argv, "HLPRh")) != -1)
81 switch (ch) {
82 case 'H':
83 Hflag = 1;
84 Lflag = 0;
85 break;
86 case 'L':
87 Lflag = 1;
88 Hflag = 0;
89 break;
90 case 'P':
91 Hflag = Lflag = 0;
92 break;
93 case 'R':
94 Rflag = 1;
95 break;
96 case 'h':
97 hflag = 1;
98 break;
99 case '?':
100 default:
101 usage();
102 }
103 argv += optind;
104 argc -= optind;
105
106 if (argc < 2)
107 usage();
108
109 fts_options = FTS_PHYSICAL;
110 if (Rflag) {
111 if (Hflag)
112 fts_options |= FTS_COMFOLLOW;
113 if (Lflag) {
114 fts_options &= ~FTS_PHYSICAL;
115 fts_options |= FTS_LOGICAL;
116 }
117 } else if (!hflag)
118 fts_options |= FTS_COMFOLLOW;
119
120 flags = *argv;
121 if (*flags >= '0' && *flags <= '7') {
122 errno = 0;
123 val = strtol(flags, &ep, 8);
124 if (val < 0)
125 errno = ERANGE;
126 if (errno)
127 err(1, "invalid flags: %s", flags);
128 if (*ep)
129 errx(1, "invalid flags: %s", flags);
130 set = val;
131 oct = 1;
132 } else {
133 if (string_to_flags(&flags, &set, &clear))
134 errx(1, "invalid flag: %s", flags);
135 clear = ~clear;
136 oct = 0;
137 }
138
139 if ((ftsp = fts_open(++argv, fts_options, NULL)) == NULL)
140 err(1, "fts_open");
141
142 for (rval = 0; (p = fts_read(ftsp)) != NULL;) {
143 change_flags = chflags;
144 switch (p->fts_info) {
145 case FTS_D:
146 if (Rflag) /* Change it at FTS_DP. */
147 continue;
148 fts_set(ftsp, p, FTS_SKIP);
149 break;
150 case FTS_DNR: /* Warn, chflag, 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 unless -h. */
160 /*
161 * All symlinks we found while doing a physical
162 * walk end up here.
163 */
164 if (!hflag)
165 continue;
166 /*
167 * Note that if we follow a symlink, fts_info is
168 * not FTS_SL but FTS_F or whatever. And we should
169 * use lchflags only for FTS_SL and should use chflags
170 * for others.
171 */
172 change_flags = lchflags;
173 break;
174 case FTS_SLNONE: /* Ignore. */
175 /*
176 * The only symlinks that end up here are ones that
177 * don't point to anything. Note that if we are
178 * doing a phisycal walk, we never reach here unless
179 * we asked to follow explicitly.
180 */
181 continue;
182 default:
183 break;
184 }
185 if (oct)
186 newflags = set;
187 else {
188 newflags = p->fts_statp->st_flags;
189 newflags |= set;
190 newflags &= clear;
191 }
192 if ((*change_flags)(p->fts_accpath, newflags)) {
193 warn("%s", p->fts_path);
194 rval = 1;
195 }
196 }
197 if (errno)
198 err(1, "fts_read");
199 exit(rval);
200 }
201
202 void
203 usage()
204 {
205
206 (void)fprintf(stderr,
207 "usage: chflags [-R [-H | -L | -P]] [-h] flags file ...\n");
208 exit(1);
209 }
210