#!/bin/bash
# Benchmark a number of kernel builds
P=kernbench
DEFAULT_VERSION=4.4
. $SHELLPACK_INCLUDE/common.sh
TIME_CMD=`which time`
if [ "$TIME_CMD" = "" ]; then
        TIMEFORMAT="%2Uuser %2Ssystem %Relapsed %P%%CPU"
        TIME_CMD="time"
fi
ITERATIONS=5

# Basic argument parser
TASKSET_SERVER=
TASKSET_CLIENT=
TASKSET_ALL=
SERVERSIDE_COMMAND=none
SERVERSIDE_NAME=`date +%Y%m%d-%H%M-%S`

while [ "$1" != "" ]; do
	case "$1" in
	-v)
		VERSION=$2
		shift 2
		;;
	--serverside-command)
		SERVERSIDE_COMMAND=$2
		shift 2
		;;
	--serverside-name)
		SERVERSIDE_NAME=$2
		shift 2
		;;
	--install-only)
		INSTALL_ONLY=yes
		shift
		;;
	--install-force)
		INSTALL_FORCE=yes
		shift
		;;
	--min-threads)
		KERNBENCH_MIN_THREADS=$2
		shift 2
		;;
	--max-threads)
		KERNBENCH_MAX_THREADS=$2
		shift 2
		;;
	--iterations)
		KERNBENCH_ITERATIONS=$2
		shift 2
		;;
	*)
		echo Unrecognised option: $1
		shift
	esac
done
if [ "$TASKSET_SERVER" != "" ]; then
	echo TASKSET_SERVER: $TASKSET_SERVER
	echo TASKSET_CLIENT: $TASKSET_CLIENT
fi
if [ -z "$VERSION" ]; then
	VERSION=$DEFAULT_VERSION
fi

# Include monitor hooks
. $SHELLPACK_INCLUDE/include-monitor.sh

WEB_LOCATION=https://www.kernel.org/pub/linux/kernel/v4.x/linux-${VERSION}.tar.gz
MIRROR_LOCATION=$WEBROOT/kernbench/linux-$VERSION.tar.gz
THREADS=$(($NUMCPUS))

if [ "$INSTALL_ONLY" = "yes" ]; then
pushd $SHELLPACK_DATA > /dev/null
rm -f ./linux-$VERSION.tar.gz
sources_fetch $WEB_LOCATION $MIRROR_LOCATION ./linux-$VERSION.tar.gz
tar xf linux-$VERSION.tar.gz || die Failed to extract
cd linux-$VERSION || die Unexpected layout
	echo kernbenc installed only as requested.
	exit $SHELLPACK_SUCCESS
fi

# Configure
cd $SHELLPACK_DATA/linux-$VERSION
yes '' | make defconfig > /dev/null 2>&1 || die Failed to make defconfig
make -j$THREADS clean > /dev/null

if [ "`uname -m`" = "aarch64" ]; then
	echo Removing broken entry from aarch64 defconfig
	sed -i -e '/CONFIG_CRYPTO_CRC32_ARM64/d' .config
	echo "# CONFIG_CRYPTO_CRC32_ARM64 is not set" >> .config
	yes '' | make oldconfig
fi

echo Warming run
make -j$THREADS vmlinux > /dev/null 2>&1 || die Failed to build vmlinux
make clean >/dev/null

# Build pass
NR_THREADS=$KERNBENCH_MIN_THREADS
if [ "$NR_THREADS" = "" ]; then
	NR_THREADS=1
fi
THREADS=$NR_THREADS
NR_THREADS=$((NR_THREADS*2))
if [ "$MMTESTS_THREAD_CUTOFF" != "" ]; then
	echo Forcing KERNBENCH_MAX_THREADS to $MMTESTS_THREAD_CUTOFF
	KERNBENCH_MAX_THREADS=$MMTESTS_THREAD_CUTOFF
fi
while [ $NR_THREADS -le $KERNBENCH_MAX_THREADS ]; do
	THREADS="$THREADS $NR_THREADS"
	NR_THREADS=$((NR_THREADS*2))
done
if [ `echo $THREADS | awk '{print $NF}'` -ne $KERNBENCH_MAX_THREADS ]; then
	THREADS="$THREADS $KERNBENCH_MAX_THREADS"
fi
for NR_THREADS in $THREADS; do
	if [ $NR_THREADS -gt $KERNBENCH_MAX_THREADS ]; then
		NR_THREADS=$KERNBENCH_MAX_THREADS
	fi
	mmtests_activity process $NR_THREADS/$KERNBENCH_MAX_THREADS
	monitor_pre_hook $LOGDIR_RESULTS $NR_THREADS
for ITERATION in `seq 1 $KERNBENCH_ITERATIONS`; do
	mmtests_activity iteration $ITERATION
		echo Starting threads $NR_THREADS/$KERNBENCH_MAX_THREADS iteration $ITERATION/$KERNBENCH_ITERATIONS
		sync

		save_rc $TIME_CMD make -j$NR_THREADS vmlinux 2>> $LOGDIR_RESULTS/kernbench-${NR_THREADS}-$ITERATION.time > /dev/null
		grep elapsed $LOGDIR_RESULTS/kernbench-${NR_THREADS}-$ITERATION.time``
		recover_rc
		if [ $? -ne 0 ]; then
			die Failed to run kernbench
		fi

		# cleanup
		make clean >/dev/null
sync
done
	monitor_post_hook $LOGDIR_RESULTS $NR_THREADS
done

# Cleanup
rm -rf $SHELLPACK_DATA/*

# clean up the tree to save space
popd > /dev/null
rm -rf $TMPDIR/kernbench

exit $SHELLPACK_SUCCESS
#### Description Benchmark based on kernel builds
#### Details kernbench-bench 39
