From 9211092c4aa914ee9f5050f9a24a3e01d06d1226 Mon Sep 17 00:00:00 2001
From: computermouth <computermouth@crunchbangplusplus.org>
Date: Mon, 5 Dec 2016 16:56:05 -0800
Subject: [PATCH] Added local flash option

---
 chip-update-firmware.sh | 39 ++++++++++++++++++++++++++++-----------
 common.sh               | 26 ++++++++++++--------------
 2 files changed, 40 insertions(+), 25 deletions(-)

diff --git a/chip-update-firmware.sh b/chip-update-firmware.sh
index 6ca92b1..5179600 100755
--- a/chip-update-firmware.sh
+++ b/chip-update-firmware.sh
@@ -25,7 +25,7 @@ UBI_PREFIX="chip"
 UBI_SUFFIX="ubi.sparse"
 UBI_TYPE="400000-4000-680"
 
-while getopts "sgpbfnrhB:N:F:" opt; do
+while getopts "sgpbfnrhB:N:F:L:" opt; do
   case $opt in
     s)
       echo "== Server selected =="
@@ -67,6 +67,10 @@ while getopts "sgpbfnrhB:N:F:" opt; do
       FORMAT="$OPTARG"
       echo "== Format ${FORMAT} selected =="
       ;;
+    L)
+      LOCALDIR="$OPTARG"
+      echo "== Local directory '${LOCALDIR}' selected =="
+      ;;
     h)
       echo ""
       echo "== Help =="
@@ -81,6 +85,7 @@ while getopts "sgpbfnrhB:N:F:" opt; do
       echo "  -B  --  Branch(optional)   [eg. -B testing]"
       echo "  -N  --  Build#(optional)   [eg. -N 150]"
       echo "  -F  --  Format(optional)   [eg. -F Toshiba_4G_MLC]"
+      echo "  -L  --  Local (optional)   [eg. -L ../img/buildroot/]"
       echo ""
       echo ""
       exit 0
@@ -104,7 +109,7 @@ function dl_probe {
     CACHENUM=$(curl -s $DL_URL/$BRANCH/$FLAVOR/latest)
   fi
 
-  if [[ ! -d "$DL_DIR/$BRANCH-$FLAVOR-b${CACHENUM}" ]]; then
+  if [[ ! -d "$DL_DIR/$BRANCH-$FLAVOR-b${CACHENUM}" ]] && [[ -z $LOCALDIR ]]; then
     echo "== New image available =="
 
     rm -rf $DL_DIR/$BRANCH-$FLAVOR*
@@ -121,52 +126,64 @@ function dl_probe {
     done
     popd > /dev/null
   else
-    echo "== Cached probe files located =="
+    echo "== Local/cached probe files located =="
   fi
 
   echo "== Staging for NAND probe =="
-  ln -s ../../$DL_DIR/${BRANCH}-${FLAVOR}-b${CACHENUM}/ $IMAGESDIR
+  if [ -z LOCALDIR ];then
+    ln -s ../../$DL_DIR/${BRANCH}-${FLAVOR}-b${CACHENUM}/ $IMAGESDIR
+  else
+    ln -s ../../$LOCALDIR $IMAGESDIR
+  fi
+
   if [[ -f ${IMAGESDIR}/ubi_type ]]; then rm ${IMAGESDIR}/ubi_type; fi
 
   if [ -z $FORMAT ]; then
-    detect_nand
+    detect_nand || exit 1
   else
     case $FORMAT in
       "Hynix_8G_MLC")
         export nand_erasesize=400000
         export nand_oobsize=680
         export nand_writesize=4000
-        UBI_TYPE="400000-4000-680"
       ;;
       "Toshiba_4G_MLC")
         export nand_erasesize=400000
         export nand_oobsize=500
         export nand_writesize=4000
-        UBI_TYPE="400000-4000-500"
       ;;
       "Toshiba_512M_MLC")
         export nand_erasesize=40000
         export nand_oobsize=100
         export nand_writesize=1000
-        UBI_TYPE="400000-1000-100"
       ;;
       \?)
     	echo "== Invalid format: $FORMAT ==" >&2
     	exit 1
     ;;
     esac
+    UBI_TYPE="$nand_erasesize-$nand_writesize-$nand_oobsize"
   fi
 
