Your Name
66 lines
1.6 KiB
Bash
66 lines
1.6 KiB
Bash
#!/bin/bash
|
|
|
|
export U2_PRESENT=( 148 149 150 151 152 153 154 155 )
|
|
export POWER_U2=( 195 196 202 199 198 197 127 126 )
|
|
export PWRGD_U2=( 161 162 163 164 165 166 167 168 )
|
|
export RST_BMC_U2=( 72 73 74 75 76 77 78 79 )
|
|
export PLUGGED=0
|
|
export I2C_BUS=8
|
|
export CHIP_ADDR=0x68
|
|
export CLOCK_GEN_VALUE=
|
|
|
|
CLOCK_GEN_VALUE=$(i2cget -y $I2C_BUS $CHIP_ADDR 0 i 2|cut -f3 -d' ')
|
|
|
|
function set_gpio_direction()
|
|
{
|
|
#$1 gpio pin, $2 'in','high','low'
|
|
echo "$2" > "/sys/class/gpio/gpio$1/direction"
|
|
}
|
|
|
|
function read_gpio_input()
|
|
{
|
|
#$1 read input gpio pin
|
|
cat "/sys/class/gpio/gpio$1/value"
|
|
}
|
|
|
|
function enable_nvme_power()
|
|
{
|
|
set_gpio_direction "${POWER_U2[$1]}" "high"
|
|
sleep 0.04
|
|
check_powergood "$1"
|
|
}
|
|
|
|
function check_powergood()
|
|
{
|
|
if [ "$(read_gpio_input "${PWRGD_U2[$1]}")" == 1 ];then
|
|
sleep 0.005
|
|
update_clock_gen_chip_register "$1" 1
|
|
sleep 0.1
|
|
set_gpio_direction "${RST_BMC_U2[$1]}" "high"
|
|
else
|
|
disable_nvme_power "$1"
|
|
fi
|
|
}
|
|
|
|
function disable_nvme_power()
|
|
{
|
|
set_gpio_direction "${RST_BMC_U2[$1]}" "low"
|
|
sleep 0.1
|
|
update_clock_gen_chip_register "$1" 0
|
|
sleep 0.005
|
|
set_gpio_direction "${POWER_U2[$1]}" "low"
|
|
}
|
|
|
|
function update_clock_gen_chip_register(){
|
|
#$1 nvme slot number, $2 enable/disable
|
|
update_value=$(printf '%x\n' "$((0x01 <<$1))")
|
|
if [ "$2" -eq 1 ];then
|
|
CLOCK_GEN_VALUE=$(printf '0x%x\n' \
|
|
"$((CLOCK_GEN_VALUE | 0x$update_value))")
|
|
else
|
|
CLOCK_GEN_VALUE=$(printf '0x%x\n' \
|
|
"$((CLOCK_GEN_VALUE & ~0x$update_value))")
|
|
fi
|
|
i2cset -y $I2C_BUS $CHIP_ADDR 0 "$CLOCK_GEN_VALUE" s
|
|
}
|