#
# Copyright 2015 EMC Corp.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are
# met:
#
# * Redistributions of source code must retain the above copyright
#   notice, this list of conditions and the following disclaimer.
# * 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 COPYRIGHT HOLDERS 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 COPYRIGHT
# OWNER 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.
#
# $FreeBSD: stable/11/usr.sbin/makefs/tests/makefs_cd9660_tests.sh 322690 2017-08-19 01:45:02Z ngie $
#

# A note on specs:
# - A copy of the ISO-9660 spec can be found here:
#   http://www.ecma-international.org/publications/files/ECMA-ST/Ecma-119.pdf
# - Any references to `rockridge` are referring to the `Rock Ridge` extensions
#   of the ISO-9660 spec. A copy of the draft `IEEE-P1282` spec can be found
#   here:
#   http://www.ymi.com/ymi/sites/default/files/pdf/Rockridge.pdf

setglobal MAKEFS = '"makefs -t cd9660'"
setglobal MOUNT = '"mount_cd9660'"

source "$(dirname "$0")/makefs_tests_common.sh"

proc common_cleanup
{
	if ! setglobal test_md_device = $[cat $TEST_MD_DEVICE_FILE] {
		echo "$TEST_MD_DEVICE_FILE could not be opened; has an md(4) device been attached?"
		return
	}

	umount -f /dev/$test_md_device || :
	mdconfig -d -u $test_md_device || :
}

proc check_base_iso9660_image_contents
{
	# Symlinks are treated like files when rockridge support isn't
	# specified
	check_image_contents @Argv -X c

	atf_check -e empty -o empty -s exit:0 test -L $TEST_INPUTS_DIR/c
	atf_check -e empty -o empty -s exit:0 test -f $TEST_MOUNT_DIR/c
}

atf_test_case D_flag cleanup
proc D_flag_body
{
	atf_skip "makefs crashes with SIGBUS with dupe mtree entries; see FreeBSD bug # 192839"

	create_test_inputs

	atf_check -e empty -o save:$TEST_SPEC_FILE -s exit:0 \
	    mtree -cp $TEST_INPUTS_DIR
	atf_check -e empty -o not-empty -s exit:0 \
	    $MAKEFS -F $TEST_SPEC_FILE -M 1m $TEST_IMAGE $TEST_INPUTS_DIR

	atf_check -e empty -o empty -s exit:0 \
	    cp $TEST_SPEC_FILE spec2.mtree
	atf_check -e empty -o save:dupe_$TEST_SPEC_FILE -s exit:0 \
	    cat $TEST_SPEC_FILE spec2.mtree

	atf_check -e empty -o not-empty -s not-exit:0 \
	    $MAKEFS -F dupe_$TEST_SPEC_FILE -M 1m $TEST_IMAGE $TEST_INPUTS_DIR
	atf_check -e empty -o not-empty -s exit:0 \
	    $MAKEFS -D -F dupe_$TEST_SPEC_FILE -M 1m $TEST_IMAGE $TEST_INPUTS_DIR
}
proc D_flag_cleanup
{
	common_cleanup
}

atf_test_case F_flag cleanup
proc F_flag_body
{
	create_test_inputs

	atf_check -e empty -o save:$TEST_SPEC_FILE -s exit:0 \
	    mtree -cp $TEST_INPUTS_DIR

	atf_check -e empty -o empty -s exit:0 \
	    $MAKEFS -F $TEST_SPEC_FILE -M 1m $TEST_IMAGE $TEST_INPUTS_DIR

	mount_image
	check_base_iso9660_image_contents
}
proc F_flag_cleanup
{
	common_cleanup
}

atf_test_case from_mtree_spec_file cleanup
proc from_mtree_spec_file_body
{
	create_test_inputs

	atf_check -e empty -o save:$TEST_SPEC_FILE -s exit:0 \
	    mtree -c -k $DEFAULT_MTREE_KEYWORDS -p $TEST_INPUTS_DIR
	cd $TEST_INPUTS_DIR
	atf_check -e empty -o empty -s exit:0 \
	    $MAKEFS $TEST_IMAGE $TEST_SPEC_FILE
	cd -

	mount_image
	check_base_iso9660_image_contents
}
proc from_mtree_spec_file_cleanup
{
	common_cleanup
}

