/*
* Copyright (c) Crackerjack Project., 2007
* Copyright (c) 2016 Fujitsu Ltd.
* Author: Xiao Yang <yangx.jy@cn.fujitsu.com>
*
* This program is free software;  you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY;  without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See
* the GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program.
*
* Test Name: quotactl01
*
* Description:
* This testcase checks the basic flag of quotactl(2) for non-XFS filesystems:
* 1) quotactl(2) succeeds to turn on quota with Q_QUOTAON flag for user.
* 2) quotactl(2) succeeds to set disk quota limits with Q_SETQUOTA flag
*    for user.
* 3) quotactl(2) succeeds to get disk quota limits with Q_GETQUOTA flag
*    for user.
* 4) quotactl(2) succeeds to set information about quotafile with Q_SETINFO
*    flag for user.
* 5) quotactl(2) succeeds to get information about quotafile with Q_GETINFO
*    flag for user.
* 6) quotactl(2) succeeds to get quota format with Q_GETFMT flag for user.
* 7) quotactl(2) succeeds to update quota usages with Q_SYNC flag for user.
* 8) quotactl(2) succeeds to turn off quota with Q_QUOTAOFF flag for user.
* 9) quotactl(2) succeeds to turn on quota with Q_QUOTAON flag for group.
* 10) quotactl(2) succeeds to set disk quota limits with Q_SETQUOTA flag
*     for group.
* 11) quotactl(2) succeeds to get disk quota limits with Q_GETQUOTA flag
*     for group.
* 12) quotactl(2) succeeds to set information about quotafile with Q_SETINFO
*     flag for group.
* 13) quotactl(2) succeeds to get information about quotafile with Q_GETINFO
*     flag for group.
* 14) quotactl(2) succeeds to get quota format with Q_GETFMT flag for group.
* 15) quotactl(2) succeeds to update quota usages with Q_SYNC flag for group.
* 16) quotactl(2) succeeds to turn off quota with Q_QUOTAOFF flag for group.
*
*/

#include <errno.h>
#include <string.h>
#include <unistd.h>
#include <stdio.h>
#include "config.h"

#include "tst_test.h"

#if defined(HAVE_QUOTAV2) || defined(HAVE_QUOTAV1)
# include <sys/quota.h>

# if defined(HAVE_QUOTAV2)
#  define _LINUX_QUOTA_VERSION  2
#  ifndef QFMT_VFS_V0
#   define QFMT_VFS_V0     2
#  endif
#  define USRPATH MNTPOINT "/aquota.user"
#  define GRPPATH MNTPOINT "/aquota.group"
#  define FMTID	QFMT_VFS_V0
# else
#  define _LINUX_QUOTA_VERSION  1
#  ifndef QFMT_VFS_OLD
#   define QFMT_VFS_OLD    1
#  endif
#  define USRPATH MNTPOINT "/quota.user"
#  define GRPPATH MNTPOINT "/quota.group"
#  define FMTID	QFMT_VFS_OLD
# endif

# define MNTPOINT	"mntpoint"

static int mount_flag;
static int32_t fmt_id = FMTID;
static int test_id;
static struct dqblk set_dq = {
	.dqb_bsoftlimit = 100,
	.dqb_valid = QIF_BLIMITS
};
static struct dqblk res_dq;
# if defined(HAVE_QUOTAV2)
static struct dqinfo set_qf = {
	.dqi_bgrace = 80,
	.dqi_valid = IIF_BGRACE
};
static struct dqinfo res_qf;
static int32_t fmt_buf;
# endif