-  if [[ ! -f "$DL_DIR/$BRANCH-$FLAVOR-b${CACHENUM}/$UBI_PREFIX-$UBI_TYPE.$UBI_SUFFIX" ]]; then
+  if [[ ! -f "$DL_DIR/$BRANCH-$FLAVOR-b${CACHENUM}/$UBI_PREFIX-$UBI_TYPE.$UBI_SUFFIX" ]] && [ -z $LOCALDIR ]; then
     echo "== Downloading new UBI, this will be cached for future flashes. =="
     pushd $DL_DIR/${BRANCH}-${FLAVOR}-b${CACHENUM} > /dev/null
     if ! $WGET $DL_URL/$BRANCH/$FLAVOR/${CACHENUM}/$UBI_PREFIX-$UBI_TYPE.$UBI_SUFFIX; then
       echo "!! download of $BRANCH-$FLAVOR-$METHOD-b${CACHENUM} failed !!"
-    exit $?
+      exit $?
     fi
     popd > /dev/null
-    else
+  else
+    if [ -z $LOCALDIR ]; then
       echo "== Cached UBI located =="
+    else
+      if [[ ! -f "$IMAGESDIR/$UBI_PREFIX-$UBI_TYPE.$UBI_SUFFIX" ]]; then
+        echo "Could not locate UBI files"
+        exit 1
+      else
+        echo "== Cached UBI located =="
+      fi
+    fi
   fi
 }
 
diff --git a/common.sh b/common.sh
index 8e02494..5084c2e 100755
--- a/common.sh
+++ b/common.sh
@@ -69,7 +69,6 @@ wait_for_fel() {
 
 #------------------------------------------------------------
 detect_nand() {
-  local RC=0
 
   local tmpdir=`mktemp -d -t chip-uboot-script-XXXXXX`
   local ubootcmds=$tmpdir/uboot.cmds
@@ -78,38 +77,37 @@ detect_nand() {
   echo "nand info
 env export -t -s 0x100 0x7c00 nand_erasesize nand_writesize nand_oobsize
 reset" > $ubootcmds
-  mkimage -A arm -T script -C none -n "detect NAND" -d $ubootcmds $ubootscr || RC=1
+  mkimage -A arm -T script -C none -n "detect NAND" -d $ubootcmds $ubootscr || return 1
 
   if ! wait_for_fel; then
     echo "ERROR: please make sure CHIP is connected and jumpered in FEL mode"
-    RC=1
-    exit 1
+    return 1
   fi
 
-  $FEL spl $IMAGESDIR/sunxi-spl.bin || RC=1
+  $FEL spl $IMAGESDIR/sunxi-spl.bin || return 1
   # wait for DRAM initialization to complete
   sleep 1
 
-  $FEL write $UBOOTMEMADDR $IMAGESDIR/u-boot-dtb.bin || RC=1
-  $FEL write $UBOOTSCRMEMADDR $ubootscr || RC=1
-  $FEL exe $UBOOTMEMADDR || RC=1
+  $FEL write $UBOOTMEMADDR $IMAGESDIR/u-boot-dtb.bin || return 1
+  $FEL write $UBOOTSCRMEMADDR $ubootscr || return 1
+  $FEL exe $UBOOTMEMADDR || return 1
 
   if ! wait_for_fel; then
     echo "ERROR: please make sure CHIP is connected and jumpered in FEL mode"
-    exit 1
+    return 1
   fi
 
-  $FEL read 0x7c00 0x100 $tmpdir/nand-info || RC=1
+  $FEL read 0x7c00 0x100 $tmpdir/nand-info || return 1
 
   echo "NAND detected:"
-  cat $tmpdir/nand-info || RC=1
+  cat $tmpdir/nand-info || return 1
   UBI_TYPE="$(cat $tmpdir/nand-info | awk -F= '/erase/ {print $2}')-$(cat $tmpdir/nand-info | awk -F= '/write/ {print $2}')-$(cat $tmpdir/nand-info | awk -F= '/oob/ {print $2}')"
-  echo "${UBI_TYPE}" > $IMAGESDIR/ubi_type || RC=1
-  source $tmpdir/nand-info || RC=1
+  echo "${UBI_TYPE}" > $IMAGESDIR/ubi_type || return 1
+  source $tmpdir/nand-info || return 1
 
   rm -rf $tmpdir
 
-  return $RC
+  return 0
 }
 
 #------------------------------------------------------------