#! /bin/bash
# SPDX-License-Identifier: GPL-2.0
# Copyright (c) 2005 Silicon Graphics, Inc.  All Rights Reserved.
#
# FS QA Test No. 106
#
# Exercise basic xfs_quota functionality (user/group/project quota)
# Use of "sync" mount option here is an attempt to get deterministic
# allocator behaviour.
#
. ./common/preamble
_begin_fstest auto quick quota

# Import common functions.
. ./common/filter
. ./common/quota

_require_scratch
_require_xfs_quota
_require_user
_require_group

_scratch_mkfs_xfs >>$seqres.full 2>&1

uqid=`id -u fsgqa`
gqid=`id -g fsgqa`
pqid=10
cat >$tmp.projects <<EOF
$pqid:$SCRATCH_MNT
EOF

cat >$tmp.projid <<EOF
root:0
fsgqa:$pqid
EOF

create_files()
{
	local bs=$1
	local inum=$2

	echo "Using type=$type id=$id" >> $seqres.full

	for ((i=0; i<$((inum-1)); i++)); do
		_file_as_id $SCRATCH_MNT/inode$i $id $type 1024 0
	done

	_file_as_id $SCRATCH_MNT/block $id $type $bs 1
}

clean_files()
{
	rm -rf $SCRATCH_MNT/* 2>/dev/null
	rm -rf $tmp.quot 2>/dev/null
	rm -rf $tmp.quota 2>/dev/null
}

filter_quot()
{
	_filter_quota | grep -v "root \|#0 " \
		| sed -e '/#[0-9]*/s/#[0-9]*/#ID/g'
}

filter_report()
{
	_filter_quota | grep -v "^root \|^#0 " \
		| sed -e '/^#[0-9]*/s/^#[0-9]*/#ID/g'
}

filter_quota()
{
	_filter_quota | sed -e "/Disk quotas for/s/([0-9]*)/(ID)/g" \
			    -e "/Disk quotas for/s/#[0-9]*/#ID/g"
}

filter_state()
{
	_filter_quota | sed -e "s/Inode: #[0-9]* (0 blocks, 0 extents)/Inode: #[INO] (0 blocks, 0 extents)/g" \
			    -e "s/Inode: #[0-9]* ([0-9]* blocks, [0-9]* extents)/Inode: #[INO] (X blocks, Y extents)/g" \
			    -e "/[0-9][0-9]:[0-9][0-9]:[0-9][0-9]/s/ [0-9][0-9]:[0-9][0-9]:[0-9][0-9]//g" \
			    -e '/max warnings:/d'
}

test_quot()
{
	local opt="$*"

	echo "checking quot command (type=$type)"
	$XFS_QUOTA_PROG -D $tmp.projects -P $tmp.projid -x \
			-c "quot -$type $opt -bi" $SCRATCH_MNT | filter_quot
}

test_report()
{
	local opt="$*"

	echo "checking report command (type=$type)"
	$XFS_QUOTA_PROG -D $tmp.projects -P $tmp.projid -x \
			-c "report -$type $opt -bi" \
			$SCRATCH_MNT | filter_report
}

test_quota()
{
	local opt="$*"

	echo "checking quota command (type=$type)"
	$XFS_QUOTA_PROG -D $tmp.projects -P $tmp.projid -x \
			-c "quota -$type $opt -bi $id" \
			$SCRATCH_MNT | filter_quota
}

test_limit()
{
	local bs=$1
	local bh=$2
	local is=$3
	local ih=$4

	echo "checking limit command (type=$type, bsoft=$bs, bhard=$bh, isoft=$is, ihard=$ih)"
	$XFS_QUOTA_PROG -D $tmp.projects -P $tmp.projid -x \
			-c "limit -$type bsoft=$bs bhard=$bh fsgqa" \
			-c "limit -$type isoft=$is ihard=$ih fsgqa" \
			$SCRATCH_MNT

	# let the timer day transition happen
	sleep 2
}

test_timer()
{
	echo "checking timer command (type=$type)"
	# set 3days+1h for time won't become 2days soon
	$XFS_QUOTA_PROG -D $tmp.projects -P $tmp.projid -x \
			-c "timer -$type -bi 73h" \
			$SCRATCH_MNT | _filter_scratch
}

test_disable()
{
	echo "checking disable command (type=$type)"
	$XFS_QUOTA_PROG -D $tmp.projects -P $tmp.projid -x \
			-c "disable -$type -v" \
			$SCRATCH_MNT | filter_state
}

