# $NetBSD: t_vnd.sh,v 1.9 2016/07/29 05:23:24 pgoyette Exp $
#
# Copyright (c) 2006, 2007, 2008 The NetBSD Foundation, Inc.
# All rights reserved.
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
# ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
# TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#

#
# Verifies that vnd works with files stored in tmpfs.
#

# Begin FreeBSD
setvar MD_DEVICE_FILE = 'md.device'
# End FreeBSD

atf_test_case basic cleanup
proc basic_head {
	atf_set "descr" "Verifies that vnd works with files stored in tmpfs"
	atf_set "require.user" "root"
}
proc basic_body {
	test_mount

	atf_check -s eq:0 -o ignore -e ignore \
	    dd if=/dev/zero of=disk.img bs=1m count=10
	# Begin FreeBSD
	if true {
		atf_check -s eq:0 -o empty -e empty mkdir mnt
		atf_check -s eq:0 -o empty -e empty mdmfs -F disk.img md mnt
		setvar md_dev = $(df mnt | awk 'NR != 1 { print $1 }' | xargs basename)
		atf_check test -c /dev/$md_dev # Sanity check
		echo -n $md_dev > $TMPDIR/$MD_DEVICE_FILE
	} else {
	# End FreeBSD
	atf_check -s eq:0 -o empty -e empty vndconfig /dev/vnd3 disk.img

	atf_check -s eq:0 -o ignore -e ignore newfs /dev/rvnd3a

	atf_check -s eq:0 -o empty -e empty mkdir mnt
	atf_check -s eq:0 -o empty -e empty mount /dev/vnd3a mnt
	# Begin FreeBSD
	}
	# End FreeBSD

	echo "Creating test files"
	for f in $(jot -w %u 100 | uniq) {
		jot 1000 >mnt/${f} || atf_fail "Failed to create file ${f}"
	}

	echo "Verifying created files"
	for f in $(jot -w %u 100 | uniq) {
		test $(md5 mnt/${f} | cut -d ' ' -f 4) = \
		    53d025127ae99ab79e8502aae2d9bea6 || \
		    atf_fail "Invalid checksum for file ${f}"
	}

	atf_check -s eq:0 -o empty -e empty umount mnt
	atf_check -s eq:0 -o empty -e empty vndconfig -u /dev/vnd3

	test_unmount
	touch done
}
proc basic_cleanup {
	# Begin FreeBSD
	if setvar md_dev = $(cat $TMPDIR/$MD_DEVICE_FILE) {
		echo "Will try disconnecting $md_dev"
	} else {
		echo "$MD_DEVICE_FILE doesn't exist in $TMPDIR; returning early"
		return 0
	}
	# End FreeBSD
	if test ! -f done {
		umount mnt 2>/dev/null 1>&2
		vndconfig -u /dev/vnd3 2>/dev/null 1>&2
	}
}

proc atf_init_test_cases {
	source $(atf_get_srcdir)/../h_funcs.subr
	source $(atf_get_srcdir)/h_funcs.subr

	atf_add_test_case basic
}