Initial commit

meta-quanta/meta-gsj/recipes-gsj/quanta-nvme-powerctrl/files/nvme_powerctrl_library.sh

@@ -0,0 +1,65 @@
+#!/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
+}