#!/bin/bash
# SPDX-License-Identifier: GPL-2.0
#
# Copyright (C) 2018 Uladzislau Rezki (Sony) <[email protected]>
#
# This is a test script for the kernel test driver to analyse vmalloc
# allocator. Therefore it is just a kernel module loader. You can specify
# and pass different parameters in order to:
# a) analyse performance of vmalloc allocations;
# b) stressing and stability check of vmalloc subsystem.
TEST_NAME="test_hmm"
DRIVER="test_hmm"
# 1 if fails
exitcode=1
# Kselftest framework requirement - SKIP code is 4.
ksft_skip=4
check_test_requirements()
{
uid=$(id -u)
if [ $uid -ne 0 ]; then
echo "$0: Must be run as root"
exit $ksft_skip
fi
if ! which modprobe > /dev/null 2>&1; then
echo "$0: You need modprobe installed"
exit $ksft_skip
fi
if ! modinfo $DRIVER > /dev/null 2>&1; then
echo "$0: You must have the following enabled in your kernel:"
echo "CONFIG_TEST_HMM=m"
exit $ksft_skip
fi
}
load_driver()
{
if [ $# -eq 0 ]; then
modprobe $DRIVER > /dev/null 2>&1
else
if [ $# -eq 2 ]; then
modprobe $DRIVER spm_addr_dev0=$1 spm_addr_dev1=$2
> /dev/null 2>&1
else
echo "Missing module parameters. Make sure pass"\
"spm_addr_dev0 and spm_addr_dev1"
usage
fi
fi
}
unload_driver()
{
modprobe -r $DRIVER > /dev/null 2>&1
}
run_smoke()
{
echo "Running smoke test. Note, this test provides basic coverage."
load_driver $1 $2
$(dirname "${BASH_SOURCE[0]}")/hmm-tests
unload_driver
}
usage()
{
echo -n "Usage: $0"
echo
echo "Example usage:"
echo
echo "# Shows help message"
echo "./${TEST_NAME}.sh"
echo
echo "# Smoke testing"
echo "./${TEST_NAME}.sh smoke"
echo
echo "# Smoke testing with SPM enabled"
echo "./${TEST_NAME}.sh smoke <spm_addr_dev0> <spm_addr_dev1>"
echo
exit 0
}
function run_test()
{
if [ $# -eq 0 ]; then
usage
else
if [ "$1" = "smoke" ]; then
run_smoke $2 $3
else
usage
fi
fi
}
check_test_requirements
run_test $@
exit 0