Added default functionality
This commit is contained in:
463
wdpostinit.sh
Executable file
463
wdpostinit.sh
Executable file
@@ -0,0 +1,463 @@
|
||||
#!/bin/bash
|
||||
#
|
||||
# Main Control script for Free/TrueNAS CORE & SCALE on Western Digital PR2100?/PR4100
|
||||
# Based off wdhws v1.0 by TFL (stefaang)
|
||||
|
||||
############### COMMAND LIST ###############
|
||||
# THING COMMAND USE
|
||||
# FAN FAN=64 Enter Hex value 01-64 (1-100%) Use at your own risk, only you can prevent forest fires
|
||||
# USB/Power LED LED=13 (See LED Guide)
|
||||
# PowerLED-Pulse PLS=01 00-off 01-on (cant change color? is always blue?)
|
||||
# PowerLED-Blink BLK=01 (See LED Guide)
|
||||
# LCDBacklight BKL=64 Enter Hex value 00-64 (0-100%)
|
||||
|
||||
################ LED GUIDE #################
|
||||
#XX-usb/pwr
|
||||
#00-off/off 01-off/blue 02-off/red 03-off/purple 04-off/green 05-off/teal 06-off/yellow 07-off/White
|
||||
#08-red/off 09-red/blue 0A-red/red 0B-red/purple 0C-red/green 0D-red/teal 0E-red/yellow 0F-red/White
|
||||
#10-blue/off 11-blue/blue 12-blue/red 13-blue/purple 14-blue/green 15-blue/teal 16-blue/yellow 17-blue/White
|
||||
#18-purple/off 19-purple/blue 1A-purple/red 1B-purple/purple 1C-purple/green 1D-purple/teal 1E-purple/yellow 1F-purple/White
|
||||
|
||||
###########################################################################
|
||||
######################### DO NOT TOUCH ################################
|
||||
###########################################################################
|
||||
fanSpeedMinimum=10 # Minimum allowed fan speed in percent
|
||||
cpuOptimalTemp=35 # Optimal (desired) temp for CPU (commie C degrees not freedom F :()
|
||||
cpuMaxTemp=80 # Maximum CPU temp before going full beans
|
||||
diskMaxTemp=40 # Maximum DISK temp before going full beans
|
||||
pmcMaxTemp=64 # Maximum PMC temp before going full beans
|
||||
ramMaxTemp=40 # Maximum RAM temp before going full beans
|
||||
updateRate=10 # How often in seconds to update temps
|
||||
|
||||
###########################################################################
|
||||
############################# VARS ####################################
|
||||
###########################################################################
|
||||
datetime=()
|
||||
hwSystem=Linux # Used to init kernal variable, gets changed based on kernal in get_sys_info()
|
||||
hwTTY=/dev/cuau3 # Used to init tty variable, gets changed based on kernal in get_sys_info()
|
||||
hwHDDArray=()
|
||||
hwCPUCoreCount=0
|
||||
hwOverTempAlarm=0
|
||||
hwOverTempArray=()
|
||||
overTemp=0
|
||||
|
||||
|
||||
|
||||
###########################################################################
|
||||
############################# FUNCS ###################################
|
||||
###########################################################################
|
||||
|
||||
check_for_dependencies(){ # Simple just-to-be-safe check that SMART Mon exists
|
||||
depenflag=0
|
||||
# S.M.A.R.T Drive Utilities
|
||||
smartctl -v >/dev/null
|
||||
if [[ $? != 1 ]]; then
|
||||
printf "\n** SMART not installed please run - sudo apt install smartmontools ** \n\n "
|
||||
(( depenflag += 1 ))
|
||||
fi
|
||||
|
||||
# lm-Sensors (For temp sensor data )
|
||||
sensors -v >/dev/null
|
||||
if [[ $? != 0 ]]; then
|
||||
printf "\n** lm-sensors not installed please run - sudo apt install smartmontools ** \n\n "
|
||||
(( depenflag+=2 ))
|
||||
fi
|
||||
|
||||
if [ $depenflag -gt 0 ]; then
|
||||
#printf "Would you like me to install? y/n:"
|
||||
#read resp
|
||||
#if [[ $resp == "y"] && [ $depenflag -gt 0 ]; then
|
||||
# if [ $depenflag -eq 1 ]; then
|
||||
# sudo apt install smartmontools
|
||||
# elif [ $depenflag -eq 2 ]; then
|
||||
# sudo apt install lm-sensors
|
||||
# elif [ $depenflag -eq 3 ]; then
|
||||
# sudo apt install smartmontools && sudo apt install lm-sensors
|
||||
# fi
|
||||
#fi
|
||||
printf "\n\n## PROGRAM TERMINATED ##\n\n"
|
||||
exit
|
||||
fi
|
||||
}
|
||||
|
||||
get_sys_info(){ # Get system info based off kernal, as BSD/LINUX has not the same commands
|
||||
case "$( uname -s )" in # Get Linux Kernal (Linux vs FreeBSD for TrueNas Scale/Core)
|
||||
Linux*) hwSystem=Linux;;
|
||||
*BSD) hwSystem=BSD;;
|
||||
Darwin*) hwSystem=MacOS;;
|
||||
CYGWIN*) hwSystem=Cygwin;;
|
||||
MINGW*) hwSystem=MinGw;;
|
||||
*) hwSystem="Other"
|
||||
esac
|
||||
if [[ ! $hwSystem =~ Linux|BSD ]]; then # If system is not Linux or *BSD Show unsupported message
|
||||
echo "Sorry, This software version for the WD PR4100 Hawrdware does not support $hwSystem platform."
|
||||
echo "Please create an issue on Github to see about gettin support added"
|
||||
exit 1
|
||||
fi
|
||||
if [ $hwSystem == BSD ]; then # If *BSD Free/TrueNAS Core
|
||||
echo '# GETINFO: Detected BSD Kernal #' # Show what kernal was identified
|
||||
hwTTY=/dev/cuau3 # FreeBSD uses /dev/cuau3 for i2C comms to PR4100 front hardware
|
||||
get_int_drives # Get location of ONLY internal bay drives
|
||||
hwCPUCoreCount=$(sysctl -n hw.ncpu) # Get how many CPU cores
|
||||
elif [ $hwSystem == Linux ]; then # If Linux Free/TrueNAS Scale
|
||||
echo '# GETINFO: Detected Linux Kernal #' # Show what kernal was identified
|
||||
hwTTY=/dev/ttyS2 # Linux uses much cooler (telatype) /dev/hwTTYS2 for i2C comms to PR4100 front hardware
|
||||
get_int_drives # Get location of ONLY internal bay drives
|
||||
hwCPUCoreCount=$(nproc) # Get how many CPU cores
|
||||
fi
|
||||
}
|
||||
|
||||
get_int_drives(){ # Gets the location of the internal bay HDD's
|
||||
for file in /dev/disk/by-id/ata* # With each HDD decice thats ata (Internal Sata)
|
||||
do
|
||||
if [[ $file != *"-part"* ]]; then # Filter out '-part$' devices as they are the same
|
||||
tmparr+=( $( ls -l "/dev/disk/by-id/ata-${file:20:100}" | awk '{print $11}' | cut -b 7-10 ) ) # Get the /dev location
|
||||
readarray -t hwHDDArray < <(for a in "${tmparr[@]}"; do echo "/dev/$a"; done | sort) # Sort
|
||||
fi
|
||||
done
|
||||
echo "# GETINFO: Detected internal bay drives: ${hwHDDArray[@]} #"
|
||||
}
|
||||
|
||||
setup_tty() { # Start i2c
|
||||
exec 4<$hwTTY 5>$hwTTY
|
||||
}
|
||||
|
||||
setup_i2c() { # load kernel modules required for the temperature sensor on the RAM modules
|
||||
if [ $hwSystem == BSD ]; then
|
||||
kldload -n iicbus smbus smb ichsmb
|
||||
fi
|
||||
}
|
||||
|
||||
send() { # Requires input - i2C send function to send commands to front panel
|
||||
setup_tty
|
||||
# send a command to the PMC module and echo the answer
|
||||
echo -ne "$1\r" >&5
|
||||
read ans <&4
|
||||
if [ "$ans" = "ALERT" ]; then
|
||||
echo -ne ALERT >&2
|
||||
exit 2
|
||||
else
|
||||
# keep this for debugging failing commands
|
||||
if [ "$ans" = "ERR" ] || [ -z "$ans" ]; then
|
||||
echo "CMD $1 gives ERR at $2" >&2
|
||||
send_empty
|
||||
ans=$(send "$1" $(($2 + 1)))
|
||||
#exit 1
|
||||
fi
|
||||
fi
|
||||
# only echo the result for retries ($2 not empty)
|
||||
if [ ! -z $2 ]; then
|
||||
echo "CMD $1 gives '$ans' at $2" >&2
|
||||
fi
|
||||
echo $ans
|
||||
|
||||
send_empty
|
||||
# deconstruct tty file pointers, otherwise this script breaks on sleep
|
||||
exec 4<&- 5>&-
|
||||
|
||||
}
|
||||
|
||||
send_empty() { # i2c send blank to clear front panel input
|
||||
# send a empty command to clear the output
|
||||
echo -ne "\r" >&5
|
||||
read ignore <&4
|
||||
}
|
||||
|
||||
get_datetime() {
|
||||
datetime[1]=$(date +"%m")
|
||||
datetime[2]=$(date +"%d")
|
||||
datetime[3]=$(date +"%y")
|
||||
datetime[4]=$(date +"%H")
|
||||
datetime[5]=$(date +"%M")
|
||||
datetime[6]=$(date +"%S")
|
||||
}
|
||||
|
||||
get_pmc() { # Requires input - Get a value from the PMC ex. inputing RPM gets fan0's rpm
|
||||
send $1 | cut -d'=' -f2
|
||||
}
|
||||
|
||||
get_disktemp() { # Requires input - Get the disks temperature only if it is active, else return status
|
||||
drivesel=$1 #$1 # For some reason i need this and cant put it in later? Makes i2c break somehow...
|
||||
smartctl -n standby -A $drivesel > /dev/null # Run command to get disk status
|
||||
getstatus=$(echo $?) # Get drive exit status
|
||||
if [ "$getstatus" == "0" ]; then # If the status of the drive is active, get its temperature
|
||||
smartctl -n standby -A $drivesel | grep Temperature_Celsius | awk '{print $10}'
|
||||
else # If the status of the drive is not active, return the exit status of the drive. Maybe its asleep/standby
|
||||
return $getstatus
|
||||
fi
|
||||
}
|
||||
|
||||
get_cpucoretemp() {
|
||||
# get the CPU temperature and strip of the Celsius
|
||||
if [ $hwSystem == BSD ]; then
|
||||
sysctl -n dev.cpu.$1.temperature | cut -d'.' -f1
|
||||
elif [ $hwSystem == Linux ]; then
|
||||
sensors | grep "Core $1" | awk '{print $3}' | cut -d'.' -f1 | cut -b 2-3
|
||||
fi
|
||||
}
|
||||
|
||||
get_ramtemp() {
|
||||
# get the memory temperature from the I2C sensor
|
||||
if [ $hwSystem == BSD ]; then
|
||||
smbmsg -s 0x98 -c 0x0$1 -i 1 -F %d
|
||||
elif [ $hwSystem == Linux ]; then
|
||||
echo "0"
|
||||
# sensors | grep "temp1" | awk '{print $2}' | cut -d'.' -f1 | cut -b 2-3
|
||||
fi
|
||||
}
|
||||
|
||||
monitor() {
|
||||
# check RPM (fan may get stuck) and convert hex to dec
|
||||
#readfanpercent=$(get_pmc FAN)
|
||||
rpmhex=$(get_pmc RPM)
|
||||
rpmdec=$((0x$rpmhex))
|
||||
echo "FAN 0 RPM: $rpmdec"
|
||||
if [ "$rpmdec" != ERR ]; then
|
||||
if [ "$rpmdec" -lt 400 ]; then
|
||||
echo "FAN 0 RPM WARNING: low RPM - $rpmdec - clean dust!"
|
||||
set_pwr_led FLASH RED
|
||||
fi
|
||||
fi
|
||||
|
||||
# Check the Temperature of the PMC and convert to hex
|
||||
tmphex=$(get_pmc TMP) # I dont need to do this in 2 steps but VSCode complains soooooo..... 2 steps wowww
|
||||
tmpdec=$((0x$tmphex)) # use tmpdec=$((0x$(get_pmc TMP))) if you want, it works, i just hate the 'error'
|
||||
if [ "$tmpdec" -gt $pmcMaxTemp ]; then
|
||||
echo "WARNING: PMC surpassed maximum ($pmcMaxTemp°C), full throttle activated!"
|
||||
hwOverTempAlarm=1
|
||||
#hwOverTempArray+=("PMC $tmp°C/$pmcMaxTemp°C")"
|
||||
fi
|
||||
|
||||
# Check the Hard Drive Temperature [adjust this for PR2100!!] (<- IDK what that means)
|
||||
highestcpucoretemp=0
|
||||
printf "|------ DISK TEMPS ------\n"
|
||||
for i in "${hwHDDArray[@]}" ; do
|
||||
tmp=$(get_disktemp $i)
|
||||
waserror=$(echo $?)
|
||||
if [ $waserror -ne "0" ]; then
|
||||
if [ $waserror == 2 ]; then
|
||||
ret=standby
|
||||
else
|
||||
ret=Error
|
||||
fi
|
||||
echo "| Drive ${i:5:5} is in $ret status"
|
||||
else
|
||||
echo "| Drive ${i:5:15} is $tmp °C"
|
||||
if [ ! -z $tmp ] && [ $tmp -gt $diskMaxTemp ]; then
|
||||
echo "| WARNING: CPU Core$i surpassed maximum ($diskMaxTemp°C), full throttle activated!"
|
||||
hwOverTempAlarm=1
|
||||
#hwOverTempArray+=("HDD$i $tmp°C/$hddMaxTemp°C")
|
||||
fi
|
||||
fi
|
||||
done
|
||||
printf "|------------------------\n"
|
||||
|
||||
# Check the Temperature of the CPU
|
||||
printf "|---- CPU CORE TEMPS ----\n"
|
||||
for i in $(seq $hwCPUCoreCount); do
|
||||
tmp=$(get_cpucoretemp $((i-1)))
|
||||
echo "| cpu core$i is $tmp °C"
|
||||
if [ $tmp -gt $cpuMaxTemp ]; then
|
||||
echo "| WARNING: CPU Core$i surpassed maximum ($cpuMaxTemp°C), full throttle activated!"
|
||||
#hwOverTempArray+=("CPU$i $tmp°C/$cpuMaxTemp°C")
|
||||
hwOverTempAlarm=1
|
||||
fi
|
||||
if [ $tmp -gt $highestcpucoretemp ]; then
|
||||
highestcpucoretemp=$tmp
|
||||
fi
|
||||
done
|
||||
printf "|------------------------\n"
|
||||
#echo "Highest CPU core temp is $highestcpucoretemp °C"
|
||||
# max-opperating=a fullfan-minfan=b b/a= fan percent per degree
|
||||
#Max-80 Optimal-35 1.5% = for every degree above 30% 80-35=45 100-30=70 70/45=1.5
|
||||
newtmp=$(("$highestcpucoretemp"-"$cpuOptimalTemp")) #MaxTemp
|
||||
setspeed=$(("$newtmp"*2+"$fanSpeedMinimum"-5))
|
||||
|
||||
# Check the installed RAM Temperature
|
||||
printf "|------ RAM TEMPS -------\n"
|
||||
if [ $hwSystem == BSD ]; then # If *BSD Free/TrueNAS Core
|
||||
for i in 0 1; do
|
||||
tmp=$(get_ramtemp $i)
|
||||
echo "| ram$i temp is $tmp °C"
|
||||
if [ "$tmp" -gt $ramMaxTemp ]; then
|
||||
#if [ "$tmp" -gt 0 ]; then
|
||||
echo "| WARNING: RAM$i surpassed maximum ($ramMaxTemp°C), full throttle activated!"
|
||||
#hwOverTempArray+=("RAM $tmp°C/$ramMaxTemp°C")
|
||||
hwOverTempAlarm=1
|
||||
fi
|
||||
done
|
||||
else
|
||||
echo "|"
|
||||
echo "| Currently Unsupported"
|
||||
echo "|"
|
||||
fi
|
||||
printf "|------------------------\n"
|
||||
|
||||
if [ ${#hwOverTempArray[@]} -gt 0 ] || [ $hwOverTempAlarm == 1 ]; then
|
||||
echo " WARNING: SYSTEM OVER LIMIT TEMPERATURE(s) FAN SET TO 100% "
|
||||
hwOverTempAlarm=1 # Flag System Over Temp-ed
|
||||
#hwLastOverTemp=$(get_datetime) # Save the time when the system over temped
|
||||
send FAN=64 # Full Beans Fan 100%
|
||||
set_pwr_led FLASH RED # Flash Power LED RED to warn
|
||||
send "LN1=WARNING"
|
||||
send "LN2=OVERHEATING!"
|
||||
overTemp=1
|
||||
else
|
||||
if [ $overTemp == 1 ]; then
|
||||
overTemp=0
|
||||
init
|
||||
fi
|
||||
if [ $setspeed -lt $fanSpeedMinimum ]; then
|
||||
echo "Calculated fan speed below minimum allowed, bumping to $fanSpeedMinimum%..."
|
||||
setspeed=$fanSpeedMinimum # Set the fan to the min allowed
|
||||
send FAN=$setspeed
|
||||
else
|
||||
echo "Setting fan speed to: $setspeed%"
|
||||
send FAN=$setspeed # Set fan to mathed speed if not overtemped
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
show_welcome() {
|
||||
# set welcome message
|
||||
# maximum "xxx xxx xxx xxx "
|
||||
send "LN1= TrueNAS "
|
||||
send "LN2= Running "
|
||||
}
|
||||
|
||||
show_ip() {
|
||||
send "LN1=eno1 Address"
|
||||
ip=$(ifconfig ""eno1"" | grep "inet " | awk '{printf $2}')
|
||||
send "LN2=$ip"
|
||||
}
|
||||
|
||||
show_cpu_temps() {
|
||||
send "LN1=CPU Temperatures"
|
||||
send "LN2=$(get_cpucoretemp 0)C $(get_cpucoretemp 1)C $(get_cpucoretemp 2)C $(get_cpucoretemp 3)C"
|
||||
}
|
||||
|
||||
|
||||
show_hdd_temps() {
|
||||
send "LN1=HDD Temperatures"
|
||||
tmp=""
|
||||
for i in "${hwHDDArray[@]}" ; do
|
||||
tmp+=$(get_disktemp $i)
|
||||
tmp+="C "
|
||||
done
|
||||
send "LN2=$tmp"
|
||||
}
|
||||
|
||||
check_btn_pressed() {
|
||||
btn=$(get_pmc ISR)
|
||||
#echo "Btn is .$btn."
|
||||
|
||||
case $btn in
|
||||
20*)
|
||||
echo "Button up pressed!"
|
||||
menu=$(( ($menu + 3) % 4 ))
|
||||
;;
|
||||
40*)
|
||||
echo "Button down pressed!"
|
||||
menu=$(( ($menu + 1) % 4 ))
|
||||
;;
|
||||
*)
|
||||
return
|
||||
esac
|
||||
|
||||
case "$menu" in
|
||||
0)
|
||||
show_welcome
|
||||
;;
|
||||
1)
|
||||
show_ip
|
||||
;;
|
||||
2)
|
||||
show_cpu_temps
|
||||
;;
|
||||
3)
|
||||
show_hdd_temps
|
||||
;;
|
||||
# if you add menu items here, update mod 3 uses above
|
||||
esac
|
||||
}
|
||||
|
||||
set_pwr_led(){
|
||||
#echo "PowerMode:$1 - PowerColor:$2 - UsbMode:$3 - UsbColor$4"
|
||||
if [ "$1" == SOLID ]; then
|
||||
send BLK=00
|
||||
send PLS=00
|
||||
if [ "$2" == BLU ]; then
|
||||
send LED=01
|
||||
elif [ "$2" == RED ]; then
|
||||
send LED=02
|
||||
elif [ "$2" == PUR ]; then
|
||||
send LED=03
|
||||
elif [ "$2" == GRE ]; then
|
||||
send LED=04
|
||||
elif [ "$2" == TEA ]; then
|
||||
send LED=05
|
||||
elif [ "$2" == YLW ]; then
|
||||
send LED=06
|
||||
elif [ "$2" == WHT ]; then
|
||||
send LED=07
|
||||
fi
|
||||
fi
|
||||
if [ "$1" == FLASH ]; then
|
||||
send LED=00
|
||||
send PLS=00
|
||||
if [ "$2" == BLU ]; then
|
||||
send BLK=01
|
||||
elif [ "$2" == RED ]; then
|
||||
send BLK=02
|
||||
elif [ "$2" == PUR ]; then
|
||||
send BLK=03
|
||||
elif [ "$2" == GRE ]; then
|
||||
send BLK=04
|
||||
elif [ "$2" == TEA ]; then
|
||||
send BLK=05
|
||||
elif [ "$2" == YLW ]; then
|
||||
send BLK=06
|
||||
elif [ "$2" == WHT ]; then
|
||||
send BLK=07
|
||||
fi
|
||||
fi
|
||||
|
||||
if [ "$1" == PULSE ]; then
|
||||
send PLS=01
|
||||
send LED=00
|
||||
send BLK=00
|
||||
fi
|
||||
}
|
||||
|
||||
init() {
|
||||
check_for_dependencies
|
||||
get_sys_info
|
||||
setup_tty
|
||||
setup_i2c
|
||||
|
||||
echo "# GETINFO: Getting system status and firmware! *"
|
||||
send VER
|
||||
send CFG
|
||||
send STA
|
||||
|
||||
show_welcome
|
||||
set_pwr_led SOLID BLU
|
||||
printf "# INIT DONE # \n\n"
|
||||
}
|
||||
|
||||
###########################################################################
|
||||
############################# MAIN ####################################
|
||||
###########################################################################
|
||||
init
|
||||
|
||||
while true; do
|
||||
# adjust fan speed every 30 seconds
|
||||
monitor
|
||||
echo "Next temp update in $updateRate seconds"
|
||||
|
||||
# check for button presses
|
||||
for i in $(seq $updateRate); do
|
||||
sleep 1
|
||||
check_btn_pressed
|
||||
done
|
||||
done
|
||||
Reference in New Issue
Block a user