[thirdparty/e2fsprogs.git] / misc / chattr.c

/*
 * chattr.c		- Change file attributes on an ext2 file system
 *
 * Copyright (C) 1993, 1994  Remy Card <card@masi.ibp.fr>
 *                           Laboratoire MASI, Institut Blaise Pascal
 *                           Universite Pierre et Marie Curie (Paris VI)
 *
 * This file can be redistributed under the terms of the GNU General
 * Public License
 */

/*
 * History:
 * 93/10/30	- Creation
 * 93/11/13	- Replace stat() calls by lstat() to avoid loops
 * 94/02/27	- Integrated in Ted's distribution
 * 98/12/29	- Ignore symlinks when working recursively (G M Sipe)
 * 98/12/29	- Display version info only when -V specified (G M Sipe)
 */

#define _LARGEFILE64_SOURCE

#include "config.h"
#include <sys/types.h>
#include <dirent.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#ifdef HAVE_ERRNO_H
#include <errno.h>
#endif
#include <sys/param.h>
#include <sys/stat.h>
#include "ext2fs/ext2_fs.h"

#ifdef __GNUC__
#define EXT2FS_ATTR(x) __attribute__(x)
#else
#define EXT2FS_ATTR(x)
#endif

#ifndef S_ISLNK			/* So we can compile even with gcc-warn */
# ifdef __S_IFLNK
#  define S_ISLNK(mode)	 __S_ISTYPE((mode), __S_IFLNK)
# else
#  define S_ISLNK(mode)  0
# endif
#endif

#include "et/com_err.h"
#include "e2p/e2p.h"
#include "support/nls-enable.h"

#include "../version.h"

static const char * program_name = "chattr";

static int add;
static int rem;
static int set;
static int set_version;

static unsigned long version;

static int set_project;
static unsigned long project;

static int recursive;
static int verbose;
static int silent;

static unsigned long af;
static unsigned long rf;
static unsigned long sf;

#ifdef _LFS64_LARGEFILE
#define LSTAT		lstat64
#define STRUCT_STAT	struct stat64
#else
#define LSTAT		lstat
#define STRUCT_STAT	struct stat
#endif

static void usage(void)
{
	fprintf(stderr,
		_("Usage: %s [-pRVf] [-+=aAcCdDeijPsStTuF] [-v version] files...\n"),
		program_name);
	exit(1);
}

struct flags_char {
	unsigned long	flag;
	char 		optchar;
};

static const struct flags_char flags_array[] = {
	{ EXT2_NOATIME_FL, 'A' },
	{ EXT2_SYNC_FL, 'S' },
	{ EXT2_DIRSYNC_FL, 'D' },
	{ EXT2_APPEND_FL, 'a' },
	{ EXT2_COMPR_FL, 'c' },
	{ EXT2_NODUMP_FL, 'd' },
	{ EXT4_EXTENTS_FL, 'e'},
	{ EXT2_IMMUTABLE_FL, 'i' },
	{ EXT3_JOURNAL_DATA_FL, 'j' },
	{ EXT4_PROJINHERIT_FL, 'P' },
	{ EXT2_SECRM_FL, 's' },
	{ EXT2_UNRM_FL, 'u' },
	{ EXT2_NOTAIL_FL, 't' },
	{ EXT2_TOPDIR_FL, 'T' },
	{ FS_NOCOW_FL, 'C' },
	{ EXT4_CASEFOLD_FL, 'F' },
	{ 0, 0 }
};

static unsigned long get_flag(char c)
{
	const struct flags_char *fp;

	for (fp = flags_array; fp->flag != 0; fp++) {
		if (fp->optchar == c)
			return fp->flag;
	}
	return 0;
}


