[thirdparty/e2fsprogs.git] / debugfs / setsuper.c

/*
 * setsuper.c --- set a superblock value
 */

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <ctype.h>
#include <string.h>
#include <time.h>
#include <sys/types.h>
#include <sys/stat.h>
#ifdef HAVE_ERRNO_H
#include <errno.h>
#endif
#include <fcntl.h>
#include <utime.h>

#include "debugfs.h"

static struct ext2_super_block set_sb;

struct super_set_info {
	const char	*name;
	void	*ptr;
	int	size;
	errcode_t (*func)(struct super_set_info *info, char *arg);
};

static errcode_t parse_uint(struct super_set_info *info, char *arg);
static errcode_t parse_int(struct super_set_info *info, char *arg);
static errcode_t parse_string(struct super_set_info *info, char *arg);
static errcode_t parse_uuid(struct super_set_info *info, char *arg);
static errcode_t parse_hashalg(struct super_set_info *info, char *arg);

static struct super_set_info super_fields[] = {
	{ "inodes_count", &set_sb.s_inodes_count, 4, parse_uint },
	{ "blocks_count", &set_sb.s_blocks_count, 4, parse_uint },
	{ "r_blocks_count", &set_sb.s_r_blocks_count, 4, parse_uint },
	{ "free_blocks_count", &set_sb.s_free_blocks_count, 4, parse_uint },
	{ "free_inodes_count", &set_sb.s_free_inodes_count, 4, parse_uint },
	{ "first_data_block", &set_sb.s_first_data_block, 4, parse_uint },
	{ "log_block_size", &set_sb.s_log_block_size, 4, parse_uint },
	{ "log_frag_size", &set_sb.s_log_frag_size, 4, parse_int },
	{ "blocks_per_group", &set_sb.s_blocks_per_group, 4, parse_uint },
	{ "frags_per_group", &set_sb.s_frags_per_group, 4, parse_uint },
	{ "inodes_per_group", &set_sb.s_inodes_per_group, 4, parse_uint },
	/* s_mtime (time_t) */
	/* s_wtime (time_t) */
	{ "mnt_count", &set_sb.s_mnt_count, 2, parse_uint },
	{ "max_mnt_count", &set_sb.s_max_mnt_count, 2, parse_int },
	/* s_magic */
	{ "state", &set_sb.s_state, 2, parse_uint },
	{ "errors", &set_sb.s_errors, 2, parse_uint },
	{ "minor_rev_level", &set_sb.s_minor_rev_level, 2, parse_uint },
	/* s_lastcheck (time_t) */
	{ "checkinterval", &set_sb.s_checkinterval, 4, parse_uint },
	{ "creator_os", &set_sb.s_creator_os, 4, parse_uint },
	{ "rev_level", &set_sb.s_rev_level, 4, parse_uint },
	{ "def_resuid", &set_sb.s_def_resuid, 2, parse_uint },
	{ "def_resgid", &set_sb.s_def_resgid, 2, parse_uint },
	{ "first_ino", &set_sb.s_first_ino, 4, parse_uint },
	{ "inode_size", &set_sb.s_inode_size, 2, parse_uint },
	{ "block_group_nr", &set_sb.s_block_group_nr, 2, parse_uint },
	{ "feature_compat", &set_sb.s_feature_compat, 4, parse_uint },
	{ "feature_incompat", &set_sb.s_feature_incompat, 4, parse_uint },
	{ "feature_ro_compat", &set_sb.s_feature_ro_compat, 4, parse_uint }, 
	{ "uuid", &set_sb.s_uuid, 16, parse_uuid },
	{ "volume_name",  &set_sb.s_volume_name, 16, parse_string },
	{ "last_mounted",  &set_sb.s_last_mounted, 64, parse_string },
	{ "lastcheck",  &set_sb.s_lastcheck, 4, parse_uint },
	{ "algorithm_usage_bitmap", &set_sb.s_algorithm_usage_bitmap, 
		  4, parse_uint },
	{ "prealloc_blocks", &set_sb.s_prealloc_blocks, 1, parse_uint },
	{ "prealloc_dir_blocks", &set_sb.s_prealloc_dir_blocks, 1,
		  parse_uint },
	/* s_padding1 */
	{ "journal_uuid", &set_sb.s_journal_uuid, 16, parse_uuid },
	{ "journal_inum", &set_sb.s_journal_inum, 4, parse_uint },
	{ "journal_dev", &set_sb.s_journal_dev, 4, parse_uint },
	{ "last_orphan", &set_sb.s_last_orphan, 4, parse_uint },
	{ "hash_seed", &set_sb.s_hash_seed, 16, parse_uuid },
	{ "def_hash_version", &set_sb.s_def_hash_version, 1, parse_hashalg },
	{ 0, 0, 0, 0 }
};

