util-linux-ng-2.17.2/tests/functions.sh - nest-learning-thermostat/5.1.5/util-linux-ng - Git at Google

 #
 # Copyright (C) 2007 Karel Zak <kzak@redhat.com>
 #
 # This file is part of util-linux-ng.
 #
 # This file 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; either version 2 of the License, or
 # (at your option) any later version.
 #
 # This file is distributed in the hope that it will 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.
 #


 function ts_abspath {
 	cd $1
 	pwd
 }

 function ts_skip_subtest {
 	echo " IGNORE ($1)"
 }

 function ts_skip {
 	ts_skip_subtest "$1"
 	if [ -n "$2" -a -b "$2" ]; then
 		ts_device_deinit "$2"
 	fi
 	exit 0
 }

 function ts_skip_nonroot {
 	if [ $UID -ne 0 ]; then
 		ts_skip "not root permissions"
 	fi
 }

 function ts_failed_subtest {
 	if [ x"$1" == x"" ]; then
 		echo " FAILED ($TS_NS)"
 	else
 		echo " FAILED ($1)"
 	fi
 }

 function ts_failed {
 	ts_failed_subtest "$1"
 	exit 1
 }

 function ts_ok_subtest {
 	if [ x"$1" == x"" ]; then
 		echo " OK"
 	else
 		echo " OK ($1)"
 	fi
 }

 function ts_ok {
 	ts_ok_subtest "$1"
 	exit 0
 }

 function ts_log {
 	echo "$1" >> $TS_OUTPUT
 	[ "$TS_VERBOSE" == "yes" ] && echo "$1"
 }

 function ts_has_option {
 	NAME="$1"
 	ALL="$2"
 	echo -n $ALL | sed 's/ //g' | awk 'BEGIN { FS="="; RS="--" } /('$NAME'$|'$NAME'=)/ { print "yes" }'
 }

 function ts_init_env {
 	local mydir=$(ts_abspath $(dirname $0))

 	export LANG="en_US.UTF-8"

 	TS_TOPDIR=$(ts_abspath $mydir/../../)
 	TS_SCRIPT="$mydir/$(basename $0)"
 	TS_SUBDIR=$(dirname $TS_SCRIPT)
 	TS_TESTNAME=$(basename $TS_SCRIPT)
 	TS_COMPONENT=$(basename $TS_SUBDIR)

 	TS_NSUBTESTS=0
 	TS_NSUBFAILED=0

 	TS_NS="$TS_COMPONENT/$TS_TESTNAME"
 	TS_SELF="$TS_SUBDIR"

 	TS_OUTDIR="$TS_TOPDIR/output/$TS_COMPONENT"
 	TS_OUTPUT="$TS_OUTDIR/$TS_TESTNAME"
 	TS_DIFFDIR="$TS_TOPDIR/diff/$TS_COMPONENT"
 	TS_DIFF="$TS_DIFFDIR/$TS_TESTNAME"
 	TS_EXPECTED="$TS_TOPDIR/expected/$TS_NS"
 	TS_MOUNTPOINT="$TS_OUTDIR/${TS_TESTNAME}-mnt"

 	TS_VERBOSE=$(ts_has_option "verbose" "$*")

 	BLKID_FILE="$TS_OUTDIR/${TS_TESTNAME}.blkidtab"

 	[ -d "$TS_OUTDIR" ]  || mkdir -p "$TS_OUTDIR"
 	[ -d "$TS_DIFFDIR" ] || mkdir -p "$TS_DIFFDIR"

 	declare -a TS_SUID_PROGS
 	declare -a TS_SUID_USER
 	declare -a TS_SUID_GROUP

 	. $TS_TOPDIR/commands.sh

 	export BLKID_FILE

 	rm -f $TS_OUTPUT
 	touch $TS_OUTPUT

 	if [ "$TS_VERBOSE" == "yes" ]; then
 		echo
 		echo "     script: $TS_SCRIPT"
 		echo "    sub dir: $TS_SUBDIR"
 		echo "    top dir: $TS_TOPDIR"
 		echo "       self: $TS_SELF"
 		echo "  test name: $TS_TESTNAME"
 		echo "  test desc: $TS_DESC"
 		echo "  component: $TS_COMPONENT"
 		echo "  namespace: $TS_NS"
 		echo "    verbose: $TS_VERBOSE"
 		echo "     output: $TS_OUTPUT"
 		echo "   expected: $TS_EXPECTED"
 		echo " mountpoint: $TS_MOUNTPOINT"
 		echo
 	fi
 }

 function ts_init_subtest {

 	TS_SUBNAME="$1"

 	TS_OUTPUT="$TS_OUTDIR/$TS_TESTNAME-$TS_SUBNAME"
 	TS_DIFF="$TS_DIFFDIR/$TS_TESTNAME-$TS_SUBNAME"
 	TS_EXPECTED="$TS_TOPDIR/expected/$TS_NS-$TS_SUBNAME"
 	TS_MOUNTPOINT="$TS_OUTDIR/${TS_TESTNAME-$TS_SUBNAME}-mnt"

 	[ $TS_NSUBTESTS -eq 0 ] && echo
 	TS_NSUBTESTS=$(( $TS_NSUBTESTS + 1 ))

 	printf "%16s: %-27s ..." "" "$TS_SUBNAME"
 }

 function ts_init {
 	local is_fake=$( ts_has_option "fake" "$*")

 	ts_init_env "$*"

 	printf "%13s: %-30s ..." "$TS_COMPONENT" "$TS_DESC"

 	[ "$is_fake" == "yes" ] && ts_skip "fake mode"
 }

 function ts_init_suid {
 	PROG="$1"
 	ct=${#TS_SUID_PROGS[*]}

 	# Save info about original setting
 	TS_SUID_PROGS[$ct]=$PROG
 	TS_SUID_USER[$ct]=$(stat --printf="%U" $PROG)
 	TS_SUID_GROUP[$ct]=$(stat --printf="%G" $PROG)

 	chown root.root $PROG &> /dev/null
 	chmod u+s $PROG &> /dev/null
 }

 function ts_gen_diff {
 	local res=0

 	if [ -s $TS_OUTPUT ]; then
 		diff -u $TS_EXPECTED $TS_OUTPUT > $TS_DIFF
 		[ -s $TS_DIFF ] && res=1
 	else
 		res=1
 	fi
 	return $res
 }

 function ts_finalize_subtest {
 	local res=0

 	if [ -s $TS_EXPECTED ]; then
 		ts_gen_diff
 		if [ $? -eq 1 ]; then
 			ts_failed_subtest "$1"
 			res=1
 		else
 			ts_ok_subtest "$1"
 		fi
 	else
 		ts_skip_subtest "output undefined"
 	fi

 	[ $res -ne 0 ] && TS_NSUBFAILED=$(( $TS_NSUBFAILED + 1 ))
 	return $res
 }

 function ts_finalize {
 	for idx in $(seq 0 $((${#TS_SUID_PROGS[*]} - 1))); do
 		PROG=${TS_SUID_PROGS[$idx]}
 		chmod a-s $PROG &> /dev/null
 		chown ${TS_SUID_USER[$idx]}.${TS_SUID_GROUP[$idx]} $PROG &> /dev/null
 	done

 	if [ $TS_NSUBTESTS -ne 0 ]; then
 		printf "%11s..."
 		if [ $TS_NSUBFAILED -ne 0 ]; then
 			ts_failed "$TS_NSUBFAILED from $TS_NSUBTESTS sub-tests"
 		else
 			ts_ok "all $TS_NSUBTESTS sub-tests PASSED"
 		fi
 	fi

 	if [ -s $TS_EXPECTED ]; then
 		ts_gen_diff
 		if [ $? -eq 1 ]; then
 			ts_failed "$1"
 		fi
 		ts_ok "$1"
 	fi

 	ts_skip "output undefined"
 }

 function ts_die {
 	ts_log "$1"
 	if [ -n "$2" ] && [ -b "$2" ]; then
 		ts_device_deinit "$2"
 		ts_fstab_clean		# for sure...
 	fi
 	ts_finalize
 }

 function ts_image_md5sum {
 	local img=${1:-"$TS_OUTDIR/${TS_TESTNAME}.img"}
 	echo $(md5sum "$img" | awk '{printf $1}') $(basename "$img")
 }

 function ts_image_init {
 	local mib=${1:-"5"}	# size in MiBs
 	local img=${2:-"$TS_OUTDIR/${TS_TESTNAME}.img"}

 	dd if=/dev/zero of="$img" bs=1M count=$mib &> /dev/null
 	echo "$img"
 	return 0
 }

 function ts_device_init {
 	local img=$(ts_image_init $1 $2)
 	local dev=$($TS_CMD_LOSETUP -s -f "$img")

 	if [ -z "$dev" ]; then
 		ts_device_deinit $dev
 		return 1		# error
 	fi

 	echo $dev
 	return 0			# succes
 }

 function ts_device_deinit {
 	local DEV="$1"

 	if [ -b "$DEV" ]; then
 		$TS_CMD_UMOUNT "$DEV" &> /dev/null
 		$TS_CMD_LOSETUP -d "$DEV" &> /dev/null
 	fi
 }

 function ts_uuid_by_devname {
 	echo $($TS_CMD_BLKID -p -s UUID -o value $1)
 }

 function ts_label_by_devname {
 	echo $($TS_CMD_BLKID -p -s LABEL -o value $1)
 }

 function ts_fstype_by_devname {
 	echo $($TS_CMD_BLKID -p -s TYPE -o value $1)
 }

 function ts_device_has {
 	local TAG="$1"
 	local VAL="$2"
 	local DEV="$3"
 	local vl=""

 	case $TAG in
 		"TYPE") vl=$(ts_fstype_by_devname $DEV);;
 		"LABEL") vl=$(ts_label_by_devname $DEV);;
 		"UUID") vl=$(ts_uuid_by_devname $DEV);;
 		*) return 1;;
 	esac

 	if [ "$vl" == "$VAL" ]; then
 		return 0
 	fi
 	return 1
 }

 function ts_device_has_uuid {
 	ts_uuid_by_devname "$1" | egrep -q '^[0-9a-z]{8}-[0-9a-z]{4}-[0-9a-z]{4}-[0-9a-z]{4}-[0-9a-z]{12}$'
 	return $?
 }

 function ts_is_mounted {
 	local DEV=$1

 	grep -q $DEV /proc/mounts && return 0

 	if [ "${DEV#/dev/loop/}" != "$DEV" ]; then
 		return grep -q "/dev/loop${DEV#/dev/loop/}" /proc/mounts
 	fi
 	return 1
 }

 function ts_swapoff {
 	local DEV="$1"

 	# swapoff doesn't exist in build tree
 	if [ ! -x "$TS_CMD_SWAPOFF" ]; then
 		ln -sf $TS_CMD_SWAPON $TS_CMD_SWAPOFF
 		REMSWAPOFF="true"
 	fi
 	LD_LIBRARY_PATH="$U_L_LIBRARY_PATH" \
 			$TS_CMD_SWAPOFF $DEV 2>&1 >> $TS_OUTPUT
 	if [ -n "$REMSWAPOFF" ]; then
 		rm -f $TS_CMD_SWAPOFF
 	fi
 }

 function ts_fstab_open {
 	echo "# <!-- util-linux-ng test entry" >> /etc/fstab
 }

 function ts_fstab_close {
 	echo "# -->" >> /etc/fstab
 }

 function ts_fstab_addline {
 	local SPEC="$1"
 	local MNT=${2:-"$TS_MOUNTPOINT"}
 	local FS=${3:-"auto"}
 	local OPT=${4:-"defaults"}

 	echo "$SPEC   $MNT   $FS   $OPT   0   0" >> /etc/fstab
 }

 function ts_fstab_add {
 	ts_fstab_open
 	ts_fstab_addline $*
 	ts_fstab_close
 }

 function ts_fstab_clean {
 	sed --in-place "
 /# <!-- util-linux-ng/!b
 :a
 /# -->/!{
   N
   ba
 }
 s/# <!-- util-linux-ng.*-->//;
 /^$/d" /etc/fstab
 }
	#
	# Copyright (C) 2007 Karel Zak <kzak@redhat.com>
	#
	# This file is part of util-linux-ng.
	#
	# This file 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; either version 2 of the License, or
	# (at your option) any later version.
	#
	# This file is distributed in the hope that it will 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.
	#


	function ts_abspath {
	cd $1
	pwd
	}

	function ts_skip_subtest {
	echo " IGNORE ($1)"
	}

	function ts_skip {
	ts_skip_subtest "$1"
	if [ -n "$2" -a -b "$2" ]; then
	ts_device_deinit "$2"
	fi
	exit 0
	}

	function ts_skip_nonroot {
	if [ $UID -ne 0 ]; then
	ts_skip "not root permissions"
	fi
	}

	function ts_failed_subtest {
	if [ x"$1" == x"" ]; then
	echo " FAILED ($TS_NS)"
	else
	echo " FAILED ($1)"
	fi
	}

	function ts_failed {
	ts_failed_subtest "$1"
	exit 1
	}

	function ts_ok_subtest {
	if [ x"$1" == x"" ]; then
	echo " OK"
	else
	echo " OK ($1)"
	fi
	}

	function ts_ok {
	ts_ok_subtest "$1"
	exit 0
	}

	function ts_log {
	echo "$1" >> $TS_OUTPUT
	[ "$TS_VERBOSE" == "yes" ] && echo "$1"
	}

	function ts_has_option {
	NAME="$1"
	ALL="$2"
	echo -n $ALL \| sed 's/ //g' \| awk 'BEGIN { FS="="; RS="--" } /('$NAME'$\|'$NAME'=)/ { print "yes" }'
	}

	function ts_init_env {
	local mydir=$(ts_abspath $(dirname $0))

	export LANG="en_US.UTF-8"

	TS_TOPDIR=$(ts_abspath $mydir/../../)
	TS_SCRIPT="$mydir/$(basename $0)"
	TS_SUBDIR=$(dirname $TS_SCRIPT)
	TS_TESTNAME=$(basename $TS_SCRIPT)
	TS_COMPONENT=$(basename $TS_SUBDIR)

	TS_NSUBTESTS=0
	TS_NSUBFAILED=0

	TS_NS="$TS_COMPONENT/$TS_TESTNAME"
	TS_SELF="$TS_SUBDIR"

	TS_OUTDIR="$TS_TOPDIR/output/$TS_COMPONENT"
	TS_OUTPUT="$TS_OUTDIR/$TS_TESTNAME"
	TS_DIFFDIR="$TS_TOPDIR/diff/$TS_COMPONENT"
	TS_DIFF="$TS_DIFFDIR/$TS_TESTNAME"
	TS_EXPECTED="$TS_TOPDIR/expected/$TS_NS"
	TS_MOUNTPOINT="$TS_OUTDIR/${TS_TESTNAME}-mnt"

	TS_VERBOSE=$(ts_has_option "verbose" "$*")

	BLKID_FILE="$TS_OUTDIR/${TS_TESTNAME}.blkidtab"

	[ -d "$TS_OUTDIR" ] \|\| mkdir -p "$TS_OUTDIR"
	[ -d "$TS_DIFFDIR" ] \|\| mkdir -p "$TS_DIFFDIR"

	declare -a TS_SUID_PROGS
	declare -a TS_SUID_USER
	declare -a TS_SUID_GROUP

	. $TS_TOPDIR/commands.sh

	export BLKID_FILE

	rm -f $TS_OUTPUT
	touch $TS_OUTPUT

	if [ "$TS_VERBOSE" == "yes" ]; then
	echo
	echo " script: $TS_SCRIPT"
	echo " sub dir: $TS_SUBDIR"
	echo " top dir: $TS_TOPDIR"
	echo " self: $TS_SELF"
	echo " test name: $TS_TESTNAME"
	echo " test desc: $TS_DESC"
	echo " component: $TS_COMPONENT"
	echo " namespace: $TS_NS"
	echo " verbose: $TS_VERBOSE"
	echo " output: $TS_OUTPUT"
	echo " expected: $TS_EXPECTED"
	echo " mountpoint: $TS_MOUNTPOINT"
	echo
	fi
	}

	function ts_init_subtest {

	TS_SUBNAME="$1"

	TS_OUTPUT="$TS_OUTDIR/$TS_TESTNAME-$TS_SUBNAME"
	TS_DIFF="$TS_DIFFDIR/$TS_TESTNAME-$TS_SUBNAME"
	TS_EXPECTED="$TS_TOPDIR/expected/$TS_NS-$TS_SUBNAME"
	TS_MOUNTPOINT="$TS_OUTDIR/${TS_TESTNAME-$TS_SUBNAME}-mnt"

	[ $TS_NSUBTESTS -eq 0 ] && echo
	TS_NSUBTESTS=$(( $TS_NSUBTESTS + 1 ))

	printf "%16s: %-27s ..." "" "$TS_SUBNAME"
	}

	function ts_init {
	local is_fake=$( ts_has_option "fake" "$*")

	ts_init_env "$*"

	printf "%13s: %-30s ..." "$TS_COMPONENT" "$TS_DESC"

	[ "$is_fake" == "yes" ] && ts_skip "fake mode"
	}

	function ts_init_suid {
	PROG="$1"
	ct=${#TS_SUID_PROGS[*]}

	# Save info about original setting
	TS_SUID_PROGS[$ct]=$PROG
	TS_SUID_USER[$ct]=$(stat --printf="%U" $PROG)
	TS_SUID_GROUP[$ct]=$(stat --printf="%G" $PROG)

	chown root.root $PROG &> /dev/null
	chmod u+s $PROG &> /dev/null
	}

	function ts_gen_diff {
	local res=0

	if [ -s $TS_OUTPUT ]; then
	diff -u $TS_EXPECTED $TS_OUTPUT > $TS_DIFF
	[ -s $TS_DIFF ] && res=1
	else
	res=1
	fi
	return $res
	}

	function ts_finalize_subtest {
	local res=0

	if [ -s $TS_EXPECTED ]; then
	ts_gen_diff
	if [ $? -eq 1 ]; then
	ts_failed_subtest "$1"
	res=1
	else
	ts_ok_subtest "$1"
	fi
	else
	ts_skip_subtest "output undefined"
	fi

	[ $res -ne 0 ] && TS_NSUBFAILED=$(( $TS_NSUBFAILED + 1 ))
	return $res
	}

	function ts_finalize {
	for idx in $(seq 0 $((${#TS_SUID_PROGS[*]} - 1))); do
	PROG=${TS_SUID_PROGS[$idx]}
	chmod a-s $PROG &> /dev/null
	chown ${TS_SUID_USER[$idx]}.${TS_SUID_GROUP[$idx]} $PROG &> /dev/null
	done

	if [ $TS_NSUBTESTS -ne 0 ]; then
	printf "%11s..."
	if [ $TS_NSUBFAILED -ne 0 ]; then
	ts_failed "$TS_NSUBFAILED from $TS_NSUBTESTS sub-tests"
	else
	ts_ok "all $TS_NSUBTESTS sub-tests PASSED"
	fi
	fi

	if [ -s $TS_EXPECTED ]; then
	ts_gen_diff
	if [ $? -eq 1 ]; then
	ts_failed "$1"
	fi
	ts_ok "$1"
	fi

	ts_skip "output undefined"
	}

	function ts_die {
	ts_log "$1"
	if [ -n "$2" ] && [ -b "$2" ]; then
	ts_device_deinit "$2"
	ts_fstab_clean # for sure...
	fi
	ts_finalize
	}

	function ts_image_md5sum {
	local img=${1:-"$TS_OUTDIR/${TS_TESTNAME}.img"}
	echo $(md5sum "$img" \| awk '{printf $1}') $(basename "$img")
	}

	function ts_image_init {
	local mib=${1:-"5"} # size in MiBs
	local img=${2:-"$TS_OUTDIR/${TS_TESTNAME}.img"}

	dd if=/dev/zero of="$img" bs=1M count=$mib &> /dev/null
	echo "$img"
	return 0
	}

	function ts_device_init {
	local img=$(ts_image_init $1 $2)
	local dev=$($TS_CMD_LOSETUP -s -f "$img")

	if [ -z "$dev" ]; then
	ts_device_deinit $dev
	return 1 # error
	fi

	echo $dev
	return 0 # succes
	}

	function ts_device_deinit {
	local DEV="$1"

	if [ -b "$DEV" ]; then
	$TS_CMD_UMOUNT "$DEV" &> /dev/null
	$TS_CMD_LOSETUP -d "$DEV" &> /dev/null
	fi
	}

	function ts_uuid_by_devname {
	echo $($TS_CMD_BLKID -p -s UUID -o value $1)
	}

	function ts_label_by_devname {
	echo $($TS_CMD_BLKID -p -s LABEL -o value $1)
	}

	function ts_fstype_by_devname {
	echo $($TS_CMD_BLKID -p -s TYPE -o value $1)
	}

	function ts_device_has {
	local TAG="$1"
	local VAL="$2"
	local DEV="$3"
	local vl=""

	case $TAG in
	"TYPE") vl=$(ts_fstype_by_devname $DEV);;
	"LABEL") vl=$(ts_label_by_devname $DEV);;
	"UUID") vl=$(ts_uuid_by_devname $DEV);;
	*) return 1;;
	esac

	if [ "$vl" == "$VAL" ]; then
	return 0
	fi
	return 1
	}

	function ts_device_has_uuid {
	ts_uuid_by_devname "$1" \| egrep -q '^[0-9a-z]{8}-[0-9a-z]{4}-[0-9a-z]{4}-[0-9a-z]{4}-[0-9a-z]{12}$'
	return $?
	}

	function ts_is_mounted {
	local DEV=$1

	grep -q $DEV /proc/mounts && return 0

	if [ "${DEV#/dev/loop/}" != "$DEV" ]; then
	return grep -q "/dev/loop${DEV#/dev/loop/}" /proc/mounts
	fi
	return 1
	}

	function ts_swapoff {
	local DEV="$1"

	# swapoff doesn't exist in build tree
	if [ ! -x "$TS_CMD_SWAPOFF" ]; then
	ln -sf $TS_CMD_SWAPON $TS_CMD_SWAPOFF
	REMSWAPOFF="true"
	fi
	LD_LIBRARY_PATH="$U_L_LIBRARY_PATH" \
	$TS_CMD_SWAPOFF $DEV 2>&1 >> $TS_OUTPUT
	if [ -n "$REMSWAPOFF" ]; then
	rm -f $TS_CMD_SWAPOFF
	fi
	}

	function ts_fstab_open {
	echo "# <!-- util-linux-ng test entry" >> /etc/fstab
	}

	function ts_fstab_close {
	echo "# -->" >> /etc/fstab
	}

	function ts_fstab_addline {
	local SPEC="$1"
	local MNT=${2:-"$TS_MOUNTPOINT"}
	local FS=${3:-"auto"}
	local OPT=${4:-"defaults"}

	echo "$SPEC $MNT $FS $OPT 0 0" >> /etc/fstab
	}

	function ts_fstab_add {
	ts_fstab_open
	ts_fstab_addline $*
	ts_fstab_close
	}

	function ts_fstab_clean {
	sed --in-place "
	/# <!-- util-linux-ng/!b
	:a
	/# -->/!{
	N
	ba
	}
	s/# <!-- util-linux-ng.*-->//;
	/^$/d" /etc/fstab
	}