static int decode_arg (int * i, int argc, char ** argv)
{
	char * p;
	char * tmp;
	unsigned long fl;

	switch (argv[*i][0])
	{
	case '-':
		for (p = &argv[*i][1]; *p; p++) {
			if (*p == 'R') {
				recursive = 1;
				continue;
			}
			if (*p == 'V') {
				verbose = 1;
				continue;
			}
			if (*p == 'f') {
				silent = 1;
				continue;
			}
			if (*p == 'p') {
				(*i)++;
				if (*i >= argc)
					usage ();
				project = strtol (argv[*i], &tmp, 0);
				if (*tmp) {
					com_err (program_name, 0,
						 _("bad project - %s\n"),
						 argv[*i]);
					usage ();
				}
				set_project = 1;
				continue;
			}
			if (*p == 'v') {
				(*i)++;
				if (*i >= argc)
					usage ();
				version = strtol (argv[*i], &tmp, 0);
				if (*tmp) {
					com_err (program_name, 0,
						 _("bad version - %s\n"),
						 argv[*i]);
					usage ();
				}
				set_version = 1;
				continue;
			}
			if ((fl = get_flag(*p)) == 0)
				usage();
			rf |= fl;
			rem = 1;
		}
		break;
	case '+':
		add = 1;
		for (p = &argv[*i][1]; *p; p++) {
			if ((fl = get_flag(*p)) == 0)
				usage();
			af |= fl;
		}
		break;
	case '=':
		set = 1;
		for (p = &argv[*i][1]; *p; p++) {
			if ((fl = get_flag(*p)) == 0)
				usage();
			sf |= fl;
		}
		break;
	default:
		return EOF;
		break;
	}
	return 1;
}

static int chattr_dir_proc(const char *, struct dirent *, void *);

static int change_attributes(const char * name)
{
	unsigned long flags;
	STRUCT_STAT	st;

	if (LSTAT (name, &st) == -1) {
		if (!silent)
			com_err (program_name, errno,
				 _("while trying to stat %s"), name);
		return -1;
	}

	if (fgetflags(name, &flags) == -1) {
		if (!silent)
			com_err(program_name, errno,
					_("while reading flags on %s"), name);
		return -1;
	}
	if (set) {
		if (verbose) {
			printf (_("Flags of %s set as "), name);
			print_flags (stdout, sf, 0);
			printf ("\n");
		}
		if (fsetflags (name, sf) == -1)
			perror (name);
	} else {
		if (rem)
			flags &= ~rf;
		if (add)
			flags |= af;
		if (verbose) {
			printf(_("Flags of %s set as "), name);
			print_flags(stdout, flags, 0);
			printf("\n");
		}
		if (!S_ISDIR(st.st_mode))
			flags &= ~EXT2_DIRSYNC_FL;
		if (fsetflags(name, flags) == -1) {
			if (!silent) {
				com_err(program_name, errno,
						_("while setting flags on %s"),
						name);
			}
			return -1;
		}
	}
	if (set_version) {
		if (verbose)
			printf (_("Version of %s set as %lu\n"), name, version);
		if (fsetversion (name, version) == -1) {
			if (!silent)
				com_err (program_name, errno,
					 _("while setting version on %s"),
					 name);
			return -1;
		}
	}
	if (set_project) {
		if (verbose)
			printf (_("Project of %s set as %lu\n"), name, project);
		if (fsetproject (name, project) == -1) {
			if (!silent)
				com_err (program_name, errno,
					 _("while setting project on %s"),
					 name);
			return -1;
		}

	}
	if (S_ISDIR(st.st_mode) && recursive)
		return iterate_on_dir (name, chattr_dir_proc, NULL);
	return 0;
}

static int chattr_dir_proc (const char * dir_name, struct dirent * de,
			    void * private EXT2FS_ATTR((unused)))
{
	int ret = 0;

	if (strcmp (de->d_name, ".") && strcmp (de->d_name, "..")) {
	        char *path;

		path = malloc(strlen (dir_name) + 1 + strlen (de->d_name) + 1);
		if (!path) {
			fprintf(stderr, "%s",
				_("Couldn't allocate path variable "
				  "in chattr_dir_proc"));
			return -1;
		}
		sprintf(path, "%s/%s", dir_name, de->d_name);
		ret = change_attributes(path);
		free(path);
	}
	return ret;
}

