selftest: cpufreq: Add support to test cpufreq modules

This patch adds support for cpufreq modules like cpufreq drivers and
cpufreq governors. The tests will insert the modules in different orders
and them perform basic cpufreq tests. The modules are then removed from
the kernel.

Signed-off-by: Viresh Kumar <viresh.kumar@linaro.org>
Signed-off-by: Shuah Khan <shuahkh@osg.samsung.com>

3 files changed, 284 insertions, 6 deletions

diff --git a/tools/testing/selftests/cpufreq/Makefile b/tools/testing/selftests/cpufreq/Makefile
index f5c6bb1965a1..80c8727dcec1 100644
--- a/tools/testing/selftests/cpufreq/Makefile
+++ b/tools/testing/selftests/cpufreq/Makefile
@@ -1,7 +1,7 @@
 all:
 
 TEST_PROGS := main.sh
-TEST_FILES := cpu.sh cpufreq.sh governor.sh
+TEST_FILES := cpu.sh cpufreq.sh governor.sh module.sh
 
 include ../lib.mk
 
diff --git a/tools/testing/selftests/cpufreq/main.sh b/tools/testing/selftests/cpufreq/main.sh
index 9ff662f67ea4..2515867ac9de 100755
--- a/tools/testing/selftests/cpufreq/main.sh
+++ b/tools/testing/selftests/cpufreq/main.sh
@@ -3,6 +3,7 @@
 source cpu.sh
 source cpufreq.sh
 source governor.sh
+source module.sh
 
 FUNC=basic      # do basic tests by default
 OUTFILE=cpufreq_selftest
@@ -12,12 +13,15 @@ CPUFREQROOT=
 
 helpme()
 {
-        printf "Usage: $0 [-h] [-to args]
+        printf "Usage: $0 [-h] [-todg args]
         [-h <help>]
         [-o <output-file-for-dump>]
         [-t <basic: Basic cpufreq testing
              suspend: suspend/resume,
-             hibernate: hibernate/resume>]
+             hibernate: hibernate/resume,
+             modtest: test driver or governor modules. Only to be used with -d or -g options>]
+        [-d <driver's module name: only with \"-t modtest>\"]
+        [-g <governor's module name: only with \"-t modtest>\"]
         \n"
         exit 2
 }
@@ -56,14 +60,14 @@ prerequisite()
 
 parse_arguments()
 {
-        while getopts ht:o: arg
+        while getopts ht:o:d:g: arg
         do
                 case $arg in
                         h) # --help
                                 helpme
                                 ;;
 
-                        t) # --func_type (Function to perform: basic, suspend, hibernate (default: basic))
+                        t) # --func_type (Function to perform: basic, suspend, hibernate, modtest (default: basic))
                                 FUNC=$OPTARG
                                 ;;
 
@@ -71,6 +75,14 @@ parse_arguments()
                                 OUTFILE=$OPTARG
                                 ;;
 
+                        d) # --driver-mod-name (Name of the driver module)
+                                DRIVER_MOD=$OPTARG
+                                ;;
+
+                        g) # --governor-mod-name (Name of the governor module)
+                                GOVERNOR_MOD=$OPTARG
+                                ;;
+
                         \?)
                                 helpme
                                 ;;
@@ -83,7 +95,7 @@ do_test()
         # Check if CPUs are managed by cpufreq or not
         count=$(count_cpufreq_managed_cpus)
 
-        if [ $count = 0 ]; then
+        if [ $count = 0 -a $FUNC != "modtest" ]; then
                 echo "No cpu is managed by cpufreq core, exiting"
                 exit 2;
         fi
@@ -101,6 +113,29 @@ do_test()
                 do_suspend "hibernate" 1
                 ;;
 
+                "modtest")
+                # Do we have modules in place?
+                if [ -z $DRIVER_MOD ] && [ -z $GOVERNOR_MOD ]; then
+                        echo "No driver or governor module passed with -d or -g"
+                        exit 2;
+                fi
+
+                if [ $DRIVER_MOD ]; then
+                        if [ $GOVERNOR_MOD ]; then
+                                module_test $DRIVER_MOD $GOVERNOR_MOD
+                        else
+                                module_driver_test $DRIVER_MOD
+                        fi
+                else
+                        if [ $count = 0 ]; then
+                                echo "No cpu is managed by cpufreq core, exiting"
+                                exit 2;
+                        fi
+
+                        module_governor_test $GOVERNOR_MOD
+                fi
+                ;;
+
                 *)
                 echo "Invalid [-f] function type"
                 helpme
diff --git a/tools/testing/selftests/cpufreq/module.sh b/tools/testing/selftests/cpufreq/module.sh
new file mode 100755
index 000000000000..8ff2244a33a1
--- /dev/null
+++ b/tools/testing/selftests/cpufreq/module.sh
@@ -0,0 +1,243 @@
+#!/bin/bash
+#
+# Modules specific tests cases
+
+# protect against multiple inclusion
+if [ $FILE_MODULE ]; then
+        return 0
+else
+        FILE_MODULE=DONE
+fi
+
+source cpu.sh
+source cpufreq.sh
+source governor.sh
+
+# Check basic insmod/rmmod
+# $1: module
+test_basic_insmod_rmmod()
+{
+        printf "** Test: Running ${FUNCNAME[0]} **\n\n"
+
+        printf "Inserting $1 module\n"
+        # insert module
+        insmod $1
+        if [ $? != 0 ]; then
+                printf "Insmod $1 failed\n"
+                exit;
+        fi
+
+        printf "Removing $1 module\n"
+        # remove module
+        rmmod $1
+        if [ $? != 0 ]; then
+                printf "rmmod $1 failed\n"
+                exit;
+        fi
+
+        printf "\n"
+}
+
+# Insert cpufreq driver module and perform basic tests
+# $1: cpufreq-driver module to insert
+# $2: If we want to play with CPUs (1) or not (0)
+module_driver_test_single()
+{
+        printf "** Test: Running ${FUNCNAME[0]} for driver $1 and cpus_hotplug=$2 **\n\n"
+
+        if [ $2 -eq 1 ]; then
+                # offline all non-boot CPUs
+                for_each_non_boot_cpu offline_cpu
+                printf "\n"
+        fi
+
+        # insert module
+        printf "Inserting $1 module\n\n"
+        insmod $1
+        if [ $? != 0 ]; then
+                printf "Insmod $1 failed\n"
+                return;
+        fi
+
+        if [ $2 -eq 1 ]; then
+                # online all non-boot CPUs
+                for_each_non_boot_cpu online_cpu
+                printf "\n"
+        fi