fastboot.sh: update scripts to flash fit image

Signed-off-by: Hongmei Gou <a0271529@ti.com>
Reviewed-by: Sam Protsenko <semen.protsenko@linaro.org>

1 files changed, 6 insertions, 1 deletions

diff --git a/fastboot.sh b/fastboot.sh
index 49556f7..7cc73bf 100755
--- a/fastboot.sh
+++ b/fastboot.sh
@@ -89,6 +89,7 @@ userdataimg="${PRODUCT_OUT}userdata.img"
 cacheimg="${PRODUCT_OUT}cache.img"
 recoveryimg="${PRODUCT_OUT}recovery.img"
 vendorimg="${PRODUCT_OUT}vendor.img"
+bootfitimg="${PRODUCT_OUT}boot_fit.img"
 
 # Verify that all the files required for the fastboot flash
 # process are available
@@ -125,6 +126,10 @@ if [ ! -e "${vendorimg}" ] ; then
   echo "Missing ${vendorimg}"
   exit -1;
 fi
+if [ ! -e "${bootfitimg}" ] ; then
+  echo "Missing ${bootfitimg}"
+  exit -1;
+fi
 
 
 echo "Create GPT partition table"
@@ -154,7 +159,7 @@ ${FASTBOOT} flash system	${systemimg}
 ${FASTBOOT} flash vendor        ${vendorimg}
 
 echo "Flashing FIT Boot Image"
-${FASTBOOT} flash boot boot_fit.img
+${FASTBOOT} flash boot ${bootfitimg}
 
 
 userdataimg_orig="${userdataimg}.orig"