SCRIPTDIR="$(readlink -f $(dirname $(type -p $0)))"
ROOTDIR="$(readlink -f $SCRIPTDIR/../../..)"
KPATCH="sudo $ROOTDIR/kpatch/kpatch"

MODULE_PREFIX="test-"
MODULE_POSTFIX=".ko"
TEST_POSTFIX="-LOADED.test"
PATCH_POSTFIX=".patch"
DISABLED_POSTFIX="${PATCH_POSTFIX}.disabled"

set -o errexit

declare -a loaded_modules

cleanup_modules() {
	for ((idx=${#loaded_modules[@]}-1 ; idx>=0 ; idx--)); do
		mod=${loaded_modules[idx]}
		$KPATCH unload $mod
	done
}

die_clean() {
	cleanup_modules
	exit 1
}

die() {
	echo "ERROR: $@" >&2
	die_clean
}

ko_to_test() {
	tmp=${1%${MODULE_POSTFIX}}${TEST_POSTFIX}
	echo ${tmp#${MODULE_PREFIX}}
}

# make sure any modules added here are disjoint
declare -a modules

for file in "${SCRIPTDIR}"/*"${TEST_POSTFIX}"; do
	name=$(basename ${file})
	skip=0
	for bname in "${blacklist[@]}"; do
		if [ "${bname}" == "${name}" ]; then
			skip=1
			break
		fi
	done
	if ! [ -e ${file/${TEST_POSTFIX}/${PATCH_POSTFIX}} ] && \
	     [ -e ${file/${TEST_POSTFIX}/${DISABLED_POSTFIX}} ]; then
		skip=1
	fi
	if [ ${skip} -eq 0 ]; then
		modules+=(${MODULE_PREFIX}${name%${TEST_POSTFIX}}${MODULE_POSTFIX})
	fi
done

for mod in "${modules[@]}"; do
	testprog=$(ko_to_test $mod)
	$SCRIPTDIR/$testprog && die "$SCRIPTDIR/$testprog succeeded before loading any modules"
done

for mod in "${modules[@]}"; do
	$KPATCH load $mod || die_clean
	loaded_modules+=($mod)
done

for mod in "${modules[@]}"; do
	testprog=$(ko_to_test $mod)
	$SCRIPTDIR/$testprog || die "$SCRIPTDIR/$testprog failed after loading modules"
done

cleanup_modules

for mod in "${modules[@]}"; do
	testprog=$(ko_to_test $mod)
	$SCRIPTDIR/$testprog && die "$SCRIPTDIR/$testprog succeeded after unloading modules"
done

exit 0