From cd7ae718e1aa8ed8b06a3c2ca50ee965abfa1ce7 Mon Sep 17 00:00:00 2001
From: Joseph Chen <chenjh@rock-chips.com>
Date: Mon, 11 Dec 2017 19:16:26 +0800
Subject: [PATCH] rockchip: make.sh: support trust and loader image pack

Change-Id: I9b2cf170138c6aba7578e14126f3010e8bffee3a
Signed-off-by: Joseph Chen <chenjh@rock-chips.com>
---
 make.sh | 109 ++++++++++++++++++++++++++++++++++++++++++++++----------
 1 file changed, 91 insertions(+), 18 deletions(-)

diff --git a/make.sh b/make.sh
index 57f292daf1..6fe2d6add8 100755
--- a/make.sh
+++ b/make.sh
@@ -1,10 +1,45 @@
 #!/bin/sh
+set -e
 BOARD=$1
 DIR=${BOARD#*-}
 DSTDIR=rockdev/${DIR}
+RKCHIP=$(echo $DIR | tr '[a-z]' '[A-Z]')
 TOOLCHAIN=arm-linux-gnueabi-
 JOB=`sed -n "N;/processor/p" /proc/cpuinfo|wc -l`
 
+prepare()
+{
+	local dst
+
+	# Check invaid args and help
+	if [ "$BOARD" = '--help' -o "$BOARD" = '-h' -o "$BOARD" = '--h' ]; then
+		echo
+		echo "Usage: ./make.sh board"
+		echo "Example:"
+		echo "./make.sh evb-rk3399     ---- build for evb-rk3399_defconfig"
+		echo "./make.sh firefly-rk3288 ---- build for firefly-rk3288_defconfig"
+		exit 1
+	elif [ ! -f configs/${BOARD}_defconfig ]; then
+		echo "Can't find: configs/${BOARD}_defconfig"
+		exit 1
+	fi
+
+	# Initialize RKBIN and RKTOOLS
+	dst=../rkbin/tools
+	if [ -d ${dst} ]; then
+		RKBIN=$(cd `dirname ${dst}`; pwd)
+		RKTOOLS=${RKBIN}/tools
+	else
+		echo
+		echo "Can't find '../rkbin/' Responsity, please download it before pack image!"
+		echo "How to obtain? 3 ways:"
+		echo "	1. Login your Rockchip gerrit account: \"Projects\" -> \"List\" -> search \"rk/rkbin\" Responsity"
+		echo "	2. Github Responsity: https://github.com/rockchip-linux/rkbin"
+		echo "	3. Download full release SDK Responsity"
+		exit 1
+	fi
+}
+
 select_toolchain()
 {
 	local dst path
@@ -25,29 +60,67 @@ select_toolchain()
 	echo toolchain: ${TOOLCHAIN}
 }
 
-pack_images()
+pack_uboot_image()
 {
-	local sys_text_base dst
+	local UBOOT_LOAD_ADDR
 
-	dst=../rkbin/tools
-	if [ -d ${dst} ]; then
-		path=$(cd `dirname ${dst}`; pwd)
-	else
-		echo "\nCan't find '../rkbin/' or '../rkbin/tools/' Responsity, please download it before pack image!"
-		echo "How to obtain, 3 ways:"
-		echo "	1. Login your gerrit account: \"Projects\" -> \"List\" -> search rkbin Responsity"
-		echo "	2. Github Responsity: https://github.com/rockchip-linux/rkbin"
-		echo "	3. Download full release SDK Responsity\n"
-		exit 1
-	fi
-
-	sys_text_base=`sed -n "/CONFIG_SYS_TEXT_BASE=/s/CONFIG_SYS_TEXT_BASE=//p" ${DSTDIR}/out/include/autoconf.mk|tr -d '\r'`
-	echo U-Boot entry point address: ${sys_text_base}
-	${path}/tools/loaderimage --pack --uboot ${DSTDIR}/out/u-boot.bin uboot.img ${sys_text_base}
+	UBOOT_LOAD_ADDR=`sed -n "/CONFIG_SYS_TEXT_BASE=/s/CONFIG_SYS_TEXT_BASE=//p" ${DSTDIR}/out/include/autoconf.mk|tr -d '\r'`
+	${RKTOOLS}/loaderimage --pack --uboot ${DSTDIR}/out/u-boot.bin uboot.img ${UBOOT_LOAD_ADDR}
 }
 
+pack_loader_image()
+{
+	cd ${RKBIN}
+	${RKTOOLS}/boot_merger ${RKBIN}/RKBOOT/${RKCHIP}MINIALL.ini
+	cd -
+	mv ${RKBIN}/*_loader_*.bin ./
+}
+
+pack_trust_image()
+{
+	local TOS TOS_TA DARM_BASE TEE_LOAD_ADDR TEE_OFFSET=0x8400000
+
+	# ARM64 uses trust_merger
+	if grep  -q '^CONFIG_ARM64=y' ${DSTDIR}/out/.config ; then
+		cd ${RKBIN}
+		${RKTOOLS}/trust_merger ${RKBIN}/RKTRUST/${RKCHIP}TRUST.ini
+		cd -
+		mv ${RKBIN}/trust.img ./trust.img
+	# ARM uses loaderimage
+	else
+		# OP-TEE is 132M(0x8400000) offset from DRAM base.
+		DARM_BASE=`sed -n "/CONFIG_SYS_SDRAM_BASE=/s/CONFIG_SYS_SDRAM_BASE=//p" ${DSTDIR}/out/include/autoconf.mk|tr -d '\r'`
+		TEE_LOAD_ADDR=$((DARM_BASE+TEE_OFFSET))
+
+		# Convert Dec to Hex
+		TEE_LOAD_ADDR=$(echo "obase=16;${TEE_LOAD_ADDR}"|bc)
+
+		TOS=`sed -n "/TOS=/s/TOS=//p" ${RKBIN}/RKTRUST/${RKCHIP}TOS.ini|tr -d '\r'`
+		TOS_TA=`sed -n "/TOSTA=/s/TOSTA=//p" ${RKBIN}/RKTRUST/${RKCHIP}TOS.ini|tr -d '\r'`
+
+		if [ $TOS_TA -a $TOS ]; then
+			${RKTOOLS}/loaderimage --pack --trustos ${RKBIN}/${TOS} ./trust.img ${TEE_LOAD_ADDR}
+			${RKTOOLS}/loaderimage --pack --trustos ${RKBIN}/${TOS_TA} ./trust_with_ta.img ${TEE_LOAD_ADDR}
+			echo "Both trust.img and trust_with_ta.img are ready"
+		elif [ $TOS ]; then
+			${RKTOOLS}/loaderimage --pack --trustos ${RKBIN}/${TOS} ./trust.img ${TEE_LOAD_ADDR}
+			echo "trust.img is ready"
+		elif [ $TOS_TA ]; then
+			${RKTOOLS}/loaderimage --pack --trustos ${RKBIN}/${TOS_TA} ./trust.img ${TEE_LOAD_ADDR}
+			echo "trust.img with ta is ready"
+		else
+			echo "Can't find any tee bin"
+			exit 1
+		fi
+	fi
+}
+
+prepare
 echo "make for ${BOARD}_defconfig by -j${JOB}"
 make ${BOARD}_defconfig O=${DSTDIR}/out
 select_toolchain
 make CROSS_COMPILE=${TOOLCHAIN}  all --jobs=${JOB} O=${DSTDIR}/out
-pack_images
+pack_loader_image
+pack_uboot_image
+pack_trust_image
+