static struct super_set_info *find_field(char *field)
{
	struct super_set_info *ss;

	if (strncmp(field, "s_", 2) == 0)
		field += 2;
	for (ss = super_fields ; ss->name ; ss++) {
		if (strcmp(ss->name, field) == 0)
			return ss;
	}
	return NULL;
}

static errcode_t parse_uint(struct super_set_info *info, char *arg)
{
	unsigned long	num;
	char *tmp;
	__u32	*ptr32;
	__u16	*ptr16;
	__u8	*ptr8;

	num = strtoul(arg, &tmp, 0);
	if (*tmp) {
		fprintf(stderr, "Couldn't parse '%s' for field %s.\n",
			arg, info->name);
		return EINVAL;
	}
	switch (info->size) {
	case 4:
		ptr32 = (__u32 *) info->ptr;
		*ptr32 = num;
		break;
	case 2:
		ptr16 = (__u16 *) info->ptr;
		*ptr16 = num;
		break;
	case 1:
		ptr8 = (__u8 *) info->ptr;
		*ptr8 = num;
		break;
	}
	return 0;
}

static errcode_t parse_int(struct super_set_info *info, char *arg)
{
	long	num;
	char *tmp;
	__s32	*ptr32;
	__s16	*ptr16;
	__s8	*ptr8;

	num = strtol(arg, &tmp, 0);
	if (*tmp) {
		fprintf(stderr, "Couldn't parse '%s' for field %s.\n",
			arg, info->name);
		return EINVAL;
	}
	switch (info->size) {
	case 4:
		ptr32 = (__s32 *) info->ptr;
		*ptr32 = num;
		break;
	case 2:
		ptr16 = (__s16 *) info->ptr;
		*ptr16 = num;
		break;
	case 1:
		ptr8 = (__s8 *) info->ptr;
		*ptr8 = num;
		break;
	}
	return 0;
}

static errcode_t parse_string(struct super_set_info *info, char *arg)
{
	char	*cp = (char *) info->ptr;

	if (strlen(arg) >= info->size) {
		fprintf(stderr, "Error maximum size for %s is %d.\n",
			info->name, info->size);
		return EINVAL;
	}
	strcpy(cp, arg);
	return 0;
}

static errcode_t parse_uuid(struct super_set_info *info, char *arg)
{
	char *	p = (char *) info->ptr;
	
	if ((strcasecmp(arg, "null") == 0) ||
	    (strcasecmp(arg, "clear") == 0)) {
		uuid_clear(p);
	} else if (strcasecmp(arg, "time") == 0) {
		uuid_generate_time(p);
	} else if (strcasecmp(arg, "random") == 0) {
		uuid_generate(p);
	} else if (uuid_parse(arg, p)) {
		fprintf(stderr, "Invalid UUID format: %s\n", arg);
		return EINVAL;
	}
	return 0;
}

static errcode_t parse_hashalg(struct super_set_info *info, char *arg)
{
	int	hashv;
	unsigned char	*p = (unsigned char *) info->ptr;

	hashv = e2p_string2hash(arg);
	if (hashv < 0) {
		fprintf(stderr, "Invalid hash algorithm: %s\n", arg);
		return EINVAL;
	}
	*p = hashv;
	return 0;
}


static void print_possible_fields(void)
{
	struct super_set_info *ss;
	const char	*type;

	printf("Superblock fields supported by the set_super_value command:\n");
	for (ss = super_fields ; ss->name ; ss++) {
		type = "unknown";
		if (ss->func == parse_string)
			type = "string";
		else if (ss->func == parse_int)
			type = "integer";
		else if (ss->func == parse_uint)
			type = "unsigned integer";
		else if (ss->func == parse_uuid)
			type = "UUID";
		else if (ss->func == parse_hashalg)
			type = "hash algorithm";
		printf("\t%-20s\t%s\n", ss->name, type);
	}
}


