#! /bin/bash
# SPDX-License-Identifier: GPL-2.0
# Copyright (c) 2013 Fusion IO.  All Rights Reserved.
#
# FS QA Test No. 022
#
# Test the basic functionality of qgroups
#
. ./common/preamble
_begin_fstest auto qgroup limit

# Import common functions.
. ./common/filter

_supported_fs btrfs
_require_scratch
_require_btrfs_qgroup_report

# Test to make sure we can actually turn it on and it makes sense
_basic_test()
{
	echo "=== basic test ===" >> $seqres.full
	_run_btrfs_util_prog subvolume create $SCRATCH_MNT/a
	_run_btrfs_util_prog quota enable $SCRATCH_MNT/a
	_run_btrfs_util_prog quota rescan -w $SCRATCH_MNT
	subvolid=$(_btrfs_get_subvolid $SCRATCH_MNT a)
	$BTRFS_UTIL_PROG qgroup show $units $SCRATCH_MNT | grep $subvolid >> \
		$seqres.full 2>&1
	[ $? -eq 0 ] || _fail "couldn't find our subvols quota group"
	run_check $FSSTRESS_PROG -d $SCRATCH_MNT/a -w -p 1 -n 2000 \
		$FSSTRESS_AVOID
	_run_btrfs_util_prog subvolume snapshot $SCRATCH_MNT/a \
		$SCRATCH_MNT/b

	# the shared values of both the original subvol and snapshot should
	# match
	a_shared=$($BTRFS_UTIL_PROG qgroup show $units $SCRATCH_MNT | grep "0/$subvolid")
	a_shared=$(echo $a_shared | $AWK_PROG '{ print $2 }')
	echo "subvol a id=$subvolid" >> $seqres.full
	subvolid=$(_btrfs_get_subvolid $SCRATCH_MNT b)
	echo "subvol b id=$subvolid" >> $seqres.full
	b_shared=$($BTRFS_UTIL_PROG qgroup show $units $SCRATCH_MNT | grep "0/$subvolid")
	b_shared=$(echo $b_shared | $AWK_PROG '{ print $2 }')
	$BTRFS_UTIL_PROG qgroup show $units $SCRATCH_MNT >> $seqres.full
	[ $b_shared -eq $a_shared ] || _fail "shared values don't match"
}

#enable quotas, do some work, check our values and then rescan and make sure we
#come up with the same answer
_rescan_test()
{
	echo "=== rescan test ===" >> $seqres.full
	# first with a blank subvol
	_run_btrfs_util_prog subvolume create $SCRATCH_MNT/a
	_run_btrfs_util_prog quota enable $SCRATCH_MNT/a
	subvolid=$(_btrfs_get_subvolid $SCRATCH_MNT a)
	run_check $FSSTRESS_PROG -d $SCRATCH_MNT/a -w -p 1 -n 2000 \
		$FSSTRESS_AVOID
	sync
	output=$($BTRFS_UTIL_PROG qgroup show $units $SCRATCH_MNT | grep "0/$subvolid")
	echo "qgroup values before rescan: $output" >> $seqres.full
	refer=$(echo $output | $AWK_PROG '{ print $2 }')
	excl=$(echo $output | $AWK_PROG '{ print $3 }')
	_run_btrfs_util_prog quota rescan -w $SCRATCH_MNT
	output=$($BTRFS_UTIL_PROG qgroup show $units $SCRATCH_MNT | grep "0/$subvolid")
	echo "qgroup values after rescan: $output" >> $seqres.full
	[ $refer -eq $(echo $output | $AWK_PROG '{ print $2 }') ] || \
		_fail "reference values don't match after rescan"
	[ $excl -eq $(echo $output | $AWK_PROG '{ print $3 }') ] || \
		_fail "exclusive values don't match after rescan"
}

#basic exceed limit testing
_limit_test_exceed()
{
	echo "=== limit exceed test ===" >> $seqres.full
	_run_btrfs_util_prog subvolume create $SCRATCH_MNT/a
	_run_btrfs_util_prog quota enable $SCRATCH_MNT
	subvolid=$(_btrfs_get_subvolid $SCRATCH_MNT a)
	_run_btrfs_util_prog qgroup limit 5M 0/$subvolid $SCRATCH_MNT
	_ddt of=$SCRATCH_MNT/a/file bs=10M count=1 >> $seqres.full 2>&1
	[ $? -ne 0 ] || _fail "quota should have limited us"
}

#basic noexceed limit testing
_limit_test_noexceed()
{
	echo "=== limit not exceed test ===" >> $seqres.full
	_run_btrfs_util_prog subvolume create $SCRATCH_MNT/a
	_run_btrfs_util_prog quota enable $SCRATCH_MNT
	subvolid=$(_btrfs_get_subvolid $SCRATCH_MNT a)
	_run_btrfs_util_prog qgroup limit 5M 0/$subvolid $SCRATCH_MNT
	_ddt of=$SCRATCH_MNT/a/file bs=4M count=1 >> $seqres.full 2>&1
	[ $? -eq 0 ] || _fail "should have been allowed to write"
}

units=`_btrfs_qgroup_units`

_scratch_mkfs > /dev/null 2>&1
_scratch_mount
_basic_test
_scratch_unmount
_check_scratch_fs

_scratch_mkfs > /dev/null 2>&1
_scratch_mount
_rescan_test
_scratch_unmount
_check_scratch_fs

_scratch_mkfs > /dev/null 2>&1
_scratch_mount
_limit_test_exceed
_scratch_unmount
_check_scratch_fs

_scratch_mkfs > /dev/null 2>&1
_scratch_mount
_limit_test_noexceed

# success, all done
echo "Silence is golden"
status=0
exit