static struct tcase {
	int cmd;
	int *id;
	void *addr;
	void *set_data;
	void *res_data;
	int sz;
	char *des;
} tcases[] = {
	{QCMD(Q_QUOTAON, USRQUOTA), &fmt_id, USRPATH,
	NULL, NULL, 0, "turn on quota for user"},

	{QCMD(Q_SETQUOTA, USRQUOTA), &test_id, &set_dq,
	NULL, NULL, 0, "set disk quota limit for user"},

	{QCMD(Q_GETQUOTA, USRQUOTA), &test_id, &res_dq,
	&set_dq.dqb_bsoftlimit, &res_dq.dqb_bsoftlimit,
	sizeof(res_dq.dqb_bsoftlimit), "get disk quota limit for user"},
# if defined(HAVE_QUOTAV2)
	{QCMD(Q_SETINFO, USRQUOTA), &test_id, &set_qf,
	NULL, NULL, 0, "set information about quotafile for user"},

	{QCMD(Q_GETINFO, USRQUOTA), &test_id, &res_qf,
	&set_qf.dqi_bgrace, &res_qf.dqi_bgrace, sizeof(res_qf.dqi_bgrace),
	"get information about quotafile for user"},

	{QCMD(Q_GETFMT, USRQUOTA), &test_id, &fmt_buf,
	&fmt_id, &fmt_buf, sizeof(fmt_buf),
	"get quota format for user"},
# endif
	{QCMD(Q_SYNC, USRQUOTA), &test_id, &res_dq,
	NULL, NULL, 0, "update quota usages for user"},

	{QCMD(Q_QUOTAOFF, USRQUOTA), &test_id, USRPATH,
	NULL, NULL, 0, "turn off quota for user"},

	{QCMD(Q_QUOTAON, GRPQUOTA), &fmt_id, GRPPATH,
	NULL, NULL, 0, "turn on quota for group"},

	{QCMD(Q_SETQUOTA, GRPQUOTA), &test_id, &set_dq,
	NULL, NULL, 0, "set disk quota limit for group"},

	{QCMD(Q_GETQUOTA, GRPQUOTA), &test_id, &res_dq, &set_dq.dqb_bsoftlimit,
	&res_dq.dqb_bsoftlimit, sizeof(res_dq.dqb_bsoftlimit),
	"set disk quota limit for group"},
# if defined(HAVE_QUOTAV2)
	{QCMD(Q_SETINFO, GRPQUOTA), &test_id, &set_qf,
	NULL, NULL, 0, "set information about quotafile for group"},

	{QCMD(Q_GETINFO, GRPQUOTA), &test_id, &res_qf, &set_qf.dqi_bgrace,
	&res_qf.dqi_bgrace, sizeof(res_qf.dqi_bgrace),
	"get information about quotafile for group"},

	{QCMD(Q_GETFMT, GRPQUOTA), &test_id, &fmt_buf,
	&fmt_id, &fmt_buf, sizeof(fmt_buf), "get quota format for group"},
# endif
	{QCMD(Q_SYNC, GRPQUOTA), &test_id, &res_dq,
	NULL, NULL, 0, "update quota usages for group"},

	{QCMD(Q_QUOTAOFF, GRPQUOTA), &test_id, GRPPATH,
	NULL, NULL, 0, "turn off quota for group"}
};

static void setup(void)
{
	const char *const cmd[] = {"quotacheck", "-ug", MNTPOINT, NULL};
	int ret;

	SAFE_MKDIR(MNTPOINT, 0755);

	SAFE_MKFS(tst_device->dev, "ext4", NULL, NULL);

	SAFE_MOUNT(tst_device->dev, MNTPOINT, "ext4", 0, "usrquota,grpquota");
	mount_flag = 1;

	ret = tst_run_cmd(cmd, NULL, NULL, 1);
	switch (ret) {
	case 255:
		tst_brk(TCONF, "quotacheck binary not installed");
	default:
		tst_brk(TBROK, "quotacheck exited with %i", ret);
	case 0:
	break;
	}

	test_id = geteuid();

	if (access(USRPATH, F_OK) == -1)
		tst_brk(TFAIL | TERRNO, "user quotafile didn't exist");

	if (access(GRPPATH, F_OK) == -1)
		tst_brk(TFAIL | TERRNO, "group quotafile didn't exist");
}

static void cleanup(void)
{
	if (mount_flag && tst_umount(MNTPOINT) < 0)
		tst_res(TWARN | TERRNO, "umount(2) failed");
}

static void verify_quota(unsigned int n)
{
	struct tcase *tc = &tcases[n];

	res_dq.dqb_bsoftlimit = 0;
	res_qf.dqi_igrace = 0;
	fmt_buf = 0;

	TEST(quotactl(tc->cmd, tst_device->dev, *tc->id, tc->addr));
	if (TEST_RETURN == -1) {
		tst_res(TFAIL | TTERRNO, "quotactl failed to %s", tc->des);
		return;
	}

	if (memcmp(tc->res_data, tc->set_data, tc->sz)) {
		tst_res(TFAIL, "quotactl failed to %s", tc->des);
		tst_res_hexd(TINFO, tc->res_data, tc->sz, "retval:   ");
		tst_res_hexd(TINFO, tc->set_data, tc->sz, "expected: ");
		return;
	}

	tst_res(TPASS, "quotactl succeeded to %s", tc->des);
}

static struct tst_test test = {
	.tid = "quotactl01",
	.needs_tmpdir = 1,
	.needs_root = 1,
	.test = verify_quota,
	.tcnt = ARRAY_SIZE(tcases),
	.needs_device = 1,
	.setup = setup,
	.cleanup = cleanup
};

#else
	TST_TEST_TCONF("This system didn't support quota");
#endif