| #!/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 |
| } |