chattr.c revision 2d328bb76d2d63bdfdba923b54c28bd686bd8fec 
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
57static const char * program_name = "chattr";
58
59static int add;
60static int rem;
61static int set;
62static int set_version;
63
64static unsigned long version;
65
66static int recursive;
67static int verbose;
68static int silent;
69
70static unsigned long af;
71static unsigned long rf;
72static 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
82static void usage(void)
83{
84	fprintf(stderr,
85		_("Usage: %s [-RVf] [-+=AacDdijsSu] [-v version] files...\n"),
86		program_name);
87	exit(1);
88}
89
90struct flags_char {
91	unsigned long	flag;
92	char 		optchar;
93};
94
95static 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	{ EXT2_IMMUTABLE_FL, 'i' },
103	{ EXT3_JOURNAL_DATA_FL, 'j' },
104	{ EXT2_SECRM_FL, 's' },
105	{ EXT2_UNRM_FL, 'u' },
106	{ EXT2_NOTAIL_FL, 't' },
107	{ EXT2_TOPDIR_FL, 'T' },
108	{ 0, 0 }
109};
110
111static unsigned long get_flag(char c)
112{
113	const struct flags_char *fp;
114
115	for (fp = flags_array; fp->flag != 0; fp++) {
116		if (fp->optchar == c)
117			return fp->flag;
118	}
119	return 0;
120}
121
122
123static int decode_arg (int * i, int argc, char ** argv)
124{
125	char * p;
126	char * tmp;
127	unsigned long fl;
128
129	switch (argv[*i][0])
130	{
131	case '-':
132		for (p = &argv[*i][1]; *p; p++) {
133			if (*p == 'R') {
134				recursive = 1;
135				continue;
136			}
137			if (*p == 'V') {
138				verbose = 1;
139				continue;
140			}
141			if (*p == 'f') {
142				silent = 1;
143				continue;
144			}
145			if (*p == 'v') {
146				(*i)++;
147				if (*i >= argc)
148					usage ();
149				version = strtol (argv[*i], &tmp, 0);
150				if (*tmp) {
151					com_err (program_name, 0,
152						 _("bad version - %s\n"),
153						 argv[*i]);
154					usage ();
155				}
156				set_version = 1;
157				continue;
158			}
159			if ((fl = get_flag(*p)) == 0)
160				usage();
161			rf |= fl;
162			rem = 1;
163		}
164		break;
165	case '+':
166		add = 1;
167		for (p = &argv[*i][1]; *p; p++) {
168			if ((fl = get_flag(*p)) == 0)
169				usage();
170			af |= fl;
171		}
172		break;
173	case '=':
174		set = 1;
175		for (p = &argv[*i][1]; *p; p++) {
176			if ((fl = get_flag(*p)) == 0)
177				usage();
178			sf |= fl;
179		}
180		break;
181	default:
182		return EOF;
183		break;
184	}
185	return 1;
186}
187
188static int chattr_dir_proc(const char *, struct dirent *, void *);
189
190static int change_attributes(const char * name)
191{
192	unsigned long flags;
193	STRUCT_STAT	st;
194
195	if (LSTAT (name, &st) == -1) {
196		if (!silent)
197			com_err (program_name, errno,
198				 _("while trying to stat %s"), name);
199		return -1;
200	}
201
202	if (set) {
203		if (verbose) {
204			printf (_("Flags of %s set as "), name);
205			print_flags (stdout, sf, 0);
206			printf ("\n");
207		}
208		if (fsetflags (name, sf) == -1)
209			perror (name);
210	} else {
211		if (fgetflags (name, &flags) == -1) {
212			if (!silent)
213				com_err (program_name, errno,
214					 _("while reading flags on %s"), name);
215			return -1;
216		} else {
217			if (rem)
218				flags &= ~rf;
219			if (add)
220				flags |= af;
221			if (verbose) {
222				printf (_("Flags of %s set as "), name);
223				print_flags (stdout, flags, 0);
224				printf ("\n");
225			}
226			if (!S_ISDIR(st.st_mode))
227				flags &= ~EXT2_DIRSYNC_FL;
228			if (fsetflags (name, flags) == -1) {
229				if (!silent)
230					com_err(program_name, errno,
231						_("while setting flags on %s"),
232						name);
233				return -1;
234			}
235		}
236	}
237	if (set_version) {
238		if (verbose)
239			printf (_("Version of %s set as %lu\n"), name, version);
240		if (fsetversion (name, version) == -1) {
241			if (!silent)
242				com_err (program_name, errno,
243					 _("while setting version on %s"),
244					 name);
245			return -1;
246		}
247	}
248	if (S_ISDIR(st.st_mode) && recursive)
249		return iterate_on_dir (name, chattr_dir_proc, NULL);
250	return 0;
251}
252
253static int chattr_dir_proc (const char * dir_name, struct dirent * de,
254			    void * private EXT2FS_ATTR((unused)))
255{
256	int ret = 0;
257
258	if (strcmp (de->d_name, ".") && strcmp (de->d_name, "..")) {
259	        char *path;
260
261		path = malloc(strlen (dir_name) + 1 + strlen (de->d_name) + 1);
262		if (!path) {
263			fprintf(stderr, _("Couldn't allocate path variable "
264					  "in chattr_dir_proc"));
265			return -1;
266		}
267		sprintf(path, "%s/%s", dir_name, de->d_name);
268		ret = change_attributes(path);
269		free(path);
270	}
271	return ret;
272}
273
274int main (int argc, char ** argv)
275{
276	int i, j;
277	int end_arg = 0;
278	int err, retval = 0;
279
280#ifdef ENABLE_NLS
281	setlocale(LC_MESSAGES, "");
282	setlocale(LC_CTYPE, "");
283	bindtextdomain(NLS_CAT_NAME, LOCALEDIR);
284	textdomain(NLS_CAT_NAME);
285#endif
286	if (argc && *argv)
287		program_name = *argv;
288	i = 1;
289	while (i < argc && !end_arg) {
290		/* '--' arg should end option processing */
291		if (strcmp(argv[i], "--") == 0) {
292			i++;
293			end_arg = 1;
294		} else if (decode_arg (&i, argc, argv) == EOF)
295			end_arg = 1;
296		else
297			i++;
298	}
299	if (i >= argc)
300		usage ();
301	if (set && (add || rem)) {
302		fputs(_("= is incompatible with - and +\n"), stderr);
303		exit (1);
304	}
305	if ((rf & af) != 0) {
306		fputs("Can't both set and unset same flag.\n", stderr);
307		exit (1);
308	}
309	if (!(add || rem || set || set_version)) {
310		fputs(_("Must use '-v', =, - or +\n"), stderr);
311		exit (1);
312	}
313	if (verbose)
314		fprintf (stderr, "chattr %s (%s)\n",
315			 E2FSPROGS_VERSION, E2FSPROGS_DATE);
316	for (j = i; j < argc; j++) {
317		err = change_attributes (argv[j]);
318		if (err)
319			retval = 1;
320	}
321	exit(retval);
322}
323