#!/bin/sh
###############################################################################
#                                                                             #
# IPFire.org - A linux based firewall                                         #
# Copyright (C) 2007-2024  IPFire Team  <info@ipfire.org>                     #
#                                                                             #
# This program is free software: you can redistribute it and/or modify        #
# it under the terms of the GNU General Public License as published by        #
# the Free Software Foundation, either version 3 of the License, or           #
# (at your option) any later version.                                         #
#                                                                             #
# This program is distributed in the hope that it will be useful,             #
# but WITHOUT ANY WARRANTY; without even the implied warranty of              #
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the               #
# GNU General Public License for more details.                                #
#                                                                             #
# You should have received a copy of the GNU General Public License           #
# along with this program.  If not, see <http://www.gnu.org/licenses/>.       #
#                                                                             #
###############################################################################

. /etc/sysconfig/rc
. ${rc_functions}

GOV="ondemand"

case "${1}" in
	start)
		boot_mesg -n "Starting cpufreq... "

		# try cpufreq hardware depend modules
		for i in $(find /lib/modules/$(uname -r)/kernel/drivers/cpufreq \
		           ! -name speedstep-lib.ko.xz ! -name p4-clockmod.ko.xz ! -name amd-pstate* ! -name "cpufreq_*" ! -name mperf.ko.xz | sort -d -r); do
			module=$(basename $i | cut -d. -f1);
			modprobe $module > /dev/null 2>&1;
			if [ ${?} = 0 ]; then
			    boot_mesg -n "$SUCCESS$module$NORMAL "
			fi
		done
		boot_mesg ""

		# load cqufreq governors
		modprobe cpufreq_conservative
		modprobe cpufreq_ondemand
		modprobe cpufreq_powersave
		modprobe cpufreq_userspace

		driver=`cat /sys/devices/system/cpu/cpu0/cpufreq/scaling_driver 2>/dev/null`

		case "$driver" in
		    *pstate*)
			# pstate not support ondemand so use powersave
			GOV="powersave"
			echo_ok;
			;;
		esac
		CPUCOUNT=`ls /sys/devices/system/cpu/cpu*/cpufreq/affected_cpus 2> /dev/null | wc -l `;
		let CPUCOUNT-=1
		# Set the governor to test if it works
		cpufreq-set -g $GOV
		if [ ${?} = 0 ]; then
			# Set the governor to ondemand for all cpus
			for i in `seq 0 $CPUCOUNT`;
			do
				cpufreq-set -c $i -g $GOV > /dev/null
			done
			echo_ok;
		else
			echo_failure;
		fi
		exit 0
		;;
	*)
		echo "Usage: ${0} {start}"
		exit 1
		;;
esac