int main (int argc, char ** argv)
{
	int i, j;
	int end_arg = 0;
	int err, retval = 0;

#ifdef ENABLE_NLS
	setlocale(LC_MESSAGES, "");
	setlocale(LC_CTYPE, "");
	bindtextdomain(NLS_CAT_NAME, LOCALEDIR);
	textdomain(NLS_CAT_NAME);
	set_com_err_gettext(gettext);
#endif
	if (argc && *argv)
		program_name = *argv;
	i = 1;
	while (i < argc && !end_arg) {
		/* '--' arg should end option processing */
		if (strcmp(argv[i], "--") == 0) {
			i++;
			end_arg = 1;
		} else if (decode_arg (&i, argc, argv) == EOF)
			end_arg = 1;
		else
			i++;
	}
	if (i >= argc)
		usage ();
	if (set && (add || rem)) {
		fputs(_("= is incompatible with - and +\n"), stderr);
		exit (1);
	}
	if ((rf & af) != 0) {
		fputs("Can't both set and unset same flag.\n", stderr);
		exit (1);
	}
	if (!(add || rem || set || set_version || set_project )) {
		fputs(_("Must use '-v', =, - or +\n"), stderr);
		exit (1);
	}
	if (verbose)
		fprintf (stderr, "chattr %s (%s)\n",
			 E2FSPROGS_VERSION, E2FSPROGS_DATE);
	for (j = i; j < argc; j++) {
		err = change_attributes (argv[j]);
		if (err)
			retval = 1;
	}
	exit(retval);
}
Commit	Line	Data
3839e657 TT	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
a88fa0c0 TT	17	* 98/12/29 - Ignore symlinks when working recursively (G M Sipe)
a88fa0c0 TT	18	* 98/12/29 - Display version info only when -V specified (G M Sipe)
3839e657 TT	19	*/
3839e657 TT	20
fff18b4e	21	#define _LARGEFILE64_SOURCE
fff18b4e	22
d1154eb4	23	#include "config.h"
a418d3ad	24	#include <sys/types.h>
3839e657 TT	25	#include <dirent.h>
3839e657 TT	26	#include <fcntl.h>
3839e657 TT	27	#include <stdio.h>
	28	#include <stdlib.h>
	29	#include <unistd.h>
19c78dc0	30	#include <string.h>
a418d3ad TT	31	#ifdef HAVE_ERRNO_H
	32	#include <errno.h>
	33	#endif
3839e657 TT	34	#include <sys/param.h>
3839e657 TT	35	#include <sys/stat.h>
54c637d4	36	#include "ext2fs/ext2_fs.h"
3839e657	37
54434927 TT	38	#ifdef __GNUC__
	39	#define EXT2FS_ATTR(x) __attribute__(x)
	40	#else
	41	#define EXT2FS_ATTR(x)
	42	#endif
	43
36caf25f TT	44	#ifndef S_ISLNK /* So we can compile even with gcc-warn */
	45	# ifdef __S_IFLNK
	46	# define S_ISLNK(mode) __S_ISTYPE((mode), __S_IFLNK)
	47	# else
	48	# define S_ISLNK(mode) 0
	49	# endif
	50	#endif
	51
3839e657 TT	52	#include "et/com_err.h"
3839e657 TT	53	#include "e2p/e2p.h"
99ceb8ec	54	#include "support/nls-enable.h"
3839e657 TT	55
	56	#include "../version.h"
	57
e1a0a3e3	58	static const char * program_name = "chattr";
3839e657	59
9a718849 TT	60	static int add;
	61	static int rem;
	62	static int set;
	63	static int set_version;
3839e657	64
e1a0a3e3	65	static unsigned long version;
3839e657	66
3627e838 LX	67	static int set_project;
	68	static unsigned long project;
	69