atf_test_case from_multiple_dirs cleanup
proc from_multiple_dirs_body
{
	setglobal test_inputs_dir2 = "$TMPDIR/inputs2"

	create_test_inputs

	atf_check -e empty -o empty -s exit:0 mkdir -p $test_inputs_dir2
	atf_check -e empty -o empty -s exit:0 \
	    touch $test_inputs_dir2/multiple_dirs_test_file

	atf_check -e empty -o empty -s exit:0 \
	    $MAKEFS $TEST_IMAGE $TEST_INPUTS_DIR $test_inputs_dir2

	mount_image
	check_base_iso9660_image_contents -d $test_inputs_dir2
}
proc from_multiple_dirs_cleanup
{
	common_cleanup
}

atf_test_case from_single_dir cleanup
proc from_single_dir_body
{
	create_test_inputs

	atf_check -e empty -o empty -s exit:0 \
	    $MAKEFS $TEST_IMAGE $TEST_INPUTS_DIR

	mount_image
	check_base_iso9660_image_contents
}
proc from_single_dir_cleanup
{
	common_cleanup
}

atf_test_case o_flag_allow_deep_trees cleanup
proc o_flag_allow_deep_trees_body
{
	create_test_inputs

	# Make sure the "more than 8 levels deep" requirement is met.
	atf_check -e empty -o empty -s exit:0 \
	    mkdir -p $TEST_INPUTS_DIR/a/b/c/d/e/f/g/h/i/j

	atf_check -e empty -o empty -s exit:0 \
	    $MAKEFS -o allow-deep-trees $TEST_IMAGE $TEST_INPUTS_DIR

	mount_image
	check_base_iso9660_image_contents
}
proc o_flag_allow_deep_trees_cleanup
{
	common_cleanup
}

atf_test_case o_flag_allow_max_name cleanup
proc o_flag_allow_max_name_body
{
	atf_expect_fail "-o allow-max-name doesn't appear to be implemented on FreeBSD's copy of makefs [yet]"

	create_test_inputs

	setglobal long_path = "$TEST_INPUTS_DIR/$[jot -s '' -b 0 37]"

	# Make sure the "37 char name" limit requirement is met.
	atf_check -e empty -o empty -s exit:0 touch $long_path

	atf_check -e empty -o empty -s exit:0 \
	    $MAKEFS -o allow-max-name $TEST_IMAGE $TEST_INPUTS_DIR

	mount_image
	check_base_iso9660_image_contents
}
proc o_flag_allow_max_name_cleanup
{
	common_cleanup
}

atf_test_case o_flag_isolevel_1 cleanup
proc o_flag_isolevel_1_body
{
	atf_expect_fail "this testcase needs work; the filenames generated seem incorrect/corrupt"

	create_test_inputs

	atf_check -e empty -o empty -s exit:0 \
	    $MAKEFS -o isolevel=1 $TEST_IMAGE $TEST_INPUTS_DIR

	mount_image
	check_base_iso9660_image_contents
}
proc o_flag_isolevel_1_cleanup
{
	common_cleanup
}

atf_test_case o_flag_isolevel_2 cleanup
proc o_flag_isolevel_2_body
{
	create_test_inputs

	atf_check -e empty -o empty -s exit:0 \
	    $MAKEFS -o isolevel=2 $TEST_IMAGE $TEST_INPUTS_DIR

	mount_image
	check_base_iso9660_image_contents
}
proc o_flag_isolevel_2_cleanup
{
	common_cleanup
}

atf_test_case o_flag_isolevel_3 cleanup
proc o_flag_isolevel_3_body
{
	create_test_inputs

	# XXX: isolevel=3 isn't implemented yet. See FreeBSD bug # 203645
	if true {
	atf_check -e match:'makefs: ISO Level 3 is greater than 2\.' -o empty -s not-exit:0 \
	    $MAKEFS -o isolevel=3 $TEST_IMAGE $TEST_INPUTS_DIR
	} else {
	atf_check -e empty -o empty -s exit:0 \
	    $MAKEFS -o isolevel=3 $TEST_IMAGE $TEST_INPUTS_DIR
	mount_image
	check_base_iso9660_image_contents
	}
}
proc o_flag_isolevel_3_cleanup
{
	common_cleanup
}

