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

/*
 * quota.c --- debugfs quota commands
 *
 * Copyright (C) 2014 Theodore Ts'o.  This file may be redistributed
 * under the terms of the GNU Public License.
 */

#include "config.h"
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <ctype.h>
#include <string.h>
#include <time.h>
#ifdef HAVE_ERRNO_H
#include <errno.h>
#endif
#include <sys/types.h>
#ifdef HAVE_GETOPT_H
#include <getopt.h>
#else
extern int optind;
extern char *optarg;
#endif

#include "debugfs.h"

const char *quota_type[] = { "user", "group", "project", NULL };

static int load_quota_ctx(char *progname)
{
	errcode_t	retval;

	if (check_fs_open(progname))
		return 1;

	if (!ext2fs_has_feature_quota(current_fs->super)) {
		com_err(progname, 0, "quota feature not enabled");
		return 1;
	}

	if (current_qctx)
		return 0;

	retval = quota_init_context(&current_qctx, current_fs, 0);
	if (retval) {
		com_err(current_fs->device_name, retval,
			"while trying to load quota information");
		return 1;
	}
	return 0;
}

static int parse_quota_type(const char *cmdname, const char *str)
{
	errcode_t	retval;
	char		*t;
	int		flags = 0;
	int		i;

	for (i = 0; i < MAXQUOTAS; i++) {
		if (strcasecmp(str, quota_type[i]) == 0)
			break;
	}
	if (i >= MAXQUOTAS) {
		i = strtol(str, &t, 0);
		if (*t)
			i = -1;
	}
	if (i < 0 || i >= MAXQUOTAS) {
		com_err(0, 0, "Invalid quota type: %s", str);
		printf("Valid quota types are: ");
		for (i = 0; i < MAXQUOTAS; i++)
			printf("%s ", quota_type[i]);
		printf("\n");
		return -1;
	}

	if (current_fs->flags & EXT2_FLAG_RW)
		flags |= EXT2_FILE_WRITE;

	retval = quota_file_open(current_qctx, NULL, 0, i, -1, flags);
	if (retval) {
		com_err(cmdname, retval,
			"while opening quota inode (type %d)", i);
		return -1;
	}
	return i;
}


static int list_quota_callback(struct dquot *dq,
			       void *cb_data EXT2FS_ATTR((unused)))
{
	printf("%10u   %8lld %8lld %8lld    %8lld %8lld %8lld\n",
	       dq->dq_id, (long long)dq->dq_dqb.dqb_curspace,
	       (long long)dq->dq_dqb.dqb_bsoftlimit,
	       (long long)dq->dq_dqb.dqb_bhardlimit,
	       (long long)dq->dq_dqb.dqb_curinodes,
	       (long long)dq->dq_dqb.dqb_isoftlimit,
	       (long long)dq->dq_dqb.dqb_ihardlimit);
	return 0;
}

void do_list_quota(int argc, char *argv[], int sci_idx EXT2FS_ATTR((unused)),
		   void *infop EXT2FS_ATTR((unused)))
{
	errcode_t	retval;
	int		type;
	struct quota_handle *qh;

	if (load_quota_ctx(argv[0]))
		return;

	if (argc != 2) {
		com_err(0, 0, "Usage: list_quota <quota_type>\n");
		return;
	}

	type = parse_quota_type(argv[0], argv[1]);
	if (type < 0)
		return;

	printf("%7s %2s   %8s %8s %8s    %8s %8s %8s\n",
	       quota_type[type], "id",
	       "blocks", "quota", "limit", "inodes", "quota", "limit");
	qh = current_qctx->quota_file[type];
	retval = qh->qh_ops->scan_dquots(qh, list_quota_callback, NULL);
	if (retval) {
		com_err(argv[0], retval, "while scanning dquots");
		return;
	}
}

void do_get_quota(int argc, char *argv[], int sci_idx EXT2FS_ATTR((unused)),
		  void *infop EXT2FS_ATTR((unused)))
{
	int		err, type;
	struct quota_handle *qh;
	struct dquot	*dq;
	qid_t		id;

	if (load_quota_ctx(argv[0]))
		return;

	if (argc != 3) {
		com_err(0, 0, "Usage: get_quota <quota_type> <id>\n");
		return;
	}

	type = parse_quota_type(argv[0], argv[1]);
	if (type < 0)
		return;

	id = parse_ulong(argv[2], argv[0], "id", &err);
	if (err)
		return;

	printf("%7s %2s   %8s %8s %8s    %8s %8s %8s\n",
	       quota_type[type], "id",
	       "blocks", "quota", "limit", "inodes", "quota", "limit");

	qh = current_qctx->quota_file[type];

	dq = qh->qh_ops->read_dquot(qh, id);
	if (dq) {
		list_quota_callback(dq, NULL);
		ext2fs_free_mem(&dq);
	} else {
		com_err(argv[0], 0, "couldn't read quota record");
	}
}
Commit	Line	Data
75fc42f1 TT	1	/*
	2	* quota.c --- debugfs quota commands
	3	*
	4	* Copyright (C) 2014 Theodore Ts'o. This file may be redistributed
	5	* under the terms of the GNU Public License.
	6	*/
	7
	8	#include "config.h"
	9	#include <stdio.h>
	10	#include <unistd.h>
	11	#include <stdlib.h>
	12	#include <ctype.h>
	13	#include <string.h>
	14	#include <time.h>
	15	#ifdef HAVE_ERRNO_H
	16	#include <errno.h>
	17	#endif
	18	#include <sys/types.h>
	19	#ifdef HAVE_GETOPT_H
	20	#include <getopt.h>
	21	#else
	22	extern int optind;
	23	extern char *optarg;
	24	#endif
	25
	26	#include "debugfs.h"
	27
