/* $NetBSD: t_quota2_remount.c,v 1.6 2021/08/20 20:25:28 andvar Exp $ */

/* $NetBSD: t_quota2_remount.c,v 1.6 2021/08/20 20:25:28 andvar Exp $ */

/*
* Basic tests for quota2
*/

#include <atf-c.h>

#include "../common/h_fsmacros.h"

#include <sys/types.h>
#include <sys/mount.h>
#include <sys/statvfs.h>

#include <stdlib.h>

#include <ufs/ufs/ufsmount.h>

#include <rump/rump.h>
#include <rump/rump_syscalls.h>

#include "h_macros.h"

static void
do_quota(const atf_tc_t *tc, int n, const char *newfs_opts, int log)
{
int i;
char buf[1024];
int res;
int fd;
struct ufs_args uargs;
struct statvfs fst;

snprintf(buf, sizeof(buf), "newfs -q user -q group -F -s 4000 -n %d "
"%s %s", (n + 3), newfs_opts, FSTEST_IMGNAME);
if (system(buf) == -1)
atf_tc_fail_errno("cannot create file system");

rump_init();
if (rump_sys_mkdir(FSTEST_MNTNAME, 0777) == -1)
atf_tc_fail_errno("mount point create");

rump_pub_etfs_register("/diskdev", FSTEST_IMGNAME, RUMP_ETFS_BLK);

uargs.fspec = __UNCONST("/diskdev");

/* read-only doesn't have quota enabled */
if (rump_sys_mount(MOUNT_FFS, FSTEST_MNTNAME, MNT_RDONLY,
&uargs, sizeof(uargs)) == -1)
atf_tc_fail_errno("mount ffs ro %s", FSTEST_MNTNAME);

if (rump_sys_statvfs1(FSTEST_MNTNAME, &fst, 0) != 0)
atf_tc_fail_errno("statbfs %s (1)", FSTEST_MNTNAME);

if ((fst.f_flag & ST_QUOTA) != 0)
atf_tc_fail("R/O filesystem has quota");

/* updating to read-write enables quota */
if (rump_sys_mount(MOUNT_FFS, FSTEST_MNTNAME,
MNT_UPDATE | (log ? MNT_LOG : 0), &uargs, sizeof(uargs)) == -1)
atf_tc_fail_errno("mount ffs rw %s", FSTEST_MNTNAME);

if (rump_sys_statvfs1(FSTEST_MNTNAME, &fst, 0) != 0)
atf_tc_fail_errno("statbfs %s (2)", FSTEST_MNTNAME);

if ((fst.f_flag & ST_QUOTA) == 0)
atf_tc_fail("R/W filesystem has no quota");

/* we can update a second time */
if (rump_sys_mount(MOUNT_FFS, FSTEST_MNTNAME,
MNT_UPDATE | (log ? MNT_LOG : 0), &uargs, sizeof(uargs)) == -1)
atf_tc_fail_errno("mount ffs rw(2) %s", FSTEST_MNTNAME);

if (rump_sys_statvfs1(FSTEST_MNTNAME, &fst, 0) != 0)
atf_tc_fail_errno("statbfs %s (3)", FSTEST_MNTNAME);

if ((fst.f_flag & ST_QUOTA) == 0)
atf_tc_fail("R/W filesystem has no quota");

/* create some files so fsck has something to check */
FSTEST_ENTER();
RL(rump_sys_chown(".", 0, 0));
for (i = 0 ; i < n; i++) {
sprintf(buf, "file%d", i);
RL(fd = rump_sys_open(buf, O_CREAT | O_RDWR, 0755));
sprintf(buf, "test file no %d", i);
RL(rump_sys_write(fd, buf, strlen(buf)));
RL(rump_sys_fchown(fd, i, i+80000));
rump_sys_close(fd);
}
FSTEST_EXIT();
if (rump_sys_unmount(FSTEST_MNTNAME, 0) != 0) {
rump_pub_vfs_mount_print(FSTEST_MNTNAME, 1);
atf_tc_fail_errno("unmount failed");
}
snprintf(buf, 1024, "fsck_ffs -fn -F %s", FSTEST_IMGNAME);
res = system(buf);
if (res != 0)
atf_tc_fail("fsck returned %d", res);
}

#define DECL_TEST(nent, newops, name, descr, log) \
ATF_TC(quota_##name); \
\
ATF_TC_HEAD(quota_##name, tc) \
{ \
atf_tc_set_md_var(tc, "descr", \
"test filesystem remount with quotas, %s", descr); \
} \
\
ATF_TC_BODY(quota_##name, tc) \
{ \
do_quota(tc, nent, newops, log); \
}

DECL_TEST(10, "-O1 -B le", 10_O1_le, "UFS1 little-endian", 0)
DECL_TEST(10, "-O1 -B be", 10_O1_be, "UFS1 big-endian", 0)

#if 0
/*
* this cause fsck to complain about summaries at the end.
* This sems to be related to -o log (reproductible on a fs with no
* quota enabled). not reproductible with a real kernel ...
*/
DECL_TEST(10, "-O1", 10_O1_log, "UFS1 log", 1)
DECL_TEST(10, "-O2", 10_O2_log, "UFS2 log", 1)
#endif

ATF_TP_ADD_TCS(tp)
{

ATF_TP_ADD_TC(tp, quota_10_O1_le);
ATF_TP_ADD_TC(tp, quota_10_O1_be);
#if 0
ATF_TP_ADD_TC(tp, quota_10_O1_log);
ATF_TP_ADD_TC(tp, quota_10_O2_log);
#endif
return atf_no_error();
}