#!/bin/bash
# SPDX-License-Identifier: GPL-2.0-or-later OR copyleft-next-0.3.1
# Copyright (C) 2023 Luis Chamberlain <[email protected]>
#
# This is a stress test script for kallsyms through find_symbol()

set -e

# Kselftest framework requirement - SKIP code is 4.
ksft_skip=4

test_reqs()
{
	if ! which modprobe 2> /dev/null > /dev/null; then
		echo "$0: You need modprobe installed" >&2
		exit $ksft_skip
	fi

	if ! which kmod 2> /dev/null > /dev/null; then
		echo "$0: You need kmod installed" >&2
		exit $ksft_skip
	fi

	if ! which perf 2> /dev/null > /dev/null; then
		echo "$0: You need perf installed" >&2
		exit $ksft_skip
	fi

	uid=$(id -u)
	if [ $uid -ne 0 ]; then
		echo $msg must be run as root >&2
		exit $ksft_skip
	fi
}

load_mod()
{
	local STATS="-e duration_time"
	STATS="$STATS -e user_time"
	STATS="$STATS -e system_time"
	STATS="$STATS -e page-faults"
	local MOD=$1

	local ARCH="$(uname -m)"
	case "${ARCH}" in
	x86_64)
		perf stat $STATS $MODPROBE $MOD
		;;
	*)
		time $MODPROBE $MOD
		exit 1
		;;
	esac
}

remove_all()
{
	$MODPROBE -r test_kallsyms_b
	for i in a b c d; do
		$MODPROBE -r test_kallsyms_$i
	done
}
test_reqs

MODPROBE=$(</proc/sys/kernel/modprobe)

remove_all
load_mod test_kallsyms_b
remove_all

# Now pollute the namespace
$MODPROBE test_kallsyms_c
load_mod test_kallsyms_b

# Now pollute the namespace with twice the number of symbols than the last time
remove_all
$MODPROBE test_kallsyms_c
$MODPROBE test_kallsyms_d
load_mod test_kallsyms_b

exit 0