9a718849 TT	70	static int recursive;
9a718849 TT	71	static int verbose;
e68594d2	72	static int silent;
3839e657	73
e1a0a3e3 TT	74	static unsigned long af;
	75	static unsigned long rf;
	76	static unsigned long sf;
3839e657	77
b705640a TT	78	#ifdef _LFS64_LARGEFILE
	79	#define LSTAT lstat64
	80	#define STRUCT_STAT struct stat64
	81	#else
	82	#define LSTAT lstat
	83	#define STRUCT_STAT struct stat
	84	#endif
	85
642935c0	86	static void usage(void)
3839e657	87	{
e68594d2	88	fprintf(stderr,
1378bb65	89	_("Usage: %s [-pRVf] [-+=aAcCdDeijPsStTuF] [-v version] files...\n"),
642935c0 TT	90	program_name);
642935c0 TT	91	exit(1);
3839e657 TT	92	}
3839e657 TT	93
9a718849 TT	94	struct flags_char {
	95	unsigned long flag;
	96	char optchar;
	97	};
	98
	99	static const struct flags_char flags_array[] = {
	100	{ EXT2_NOATIME_FL, 'A' },
	101	{ EXT2_SYNC_FL, 'S' },
88372d5c	102	{ EXT2_DIRSYNC_FL, 'D' },
9a718849 TT	103	{ EXT2_APPEND_FL, 'a' },
	104	{ EXT2_COMPR_FL, 'c' },
	105	{ EXT2_NODUMP_FL, 'd' },
7c8da6e3	106	{ EXT4_EXTENTS_FL, 'e'},
9a718849 TT	107	{ EXT2_IMMUTABLE_FL, 'i' },
9a718849 TT	108	{ EXT3_JOURNAL_DATA_FL, 'j' },
e1cec446	109	{ EXT4_PROJINHERIT_FL, 'P' },
9a718849 TT	110	{ EXT2_SECRM_FL, 's' },
9a718849 TT	111	{ EXT2_UNRM_FL, 'u' },
b3f5b4c2	112	{ EXT2_NOTAIL_FL, 't' },
15f9011a	113	{ EXT2_TOPDIR_FL, 'T' },
0796e660	114	{ FS_NOCOW_FL, 'C' },
1378bb65	115	{ EXT4_CASEFOLD_FL, 'F' },
9a718849 TT	116	{ 0, 0 }
	117	};
	118
	119	static unsigned long get_flag(char c)
	120	{
	121	const struct flags_char *fp;
efc6f628	122
9a718849 TT	123	for (fp = flags_array; fp->flag != 0; fp++) {
	124	if (fp->optchar == c)
	125	return fp->flag;
	126	}
	127	return 0;
	128	}
	129
	130
