#! /bin/bash
# FSQA Test No. btrfs/074
#
# Run btrfs defrag operations and remount with different compress algorithms
# simultaneously with fsstress running in background.
#
#-----------------------------------------------------------------------
# Copyright (C) 2014 Red Hat Inc. All rights reserved.
#
# 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.
#
# This program is distributed in the hope that it would 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; if not, write the Free Software Foundation,
# Inc.,  51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
#
#-----------------------------------------------------------------------
#

seq=`basename $0`
seqres=$RESULT_DIR/$seq
echo "QA output created by $seq"

here=`pwd`
tmp=/tmp/$$
status=1
trap "_cleanup; exit \$status" 0 1 2 3 15

_cleanup()
{
	cd /
	rm -f $tmp.*
}

# get standard environment, filters and checks
. ./common/rc
. ./common/filter

# real QA test starts here
_supported_fs btrfs
_supported_os Linux
# we check scratch dev after each loop
_require_scratch_nocheck
_require_scratch_dev_pool 4
_btrfs_get_profile_configs

rm -f $seqres.full

run_test()
{
	local mkfs_opts=$1
	local with_compress=$2

	echo "Test $mkfs_opts with $with_compress" >>$seqres.full

	_scratch_pool_mkfs $mkfs_opts >>$seqres.full 2>&1
	# make sure we created btrfs with desired options
	if [ $? -ne 0 ]; then
		echo "mkfs $mkfs_opts failed"
		return
	fi
	_scratch_mount >>$seqres.full 2>&1

	args=`_scale_fsstress_args -p 20 -n 100 $FSSTRESS_AVOID -d $SCRATCH_MNT/stressdir`
	echo "Run fsstress $args" >>$seqres.full
	$FSSTRESS_PROG $args >/dev/null 2>&1 &
	fsstress_pid=$!

	echo -n "Start defrag worker: " >>$seqres.full
	_btrfs_stress_defrag $SCRATCH_MNT $with_compress >/dev/null 2>&1 &
	defrag_pid=$!
	echo "$defrag_pid" >>$seqres.full

	echo -n "Start remount worker: " >>$seqres.full
	_btrfs_stress_remount_compress $SCRATCH_MNT >/dev/null 2>&1 &
	remount_pid=$!
	echo "$remount_pid" >>$seqres.full

	echo "Wait for fsstress to exit and kill all background workers" >>$seqres.full
	wait $fsstress_pid
	kill $defrag_pid $remount_pid
	wait
	# wait for the defrag and remount operations to finish
	while ps aux | grep "btrfs filesystem defrag" | grep -qv grep; do
		sleep 1
	done
	while ps aux | grep "mount.*$SCRATCH_MNT" | grep -qv grep; do
		sleep 1
	done

	echo "Scrub the filesystem" >>$seqres.full
	$BTRFS_UTIL_PROG scrub start -B $SCRATCH_MNT >>$seqres.full 2>&1
	if [ $? -ne 0 ]; then
		echo "Scrub find errors in \"$mkfs_opts\" test" | tee -a $seqres.full
	fi

	_scratch_unmount
	# we called _require_scratch_nocheck instead of _require_scratch
	# do check after test for each profile config
	_check_scratch_fs
}

echo "Silence is golden"
for t in "${_btrfs_profile_configs[@]}"; do
	run_test "$t" nocompress
	run_test "$t" compress
done

status=0
exit
