# Common settings and functions for the various resize_ffs tests.
# 

# called from atf_init_test_cases
proc setupvars {
	setvar IMG = 'fsimage'
	setvar TDBASE64 = "$(atf_get_srcdir)/testdata.tar.gz.base64"
	setvar GOODMD5 = "$(atf_get_srcdir)/testdata.md5"
	# set BYTESWAP to opposite-endian.
	if test $(sysctl -n hw.byteorder) = "1234" {
		setvar BYTESWAP = 'be'
	} else {
		setvar BYTESWAP = 'le'
	}
}

# test_case() taken from the tests/ipf/h_common.sh
# Used to declare the atf goop for a test.
proc test_case {
	local name="${1}"; shift
	local check_function="${1}"; shift

	atf_test_case ${name} cleanup
	eval "${name}_head() { \
		atf_set "require.user" "root" ; \
		atf_set "require.progs" "rump_ffs" ; \
	}"
	eval "${name}_body() { \
		${check_function} " ${@} "; \
	}"
	eval "${name}_cleanup() { \
		umount -f mnt  ; \
		: reset error ; \
	}"
}

# Used to declare the atf goop for a test expected to fail.
proc test_case_xfail {
	local name="${1}"; shift
	local reason="${1}"; shift
	local check_function="${1}"; shift

	atf_test_case ${name} cleanup
	eval "${name}_head() { \
		atf_set "require.user" "root" ; \
	}"
	eval "${name}_body() { \
		atf_expect_fail "${reason}" ; \
		${check_function} " ${@} "; \
	}"
	eval "${name}_cleanup() { \
		umount -f mnt  ; \
		: reset error ; \
	}"
}

# copy_data requires the mount already done;  makes one copy of the test data
proc copy_data {
	uudecode -p ${TDBASE64} | shell {cd mnt; tar xzf - -s/testdata/TD$1/}
}

proc copy_multiple {
	local i
	for i in $(seq $1) {
		copy_data $i
	}
}

# remove_data removes one directory worth of test data; the purpose
# is to ensure data exists near the end of the fs under test.
proc remove_data {
	rm -rf mnt/TD$1
}

proc remove_multiple {
	local i
	for i in $(seq $1) {
		remove_data $i
	}
}

# verify that the data in a particular directory is still OK
# generated md5 file doesn't need explicit cleanup thanks to ATF
proc check_data {
	shell {cd mnt/TD$1 && md5 *} > TD$1.md5
	atf_check diff -u ${GOODMD5} TD$1.md5
}

# supply begin and end arguments
proc check_data_range {
	local i
	for i in $(seq $1 $2) {
		check_data $i
	}
}


proc resize_ffs {
	echo "in resize_ffs:" ${@}
	local bs=$1
	local fragsz=$2
	local osize=$3
	local nsize=$4
	local fslevel=$5
	local numdata=$6
	local swap=$7
	mkdir -p mnt
	echo "bs is ${bs} numdata is ${numdata}"
	echo "****resizing fs with blocksize ${bs}"

	# we want no more than 16K/inode to allow test files to copy.
	local fpi=$((fragsz * 4))
	local i
	if test $fpi -gt 16384 {
		setvar i = ""-i 16384""
	}
	if test x$swap != x {
		newfs -B ${BYTESWAP} -O${fslevel} -b ${bs} -f ${fragsz} \
			-s ${osize} ${i} -F ${IMG}
	} else {
		newfs -O${fslevel} -b ${bs} -f ${fragsz} -s ${osize} ${i} \
			-F ${IMG}
	}

	# we're specifying relative paths, so rump_ffs warns - ignore.
	atf_check -s exit:0 -e ignore rump_ffs ${IMG} mnt
	copy_multiple ${numdata}

	if test ${nsize} -lt ${osize} {
	    # how much data to remove so fs can be shrunk
	    local remove=$((numdata-numdata*nsize/osize))
	    local dataleft=$((numdata-remove))
	    echo remove is $remove dataleft is $dataleft
	    remove_multiple ${remove}
	}

	umount mnt
	# Check that resize needed
	atf_check -s exit:0 -o ignore resize_ffs -c -y -s ${nsize} ${IMG}
	atf_check -s exit:0 -o ignore resize_ffs -y -s ${nsize} ${IMG}
	atf_check -s exit:0 -o ignore fsck_ffs -f -n -F ${IMG}
	atf_check -s exit:0 -e ignore rump_ffs ${IMG} mnt
	if test ${nsize} -lt ${osize} {
	    check_data_range $((remove + 1)) ${numdata}
	} else {
	    # checking everything because we don't delete on grow
	    check_data_range 1 ${numdata}
	}
	# Check that no resize needed
	atf_check -s exit:1 -o ignore resize_ffs -c -y -s ${nsize} ${IMG}
	umount mnt
}