3839e657 TT	131	static int decode_arg (int * i, int argc, char ** argv)
	132	{
	133	char * p;
	134	char * tmp;
9a718849	135	unsigned long fl;
3839e657 TT	136
	137	switch (argv[*i][0])
	138	{
	139	case '-':
9a718849 TT	140	for (p = &argv[i][1]; p; p++) {
9a718849 TT	141	if (*p == 'R') {
3839e657	142	recursive = 1;
9a718849 TT	143	continue;
	144	}
	145	if (*p == 'V') {
3839e657	146	verbose = 1;
9a718849 TT	147	continue;
9a718849 TT	148	}
e68594d2 TT	149	if (*p == 'f') {
	150	silent = 1;
	151	continue;
	152	}
3627e838 LX	153	if (*p == 'p') {
	154	(*i)++;
	155	if (*i >= argc)
	156	usage ();
	157	project = strtol (argv[*i], &tmp, 0);
	158	if (*tmp) {
	159	com_err (program_name, 0,
	160	_("bad project - %s\n"),
	161	argv[*i]);
	162	usage ();
	163	}
	164	set_project = 1;
	165	continue;
	166	}
9a718849	167	if (*p == 'v') {
1e3472c5	168	(*i)++;
3839e657 TT	169	if (*i >= argc)
3839e657 TT	170	usage ();
3839e657	171	version = strtol (argv[*i], &tmp, 0);
9a718849	172	if (*tmp) {
3839e657	173	com_err (program_name, 0,
e68594d2	174	_("bad version - %s\n"),
9a718849	175	argv[*i]);
3839e657 TT	176	usage ();
	177	}
	178	set_version = 1;
9a718849	179	continue;
3839e657	180	}
9a718849 TT	181	if ((fl = get_flag(*p)) == 0)
	182	usage();
	183	rf \|= fl;
	184	rem = 1;
	185	}
3839e657 TT	186	break;
	187	case '+':
	188	add = 1;
9a718849 TT	189	for (p = &argv[i][1]; p; p++) {
	190	if ((fl = get_flag(*p)) == 0)
	191	usage();
	192	af \|= fl;
	193	}
3839e657 TT	194	break;
	195	case '=':
	196	set = 1;
9a718849 TT	197	for (p = &argv[i][1]; p; p++) {
	198	if ((fl = get_flag(*p)) == 0)
	199	usage();
	200	sf \|= fl;
	201	}
3839e657 TT	202	break;
	203	default:
	204	return EOF;
	205	break;
	206	}
	207	return 1;
	208	}
	209
