#!/bin/bash
#
# Open-iSCSI Regression Test Utility
# Copyright (C) 2004 Dmitry Yusupov
# maintained by open-iscsi@googlegroups.com
#
# This program 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 program 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.
#
# See the file COPYING included with this distribution for more details.
#

PATH=".:${PATH}"
FSTYPE="${FSTYPE:-ext3}"
DEFAULTMOUNTOPTS='-o _netdev'
[ -z "${MOUNTOPTS}" ] && MOUNTOPTS="${DEFAULTMOUNTOPTS}"
# to avoid mount looking for fstype
MOUNTOPTIONS="${MOUNTOPTIONS} -t ${FSTYPE}"
MKFSCMD="${MKFSCMD:-mkfs.${FSTYPE}} ${MKFSOPTS}"
PARTITIONSUFFIX="1"
BONNIEPARAMS="${BONNIEPARAMS:--r0 -n10:0:0 -s16 -uroot -f -q}"

trap regress_signal INT QUIT TERM
regress_signal() {
    printf "\nterminating, restore defaults: "
	# use the other function to clean up
	imm_data_en="Yes"
	initial_r2t_en="No"
	hdrdgst_en="None,CRC32C"
	datdgst_en="None,CRC32C"
	c="${iscsiadm} -m node -T $target -p $ipnr -o update"
	first_burst="$((256*1024))"
	max_burst="$((16*1024*1024-1024))"
	max_recv_dlength="$((128*1024))"
	max_r2t="1"
	update_cfg
	${iscsiadm} -m node -T $target -p $ipnr --logout 2>/dev/null >/dev/null
    printf "done\n"
    exit 0
}

function update_cfg() {
	c="${iscsiadm} -m node -T $target -p $ipnr -o update"
	$c -n node.session.iscsi.ImmediateData -v $imm_data_en
	$c -n node.session.iscsi.InitialR2T -v $initial_r2t_en
	$c -n node.conn[0].iscsi.HeaderDigest -v $hdrdgst_en
	$c -n node.conn[0].iscsi.DataDigest -v $datdgst_en
	$c -n node.session.iscsi.FirstBurstLength -v $first_burst
	$c -n node.session.iscsi.MaxBurstLength -v $max_burst
	$c -n node.conn[0].iscsi.MaxRecvDataSegmentLength -v $max_recv_dlength
	$c -n node.session.iscsi.MaxOutstandingR2T -v $max_r2t
}

function disktest_run() {
	bsizes="512 1024 2048 4096 8192 16384 32768 65536 131072 1000000"
	test "x$bsize" != x && bsizes=$bsize
	test "x$bsize" = xbonnie && return 0;
	for bs in $bsizes; do
		echo -n "disktest -T2 -K8 -B$bs -r -ID $device: "
		#if ! ${disktest} -T2 -K8 -B$bs -r -ID $device >/dev/null; then
		if ! ${disktest} -T2 -K8 -B$bs -r -ID $device; then
			echo "FAILED"
			return 1;
		fi
		echo "PASSED"
		#echo -n "disktest -T2 -K8 -B$bs -E16 -w -ID $device: "
		#if ! ${disktest} -T2 -K8 -B$bs -E16 -w -ID $device >/dev/null;then
		echo -n "disktest -T2 -K8 -B$bs -E16 -ID $device: "
		if ! ${disktest} -T2 -K8 -B$bs -E16 -ID $device; then
			echo "FAILED"
			return 1;
		fi
		echo "PASSED"
	done
	return 0;
}

function fdisk_run() {
	echo -n "sfdisk -qf $device: "
	#sfdisk -Lqf $device >/dev/null 2>/dev/null <<-EOF
	sfdisk -Lqf $device <<-EOF
	,
	quit
	EOF
	rc=$?
	if [ $rc -ne 0 ]; then
		echo "FAILED"
		return 1;
	fi
	echo "PASSED"
	return 0;
}

function mkfs_run() {
	echo -n "${MKFSCMD} $device_partition: "
	#if ! ${MKFSCMD} $device_partition 2>/dev/null >/dev/null; then
	if ! ${MKFSCMD} $device_partition ; then
		echo "FAILED"
		return 1;
	fi
	echo "PASSED"
	return 0;
}

function bonnie_run() {
	dir="/tmp/iscsi.bonnie.regression.$record.$RANDOM"
	umount $dir 2>/dev/null >/dev/null
	rm -rf $dir; mkdir $dir
	echo -n "mount $dir: "
	if ! mount ${MOUNTOPTIONS} $device_partition $dir; then
		echo "FAILED"
		return 1;
	fi
	echo "PASSED"
	echo -n "bonnie++ ${BONNIEPARAMS}: "
	pushd $dir >/dev/null
	${bonnie} ${BONNIEPARAMS} 2>/dev/null >/dev/null
	rc=$?
	popd >/dev/null
	umount $dir 2>/dev/null >/dev/null
	rmdir ${dir}
	if [ $rc -ne 0 ]; then
		echo "FAILED"
		return 1;
	fi
	echo "PASSED"
	return 0;
}

