From 3bcf0b7c0e5c2d54fae431c2586315c9bc44125c Mon Sep 17 00:00:00 2001
From: Joseph Chen <chenjh@rock-chips.com>
Date: Thu, 17 Dec 2020 18:07:47 +0800
Subject: [PATCH] scripts: pack_resource.sh: support android and fit image

./scripts/pack_resource.sh boot.img
./scripts/pack_resource.sh recovery.img

Signed-off-by: Joseph Chen <chenjh@rock-chips.com>
Change-Id: I067b00ecbadaab78962605e0caa03e04ff4f3c02
---
 scripts/pack_resource.sh | 35 +++++++++++++++++++++++++++++++++--
 1 file changed, 33 insertions(+), 2 deletions(-)

diff --git a/scripts/pack_resource.sh b/scripts/pack_resource.sh
index 2f46a07596..a6e7baef7c 100755
--- a/scripts/pack_resource.sh
+++ b/scripts/pack_resource.sh
@@ -7,9 +7,14 @@
 
 RSCE_OLD=$1
 RSCE_NEW=resource.img
+
+BOOT_OLD=$1
+BOOT_NEW=boot.img
+
 TOOL=../rkbin/tools/resource_tool
 IMAGES=./tools/images/
 TMP_DIR=.resource_tmp
+TMP_DIR2=.boot_tmp
 
 usage()
 {
@@ -61,6 +66,32 @@ append_images_to_resource()
 	echo "./resource.img with battery images is ready"
 }
 
-prepare
-append_images_to_resource
+append_images_to_android_img()
+{
+	./scripts/unpack_bootimg --boot_img ${BOOT_OLD} --out ${TMP_DIR2}/
+	RSCE_OLD="${TMP_DIR2}/second"
+	append_images_to_resource
+	./scripts/repack-bootimg --boot_img ${BOOT_OLD} --second ${RSCE_NEW} -o ${BOOT_NEW}
+	rm -rf ${TMP_DIR2}
+}
+
+append_images_to_fit_img()
+{
+	./scripts/fit-unpack.sh -f ${BOOT_OLD} -o ${TMP_DIR2}/
+	RSCE_OLD="${TMP_DIR2}/resource"
+	append_images_to_resource
+	rm -rf ${TMP_DIR2}/*
+	mv ${RSCE_NEW} ${TMP_DIR2}/resource
+	 ./scripts/fit-repack.sh -f ${BOOT_OLD} -d ${TMP_DIR2}
+	rm -rf ${TMP_DIR2}
+}
+
+prepare
+if file ${RSCE_OLD} | grep 'Android bootimg' >/dev/null 2>&1 ; then
+	append_images_to_android_img
+elif file ${RSCE_OLD} | grep 'Device Tree Blob' >/dev/null 2>&1 ; then
+	append_images_to_fit_img
+elif strings ${RSCE_OLD} | grep "RSCE" >/dev/null 2>&1 ; then
+	append_images_to_resource
+fi