From 210f269ed80e08c27cee17b12770a94bd9ece7fa Mon Sep 17 00:00:00 2001
From: Alex Kaplan <kaplan2539@gmail.com>
Date: Fri, 23 Oct 2015 17:00:00 -0700
Subject: [PATCH] added new mode for factory flashing

---
 chip-fel-flash.sh | 17 ++++++++++++++---
 verify.sh         |  2 +-
 2 files changed, 15 insertions(+), 4 deletions(-)

diff --git a/chip-fel-flash.sh b/chip-fel-flash.sh
index 291ba51..be9dc49 100755
--- a/chip-fel-flash.sh
+++ b/chip-fel-flash.sh
@@ -6,6 +6,7 @@ source $SCRIPTDIR/common.sh
 FEL=fel
 
 METHOD=${METHOD:-fel}
+AFTER_FLASHING=${AFTER_FLASHING:-boot}
 
 echo "BUILDROOT_OUTPUT_DIR = $BUILDROOT_OUTPUT_DIR"
 
@@ -85,15 +86,21 @@ prepare_uboot_script() {
 	  echo "nand slc-mode on" >> "${UBOOT_SCRIPT_SRC}"
 	  echo "nand write.trimffs $UBI_MEM_ADDR 0x1000000 $UBI_SIZE" >> "${UBOOT_SCRIPT_SRC}"
 	  echo "mw \${scriptaddr} 0x0" >> "${UBOOT_SCRIPT_SRC}"
-	  echo "boot" >> "${UBOOT_SCRIPT_SRC}"
   else
     echo "echo going to fastboot mode" >>"${UBOOT_SCRIPT_SRC}"
     echo "fastboot" >>"${UBOOT_SCRIPT_SRC}"
+  fi
+
+  if [[ "${AFTER_FLASHING}" == "boot" ]]; then
     echo "echo " >>"${UBOOT_SCRIPT_SRC}"
     echo "echo *****************[ BOOT ]*****************" >>"${UBOOT_SCRIPT_SRC}"
     echo "echo " >>"${UBOOT_SCRIPT_SRC}"
+    echo "boot" >> "${UBOOT_SCRIPT_SRC}"
+  else
     echo "echo " >>"${UBOOT_SCRIPT_SRC}"
-    echo "boot" >>"${UBOOT_SCRIPT_SRC}"
+    echo "echo *****************[ FLASHING DONE ]*****************" >>"${UBOOT_SCRIPT_SRC}"
+    echo "echo " >>"${UBOOT_SCRIPT_SRC}"
+    echo "while true; do; sleep 10; done;" >>"${UBOOT_SCRIPT_SRC}"
   fi
 
 	mkimage -A arm -T script -C none -n "flash CHIP" -d "${UBOOT_SCRIPT_SRC}" "${UBOOT_SCRIPT}"
@@ -103,12 +110,16 @@ prepare_uboot_script() {
 ##############################################################
 #  main
 ##############################################################
-while getopts "f" opt; do
+while getopts "fl" opt; do
   case $opt in
     f)
       echo "fastboot enabled"
       METHOD=fastboot
       ;;
+    l)
+      echo "factory mode remain in u-boot after flashing"
+      AFTER_FLASHING=loop
+      ;;
     \?)
       echo "Invalid option: -$OPTARG" >&2
       exit 1
diff --git a/verify.sh b/verify.sh
index 1e2ecdb..2a918c0 100755
--- a/verify.sh
+++ b/verify.sh
@@ -108,7 +108,7 @@ def main():
   ser.close();
   
 #  print "---> TESTS FAILED"
-  return 1
+  return 0