• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * chattr.c		- Change file attributes on an ext2 file system
3  *
4  * Copyright (C) 1993, 1994  Remy Card <card@masi.ibp.fr>
5  *                           Laboratoire MASI, Institut Blaise Pascal
6  *                           Universite Pierre et Marie Curie (Paris VI)
7  *
8  * This file can be redistributed under the terms of the GNU General
9  * Public License
10  */
11 
12 /*
13  * History:
14  * 93/10/30	- Creation
15  * 93/11/13	- Replace stat() calls by lstat() to avoid loops
16  * 94/02/27	- Integrated in Ted's distribution
17  * 98/12/29	- Ignore symlinks when working recursively (G M Sipe)
18  * 98/12/29	- Display version info only when -V specified (G M Sipe)
19  */
20 
21 #define _LARGEFILE64_SOURCE
22 
23 #include <sys/types.h>
24 #include <dirent.h>
25 #include <fcntl.h>
26 #include <stdio.h>
27 #include <stdlib.h>
28 #include <unistd.h>
29 #include <string.h>
30 #ifdef HAVE_ERRNO_H
31 #include <errno.h>
32 #endif
33 #include <sys/param.h>
34 #include <sys/stat.h>
35 #include "ext2fs/ext2_fs.h"
36 
37 #ifdef __GNUC__
38 #define EXT2FS_ATTR(x) __attribute__(x)
39 #else
40 #define EXT2FS_ATTR(x)
41 #endif
42 
43 #ifndef S_ISLNK			/* So we can compile even with gcc-warn */
44 # ifdef __S_IFLNK
45 #  define S_ISLNK(mode)	 __S_ISTYPE((mode), __S_IFLNK)
46 # else
47 #  define S_ISLNK(mode)  0
48 # endif
49 #endif
50 
51 #include "et/com_err.h"
52 #include "e2p/e2p.h"
53 
54 #include "../version.h"
55 #include "nls-enable.h"
56 
57 static const char * program_name = "chattr";
58 
59 static int add;
60 static int rem;
61 static int set;
62 static int set_version;
63 
64 static unsigned long version;
65 
66 static int recursive;
67 static int verbose;
68 static int silent;
69 
70 static unsigned long af;
71 static unsigned long rf;
72 static unsigned long sf;
73 
74 #ifdef _LFS64_LARGEFILE
75 #define LSTAT		lstat64
76 #define STRUCT_STAT	struct stat64
77 #else
78 #define LSTAT		lstat
79 #define STRUCT_STAT	struct stat
80 #endif
81 
usage(void)82 static void usage(void)
83 {
84 	fprintf(stderr,
85 		_("Usage: %s [-RVf] [-+=AacDdeijsSu] [-v version] files...\n"),
86 		program_name);
87 	exit(1);
88 }
89 
90 struct flags_char {
91 	unsigned long	flag;
92 	char 		optchar;
93 };
94 
95 static const struct flags_char flags_array[] = {
96 	{ EXT2_NOATIME_FL, 'A' },
97 	{ EXT2_SYNC_FL, 'S' },
98 	{ EXT2_DIRSYNC_FL, 'D' },
99 	{ EXT2_APPEND_FL, 'a' },
100 	{ EXT2_COMPR_FL, 'c' },
101 	{ EXT2_NODUMP_FL, 'd' },
102 	{ EXT4_EXTENTS_FL, 'e'},
103 	{ EXT2_IMMUTABLE_FL, 'i' },
104 	{ EXT3_JOURNAL_DATA_FL, 'j' },
105 	{ EXT2_SECRM_FL, 's' },
106 	{ EXT2_UNRM_FL, 'u' },
107 	{ EXT2_NOTAIL_FL, 't' },
108 	{ EXT2_TOPDIR_FL, 'T' },
109 	{ 0, 0 }
110 };
111 
get_flag(char c)112 static unsigned long get_flag(char c)
113 {
114 	const struct flags_char *fp;
115 
116 	for (fp = flags_array; fp->flag != 0; fp++) {
117 		if (fp->optchar == c)
118 			return fp->flag;
119 	}
120 	return 0;
121 }
122 
123 
decode_arg(int * i,int argc,char ** argv)124 static int decode_arg (int * i, int argc, char ** argv)
125 {
126 	char * p;
127 	char * tmp;
128 	unsigned long fl;
129 
130 	switch (argv[*i][0])
131 	{
132 	case '-':
133 		for (p = &argv[*i][1]; *p; p++) {
134 			if (*p == 'R') {
135 				recursive = 1;
136 				continue;
137 			}
138 			if (*p == 'V') {
139 				verbose = 1;
140 				continue;
141 			}
142 			if (*p == 'f') {
143 				silent = 1;
144 				continue;
145 			}
146 			if (*p == 'v') {
147 				(*i)++;
148 				if (*i >= argc)
149 					usage ();
150 				version = strtol (argv[*i], &tmp, 0);
151 				if (*tmp) {
152 					com_err (program_name, 0,
153 						 _("bad version - %s\n"),
154 						 argv[*i]);
155 					usage ();
156 				}
157 				set_version = 1;
158 				continue;
159 			}
160 			if ((fl = get_flag(*p)) == 0)
161 				usage();
162 			rf |= fl;
163 			rem = 1;
164 		}
165 		break;
166 	case '+':
167 		add = 1;
168 		for (p = &argv[*i][1]; *p; p++) {
169 			if ((fl = get_flag(*p)) == 0)
170 				usage();
171 			af |= fl;
172 		}
173 		break;
174 	case '=':
175 		set = 1;
176 		for (p = &argv[*i][1]; *p; p++) {
177 			if ((fl = get_flag(*p)) == 0)
178 				usage();
179 			sf |= fl;
180 		}
181 		break;
182 	default:
183 		return EOF;
184 		break;
185 	}
186 	return 1;
187 }
188 
189 static int chattr_dir_proc(const char *, struct dirent *, void *);
190 
change_attributes(const char * name)191 static int change_attributes(const char * name)
192 {
193 	unsigned long flags;
194 	STRUCT_STAT	st;
195 	int extent_file = 0;
196 
197 	if (LSTAT (name, &st) == -1) {
198 		if (!silent)
199 			com_err (program_name, errno,
200 				 _("while trying to stat %s"), name);
201 		return -1;
202 	}
203 
204 	if (fgetflags(name, &flags) == -1) {
205 		if (!silent)
206 			com_err(program_name, errno,
207 					_("while reading flags on %s"), name);
208 		return -1;
209 	}
210 	if (flags & EXT4_EXTENTS_FL)
211 		extent_file = 1;
212 	if (set) {
213 		if (extent_file && !(sf & EXT4_EXTENTS_FL)) {
214 			if (!silent)
215 				com_err(program_name, 0,
216 				_("Clearing extent flag not supported on %s"),
217 					name);
218 			return -1;
219 		}
220 		if (verbose) {
221 			printf (_("Flags of %s set as "), name);
222 			print_flags (stdout, sf, 0);
223 			printf ("\n");
224 		}
225 		if (fsetflags (name, sf) == -1)
226 			perror (name);
227 	} else {
228 		if (rem)
229 			flags &= ~rf;
230 		if (add)
231 			flags |= af;
232 		if (extent_file && !(flags & EXT4_EXTENTS_FL)) {
233 			if (!silent)
234 				com_err(program_name, 0,
235 				_("Clearing extent flag not supported on %s"),
236 					name);
237 			return -1;
238 		}
239 		if (verbose) {
240 			printf(_("Flags of %s set as "), name);
241 			print_flags(stdout, flags, 0);
242 			printf("\n");
243 		}
244 		if (!S_ISDIR(st.st_mode))
245 			flags &= ~EXT2_DIRSYNC_FL;
246 		if (fsetflags(name, flags) == -1) {
247 			if (!silent) {
248 				com_err(program_name, errno,
249 						_("while setting flags on %s"),
250 						name);
251 			}
252 			return -1;
253 		}
254 	}
255 	if (set_version) {
256 		if (verbose)
257 			printf (_("Version of %s set as %lu\n"), name, version);
258 		if (fsetversion (name, version) == -1) {
259 			if (!silent)
260 				com_err (program_name, errno,
261 					 _("while setting version on %s"),
262 					 name);
263 			return -1;
264 		}
265 	}
266 	if (S_ISDIR(st.st_mode) && recursive)
267 		return iterate_on_dir (name, chattr_dir_proc, NULL);
268 	return 0;
269 }
270 
chattr_dir_proc(const char * dir_name,struct dirent * de,void * private EXT2FS_ATTR ((unused)))271 static int chattr_dir_proc (const char * dir_name, struct dirent * de,
272 			    void * private EXT2FS_ATTR((unused)))
273 {
274 	int ret = 0;
275 
276 	if (strcmp (de->d_name, ".") && strcmp (de->d_name, "..")) {
277 	        char *path;
278 
279 		path = malloc(strlen (dir_name) + 1 + strlen (de->d_name) + 1);
280 		if (!path) {
281 			fprintf(stderr, _("Couldn't allocate path variable "
282 					  "in chattr_dir_proc"));
283 			return -1;
284 		}
285 		sprintf(path, "%s/%s", dir_name, de->d_name);
286 		ret = change_attributes(path);
287 		free(path);
288 	}
289 	return ret;
290 }
291 
main(int argc,char ** argv)292 int main (int argc, char ** argv)
293 {
294 	int i, j;
295 	int end_arg = 0;
296 	int err, retval = 0;
297 
298 #ifdef ENABLE_NLS
299 	setlocale(LC_MESSAGES, "");
300 	setlocale(LC_CTYPE, "");
301 	bindtextdomain(NLS_CAT_NAME, LOCALEDIR);
302 	textdomain(NLS_CAT_NAME);
303 #endif
304 	if (argc && *argv)
305 		program_name = *argv;
306 	i = 1;
307 	while (i < argc && !end_arg) {
308 		/* '--' arg should end option processing */
309 		if (strcmp(argv[i], "--") == 0) {
310 			i++;
311 			end_arg = 1;
312 		} else if (decode_arg (&i, argc, argv) == EOF)
313 			end_arg = 1;
314 		else
315 			i++;
316 	}
317 	if (i >= argc)
318 		usage ();
319 	if (set && (add || rem)) {
320 		fputs(_("= is incompatible with - and +\n"), stderr);
321 		exit (1);
322 	}
323 	if ((rf & af) != 0) {
324 		fputs("Can't both set and unset same flag.\n", stderr);
325 		exit (1);
326 	}
327 	if (!(add || rem || set || set_version)) {
328 		fputs(_("Must use '-v', =, - or +\n"), stderr);
329 		exit (1);
330 	}
331 	if (verbose)
332 		fprintf (stderr, "chattr %s (%s)\n",
333 			 E2FSPROGS_VERSION, E2FSPROGS_DATE);
334 	for (j = i; j < argc; j++) {
335 		err = change_attributes (argv[j]);
336 		if (err)
337 			retval = 1;
338 	}
339 	exit(retval);
340 }
341