1 /*-
2 * SPDX-License-Identifier: BSD-3-Clause
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. Neither the name of the University nor the names of its contributors
16 * may be used to endorse or promote products derived from this software
17 * without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31
32 #if 0
33 #ifndef lint
34 static const char copyright[] =
35 "@(#) Copyright (c) 1992, 1993, 1994\n\
36 The Regents of the University of California. All rights reserved.\n";
37 #endif
38
39 #ifndef lint
40 static char sccsid[] = "@(#)chflags.c 8.5 (Berkeley) 4/1/94";
41 #endif
42 #endif
43
44 #include <sys/cdefs.h>
45 __FBSDID("$FreeBSD$");
46
47 #include <sys/types.h>
48 #include <sys/stat.h>
49
50 #include <err.h>
51 #include <errno.h>
52 #include <fcntl.h>
53 #include <fts.h>
54 #include <signal.h>
55 #include <stdio.h>
56 #include <stdlib.h>
57 #include <string.h>
58 #include <unistd.h>
59
60 static volatile sig_atomic_t siginfo;
61
62 static void usage(void);
63
64 static void
siginfo_handler(int sig __unused)65 siginfo_handler(int sig __unused)
66 {
67
68 siginfo = 1;
69 }
70
71 int
main(int argc,char * argv[])72 main(int argc, char *argv[])
73 {
74 FTS *ftsp;
75 FTSENT *p;
76 u_long clear, newflags, set;
77 long val;
78 int Hflag, Lflag, Rflag, fflag, hflag, vflag, xflag;
79 int ch, fts_options, oct, rval;
80 char *flags, *ep;
81
82 Hflag = Lflag = Rflag = fflag = hflag = vflag = xflag = 0;
83 while ((ch = getopt(argc, argv, "HLPRfhvx")) != -1)
84 switch (ch) {
85 case 'H':
86 Hflag = 1;
87 Lflag = 0;
88 break;
89 case 'L':
90 Lflag = 1;
91 Hflag = 0;
92 break;
93 case 'P':
94 Hflag = Lflag = 0;
95 break;
96 case 'R':
97 Rflag = 1;
98 break;
99 case 'f':
100 fflag = 1;
101 break;
102 case 'h':
103 hflag = 1;
104 break;
105 case 'v':
106 vflag++;
107 break;
108 case 'x':
109 xflag = 1;
110 break;
111 case '?':
112 default:
113 usage();
114 }
115 argv += optind;
116 argc -= optind;
117
118 if (argc < 2)
119 usage();
120
121 (void)signal(SIGINFO, siginfo_handler);
122
123 if (Rflag) {
124 if (hflag)
125 errx(1, "the -R and -h options may not be "
126 "specified together.");
127 if (Lflag) {
128 fts_options = FTS_LOGICAL;
129 } else {
130 fts_options = FTS_PHYSICAL;
131
132 if (Hflag) {
133 fts_options |= FTS_COMFOLLOW;
134 }
135 }
136 } else if (hflag) {
137 fts_options = FTS_PHYSICAL;
138 } else {
139 fts_options = FTS_LOGICAL;
140 }
141 if (xflag)
142 fts_options |= FTS_XDEV;
143
144 flags = *argv;
145 if (*flags >= '0' && *flags <= '7') {
146 errno = 0;
147 val = strtol(flags, &ep, 8);
148 if (val < 0)
149 errno = ERANGE;
150 if (errno)
151 err(1, "invalid flags: %s", flags);
152 if (*ep)
153 errx(1, "invalid flags: %s", flags);
154 set = val;
155 oct = 1;
156 } else {
157 if (strtofflags(&flags, &set, &clear))
158 errx(1, "invalid flag: %s", flags);
159 clear = ~clear;
160 oct = 0;
161 }
162
163 if ((ftsp = fts_open(++argv, fts_options , 0)) == NULL)
164 err(1, NULL);
165
166 for (rval = 0; errno = 0, (p = fts_read(ftsp)) != NULL;) {
167 int atflag;
168
169 if ((fts_options & FTS_LOGICAL) ||
170 ((fts_options & FTS_COMFOLLOW) &&
171 p->fts_level == FTS_ROOTLEVEL))
172 atflag = 0;
173 else
174 atflag = AT_SYMLINK_NOFOLLOW;
175
176 switch (p->fts_info) {
177 case FTS_D: /* Change it at FTS_DP if we're recursive. */
178 if (!Rflag)
179 fts_set(ftsp, p, FTS_SKIP);
180 continue;
181 case FTS_DNR: /* Warn, chflags. */
182 warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
183 rval = 1;
184 break;
185 case FTS_ERR: /* Warn, continue. */
186 case FTS_NS:
187 warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
188 rval = 1;
189 continue;
190 default:
191 break;
192 }
193 if (oct)
194 newflags = set;
195 else
196 newflags = (p->fts_statp->st_flags | set) & clear;
197 if (newflags == p->fts_statp->st_flags)
198 continue;
199 if (chflagsat(AT_FDCWD, p->fts_accpath, newflags,
200 atflag) == -1 && !fflag) {
201 warn("%s", p->fts_path);
202 rval = 1;
203 } else if (vflag || siginfo) {
204 (void)printf("%s", p->fts_path);
205 if (vflag > 1 || siginfo)
206 (void)printf(": 0%lo -> 0%lo",
207 (u_long)p->fts_statp->st_flags,
208 newflags);
209 (void)printf("\n");
210 siginfo = 0;
211 }
212 }
213 if (errno)
214 err(1, "fts_read");
215 exit(rval);
216 }
217
218 static void
usage(void)219 usage(void)
220 {
221 (void)fprintf(stderr,
222 "usage: chflags [-fhvx] [-R [-H | -L | -P]] flags file ...\n");
223 exit(1);
224 }
225