commit 6221a4a06b0b8400fa190914227996bde937daa3
author Tirupathi Reddy Gillella <tgillella@chromium.org> Thu Jan 18 16:51:53 2018 -0800
committer chrome-bot <chrome-bot@chromium.org> Tue Jan 23 17:42:00 2018 -0800
tree 9c3ac11d6a2e7b72add4e5f47878938970c44c01
parent 8b4432fdd8523bd1fc4e56214327619626572f67

Updated script to check WP after each key increment

BUG=b:35583826
TEST=Tested locally

Change-Id: I5469d34b7b86e28bf8436cc822e3f2344d8ff4d1
Reviewed-on: https://chromium-review.googlesource.com/875159
Commit-Ready: TIRUPATHI REDDY GILLELLA <tgillella@chromium.org>
Tested-by: TIRUPATHI REDDY GILLELLA <tgillella@chromium.org>
Reviewed-by: Kevin Shelton <kmshelton@chromium.org>

provingground/firmware/keyinc_setup.sh

diff --git a/provingground/firmware/keyinc_setup.sh b/provingground/firmware/keyinc_setup.sh
index 764e9da..6af3b4d 100755
--- a/provingground/firmware/keyinc_setup.sh
+++ b/provingground/firmware/keyinc_setup.sh
@@ -37,10 +37,13 @@
 # CHANNEL=dev
 EOF
 
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+NC='\033[0m'
 
 log()
 {
-  echo -e $(date '+[%H%M%S-%m%d]') "$*"
+  echo -e $(date '+[%H%M%S-%m%d]') "${GREEN}$*${NC}"
 }
 
 generate_payload()
@@ -324,7 +327,7 @@
 
 exit_on_error()
 {
-  echo $(date '+[%H%M%S-%m%d]') ERROR:- "$*"
+  echo -e $(date '+[%H%M%S-%m%d]') "${RED}ERROR:- $*${NC}"
   exit 1
 }
 
@@ -395,6 +398,33 @@
 
 }
 
+check_write_protects()
+{
+  log "---------------------WRITE PROTECTS ARE------------------------"
+  cat <<EOF | ssh -T root@$DUT_IP
+  crossystem 2>/dev/null |grep ^wpsw_
+  echo -n "HOST WRITE PROTECT IS: "
+  echo \$(flashrom -p host --wp-status 2>/dev/null | grep "write protect" \
+          | tr '\n' ' ' | awk '{print \$5, \$(NF-1), \$NF}')
+
+  ectool version > /dev/null 2>&1
+  if [ \$? -eq 0 ]; then
+    echo -n "EC WRITE PROTECT IS:   "
+    echo \$(flashrom -p ec --wp-status 2>/dev/null |grep "write protect" \
+            | tr '\n' ' ' | awk '{print \$5, \$(NF-1), \$NF}')
+  fi
+
+  ectool --dev=1 version > /dev/null 2>&1
+  if [ \$? -eq 0 ]; then
+    echo -n "PD WRITE PROTECT IS:   "
+    echo \$(flashrom -p ec:dev=1 --wp-status 2>/dev/null |grep "write protect" \
+            | tr '\n' ' ' | awk '{print \$5, \$(NF-1), \$NF}')
+  fi
+
+EOF
+  log "---------------------------------------------------------------"
+}
+
 
 verify_keyincrement()
 {
@@ -420,6 +450,7 @@
 
     sleep 45
     crossystem_verification $auto_update
+    check_write_protects
     sleep 60
 
     ssh root@$DUT_IP "grep '$SUCCESSFUL_UPDATE_TOKEN' $UPDATE_ENGINE_LOG"
@@ -436,6 +467,7 @@
     ssh root@$DUT_IP "reboot"
 
     sleep 90
+    check_write_protects
     get_logs $auto_update "Autoupdate $auto_update completed"
 
   done