2d328bb7	210	static int chattr_dir_proc(const char , struct dirent , void *);
3839e657	211
2d328bb7	212	static int change_attributes(const char * name)
3839e657 TT	213	{
3839e657 TT	214	unsigned long flags;
b705640a	215	STRUCT_STAT st;
3839e657	216
b705640a	217	if (LSTAT (name, &st) == -1) {
e68594d2 TT	218	if (!silent)
	219	com_err (program_name, errno,
	220	_("while trying to stat %s"), name);
	221	return -1;
3839e657	222	}
e1a0a3e3	223
7c8da6e3 AK	224	if (fgetflags(name, &flags) == -1) {
	225	if (!silent)
	226	com_err(program_name, errno,
	227	_("while reading flags on %s"), name);
	228	return -1;
	229	}
e1a0a3e3 TT	230	if (set) {
e1a0a3e3 TT	231	if (verbose) {
d9c56d3c	232	printf (_("Flags of %s set as "), name);
f3db3566	233	print_flags (stdout, sf, 0);
3839e657 TT	234	printf ("\n");
	235	}
	236	if (fsetflags (name, sf) == -1)
	237	perror (name);
e1a0a3e3	238	} else {
7c8da6e3 AK	239	if (rem)
	240	flags &= ~rf;
	241	if (add)
	242	flags \|= af;
7c8da6e3 AK	243	if (verbose) {
	244	printf(_("Flags of %s set as "), name);
	245	print_flags(stdout, flags, 0);
	246	printf("\n");
	247	}
	248	if (!S_ISDIR(st.st_mode))
	249	flags &= ~EXT2_DIRSYNC_FL;
	250	if (fsetflags(name, flags) == -1) {
	251	if (!silent) {
	252	com_err(program_name, errno,
e68594d2 TT	253	_("while setting flags on %s"),
e68594d2 TT	254	name);
e68594d2	255	}
7c8da6e3	256	return -1;
3839e657 TT	257	}
3839e657 TT	258	}
e1a0a3e3	259	if (set_version) {
3839e657	260	if (verbose)
d9c56d3c	261	printf (_("Version of %s set as %lu\n"), name, version);
e68594d2 TT	262	if (fsetversion (name, version) == -1) {
	263	if (!silent)
	264	com_err (program_name, errno,
	265	_("while setting version on %s"),
	266	name);
	267	return -1;
	268	}
3839e657	269	}
3627e838 LX	270	if (set_project) {
3627e838 LX	271	if (verbose)
25040c04	272	printf (_("Project of %s set as %lu\n"), name, project);
3627e838 LX	273	if (fsetproject (name, project) == -1) {
	274	if (!silent)
	275	com_err (program_name, errno,
	276	_("while setting project on %s"),
	277	name);
	278	return -1;
	279	}
	280
	281	}
3839e657	282	if (S_ISDIR(st.st_mode) && recursive)
e68594d2 TT	283	return iterate_on_dir (name, chattr_dir_proc, NULL);
e68594d2 TT	284	return 0;
3839e657 TT	285	}
3839e657 TT	286
e1a0a3e3	287	static int chattr_dir_proc (const char * dir_name, struct dirent * de,
54434927	288	void * private EXT2FS_ATTR((unused)))
3839e657	289	{
e68594d2 TT	290	int ret = 0;
e68594d2 TT	291
e1a0a3e3	292	if (strcmp (de->d_name, ".") && strcmp (de->d_name, "..")) {
a418d3ad TT	293	char *path;
	294
	295	path = malloc(strlen (dir_name) + 1 + strlen (de->d_name) + 1);
642935c0	296	if (!path) {
45ff69ff AD	297	fprintf(stderr, "%s",
	298	_("Couldn't allocate path variable "
	299	"in chattr_dir_proc"));
e68594d2	300	return -1;
642935c0	301	}
e68594d2	302	sprintf(path, "%s/%s", dir_name, de->d_name);
2d328bb7	303	ret = change_attributes(path);
a418d3ad	304	free(path);
3839e657	305	}
e68594d2	306	return ret;
3839e657 TT	307	}
3839e657 TT	308
00e5433e	309	int main (int argc, char ** argv)
3839e657 TT	310	{
	311	int i, j;
	312	int end_arg = 0;
e68594d2	313	int err, retval = 0;
3839e657	314
d9c56d3c TT	315	#ifdef ENABLE_NLS
d9c56d3c TT	316	setlocale(LC_MESSAGES, "");
14308a53	317	setlocale(LC_CTYPE, "");
d9c56d3c TT	318	bindtextdomain(NLS_CAT_NAME, LOCALEDIR);
d9c56d3c TT	319	textdomain(NLS_CAT_NAME);
9d4507c5	320	set_com_err_gettext(gettext);
d9c56d3c	321	#endif
3839e657 TT	322	if (argc && *argv)
	323	program_name = *argv;
	324	i = 1;
e1a0a3e3	325	while (i < argc && !end_arg) {
2293a4d5 TT	326	/* '--' arg should end option processing */
	327	if (strcmp(argv[i], "--") == 0) {
	328	i++;
	329	end_arg = 1;
	330	} else if (decode_arg (&i, argc, argv) == EOF)
3839e657 TT	331	end_arg = 1;
	332	else
	333	i++;
	334	}
	335	if (i >= argc)
	336	usage ();
e1a0a3e3	337	if (set && (add \|\| rem)) {
54434927	338	fputs(_("= is incompatible with - and +\n"), stderr);
3839e657 TT	339	exit (1);
3839e657 TT	340	}
e1a0a3e3	341	if ((rf & af) != 0) {
54434927	342	fputs("Can't both set and unset same flag.\n", stderr);
e1a0a3e3 TT	343	exit (1);
e1a0a3e3 TT	344	}
3627e838	345	if (!(add \|\| rem \|\| set \|\| set_version \|\| set_project )) {
54434927	346	fputs(_("Must use '-v', =, - or +\n"), stderr);
3839e657 TT	347	exit (1);
3839e657 TT	348	}
a88fa0c0	349	if (verbose)
0f8973fb TT	350	fprintf (stderr, "chattr %s (%s)\n",
0f8973fb TT	351	E2FSPROGS_VERSION, E2FSPROGS_DATE);
e68594d2	352	for (j = i; j < argc; j++) {
2d328bb7	353	err = change_attributes (argv[j]);
e68594d2 TT	354	if (err)
	355	retval = 1;
	356	}
	357	exit(retval);
3839e657	358	}