void do_set_super(int argc, char *argv[])
{
	const char *usage = "<field> <value>\n"
		"\t\"set_super_value -l\" will list the names of "
		"superblock fields\n\twhich can be set.";
	static struct super_set_info *ss;
	
	if ((argc == 2) && !strcmp(argv[1], "-l")) {
		print_possible_fields();
		return;
	}

	if (common_args_process(argc, argv, 3, 3, "set_super_value",
				usage, CHECK_FS_RW))
		return;

	if ((ss = find_field(argv[1])) == 0) {
		com_err(argv[0], 0, "invalid field specifier: %s", argv[1]);
		return;
	}
	set_sb = *current_fs->super;
	if (ss->func(ss, argv[2]) == 0) {
		*current_fs->super = set_sb;
		ext2fs_mark_super_dirty(current_fs);
	}
}
Commit	Line	Data
c6bd0c97 TT	1	/*
	2	* setsuper.c --- set a superblock value
	3	*/
	4
	5	#include <stdio.h>
	6	#include <unistd.h>
	7	#include <stdlib.h>
	8	#include <ctype.h>
	9	#include <string.h>
	10	#include <time.h>
	11	#include <sys/types.h>
	12	#include <sys/stat.h>
	13	#ifdef HAVE_ERRNO_H
	14	#include <errno.h>
	15	#endif
	16	#include <fcntl.h>
	17	#include <utime.h>
	18
	19	#include "debugfs.h"
	20
	21	static struct ext2_super_block set_sb;
	22
	23	struct super_set_info {
b044c2e0	24	const char *name;
c6bd0c97 TT	25	void *ptr;
	26	int size;
	27	errcode_t (func)(struct super_set_info info, char *arg);
	28	};
	29
	30	static errcode_t parse_uint(struct super_set_info info, char arg);
	31	static errcode_t parse_int(struct super_set_info info, char arg);
	32	static errcode_t parse_string(struct super_set_info info, char arg);