83628f38	28	const char *quota_type[] = { "user", "group", "project", NULL };
75fc42f1 TT	29
	30	static int load_quota_ctx(char *progname)
	31	{
	32	errcode_t retval;
	33
	34	if (check_fs_open(progname))
	35	return 1;
	36
4ee26699	37	if (!ext2fs_has_feature_quota(current_fs->super)) {
e6847693	38	com_err(progname, 0, "quota feature not enabled");
75fc42f1 TT	39	return 1;
	40	}
	41
	42	if (current_qctx)
	43	return 0;
	44
5c2a665a	45	retval = quota_init_context(&current_qctx, current_fs, 0);
75fc42f1 TT	46	if (retval) {
	47	com_err(current_fs->device_name, retval,
	48	"while trying to load quota information");
	49	return 1;
	50	}
	51	return 0;
	52	}
	53
	54	static int parse_quota_type(const char cmdname, const char str)
	55	{
	56	errcode_t retval;
	57	char *t;
	58	int flags = 0;
	59	int i;
	60
	61	for (i = 0; i < MAXQUOTAS; i++) {
	62	if (strcasecmp(str, quota_type[i]) == 0)
	63	break;
	64	}
	65	if (i >= MAXQUOTAS) {
	66	i = strtol(str, &t, 0);
	67	if (*t)
	68	i = -1;
	69	}
	70	if (i < 0 \|\| i >= MAXQUOTAS) {
	71	com_err(0, 0, "Invalid quota type: %s", str);
	72	printf("Valid quota types are: ");
	73	for (i = 0; i < MAXQUOTAS; i++)
	74	printf("%s ", quota_type[i]);
	75	printf("\n");
	76	return -1;
	77	}
	78
	79	if (current_fs->flags & EXT2_FLAG_RW)
	80	flags \|= EXT2_FILE_WRITE;
	81
	82	retval = quota_file_open(current_qctx, NULL, 0, i, -1, flags);
	83	if (retval) {
	84	com_err(cmdname, retval,
	85	"while opening quota inode (type %d)", i);
	86	return -1;
	87	}
	88	return i;
	89	}
	90
	91
25f291c9 TT	92	static int list_quota_callback(struct dquot *dq,
25f291c9 TT	93	void *cb_data EXT2FS_ATTR((unused)))
75fc42f1	94	{
83628f38	95	printf("%10u %8lld %8lld %8lld %8lld %8lld %8lld\n",
999d38d9 AD	96	dq->dq_id, (long long)dq->dq_dqb.dqb_curspace,
	97	(long long)dq->dq_dqb.dqb_bsoftlimit,
	98	(long long)dq->dq_dqb.dqb_bhardlimit,
	99	(long long)dq->dq_dqb.dqb_curinodes,
	100	(long long)dq->dq_dqb.dqb_isoftlimit,
	101	(long long)dq->dq_dqb.dqb_ihardlimit);
75fc42f1 TT	102	return 0;
	103	}
	104
2fcbcb1b TT	105	void do_list_quota(int argc, char *argv[], int sci_idx EXT2FS_ATTR((unused)),
2fcbcb1b TT	106	void *infop EXT2FS_ATTR((unused)))
75fc42f1 TT	107	{
75fc42f1 TT	108	errcode_t retval;
999d38d9	109	int type;
75fc42f1 TT	110	struct quota_handle *qh;
	111
	112	if (load_quota_ctx(argv[0]))
	113	return;
	114
	115	if (argc != 2) {
	116	com_err(0, 0, "Usage: list_quota <quota_type>\n");
	117	return;
	118	}
	119
	120	type = parse_quota_type(argv[0], argv[1]);
	121	if (type < 0)
	122	return;
	123
83628f38 EW	124	printf("%7s %2s %8s %8s %8s %8s %8s %8s\n",
83628f38 EW	125	quota_type[type], "id",
75fc42f1 TT	126	"blocks", "quota", "limit", "inodes", "quota", "limit");
	127	qh = current_qctx->quota_file[type];
	128	retval = qh->qh_ops->scan_dquots(qh, list_quota_callback, NULL);
	129	if (retval) {
	130	com_err(argv[0], retval, "while scanning dquots");
	131	return;
	132	}
	133	}
	134
2fcbcb1b TT	135	void do_get_quota(int argc, char *argv[], int sci_idx EXT2FS_ATTR((unused)),
2fcbcb1b TT	136	void *infop EXT2FS_ATTR((unused)))
75fc42f1	137	{
999d38d9	138	int err, type;
75fc42f1 TT	139	struct quota_handle *qh;
	140	struct dquot *dq;
	141	qid_t id;
	142
	143	if (load_quota_ctx(argv[0]))
	144	return;
	145
	146	if (argc != 3) {
	147	com_err(0, 0, "Usage: get_quota <quota_type> <id>\n");
	148	return;
	149	}
	150
	151	type = parse_quota_type(argv[0], argv[1]);
	152	if (type < 0)
	153	return;
	154
	155	id = parse_ulong(argv[2], argv[0], "id", &err);
	156	if (err)
	157	return;
	158
83628f38 EW	159	printf("%7s %2s %8s %8s %8s %8s %8s %8s\n",
83628f38 EW	160	quota_type[type], "id",
75fc42f1 TT	161	"blocks", "quota", "limit", "inodes", "quota", "limit");
	162
	163	qh = current_qctx->quota_file[type];
	164
	165	dq = qh->qh_ops->read_dquot(qh, id);
	166	if (dq) {
	167	list_quota_callback(dq, NULL);
	168	ext2fs_free_mem(&dq);
999d38d9	169	} else {
75fc42f1	170	com_err(argv[0], 0, "couldn't read quota record");
999d38d9	171	}
75fc42f1	172	}