function fatal() {
	echo "regression.sh: $1"
	echo "Usage: regression.sh [-f | <targetname> <ipnumber#> ] <device> [test#[:#]] [bsize]"
	exit 1
}

############################ main ###################################

disktest=`which disktest`
iscsiadm=`which iscsiadm`
bonnie=`which bonnie++`
datfile=`dirname $0`"/regression.dat"
test ! -e ${datfile} && fatal "can not find regression.dat"
test ! -e ${disktest} && fatal "can not find disktest"
test ! -e ${iscsiadm} && fatal "can not find iscsiadm"
test ! -e ${bonnie} && fatal "can not find bonnie++"

if test x$1 = "x-f" -o x$1 = "x--format"; then
	test x$2 = x && fatal "SCSI device parameter error"
	device=$2
else
	test x$1 = x && fatal "target name parameter error"
	test x$2 = x && fatal "ipnumber parameter error"
	test x$3 = x && fatal "SCSI device parameter error"

	target="$1"
	ipnr="$2"
	device=$3
fi

device_dir="$(dirname ${device})"
device_partition=''
case "${device_dir}" in
	# /dev/sdaX
	/dev) device_partition="${device}1" ;;
	# /dev/disk/by-id/scsi-${ID_SERIAL}-part1
	# where ID_SERIAL is SCSI disk SERIAL from scsi_id
	/dev/disk/by-id|/dev/disk/by-path) device_partition="${device}-part1" ;;
	# upcoming stuff
	/dev/iscsi/*) device_partition="${device}-part1" ;;
esac

if test x$1 = "x-f" -o x$1 = "x--format"; then
	mkfs_run
	exit
fi

if [ -z "${device_partition}" ]; then
	echo 'Unable to find device name for first partition.' >&2
	exit 1
fi

test "x$4" != x && begin="$4"
test "x$5" != x && bsize="$5"

if test "x$begin" != "x"; then
	end="${begin/*:}"
	begin="${begin/:*}"
fi

# don't say we didn't warn you
if [ -z "${SKIP_WARNING}" ]; then
	cat <<-EOF
	BIG FAT WARNING!
	
	Open-iSCSI Regression Test Suite is about to start. It is going
	to use "$device" for its testing. iSCSI session could be re-opened
	during the tests several times and as the result device name could
	not match provided device name if some other SCSI activity happened
	during the test.
	
	Are you sure you want to continue? [y/n]:
	EOF
	read line
	if test x$line = xn -o x$line = xN -o x$line = xno -o x$line = xNO; then
		echo "aborting..."
		exit
	fi
fi

i=0
cat ${datfile} | while read line; do
	if echo $line | grep "^#" >/dev/null; then continue; fi
	if echo $line | grep "^$" >/dev/null; then continue; fi
	if test x$begin != x; then
		if test x$begin != x$i -a x$end = x; then
			let i=i+1
			continue
		elif test x$begin != x -a x$end != x; then
			if test $i -lt $begin -o $i -gt $end; then
				let i=i+1
				continue
			fi
		fi
	fi
	imm_data_en=`echo $line | awk '/^[YesNo]+/ {print $1}'`
	if test x$imm_data_en = x; then continue; fi
	initial_r2t_en=`echo $line | awk '{print $2}'`
	hdrdgst_en=`echo $line | awk '{print $3}'`
	datdgst_en=`echo $line | awk '{print $4}'`
	first_burst=`echo $line | awk '{print $5}'`
	max_burst=`echo $line | awk '{print $6}'`
	max_recv_dlength=`echo $line | awk '{print $7}'`
	max_r2t=`echo $line | awk '{print $8}'`
	# ensure we are logged out
	${iscsiadm} -m node -T $target -p $ipnr --logout 2>/dev/null >/dev/null
	# set parameters for next run
	update_cfg
	echo "================== TEST #$i BEGIN ===================="
	echo "ImmediateData = $imm_data_en"
	echo "InitialR2T = $initial_r2t_en"
	echo "HeaderDigest = $hdrdgst_en"
	echo "DataDigest = $datdgst_en"
	echo "FirstBurstLength = $first_burst"
	echo "MaxBurstLength = $max_burst"
	echo "MaxRecvDataSegmentLength = $max_recv_dlength"
	echo "MaxOutstandingR2T = $max_r2t"
	# login for new test
	# catch errors on this
	if ! ${iscsiadm} -m node -T $target -p $ipnr --login; then break; fi
	while [ ! -e $device ] ; do sleep 1 ; done
	if ! disktest_run; then break; fi
	if ! fdisk_run; then break; fi
	if ! mkfs_run; then break; fi
	if ! bonnie_run; then break; fi
	let i=i+1
done
regress_signal
echo
echo "===================== THE END ========================"