f61fc0b5 TT	33	static errcode_t parse_uuid(struct super_set_info info, char arg);
f61fc0b5 TT	34	static errcode_t parse_hashalg(struct super_set_info info, char arg);
c6bd0c97 TT	35
	36	static struct super_set_info super_fields[] = {
	37	{ "inodes_count", &set_sb.s_inodes_count, 4, parse_uint },
	38	{ "blocks_count", &set_sb.s_blocks_count, 4, parse_uint },
	39	{ "r_blocks_count", &set_sb.s_r_blocks_count, 4, parse_uint },
	40	{ "free_blocks_count", &set_sb.s_free_blocks_count, 4, parse_uint },
	41	{ "free_inodes_count", &set_sb.s_free_inodes_count, 4, parse_uint },
	42	{ "first_data_block", &set_sb.s_first_data_block, 4, parse_uint },
	43	{ "log_block_size", &set_sb.s_log_block_size, 4, parse_uint },
	44	{ "log_frag_size", &set_sb.s_log_frag_size, 4, parse_int },
	45	{ "blocks_per_group", &set_sb.s_blocks_per_group, 4, parse_uint },
	46	{ "frags_per_group", &set_sb.s_frags_per_group, 4, parse_uint },
	47	{ "inodes_per_group", &set_sb.s_inodes_per_group, 4, parse_uint },
	48	/* s_mtime (time_t) */
	49	/* s_wtime (time_t) */
	50	{ "mnt_count", &set_sb.s_mnt_count, 2, parse_uint },
	51	{ "max_mnt_count", &set_sb.s_max_mnt_count, 2, parse_int },
	52	/* s_magic */
	53	{ "state", &set_sb.s_state, 2, parse_uint },
	54	{ "errors", &set_sb.s_errors, 2, parse_uint },
	55	{ "minor_rev_level", &set_sb.s_minor_rev_level, 2, parse_uint },
	56	/* s_lastcheck (time_t) */
	57	{ "checkinterval", &set_sb.s_checkinterval, 4, parse_uint },
	58	{ "creator_os", &set_sb.s_creator_os, 4, parse_uint },
	59	{ "rev_level", &set_sb.s_rev_level, 4, parse_uint },
	60	{ "def_resuid", &set_sb.s_def_resuid, 2, parse_uint },
	61	{ "def_resgid", &set_sb.s_def_resgid, 2, parse_uint },
	62	{ "first_ino", &set_sb.s_first_ino, 4, parse_uint },
426d734f	63	{ "inode_size", &set_sb.s_inode_size, 2, parse_uint },
c6bd0c97 TT	64	{ "block_group_nr", &set_sb.s_block_group_nr, 2, parse_uint },
	65	{ "feature_compat", &set_sb.s_feature_compat, 4, parse_uint },
	66	{ "feature_incompat", &set_sb.s_feature_incompat, 4, parse_uint },
	67	{ "feature_ro_compat", &set_sb.s_feature_ro_compat, 4, parse_uint },
f61fc0b5	68	{ "uuid", &set_sb.s_uuid, 16, parse_uuid },
c6bd0c97 TT	69	{ "volume_name", &set_sb.s_volume_name, 16, parse_string },
c6bd0c97 TT	70	{ "last_mounted", &set_sb.s_last_mounted, 64, parse_string },
364a373e	71	{ "lastcheck", &set_sb.s_lastcheck, 4, parse_uint },
c6bd0c97 TT	72	{ "algorithm_usage_bitmap", &set_sb.s_algorithm_usage_bitmap,
	73	4, parse_uint },
	74	{ "prealloc_blocks", &set_sb.s_prealloc_blocks, 1, parse_uint },
	75	{ "prealloc_dir_blocks", &set_sb.s_prealloc_dir_blocks, 1,
	76	parse_uint },
	77	/* s_padding1 */
f61fc0b5	78	{ "journal_uuid", &set_sb.s_journal_uuid, 16, parse_uuid },
c6bd0c97 TT	79	{ "journal_inum", &set_sb.s_journal_inum, 4, parse_uint },
	80	{ "journal_dev", &set_sb.s_journal_dev, 4, parse_uint },
	81	{ "last_orphan", &set_sb.s_last_orphan, 4, parse_uint },
f61fc0b5 TT	82	{ "hash_seed", &set_sb.s_hash_seed, 16, parse_uuid },
f61fc0b5 TT	83	{ "def_hash_version", &set_sb.s_def_hash_version, 1, parse_hashalg },
c6bd0c97 TT	84	{ 0, 0, 0, 0 }
	85	};
	86
	87	static struct super_set_info find_field(char field)
	88	{
	89	struct super_set_info *ss;
	90
3db9305a TT	91	if (strncmp(field, "s_", 2) == 0)
3db9305a TT	92	field += 2;
c6bd0c97 TT	93	for (ss = super_fields ; ss->name ; ss++) {
	94	if (strcmp(ss->name, field) == 0)
	95	return ss;
	96	}
	97	return NULL;
	98	}
	99
	100	static errcode_t parse_uint(struct super_set_info info, char arg)
	101	{
	102	unsigned long num;
	103	char *tmp;
	104	__u32 *ptr32;
	105	__u16 *ptr16;
	106	__u8 *ptr8;
	107
	108	num = strtoul(arg, &tmp, 0);
	109	if (*tmp) {
	110	fprintf(stderr, "Couldn't parse '%s' for field %s.\n",
	111	arg, info->name);
	112	return EINVAL;
	113	}
	114	switch (info->size) {
	115	case 4:
	116	ptr32 = (__u32 *) info->ptr;
	117	*ptr32 = num;
	118	break;
	119	case 2:
	120	ptr16 = (__u16 *) info->ptr;
	121	*ptr16 = num;
	122	break;
	123	case 1:
	124	ptr8 = (__u8 *) info->ptr;
	125	*ptr8 = num;
	126	break;
	127	}
	128	return 0;
	129	}
	130
	131	static errcode_t parse_int(struct super_set_info info, char arg)
	132	{
	133	long num;
	134	char *tmp;
	135	__s32 *ptr32;
	136	__s16 *ptr16;
	137	__s8 *ptr8;
	138
	139	num = strtol(arg, &tmp, 0);
	140	if (*tmp) {
	141	fprintf(stderr, "Couldn't parse '%s' for field %s.\n",
	142	arg, info->name);
	143	return EINVAL;
	144	}
	145	switch (info->size) {
	146	case 4:
	147	ptr32 = (__s32 *) info->ptr;
	148	*ptr32 = num;
	149	break;
	150	case 2:
	151	ptr16 = (__s16 *) info->ptr;
	152	*ptr16 = num;
	153	break;
	154	case 1:
	155	ptr8 = (__s8 *) info->ptr;
	156	*ptr8 = num;
157	break;
158	}
159	return 0;
160	}
161
162	static errcode_t parse_string(struct super_set_info info, char arg)
163	{
164	char cp = (char ) info->ptr;
165
166	if (strlen(arg) >= info->size) {
167	fprintf(stderr, "Error maximum size for %s is %d.\n",
168	info->name, info->size);
169	return EINVAL;
170	}
171	strcpy(cp, arg);
172	return 0;
173	}
174
f61fc0b5 TT	175	static errcode_t parse_uuid(struct super_set_info info, char arg)
	176	{
	177	char * p = (char *) info->ptr;
	178
	179	if ((strcasecmp(arg, "null") == 0) \|\|
	180	(strcasecmp(arg, "clear") == 0)) {
	181	uuid_clear(p);
	182	} else if (strcasecmp(arg, "time") == 0) {
	183	uuid_generate_time(p);
	184	} else if (strcasecmp(arg, "random") == 0) {
	185	uuid_generate(p);
	186	} else if (uuid_parse(arg, p)) {
	187	fprintf(stderr, "Invalid UUID format: %s\n", arg);
	188	return EINVAL;
	189	}
	190	return 0;
	191	}
	192
	193	static errcode_t parse_hashalg(struct super_set_info info, char arg)
	194	{
	195	int hashv;
	196	unsigned char p = (unsigned char ) info->ptr;
	197
	198	hashv = e2p_string2hash(arg);
	199	if (hashv < 0) {
	200	fprintf(stderr, "Invalid hash algorithm: %s\n", arg);
	201	return EINVAL;
	202	}
	203	*p = hashv;
	204	return 0;
	205	}
	206
	207