atf_test_case o_flag_preparer
proc o_flag_preparer_head
{
	atf_set "require.progs" "strings"
}
proc o_flag_preparer_body
{
	create_test_dirs

	setglobal preparer = ''My Very First ISO''
	setglobal preparer_uppercase = $[echo $preparer | tr '[[:lower:]]' '[[:upper:]]]

	atf_check -e empty -o empty -s exit:0 touch $TEST_INPUTS_DIR/dummy_file
	atf_check -e empty -o empty -s exit:0 \
	    $MAKEFS -o preparer="$preparer" $TEST_IMAGE $TEST_INPUTS_DIR
	atf_check -e empty -o match:"$preparer_uppercase" -s exit:0 \
	    strings $TEST_IMAGE
}

atf_test_case o_flag_publisher
proc o_flag_publisher_head
{
	atf_set "require.progs" "strings"
}
proc o_flag_publisher_body
{
	create_test_dirs

	setglobal publisher = ''My Super Awesome Publishing Company LTD''
	setglobal publisher_uppercase = $[echo $publisher | tr '[[:lower:]]' '[[:upper:]]]

	atf_check -e empty -o empty -s exit:0 touch $TEST_INPUTS_DIR/dummy_file
	atf_check -e empty -o empty -s exit:0 \
	    $MAKEFS -o publisher="$publisher" $TEST_IMAGE $TEST_INPUTS_DIR
	atf_check -e empty -o match:"$publisher_uppercase" -s exit:0 \
	    strings $TEST_IMAGE
}

atf_test_case o_flag_rockridge cleanup
proc o_flag_rockridge_body
{
	create_test_dirs

	# Make sure the "more than 8 levels deep" requirement is met.
	atf_check -e empty -o empty -s exit:0 \
	    mkdir -p $TEST_INPUTS_DIR/a/b/c/d/e/f/g/h/i/j

	# Make sure the "pathname larger than 255 chars" requirement is met.
	#
	# $long_path's needs to be nested in a directory, as creating it
	# outright as a 256 char filename via touch will fail with ENAMETOOLONG
	setglobal long_path = "$TEST_INPUTS_DIR/$[jot -s '/' -b $[jot -s '' -b 0 64] 4]"
	atf_check -e empty -o empty -s exit:0 mkdir -p $[dirname $long_path]
	atf_check -e empty -o empty -s exit:0 touch $long_path

	atf_check -e empty -o empty -s exit:0 \
	    $MAKEFS -o rockridge $TEST_IMAGE $TEST_INPUTS_DIR

	mount_image
	check_image_contents -X .rr_moved

	# .rr_moved is a special directory created when you have deep directory
	# trees with rock ridge extensions on
	atf_check -e empty -o empty -s exit:0 \
	    test -d $TEST_MOUNT_DIR/.rr_moved
}
proc o_flag_rockridge_cleanup
{
	common_cleanup
}

atf_test_case o_flag_rockridge_dev_nodes cleanup
proc o_flag_rockridge_dev_nodes_head
{
	atf_set "descr" "Functional tests to ensure that dev nodes are handled properly with rockridge extensions (NetBSD kern/48852; FreeBSD bug 203648)"
}
proc o_flag_rockridge_dev_nodes_body
{
	create_test_dirs

	shell {tar -cvf - -C /dev null && touch .tar_ok} | \
	atf_check -e not-empty -o empty -s exit:0 tar -xvf - -C $TEST_INPUTS_DIR

	atf_check -e empty -o empty -s exit:0 test -c $TEST_INPUTS_DIR/null
	atf_check -e empty -o empty -s exit:0 test -f .tar_ok

	atf_check -e empty -o empty -s exit:0 \
	    $MAKEFS -o rockridge $TEST_IMAGE $TEST_INPUTS_DIR

	mount_image
	check_image_contents
}
proc o_flag_rockridge_dev_nodes_cleanup
{
	common_cleanup
}

proc atf_init_test_cases
{
	atf_add_test_case D_flag
	atf_add_test_case F_flag

	atf_add_test_case from_mtree_spec_file
	atf_add_test_case from_multiple_dirs
	atf_add_test_case from_single_dir

	atf_add_test_case o_flag_allow_deep_trees
	atf_add_test_case o_flag_allow_max_name
	atf_add_test_case o_flag_isolevel_1
	atf_add_test_case o_flag_isolevel_2
	atf_add_test_case o_flag_isolevel_3
	atf_add_test_case o_flag_preparer
	atf_add_test_case o_flag_publisher
	atf_add_test_case o_flag_rockridge
	atf_add_test_case o_flag_rockridge_dev_nodes
}