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