test_enable()
{
	echo "checking enable command (type=$type)"
	$XFS_QUOTA_PROG -D $tmp.projects -P $tmp.projid -x \
			-c "enable -$type -v" $SCRATCH_MNT | filter_state
}

test_off()
{
	echo "turning quota off by remounting"
	_scratch_unmount
	_qmount_option "noquota"
	_qmount
}

test_remove()
{
	echo "checking remove command (type=$type)"
	$XFS_QUOTA_PROG -D $tmp.projects -P $tmp.projid -x \
			-c "remove -$type -v" \
			$SCRATCH_MNT | _filter_scratch
}

test_state()
{
	echo "checking state command (type=$type)"
	$XFS_QUOTA_PROG -D $tmp.projects -P $tmp.projid -x \
			-c "state -$type" $SCRATCH_MNT | filter_state
}

test_dump()
{
	echo "checking dump command (type=$type)"
	rm -f $tmp.backup 2>>/dev/null
	$XFS_QUOTA_PROG -D $tmp.projects -P $tmp.projid -x \
			-c "dump -$type -f $tmp.backup" \
			$SCRATCH_MNT | _filter_scratch
}

test_restore()
{
	echo "checking restore command (type=$type)"
	$XFS_QUOTA_PROG -D $tmp.projects -P $tmp.projid -x \
			-c "restore -$type -f $tmp.backup" \
			$SCRATCH_MNT | _filter_scratch
}

test_xfs_quota()
{
	_qmount_option $1
	_qmount
	_force_vfs_quota_testing $SCRATCH_MNT	 # golden output encodes block usage

	if [ $type == "p" ]; then
		_require_prjquota $SCRATCH_DEV
		$XFS_QUOTA_PROG -D $tmp.projects -P $tmp.projid -x \
			-c "project -s $id" $SCRATCH_MNT > /dev/null
	fi

	# init quota
	echo "init quota limit and timer, and dump it"
	echo "create_files 1024k 15"; create_files 1024k 15
	echo "quota remount"; _qmount
	echo ; test_quot
	echo ; test_timer
	echo ; test_limit 512k 2048k 10 20
	echo ; test_dump

	# report options test
	echo "report options test"
	echo ; test_report
	echo "-N option"; test_report -N
	echo "-L -U options"; test_report -L $id -U $id
	echo "-t option"; test_report -t
	echo "-n option"; test_report -n
	echo "-h option"; test_report -h

	# quot options test
	echo "quot options test"
	echo ; test_quot
	echo "-f option"; test_quot -f $tmp.quot
	cat $tmp.quot | filter_quot
	echo "-n option"; test_quot -n

	# quota options test
	echo ; test_quota
	echo "-f option"; test_quota -f $tmp.quota
	cat $tmp.quota | filter_quota
	echo "-N option"; test_quota -N
	echo "-n option"; test_quota -n
	echo "-h option"; test_quota -h

	# disable/enable test
	echo "disable quota"
	echo ; test_disable
	echo ; test_report -N
	echo "expect a remove error at here"; test_remove
	echo ; test_enable
	echo ; test_report -N

	# off and remove test
	echo "disable and remove test"
	echo ; test_limit 100m 100m 100 100
	echo ; test_quota -N
	echo ; test_disable
	echo ; test_state
	echo ; test_off
	echo ; test_remove
	echo ; test_report -N
	_scratch_unmount

	echo "quota remount";
	_qmount_option $1
	_qmount
	echo ; test_report -N

	# restore test
	echo "restore quota"
	echo ; test_restore
	echo ; test_report -N
	echo ; test_state
	echo "cleanup files"; clean_files
}

echo "----------------------- uquota,sync ---------------------------"
type=u
id=$uqid
test_xfs_quota "uquota,sync"

echo "----------------------- gquota,sync ---------------------------"
type=g
id=$gqid
test_xfs_quota "gquota,sync"

echo "----------------------- pquota,sync ---------------------------"
# Need to clean the group quota before test project quota, because
# V4 xfs doesn't support separate project inode. So mkfs at here.
_scratch_unmount
_scratch_mkfs_xfs >>$seqres.full 2>&1
type=p
id=$pqid
test_xfs_quota "pquota,sync"

_scratch_unmount
# success, all done
status=0
exit
