# $NetBSD: t_modload.sh,v 1.14 2024/04/28 07:27:41 rillig Exp $
#
# Copyright (c) 2008 The NetBSD Foundation, Inc.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
#    notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
#    notice, this list of conditions and the following disclaimer in the
#    documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
# ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
# TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
# PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
# BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#

check_sysctl() {
       echo "${1} = ${2}" >expout
       atf_check -s exit:0 -o file:expout -e empty sysctl ${1}
}

atf_test_case plain cleanup
plain_head() {
       atf_set "descr" "Test load without arguments"
       atf_set "require.user" "root"
}
plain_body() {

       # XXX: Adjust when modctl(8) fails consistently.
       #
       $(atf_get_srcdir)/k_helper3 \
               "%s/k_helper/k_helper.kmod" $(atf_get_srcdir)

       if [ $? -eq 1 ] || [ $? -eq 78 ]; then
               atf_skip "host does not support modules"
       fi

       cat >experr <<EOF
modload: No such file or directory
EOF
       atf_check -s exit:1 -o empty -e ignore modload non-existent.o

       atf_check -s exit:0 -o empty -e empty \
           modload $(atf_get_srcdir)/k_helper/k_helper.kmod
       check_sysctl vendor.k_helper.present 1
       check_sysctl vendor.k_helper.prop_int_ok 0
       check_sysctl vendor.k_helper.prop_str_ok 0
       atf_check -s exit:0 -o empty -e empty modunload k_helper
       touch done
}
plain_cleanup() {
       test -f done || modunload k_helper >/dev/null 2>&1
}

atf_test_case bflag cleanup
bflag_head() {
       atf_set "descr" "Test the -b flag"
       atf_set "require.user" "root"
}
bflag_body() {
       echo "Checking error conditions"

       atf_check -s exit:1 -o empty -e save:stderr \
           modload -b foo k_helper.kmod
       atf_check -s exit:0 -o ignore -e empty \
           grep 'Invalid parameter.*foo' stderr

       atf_check -s exit:1 -o empty -e save:stderr \
           modload -b foo= k_helper.kmod
       atf_check -s exit:0 -o ignore -e empty \
           grep 'Invalid boolean value' stderr

       atf_check -s exit:1 -o empty -e save:stderr \
           modload -b foo=bar k_helper.kmod
       atf_check -s exit:0 -o ignore -e empty \
           grep 'Invalid boolean value.*bar' stderr

       atf_check -s exit:1 -o empty -e save:stderr \
           modload -b foo=falsea k_helper.kmod
       atf_check -s exit:0 -o ignore -e empty \
           grep 'Invalid boolean value.*falsea' stderr

       atf_check -s exit:1 -o empty -e save:stderr \
           modload -b foo=truea k_helper.kmod
       atf_check -s exit:0 -o ignore -e empty \
           grep 'Invalid boolean value.*truea' stderr

       # TODO Once sysctl(8) supports CTLTYPE_BOOL nodes.
       #echo "Checking valid values"
}
bflag_cleanup() {
       modunload k_helper >/dev/null 2>&1 || true
}

atf_test_case iflag cleanup
iflag_head() {
       atf_set "descr" "Test the -i flag"
       atf_set "require.user" "root"
}
iflag_body() {

       # XXX: Adjust when modctl(8) fails consistently.
       #
       $(atf_get_srcdir)/k_helper3 \
               "%s/k_helper/k_helper.kmod" $(atf_get_srcdir)

       if [ $? -eq 1 ] || [ $? -eq 78 ]; then
               atf_skip "host does not support modules"
       fi

       echo "Checking error conditions"

       atf_check -s exit:1 -o empty -e save:stderr modload -i foo \
           k_helper/k_helper.kmod
       atf_check -s exit:0 -o ignore -e empty \
           grep 'Invalid parameter.*foo' stderr

       atf_check -s exit:1 -o empty -e save:stderr modload -i foo= \
           k_helper/k_helper.kmod
       atf_check -s exit:0 -o ignore -e empty \
           grep 'Invalid integer value' stderr

       atf_check -s exit:1 -o empty -e save:stderr \
           modload -i foo=bar k_helper/k_helper.kmod
       atf_check -s exit:0 -o ignore -e empty \
           grep 'Invalid integer value.*bar' stderr

       atf_check -s exit:1 -o empty -e save:stderr \
           modload -i foo=123a k_helper/k_helper.kmod
       atf_check -s exit:0 -o ignore -e empty \
           grep 'Invalid integer value.*123a' stderr

       echo "Checking valid values"

       for v in 5 10; do
               atf_check -s exit:0 -o empty -e empty \
                   modload -i prop_int="${v}" \
                   $(atf_get_srcdir)/k_helper/k_helper.kmod
               check_sysctl vendor.k_helper.prop_int_ok 1
               check_sysctl vendor.k_helper.prop_int_val "${v}"
               atf_check -s exit:0 -o empty -e empty modunload k_helper
       done
       touch done
}
iflag_cleanup() {
       test -f done || modunload k_helper >/dev/null 2>&1
}

atf_test_case sflag cleanup
sflag_head() {
       atf_set "descr" "Test the -s flag"
       atf_set "require.user" "root"
}
sflag_body() {

       # XXX: Adjust when modctl(8) fails consistently.
       #
       $(atf_get_srcdir)/k_helper3 \
               "%s/k_helper/k_helper.kmod" $(atf_get_srcdir)

       if [ $? -eq 1 ] || [ $? -eq 78 ]; then
               atf_skip "host does not support modules"
       fi

       echo "Checking error conditions"

       atf_check -s exit:1 -o empty -e save:stderr modload -s foo \
           k_helper/k_helper.kmod
       atf_check -s exit:0 -o ignore -e empty \
           grep 'Invalid parameter.*foo' stderr

       echo "Checking valid values"

       for v in '1st string' '2nd string'; do
               atf_check -s exit:0 -o empty -e empty \
                   modload -s prop_str="${v}" \
                   $(atf_get_srcdir)/k_helper/k_helper.kmod
               check_sysctl vendor.k_helper.prop_str_ok 1
               check_sysctl vendor.k_helper.prop_str_val "${v}"
               atf_check -s exit:0 -o empty -e empty modunload k_helper
       done
       touch done
}
sflag_cleanup() {
       test -f done || modunload k_helper >/dev/null 2>&1
}

atf_init_test_cases()
{
       atf_add_test_case plain
       atf_add_test_case bflag
       atf_add_test_case iflag
       atf_add_test_case sflag
}