e1018eea	208	static void print_possible_fields(void)
6e96adcb TT	209	{
6e96adcb TT	210	struct super_set_info *ss;
4ea7bd04	211	const char *type;
6e96adcb TT	212
	213	printf("Superblock fields supported by the set_super_value command:\n");
	214	for (ss = super_fields ; ss->name ; ss++) {
	215	type = "unknown";
	216	if (ss->func == parse_string)
	217	type = "string";
	218	else if (ss->func == parse_int)
	219	type = "integer";
	220	else if (ss->func == parse_uint)
	221	type = "unsigned integer";
f61fc0b5 TT	222	else if (ss->func == parse_uuid)
	223	type = "UUID";
	224	else if (ss->func == parse_hashalg)
	225	type = "hash algorithm";
6e96adcb TT	226	printf("\t%-20s\t%s\n", ss->name, type);
	227	}
	228	}
	229
	230
c6bd0c97 TT	231	void do_set_super(int argc, char *argv[])
c6bd0c97 TT	232	{
e1018eea TT	233	const char *usage = "<field> <value>\n"
	234	"\t\"set_super_value -l\" will list the names of "
	235	"superblock fields\n\twhich can be set.";
c6bd0c97 TT	236	static struct super_set_info *ss;
c6bd0c97 TT	237
6e96adcb TT	238	if ((argc == 2) && !strcmp(argv[1], "-l")) {
	239	print_possible_fields();
	240	return;
	241	}
	242
e1018eea TT	243	if (common_args_process(argc, argv, 3, 3, "set_super_value",
e1018eea TT	244	usage, CHECK_FS_RW))
c6bd0c97 TT	245	return;
c6bd0c97 TT	246
c6bd0c97 TT	247	if ((ss = find_field(argv[1])) == 0) {
	248	com_err(argv[0], 0, "invalid field specifier: %s", argv[1]);
	249	return;
	250	}
	251	set_sb = *current_fs->super;
	252	if (ss->func(ss, argv[2]) == 0) {
	253	*current_fs->super = set_sb;
	254	ext2fs_mark_super_dirty(current_fs);
	255	}
	256	}