You've already forked UGREEN-NAS
This commit is contained in:
Cantibra
260
root/usr/sbin/ugreen-led-disk
Executable file
260
root/usr/sbin/ugreen-led-disk
Executable file
@@ -0,0 +1,260 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
|
||||
###
|
||||
#
|
||||
# Options Section
|
||||
#
|
||||
###
|
||||
|
||||
set -e
|
||||
set -o pipefail
|
||||
|
||||
|
||||
###
|
||||
#
|
||||
# Variables Section
|
||||
#
|
||||
###
|
||||
|
||||
if [[ -f '/etc/ugreen-nas.conf' ]]; then
|
||||
source '/etc/ugreen-nas.conf'
|
||||
fi
|
||||
|
||||
DISK_ZFS="${DISK_ZFS:='false'}"
|
||||
DISK_SMART="${DISK_SMART:='false'}"
|
||||
DISK_LED_BRIGHTNESS="${DISK_LED_BRIGHTNESS:='255'}"
|
||||
DISK_LED_COLOR_HEALTH="${DISK_LED_COLOR_HEALTH:='0 255 0'}"
|
||||
DISK_LED_COLOR_FAILED="${DISK_LED_COLOR_FAILED:='255 0 0'}"
|
||||
DISK_LED_COLOR_STANDBY="${DISK_LED_COLOR_STANDBY:='0 0 255'}"
|
||||
|
||||
MODEL=$(/usr/sbin/dmidecode --string 'system-product-name')
|
||||
SERIAL_MAP=(${DISK_SERIAL})
|
||||
|
||||
declare -A DEVICES
|
||||
declare -A DEV_MAP
|
||||
declare -A DEV_LED_MAP
|
||||
declare -A ZFS_LED_MAP
|
||||
|
||||
|
||||
###
|
||||
#
|
||||
# Function Section
|
||||
#
|
||||
###
|
||||
|
||||
function trap_exit () {
|
||||
if [[ -n "${PID_ZFS}" ]]; then
|
||||
/usr/bin/kill "${PID_ZFS}" &> '/dev/null'
|
||||
fi
|
||||
if [[ -n "${PID_SMART}" ]]; then
|
||||
/usr/bin/kill "${PID_SMART}" &> '/dev/null'
|
||||
fi
|
||||
if [[ -n "${PID_DISK}" ]]; then
|
||||
/usr/bin/kill "${PID_DISK}" &> '/dev/null'
|
||||
fi
|
||||
if [[ -n "${PID_DISK_STANDBY}" ]]; then
|
||||
/usr/bin/kill "${PID_DISK_STANDBY}" &> '/dev/null'
|
||||
fi
|
||||
}
|
||||
|
||||
function load_module () {
|
||||
MODULE="${1}"
|
||||
if ! /usr/sbin/lsmod | /usr/bin/grep --quiet "${MODULE}"; then
|
||||
/usr/sbin/modprobe "${MODULE}"
|
||||
fi
|
||||
}
|
||||
|
||||
function check_disk_health () {
|
||||
if [[ "${1}" == "${DISK_LED_COLOR_HEALTH}" || \
|
||||
"${1}" == "${DISK_LED_COLOR_STANDBY}" ]]; then
|
||||
return 0
|
||||
else
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
|
||||
function check_disk_zfs () {
|
||||
while true; do
|
||||
while read LINE; do
|
||||
LINE=(${LINE})
|
||||
DEV_NAME="${LINE[0]}"
|
||||
DEV_STATE="${LINE[1]}"
|
||||
if [[ -v "ZFS_LED_MAP[${DEV_NAME}]" ]]; then
|
||||
LED="${ZFS_LED_MAP[${DEV_NAME}]}"
|
||||
LED_COLOR=$(/usr/bin/cat "/sys/class/leds/${LED}/color")
|
||||
if ! check_disk_health "${LED_COLOR}"; then
|
||||
continue;
|
||||
fi
|
||||
if [[ "${DEV_STATE}" != "ONLINE" ]]; then
|
||||
/usr/bin/echo "${COLOR_ZPOOL_FAIL}" > "/sys/class/leds/${LED}/color"
|
||||
fi
|
||||
fi
|
||||
done <<< $(/usr/bin/zpool status -L | /usr/bin/grep --extended-regexp ^\\s*\(nvme\|dm\))
|
||||
/usr/bin/sleep '15s'
|
||||
done
|
||||
}
|
||||
|
||||
function disk_parameters () {
|
||||
for LED in "${!DEVICES[@]}"; do
|
||||
/usr/bin/echo "${DEVICES[${LED}]} ${LED}"
|
||||
done
|
||||
}
|
||||
|
||||
|
||||
###
|
||||
#
|
||||
# Runtime Environment
|
||||
#
|
||||
###
|
||||
|
||||
if [[ -z "${DISK_SERIAL}" ]]; then
|
||||
/usr/bin/echo "The serial number of the hard drives could not be found in the configuration."
|
||||
exit 0
|
||||
fi
|
||||
|
||||
trap 'trap_exit' EXIT
|
||||
|
||||
case "${MODEL}" in
|
||||
DXP2800*)
|
||||
LED_MAP=(disk1 disk2)
|
||||
;;
|
||||
DH4300*)
|
||||
LED_MAP=(disk1 disk2 disk3 disk4)
|
||||
;;
|
||||
DXP4800*)
|
||||
LED_MAP=(disk1 disk2 disk3 disk4)
|
||||
;;
|
||||
DXP6800*)
|
||||
LED_MAP=(disk1 disk2 disk3 disk4 disk5 disk6)
|
||||
;;
|
||||
DXP8800*)
|
||||
LED_MAP=(disk1 disk2 disk3 disk4 disk5 disk6 disk7 disk8)
|
||||
;;
|
||||
*)
|
||||
/usr/bin/echo "The UGREEN NAS ${MODEL} is not supported."
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
|
||||
load_module 'ledtrig_oneshot'
|
||||
|
||||
while read LINE; do
|
||||
LINE=(${LINE})
|
||||
SERIAL="${LINE[1]}"
|
||||
DISK="${LINE[0]}"
|
||||
DEV_MAP[${SERIAL}]="${DISK}"
|
||||
done <<< $(/usr/bin/lsblk --all --output 'name,serial,tran' | /usr/bin/grep --extended-regexp '^nvme[0-9]+n[0-9]+ ')
|
||||
|
||||
for MAP in "${!LED_MAP[@]}"; do
|
||||
LED="${LED_MAP[MAP]}"
|
||||
if [[ -d "/sys/class/leds/${LED}" ]]; then
|
||||
/usr/bin/echo 'oneshot' > "/sys/class/leds/${LED}/trigger"
|
||||
/usr/bin/echo '1' > "/sys/class/leds/${LED}/invert"
|
||||
/usr/bin/echo '100' > "/sys/class/leds/${LED}/delay_on"
|
||||
/usr/bin/echo '100' > "/sys/class/leds/${LED}/delay_off"
|
||||
/usr/bin/echo "${DISK_LED_COLOR_HEALTH}" > "/sys/class/leds/${LED}/color"
|
||||
/usr/bin/echo "${DISK_LED_BRIGHTNESS}" > "/sys/class/leds/${LED}/brightness"
|
||||
TMP_STR="SERIAL_MAP[@]"
|
||||
TMP_ARR=("${!TMP_STR}")
|
||||
if [[ -v "DEV_MAP[${TMP_ARR[MAP]}]" ]]; then
|
||||
DEV="${DEV_MAP[${TMP_ARR[MAP]}]}"
|
||||
if [[ -f "/sys/class/block/${DEV}/stat" ]]; then
|
||||
DEVICES[${LED}]="${DEV}"
|
||||
DEV_LED_MAP[${DEV}]="${LED}"
|
||||
else
|
||||
/usr/bin/echo '0' > "/sys/class/leds/${LED}/brightness"
|
||||
/usr/bin/echo 'none' > "/sys/class/leds/${LED}/trigger"
|
||||
fi
|
||||
else
|
||||
/usr/bin/echo '0' > "/sys/class/leds/${LED}/brightness"
|
||||
/usr/bin/echo 'none' > "/sys/class/leds/${LED}/trigger"
|
||||
fi
|
||||
fi
|
||||
done
|
||||
|
||||
if [[ "${DISK_ZFS}" == 'true' ]]; then
|
||||
while read LINE; do
|
||||
LINE=(${LINE})
|
||||
DEV_NAME="${LINE[0]}"
|
||||
SCSI_DEV_NAME="unknown"
|
||||
case "${DEV_NAME}" in
|
||||
nvme*)
|
||||
SCSI_DEV_NAME="${DEV_NAME}"
|
||||
;;
|
||||
dm*)
|
||||
DM_SLAVES=($(/usr/bin/ls "/sys/block/${DEV_NAME}/slaves"))
|
||||
SCSI_DEV_NAME="${DM_SLAVES[0]}"
|
||||
;;
|
||||
*)
|
||||
break
|
||||
;;
|
||||
esac
|
||||
if [[ -v "DEV_LED_MAP[${SCSI_DEV_NAME}]" ]]; then
|
||||
ZFS_LED_MAP[${DEV_NAME}]="${DEV_LED_MAP[${SCSI_DEV_NAME}]}"
|
||||
fi
|
||||
done <<< $(/usr/bin/zpool status -L | /usr/bin/grep --extended-regexp ^\\s*\(nvme\|dm\))
|
||||
check_disk_zfs &
|
||||
PID_ZFS="${!}"
|
||||
fi
|
||||
|
||||
if [[ "${DISK_SMART}" == 'true' ]]; then
|
||||
(
|
||||
while true; do
|
||||
for LED in "${!DEVICES[@]}"; do
|
||||
LED_COLOR=$(/usr/bin/cat "/sys/class/leds/${LED}/color")
|
||||
if ! check_disk_health "${LED_COLOR}"; then
|
||||
continue;
|
||||
fi
|
||||
DEV="${DEVICES[${LED}]}"
|
||||
/usr/sbin/smartctl --health "/dev/${DEV}" --nocheck='standby,0' &> '/dev/null'
|
||||
RETURN="${?}"
|
||||
if (( "${RETURN}" & ~32 )); then
|
||||
/usr/bin/echo "${DISK_LED_COLOR_FAILED}" > "/sys/class/leds/${LED}/color"
|
||||
continue
|
||||
fi
|
||||
done
|
||||
/usr/bin/sleep '900s'
|
||||
done
|
||||
) &
|
||||
PID_SMART="${!}"
|
||||
fi
|
||||
|
||||
(
|
||||
while true; do
|
||||
for LED in "${!DEVICES[@]}"; do
|
||||
DEV="${DEVICES[${LED}]}"
|
||||
LED_COLOR=$(/usr/bin/cat "/sys/class/leds/${LED}/color")
|
||||
if ! check_disk_health "${LED_COLOR}"; then
|
||||
continue;
|
||||
fi
|
||||
if [[ ! -f "/sys/class/block/${DEV}/stat" ]]; then
|
||||
/usr/bin/echo "${DISK_LED_COLOR_FAILED}" > "/sys/class/leds/${LED}/color"
|
||||
continue
|
||||
fi
|
||||
done
|
||||
/usr/bin/sleep '15s'
|
||||
done
|
||||
) &
|
||||
PID_DISK="${!}"
|
||||
|
||||
if [[ -x '/usr/sbin/ugreen-led-disk-standby' ]]; then
|
||||
/usr/sbin/ugreen-led-disk-standby '15' "${DISK_LED_COLOR_STANDBY}" "${DISK_LED_COLOR_HEALTH}" $(disk_parameters) &
|
||||
PID_DISK_STANDBY="${!}"
|
||||
fi
|
||||
|
||||
if [[ -x '/usr/sbin/ugreen-led-disk-blink' ]]; then
|
||||
/usr/sbin/ugreen-led-disk-blink '0.1' $(disk_parameters)
|
||||
else
|
||||
declare -A DISKIO_RW
|
||||
while true; do
|
||||
for LED in "${!DEVICES[@]}"; do
|
||||
DISKIO_RW_CMP=$(/usr/bin/cat "/sys/block/${DEVICES[${LED}]}/stat" 2> '/dev/null')
|
||||
if [[ "${DISKIO_RW[${LED}]}" != "${DISKIO_RW_COMP}" ]]; then
|
||||
/usr/bin/echo '1' > "/sys/class/leds/${LED}/shot"
|
||||
fi
|
||||
DISKIO_RW[${LED}]="${DISKIO_RW_CMP}"
|
||||
done
|
||||
/usr/bin/sleep '0.1s'
|
||||
done
|
||||
fi
|
||||
51
root/usr/sbin/ugreen-led-init
Executable file
51
root/usr/sbin/ugreen-led-init
Executable file
@@ -0,0 +1,51 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
|
||||
###
|
||||
#
|
||||
# Options Section
|
||||
#
|
||||
###
|
||||
|
||||
set -u
|
||||
set -o pipefail
|
||||
|
||||
|
||||
###
|
||||
#
|
||||
# Function Section
|
||||
#
|
||||
###
|
||||
|
||||
function load_module () {
|
||||
MODULE="${1}"
|
||||
if ! /usr/sbin/lsmod | /usr/bin/grep --quiet "${MODULE}"; then
|
||||
/usr/sbin/modprobe "${MODULE}"
|
||||
fi
|
||||
}
|
||||
|
||||
|
||||
###
|
||||
#
|
||||
# Runtime Environment
|
||||
#
|
||||
###
|
||||
|
||||
load_module 'i2c_dev'
|
||||
load_module 'led_ugreen'
|
||||
|
||||
I2C_DEV=$(/usr/sbin/i2cdetect -l | /usr/bin/grep 'SMBus I801 adapter' | grep --perl-regexp --only-matching 'i2c-\d+')
|
||||
|
||||
if [[ -n "${I2C_DEV}" ]]; then
|
||||
DEV_PATH="/sys/bus/i2c/devices/$I2C_DEV/${I2C_DEV/i2c-/}-003a"
|
||||
if [[ -d "${DEV_PATH}" ]]; then
|
||||
DEV_NAME=$(/usr/bin/cat "${DEV_PATH}/name")
|
||||
fi
|
||||
if [[ ! -d "${DEV_PATH}" ]]; then
|
||||
/usr/bin/echo 'led-ugreen 0x3a' > "/sys/bus/i2c/devices/${I2C_DEV}/new_device"
|
||||
elif [[ "${DEV_NAME}" != 'led-ugreen' ]]; then
|
||||
exit 1
|
||||
fi
|
||||
else
|
||||
exit 1
|
||||
fi
|
||||
87
root/usr/sbin/ugreen-led-netdev
Executable file
87
root/usr/sbin/ugreen-led-netdev
Executable file
@@ -0,0 +1,87 @@
|
||||
#!/usr/bin/bash
|
||||
|
||||
|
||||
###
|
||||
#
|
||||
# Options Section
|
||||
#
|
||||
###
|
||||
|
||||
set -e
|
||||
set -u
|
||||
set -o pipefail
|
||||
|
||||
|
||||
###
|
||||
#
|
||||
# Variables Section
|
||||
#
|
||||
###
|
||||
|
||||
if [[ -f '/etc/ugreen-nas.conf' ]]; then
|
||||
source '/etc/ugreen-nas.conf'
|
||||
fi
|
||||
|
||||
NET_DEVICE="${1}"
|
||||
NET_SPEED_MIN=${NET_SPEED_MIN:='0'}
|
||||
NET_SPEED_MAX=${NET_SPEED_MAX:='10000'}
|
||||
NET_LED_COLOR_SPEED_MIN=${NET_LED_COLOR_SPEED_MIN:='255 0 0'}
|
||||
NET_LED_COLOR_SPEED_MAX=${NET_LED_COLOR_SPEED_MAX:='0 255 0'}
|
||||
|
||||
|
||||
###
|
||||
#
|
||||
# Function Section
|
||||
#
|
||||
###
|
||||
|
||||
function load_module () {
|
||||
local MODULE
|
||||
MODULE="${1}"
|
||||
if ! /usr/sbin/lsmod | /usr/bin/grep --quiet "${MODULE}"; then
|
||||
/usr/sbin/modprobe "${MODULE}"
|
||||
fi
|
||||
}
|
||||
|
||||
function set_net_led_color () {
|
||||
local IFS
|
||||
NET_SPEED=$(/usr/bin/cat "/sys/class/net/${NET_DEVICE}/speed")
|
||||
NET_SPEED_PERCENTAGE=$(/usr/bin/bc --mathlib <<< "(${NET_SPEED} - ${NET_SPEED_MIN}) / (${NET_SPEED_MAX} - ${NET_SPEED_MIN})")
|
||||
IFS=' '
|
||||
read -r RGB_RED_MIN RGB_GREEN_MIN RGB_BLUE_MIN <<< "${NET_LED_COLOR_SPEED_MIN}"
|
||||
read -r RGB_RED_MAX RGB_GREEN_MAX RGB_BLUE_MAX <<< "${NET_LED_COLOR_SPEED_MAX}"
|
||||
RGB_RED=$(/usr/bin/bc --mathlib <<< "scale=0; ${RGB_RED_MIN} + ${NET_SPEED_PERCENTAGE} * (${RGB_RED_MAX} - ${RGB_RED_MIN})/1")
|
||||
RGB_GREEN=$(/usr/bin/bc --mathlib <<< "scale=0; ${RGB_GREEN_MIN} + ${NET_SPEED_PERCENTAGE} * (${RGB_GREEN_MAX} - ${RGB_GREEN_MIN})/1")
|
||||
RGB_BLUE=$(/usr/bin/bc --mathlib <<< "scale=0; ${RGB_BLUE_MIN} + ${NET_SPEED_PERCENTAGE} * (${RGB_BLUE_MAX} - ${RGB_BLUE_MIN})/1")
|
||||
/usr/bin/echo "${RGB_RED} ${RGB_GREEN} ${RGB_BLUE}" > '/sys/class/leds/netdev/color'
|
||||
}
|
||||
|
||||
|
||||
###
|
||||
#
|
||||
# Runtime Environment
|
||||
#
|
||||
###
|
||||
|
||||
load_module 'ledtrig_netdev'
|
||||
|
||||
/usr/bin/echo 'netdev' > '/sys/class/leds/netdev/trigger'
|
||||
/usr/bin/echo "${NET_DEVICE}" > '/sys/class/leds/netdev/device_name'
|
||||
/usr/bin/echo '1' > '/sys/class/leds/netdev/link'
|
||||
if [[ "${NET_LED_TX_RX}" == 'true' ]]; then
|
||||
/usr/bin/echo '1' > '/sys/class/leds/netdev/tx'
|
||||
/usr/bin/echo '1' > '/sys/class/leds/netdev/rx'
|
||||
fi
|
||||
/usr/bin/echo '250' > '/sys/class/leds/netdev/interval'
|
||||
/usr/bin/echo '255 255 255' > '/sys/class/leds/netdev/color'
|
||||
/usr/bin/echo "${NET_LED_BRIGHTNESS}" > '/sys/class/leds/netdev/brightness'
|
||||
|
||||
while true; do
|
||||
GATEWAY=$(/usr/bin/ip route | /usr/bin/mawk '/default/ { print $3 }')
|
||||
if /usr/bin/ping -q -c '1' -W '1' "${GATEWAY}" &> '/dev/null'; then
|
||||
set_net_led_color
|
||||
else
|
||||
/usr/bin/echo '255 0 0' > '/sys/class/leds/netdev/color'
|
||||
fi
|
||||
/usr/bin/sleep '1m'
|
||||
done
|
||||
40
root/usr/sbin/ugreen-led-power
Normal file
40
root/usr/sbin/ugreen-led-power
Normal file
@@ -0,0 +1,40 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
|
||||
###
|
||||
#
|
||||
# Options Section
|
||||
#
|
||||
###
|
||||
|
||||
set -e
|
||||
set -o pipefail
|
||||
|
||||
|
||||
###
|
||||
#
|
||||
# Variables Section
|
||||
#
|
||||
###
|
||||
|
||||
if [[ -f '/etc/ugreen-nas.conf' ]]; then
|
||||
source '/etc/ugreen-nas.conf'
|
||||
fi
|
||||
|
||||
POWER_LED_BLINK_PATTERN=${POWER_LED_BLINK_PATTERN:='none'}
|
||||
POWER_LED_BRIGHTNESS=${POWER_LED_BRIGHTNESS:='255'}
|
||||
POWER_LED_COLOR=${POWER_LED_COLOR:='255 255 255'}
|
||||
|
||||
|
||||
###
|
||||
#
|
||||
# Runtime Environment
|
||||
#
|
||||
###
|
||||
|
||||
|
||||
if [[ -d '/sys/class/leds/power' ]]; then
|
||||
/usr/bin/echo "${POWER_LED_BLINK_PATTERN}" > '/sys/class/leds/power/blink_type'
|
||||
/usr/bin/echo "${POWER_LED_BRIGHTNESS}" > '/sys/class/leds/power/brightness'
|
||||
/usr/bin/echo "${POWER_LED_COLOR}" > '/sys/class/leds/power/color'
|
||||
fi
|
||||
Reference in New Issue
Block a user