From 299a3d8f1a3a6216a2b8f2a38c2d97f89e22b521 Mon Sep 17 00:00:00 2001
From: Mike Choi <mikechoi@fb.com>
Date: Mon, 22 Oct 2018 16:24:16 -0700
Subject: [PATCH] Adding PIM reset feature to wedge_power.sh

Summary:
[YAMP] Adding PIM reset feature to wedge_power.sh
Internal logic for this is totally different from that of Accton Minipack,
but we still add the same UI (the same script parameter) as Minipack for tool compatibility

Test Plan: Tested on MPK17 Glacier #17

Reviewed By: benwei13

fbshipit-source-id: 08999f67e
---
 .../openbmc-utils/files/wedge_power.sh        | 83 +++++++++++++++++++
 1 file changed, 83 insertions(+)

diff --git a/meta-facebook/meta-yamp/recipes-utils/openbmc-utils/files/wedge_power.sh b/meta-facebook/meta-yamp/recipes-utils/openbmc-utils/files/wedge_power.sh
index cb0710555b5a..29956efd8b70 100644
--- a/meta-facebook/meta-yamp/recipes-utils/openbmc-utils/files/wedge_power.sh
+++ b/meta-facebook/meta-yamp/recipes-utils/openbmc-utils/files/wedge_power.sh
@@ -41,6 +41,11 @@ usage() {
     echo "    options:"
     echo "      -s: Power reset whole yamp system ungracefully"
     echo
+    echo "  pimreset: Power-cycle one or all PIM(s)"
+    echo "    options:"
+    echo "      -a  : Reset all PIMs or "
+    echo "      -2 , -3 , ... , -9 : Reset a single PIM (2, 3 ... 9) "
+    echo
 }
 
 do_status() {
@@ -137,6 +142,81 @@ do_reset() {
     fi
 }
 
+toggle_pim_reset() {
+  pim=$1
+  # First, put the PIMs we want into reset
+  for slot in 2 3 4 5 6 7 8 9; do
+    if [ $pim -eq 0 ] || [ $slot -eq $pim ]; then
+      # Unlike Minipack, YAMP pim index is 1 based
+      index=$(expr $slot - 1)
+      # Unlike Minipack, YAMP uses GPIO
+      echo Power-cycling PIM in slot $slot
+      echo out > /tmp/gpionames/LC${index}_LC_PWR_CYC/direction
+      echo high > /tmp/gpionames/LC${index}_LC_PWR_CYC/direction
+    fi
+  done
+  # Sleep 1 sec
+  sleep 1
+  # Finally, put the LC_PWR_CYC bit to 0
+  for slot in 2 3 4 5 6 7 8 9; do
+    if [ $pim -eq 0 ] || [ $slot -eq $pim ]; then
+      # Unlike Minipack, YAMP pim index is 1 based
+      index=$(expr $slot - 1)
+      # Unlike Minipack, YAMP uses GPIO
+      echo Putting PIM in $slot into normal operation
+      echo low > /tmp/gpionames/LC${index}_LC_PWR_CYC/direction
+    fi
+  done
+}
+do_pimreset() {
+    local pim opt retval rc
+    retval=0
+    pim=-1
+    while getopts "23456789a" opt; do
+        case $opt in
+            a)
+                pim=0
+                ;;
+            2)
+                pim=2
+                ;;
+            3)
+                pim=3
+                ;;
+            4)
+                pim=4
+                ;;
+            5)
+                pim=5
+                ;;
+            6)
+                pim=6
+                ;;
+            7)
+                pim=7
+                ;;
+            8)
+                pim=8
+                ;;
+            9)
+                pim=9
+                ;;
+            *)
+                usage
+                exit -1
+                ;;
+        esac
+    done
+    if [ $pim -eq -1 ]; then
+      usage
+      exit -1
+    fi
+
+    toggle_pim_reset $pim
+
+    return $retval
+}
+
 if [ $# -lt 1 ]; then
     usage
     exit -1
@@ -158,6 +238,9 @@ case "$command" in
     reset)
         do_reset $@
         ;;
+    pimreset)
+        do_pimreset $@
+        ;;
     *)
         usage
         exit -1