xref: /rk3399_rockchip-uboot/make.sh (revision b9fb3eda732297f8c84a06ded62543f5579fc0eb)
1#!/bin/bash
2#
3# Copyright (c) 2019 Fuzhou Rockchip Electronics Co., Ltd
4#
5# SPDX-License-Identifier: GPL-2.0
6#
7
8set -e
9JOB=`sed -n "N;/processor/p" /proc/cpuinfo|wc -l`
10SUPPORT_LIST=`ls configs/*[r,p][x,v,k][0-9][0-9]*_defconfig`
11CMD_ARGS=$1
12
13########################################### User can modify #############################################
14RKBIN_TOOLS=../rkbin/tools
15CROSS_COMPILE_ARM32=../prebuilts/gcc/linux-x86/arm/gcc-linaro-6.3.1-2017.05-x86_64_arm-linux-gnueabihf/bin/arm-linux-gnueabihf-
16CROSS_COMPILE_ARM64=../prebuilts/gcc/linux-x86/aarch64/gcc-linaro-6.3.1-2017.05-x86_64_aarch64-linux-gnu/bin/aarch64-linux-gnu-
17########################################### User not touch #############################################
18# Declare global INI file searching index name for every chip, update in select_chip_info()
19RKCHIP=
20RKCHIP_LABEL=
21RKCHIP_LOADER=
22RKCHIP_TRUST=
23INI_TRUST=
24INI_LOADER=
25
26# Declare rkbin repository path, updated in prepare()
27RKBIN=
28
29# Declare global toolchain path for CROSS_COMPILE, updated in select_toolchain()
30TOOLCHAIN=
31TOOLCHAIN_NM=
32TOOLCHAIN_OBJDUMP=
33TOOLCHAIN_ADDR2LINE=
34
35# Declare global plaform configure, updated in fixup_platform_configure()
36PLAT_RSA=
37PLAT_SHA=
38PLAT_UBOOT_SIZE=
39PLAT_TRUST_SIZE=
40PLAT_TYPE="RKFW" # default
41
42SRCTREE=`pwd`
43SCRIPT_FIT="${SRCTREE}/scripts/fit.sh"
44SCRIPT_ATF="${SRCTREE}/scripts/atf.sh"
45SCRIPT_TOS="${SRCTREE}/scripts/tos.sh"
46SCRIPT_SPL="${SRCTREE}/scripts/spl.sh"
47SCRIPT_UBOOT="${SRCTREE}/scripts/uboot.sh"
48SCRIPT_LOADER="${SRCTREE}/scripts/loader.sh"
49SCRIPT_DECOMP="${SRCTREE}/scripts/decomp.sh"
50CC_FILE=".cc"
51REP_DIR="./rep"
52#########################################################################################################
53function help()
54{
55	echo
56	echo "Usage:"
57	echo "	./make.sh [board|sub-command]"
58	echo
59	echo "	 - board:        board name of defconfig"
60	echo "	 - sub-command:  elf*|loader|trust|uboot|--spl|--tpl|itb|map|sym|<addr>"
61	echo "	 - ini:          ini file to pack trust/loader"
62	echo
63	echo "Output:"
64	echo "	 When board built okay, there are uboot/trust/loader images in current directory"
65	echo
66	echo "Example:"
67	echo
68	echo "1. Build:"
69	echo "	./make.sh evb-rk3399               --- build for evb-rk3399_defconfig"
70	echo "	./make.sh firefly-rk3288           --- build for firefly-rk3288_defconfig"
71	echo "	./make.sh EXT_DTB=rk-kernel.dtb    --- build with exist .config and external dtb"
72	echo "	./make.sh                          --- build with exist .config"
73	echo "	./make.sh env                      --- build envtools"
74	echo
75	echo "2. Pack:"
76	echo "	./make.sh uboot                    --- pack uboot.img"
77	echo "	./make.sh trust                    --- pack trust.img"
78	echo "	./make.sh trust <ini>              --- pack trust img with assigned ini file"
79	echo "	./make.sh loader                   --- pack loader bin"
80	echo "	./make.sh loader <ini>             --- pack loader img with assigned ini file"
81	echo "	./make.sh --spl                    --- pack loader with u-boot-spl.bin"
82	echo "	./make.sh --tpl                    --- pack loader with u-boot-tpl.bin"
83	echo "	./make.sh --tpl --spl              --- pack loader with u-boot-tpl.bin and u-boot-spl.bin"
84	echo
85	echo "3. Debug:"
86	echo "	./make.sh elf                      --- dump elf file with -D(default)"
87	echo "	./make.sh elf-S                    --- dump elf file with -S"
88	echo "	./make.sh elf-d                    --- dump elf file with -d"
89	echo "	./make.sh elf-*                    --- dump elf file with -*"
90	echo "	./make.sh <no reloc_addr>          --- unwind address(no relocated)"
91	echo "	./make.sh <reloc_addr-reloc_off>   --- unwind address(relocated)"
92	echo "	./make.sh map                      --- cat u-boot.map"
93	echo "	./make.sh sym                      --- cat u-boot.sym"
94}
95
96function prepare()
97{
98	if [ -d ${RKBIN_TOOLS} ]; then
99		absolute_path=$(cd `dirname ${RKBIN_TOOLS}`; pwd)
100		RKBIN=${absolute_path}
101	else
102		echo "ERROR: No ../rkbin repository"
103		exit 1
104	fi
105
106	if grep -Eq ''^CONFIG_ARM64=y'|'^CONFIG_ARM64_BOOT_AARCH32=y'' .config ; then
107		ARM64_TRUSTZONE="y"
108	fi
109
110	if grep  -q '^CONFIG_ROCKCHIP_FIT_IMAGE_PACK=y' .config ; then
111		PLAT_TYPE="FIT"
112	elif grep  -q '^CONFIG_SPL_DECOMP_HEADER=y' .config ; then
113		PLAT_TYPE="DECOMP"
114	fi
115}
116
117function process_args()
118{
119	while [ $# -gt 0 ]; do
120		case $1 in
121			*help|--h|-h)
122				help
123				exit 0
124				;;
125			CROSS_COMPILE=*)  # set CROSS_COMPILE
126				ARG_COMPILE="y"
127				CROSS_COMPILE_ARM32=${1#*=}
128				CROSS_COMPILE_ARM64=${1#*=}
129				if [ ${CMD_ARGS} == $1 ]; then
130					shift 1
131					CMD_ARGS=$1
132				else
133					shift 1
134				fi
135				;;
136			# '': build with exist .config
137			# loader|trust|uboot: pack image
138			# debug*|map|sym|elf*|nm: debug command
139			# env: build env tool
140			# itb: pack itb image
141			# fit: pack non-secure uboot.img && boot.img
142			''|loader|trust|uboot|debug*|itb|env|fit|map|sym|elf*|nm)
143				if [ "$2" == "spl" -o "$2" == "tpl" ]; then
144					ARG_TSPL=$2
145					shift 1
146				fi
147				shift 1
148				;;
149			# trust/loader ini files for packing trust.img/loader.img
150			*.ini|*.INI)
151				if [ ! -f $1 ]; then
152					echo "ERROR: No $1"
153				fi
154				if grep -q 'CODE471_OPTION' $1 ; then
155					ARG_INI_LOADER=$1
156				elif grep -Eq ''BL31_OPTION'|'TOS'' $1 ; then
157					ARG_INI_TRUST=$1
158				fi
159				shift 1
160				;;
161			--sz-trust) # set trust size
162				ARG_TRUST_SIZE="--size $2 $3"
163				shift 3
164				;;
165			--sz-uboot) # set uboot size
166				ARG_UBOOT_SIZE="--size $2 $3"
167				shift 3
168				;;
169			--raw-compile)  # FIT: build but not pack image
170				ARG_RAW_COMPILE="y"
171				shift 1
172				;;
173			--no-uboot) # FIT: pack uboot.img without u-boot
174				ARG_NO_UBOOT="y"
175				shift 1
176				;;
177			--idblock)  # pack idblock.bin
178				shift 1
179				;;
180			--tpl|tpl)  # use tpl file
181				ARG_TPL_BIN="tpl/u-boot-tpl.bin"
182				shift 1
183				;;
184			--spl|spl*) # use spl file
185				ARG_SPL_BIN="spl/u-boot-spl.bin"
186				shift 1
187				;;
188			--uboot|--fdt|--optee|--mcu|--bl31) # uboot.img components
189				mkdir -p ${REP_DIR}
190				if [ ! -f $2 ]; then
191					echo "ERROR: No $2"
192					exit 1
193				fi
194				if [ "$1" == "--uboot" ]; then
195					cp $2 ${REP_DIR}/u-boot-nodtb.bin
196				elif [ "$1" == "--fdt" ]; then
197					cp $2 ${REP_DIR}/u-boot.dtb
198				elif [ "$1" == "--optee" ]; then
199					cp $2 ${REP_DIR}/tee.bin
200				elif [ "$1" == "--mcu" ]; then
201					cp $2 ${REP_DIR}/mcu.bin
202				elif [ "$1" == "--bl31" ]; then
203					if ! file $2 | grep 'ELF ' >/dev/null 2>&1 ; then
204						echo "ERROR: $2 is not a bl31.elf file"
205						exit 1
206					fi
207					cp $2 ${REP_DIR}/bl31.elf
208				fi
209				shift 2
210				;;
211			*)
212				#1. FIT scripts args
213				NUM=$(${SCRIPT_FIT} --args $1)
214				if  [ ${NUM} -ne 0 ]; then
215					[ ${NUM} -eq 1 ] && ARG_LIST_FIT="${ARG_LIST_FIT} $1"
216					[ ${NUM} -eq 2 ] && ARG_LIST_FIT="${ARG_LIST_FIT} $1 $2"
217					shift ${NUM}
218					continue
219				#2. unwind function address
220				elif [ -z $(echo $1 | sed 's/[0-9,a-f,A-F,x,X,-]//g') ]; then
221					ARG_FUNCADDR=$1
222				#3. make defconfig
223				else
224					ARG_BOARD=$1
225					if [ ! -f configs/${ARG_BOARD}_defconfig -a ! -f configs/${ARG_BOARD}.config ]; then
226						echo -e "\n${SUPPORT_LIST}\n"
227						echo "ERROR: No configs/${ARG_BOARD}_defconfig"
228						exit 1
229					elif [ -f configs/${ARG_BOARD}.config ]; then
230						BASE1_DEFCONFIG=`sed -n "/CONFIG_BASE_DEFCONFIG=/s/CONFIG_BASE_DEFCONFIG=//p" configs/${ARG_BOARD}.config |tr -d '\r' | tr -d '"'`
231						BASE0_DEFCONFIG=`sed -n "/CONFIG_BASE_DEFCONFIG=/s/CONFIG_BASE_DEFCONFIG=//p" configs/${BASE1_DEFCONFIG} |tr -d '\r' | tr -d '"'`
232						MAKE_CMD="make ${BASE0_DEFCONFIG} ${BASE1_DEFCONFIG} ${ARG_BOARD}.config -j${JOB}"
233						echo "## ${MAKE_CMD}"
234						make ${BASE0_DEFCONFIG} ${BASE1_DEFCONFIG} ${ARG_BOARD}.config ${OPTION}
235						rm -f ${CC_FILE}
236					else
237						MAKE_CMD="make ${ARG_BOARD}_defconfig -j${JOB}"
238						echo "## ${MAKE_CMD}"
239						make ${ARG_BOARD}_defconfig ${OPTION}
240						rm -f ${CC_FILE}
241					fi
242				fi
243				shift 1
244				;;
245		esac
246	done
247
248	if [ ! -f .config ]; then
249		echo
250		echo "ERROR: No .config"
251		help
252		exit 1
253	fi
254}
255
256function select_toolchain()
257{
258	# If no outer CROSS_COMPILE, look for it from CC_FILE.
259	if [ "${ARG_COMPILE}" != "y" ]; then
260		if [ -f ${CC_FILE} ]; then
261			CROSS_COMPILE_ARM32=`cat ${CC_FILE}`
262			CROSS_COMPILE_ARM64=`cat ${CC_FILE}`
263		else
264			if grep -q '^CONFIG_ARM64=y' .config ; then
265				CROSS_COMPILE_ARM64=$(cd `dirname ${CROSS_COMPILE_ARM64}`; pwd)"/aarch64-linux-gnu-"
266			else
267				CROSS_COMPILE_ARM32=$(cd `dirname ${CROSS_COMPILE_ARM32}`; pwd)"/arm-linux-gnueabihf-"
268			fi
269		fi
270	fi
271
272	if grep -q '^CONFIG_ARM64=y' .config ; then
273		TOOLCHAIN=${CROSS_COMPILE_ARM64}
274		TOOLCHAIN_NM=${CROSS_COMPILE_ARM64}nm
275		TOOLCHAIN_OBJDUMP=${CROSS_COMPILE_ARM64}objdump
276		TOOLCHAIN_ADDR2LINE=${CROSS_COMPILE_ARM64}addr2line
277	else
278		TOOLCHAIN=${CROSS_COMPILE_ARM32}
279		TOOLCHAIN_NM=${CROSS_COMPILE_ARM32}nm
280		TOOLCHAIN_OBJDUMP=${CROSS_COMPILE_ARM32}objdump
281		TOOLCHAIN_ADDR2LINE=${CROSS_COMPILE_ARM32}addr2line
282	fi
283
284	if [ ! `which ${TOOLCHAIN}gcc` ]; then
285		echo "ERROR: No find ${TOOLCHAIN}gcc"
286		exit 1
287	fi
288
289	# save to CC_FILE
290	if [ "${ARG_COMPILE}" == "y" ]; then
291		echo "${TOOLCHAIN}" > ${CC_FILE}
292	fi
293}
294
295#
296# We select chip info to do:
297#	1. RKCHIP:        fixup platform configure
298#	2. RKCHIP_LOADER: search ini file to pack loader
299#	3. RKCHIP_TRUST:  search ini file to pack trust
300#	4. RKCHIP_LABEL:  show build message
301#
302function select_chip_info()
303{
304	# Read RKCHIP firstly from .config
305	# The regular expression that matching:
306	#  - PX30, PX3SE
307	#  - RK????, RK????X
308	#  - RV????
309	CHIP_PATTERN='^CONFIG_ROCKCHIP_[R,P][X,V,K][0-9ESX]{1,5}'
310	RKCHIP=`egrep -o ${CHIP_PATTERN} .config`
311
312	# default
313	RKCHIP=${RKCHIP##*_}
314	RKCHIP_LOADER=${RKCHIP}
315	RKCHIP_TRUST=${RKCHIP}
316	RKCHIP_LABEL=`sed -n "/CONFIG_CHIP_NAME=/s/CONFIG_CHIP_NAME=//p" .config |tr -d '\r' | tr -d '"'`
317	if [ -z "${RKCHIP_LABEL}" ]; then
318		RKCHIP_LABEL=${RKCHIP}
319	fi
320}
321
322# Priority: default < CHIP_CFG_FIXUP_TABLE() < make.sh args
323function fixup_platform_configure()
324{
325	U_KB=`sed -n "/CONFIG_UBOOT_SIZE_KB=/s/CONFIG_UBOOT_SIZE_KB=//p" .config |tr -d '\r' | tr -d '"'`
326	U_NUM=`sed -n "/CONFIG_UBOOT_NUM=/s/CONFIG_UBOOT_NUM=//p" .config |tr -d '\r' | tr -d '"'`
327	T_KB=`sed -n "/CONFIG_TRUST_SIZE_KB=/s/CONFIG_TRUST_SIZE_KB=//p" .config |tr -d '\r' | tr -d '"'`
328	T_NUM=`sed -n "/CONFIG_TRUST_NUM=/s/CONFIG_TRUST_NUM=//p" .config |tr -d '\r' | tr -d '"'`
329	SHA=`sed -n "/CONFIG_TRUST_SHA_MODE=/s/CONFIG_TRUST_SHA_MODE=//p" .config |tr -d '\r' | tr -d '"'`
330	RSA=`sed -n "/CONFIG_TRUST_RSA_MODE=/s/CONFIG_TRUST_RSA_MODE=//p" .config |tr -d '\r' | tr -d '"'`
331
332	# .config
333	PLAT_UBOOT_SIZE="--size ${U_KB} ${U_NUM}"
334	PLAT_TRUST_SIZE="--size ${T_KB} ${T_NUM}"
335	PLAT_SHA="--sha ${SHA}"
336	PLAT_RSA="--rsa ${RSA}"
337
338	# ./make.sh args
339	if [ ! -z "${ARG_UBOOT_SIZE}" ]; then
340		PLAT_UBOOT_SIZE=${ARG_UBOOT_SIZE}
341	fi
342	if [ ! -z "${ARG_TRUST_SIZE}" ]; then
343		PLAT_TRUST_SIZE=${ARG_TRUST_SIZE}
344	fi
345}
346
347function select_ini_file()
348{
349	# default
350	INI_LOADER=${RKBIN}/RKBOOT/${RKCHIP_LOADER}MINIALL.ini
351	if [ "${ARM64_TRUSTZONE}" == "y" ]; then
352		INI_TRUST=${RKBIN}/RKTRUST/${RKCHIP_TRUST}TRUST.ini
353	else
354		INI_TRUST=${RKBIN}/RKTRUST/${RKCHIP_TRUST}TOS.ini
355	fi
356
357	# defconfig
358	NAME=`sed -n "/CONFIG_LOADER_INI=/s/CONFIG_LOADER_INI=//p" .config |tr -d '\r' | tr -d '"'`
359	if [ ! -z "${NAME}" ]; then
360		INI_LOADER=${RKBIN}/RKBOOT/${NAME}
361	fi
362	NAME=`sed -n "/CONFIG_TRUST_INI=/s/CONFIG_TRUST_INI=//p" .config |tr -d '\r' | tr -d '"'`
363	if [ ! -z "${NAME}" ]; then
364		INI_TRUST=${RKBIN}/RKTRUST/${NAME}
365	fi
366
367	# args
368	if [ ! -z "${ARG_INI_TRUST}" ]; then
369		INI_TRUST=${ARG_INI_TRUST}
370	fi
371	if [ ! -z "${ARG_INI_LOADER}" ]; then
372		INI_LOADER=${ARG_INI_LOADER}
373	fi
374}
375
376function sub_commands()
377{
378	# skip "--" parameter, such as "--rollback-index-..."
379	if [[ ${CMD_ARGS} != --* ]]; then
380		CMD=${CMD_ARGS%-*}
381		ARG=${CMD_ARGS#*-}
382	else
383		CMD=${CMD_ARGS}
384	fi
385
386	if [ "${ARG_TSPL}" == "tpl" -o "${ARG_TSPL}" == "spl" ]; then
387		ELF=`find -name u-boot-${ARG_TSPL}`
388		MAP=`find -name u-boot-${ARG_TSPL}.map`
389		SYM=`find -name u-boot-${ARG_TSPL}.sym`
390	else
391		ELF=u-boot
392		MAP=u-boot.map
393		SYM=u-boot.sym
394	fi
395
396	case ${CMD} in
397		elf|nm)
398			if [ "${CMD}" == "nm" ]; then
399				echo -e "\n${ELF}:     file format elf\n"
400				${TOOLCHAIN_NM} -r --size ${ELF} | grep -iv 'b' | less
401			else
402				if [ "${CMD}" == "elf" -a "${ARG}" == "elf" ]; then
403					ARG=D # default
404				fi
405				${TOOLCHAIN_OBJDUMP} -${ARG} ${ELF} | less
406			fi
407			exit 0
408			;;
409		map|sym)
410			if [ ${CMD} == "map" ]; then
411				cat ${MAP} | less
412			else
413				cat ${SYM} | less
414			fi
415			exit 0
416			;;
417		debug)
418			./scripts/rkpatch.sh ${ARG}
419			exit 0
420			;;
421		fit)
422			# Non-secure
423			${SCRIPT_FIT} --boot_img_dir images/ ${ARG_LIST_FIT}
424			exit 0
425			;;
426		uboot)
427			pack_uboot_image
428			exit 0
429			;;
430		trust)
431			pack_trust_image
432			exit 0
433			;;
434		loader)
435			pack_loader_image
436			exit 0
437			;;
438		itb)
439			pack_uboot_itb_image
440			exit 0
441			;;
442		env)
443			make CROSS_COMPILE=${TOOLCHAIN} envtools
444			exit 0
445			;;
446		--idblock)
447			pack_idblock
448			exit 0
449			;;
450		--tpl|--spl|tpl|spl)
451			pack_spl_loader_image
452			exit 0
453			;;
454		*)
455			unwind_addr_or_continue
456			;;
457	esac
458}
459
460function unwind_addr_or_continue()
461{
462	FUNCADDR=${ARG_FUNCADDR}
463	RELOCOFF=${FUNCADDR#*-}
464	FUNCADDR=${FUNCADDR%-*}
465
466	if [ -z $(echo ${FUNCADDR} | sed 's/[0-9,a-f,A-F,x,X,-]//g') ] && [ ${FUNCADDR} ]; then
467		# With prefix: '0x' or '0X'
468		if [ `echo ${FUNCADDR} | sed -n "/0[x,X]/p" | wc -l` -ne 0 ]; then
469			FUNCADDR=`echo ${FUNCADDR} | awk '{ print strtonum($0) }'`
470			FUNCADDR=`echo "obase=16;${FUNCADDR}"|bc | tr '[A-Z]' '[a-z]'`
471		fi
472		if [ `echo ${RELOCOFF} | sed -n "/0[x,X]/p" | wc -l` -ne 0 ] && [ ${RELOCOFF} ]; then
473			RELOCOFF=`echo ${RELOCOFF} | awk '{ print strtonum($0) }'`
474			RELOCOFF=`echo "obase=16;${RELOCOFF}"|bc | tr '[A-Z]' '[a-z]'`
475		fi
476
477		# If reloc address is assigned, do sub
478		if [ "${FUNCADDR}" != "${RELOCOFF}" ]; then
479			# Hex -> Dec -> SUB -> Hex
480			FUNCADDR=`echo $((16#${FUNCADDR}))`
481			RELOCOFF=`echo $((16#${RELOCOFF}))`
482			FUNCADDR=$((FUNCADDR-RELOCOFF))
483			FUNCADDR=$(echo "obase=16;${FUNCADDR}"|bc | tr '[A-Z]' '[a-z]')
484		fi
485
486		echo
487		sed -n "/${FUNCADDR}/p" ${SYM}
488		${TOOLCHAIN_ADDR2LINE} -e ${ELF} ${FUNCADDR}
489		exit 0
490	fi
491}
492
493function pack_idblock()
494{
495	INI=${INI_LOADER}
496	if [ ! -f ${INI} ]; then
497		echo "ERROR: No ${INI}"
498		exit 1
499	fi
500
501	# chip
502	COMMON_H=`grep "_common.h:" include/autoconf.mk.dep | awk -F "/" '{ printf $3 }'`
503	PLAT=${COMMON_H%_*}
504
505	# file
506	SPL_BIN=${RKBIN}/`sed -n "/FlashBoot=/s/FlashBoot=//p" ${INI} | tr -d '\r'`
507	TPL_BIN=${RKBIN}/`sed -n "/FlashData=/s/FlashData=//p" ${INI} | tr -d '\r'`
508	if [ ! -z "${ARG_SPL_BIN}" ]; then
509		SPL_BIN=${ARG_SPL_BIN}
510	fi
511	if [ ! -z "${ARG_TPL_BIN}" ]; then
512		TPL_BIN=${ARG_TPL_BIN}
513	fi
514
515	# pack
516	rm idblock.bin -f
517	./tools/mkimage -n ${PLAT} -T rksd -d ${TPL_BIN}:${SPL_BIN} idblock.bin
518	echo "Input:"
519	echo "    ${INI}"
520	echo "    ${TPL_BIN}"
521	echo "    ${SPL_BIN}"
522	echo
523	echo "Pack ${PLAT} idblock.bin okay!"
524	echo
525}
526
527function pack_uboot_itb_image()
528{
529	INI=${INI_TRUST}
530	if [ ! -f ${INI} ]; then
531		echo "ERROR: No ${INI}"
532		exit 1
533	fi
534
535	if [ "${ARM64_TRUSTZONE}" == "y" ]; then
536		BL31_ELF=`sed -n '/_bl31_/s/PATH=//p' ${INI} | tr -d '\r'`
537		BL32_BIN=`sed -n '/_bl32_/s/PATH=//p' ${INI} | tr -d '\r'`
538		rm bl31.elf tee.bin -rf
539		cp ${RKBIN}/${BL31_ELF} bl31.elf
540		if grep BL32_OPTION -A 1 ${INI} | grep SEC=1 ; then
541			cp ${RKBIN}/${BL32_BIN} tee.bin
542			TEE_OFFSET=`grep BL32_OPTION -A 3 ${INI} | grep ADDR= | awk -F "=" '{ printf $2 }' | tr -d '\r'`
543			TEE_ARG="-t ${TEE_OFFSET}"
544		fi
545	else
546		# TOS
547		TOS=`sed -n "/TOS=/s/TOS=//p" ${INI} | tr -d '\r'`
548		TOSTA=`sed -n "/TOSTA=/s/TOSTA=//p" ${INI} | tr -d '\r'`
549		if [ ! -z "${TOSTA}" ]; then
550			cp ${RKBIN}/${TOSTA} tee.bin
551		elif [ ! -z "${TOS}" ]; then
552			cp ${RKBIN}/${TOS}   tee.bin
553		else
554			echo "WARN: No tee bin"
555		fi
556
557		TEE_OFFSET=`sed -n "/ADDR=/s/ADDR=//p" ${INI} | tr -d '\r'`
558		if [ "${TEE_OFFSET}" == "" ]; then
559			TEE_OFFSET=0x8400000
560		fi
561		TEE_ARG="-t ${TEE_OFFSET}"
562	fi
563
564	# MCUs
565	for ((i=0; i<5; i++))
566	do
567		MCU_BIN="mcu${i}.bin"
568		MCU_IDX="MCU${i}"
569
570		# compatible: use "MCU" to replace "MCU0" if "MCU" is present.
571		ENABLED=`awk -F"," '/MCU=/  { printf $3 }' ${INI} | tr -d ' '`
572		if [ ${i} -eq 0 ]; then
573			ENABLED=`awk -F"," '/MCU=/  { printf $3 }' ${INI} | tr -d ' '`
574			if [ ! -z ${ENABLED} ]; then
575				MCU_IDX="MCU"
576			fi
577		fi
578
579		ENABLED=`awk -F "," '/'${MCU_IDX}'=/  { printf $3 }' ${INI} | tr -d ' '`
580		if [ "${ENABLED}" == "enabled" -o "${ENABLED}" == "okay" ]; then
581			NAME=`awk -F "," '/'${MCU_IDX}'=/ { printf $1 }' ${INI} | tr -d ' ' | awk -F "=" '{ print $2 }'`
582			OFFS=`awk -F "," '/'${MCU_IDX}'=/ { printf $2 }' ${INI} | tr -d ' '`
583			cp ${RKBIN}/${NAME} ${MCU_BIN}
584			if [ -z ${OFFS} ]; then
585				echo "ERROR: No ${MCU_BIN} address in ${INI}"
586				exit 1
587			fi
588			MCU_ARG=${MCU_ARG}" -m${i} ${OFFS}"
589		fi
590	done
591
592	# Loadables
593	for ((i=0; i<5; i++))
594	do
595		LOAD_BIN="load${i}.bin"
596		LOAD_IDX="LOAD${i}"
597		ENABLED=`awk -F "," '/'${LOAD_IDX}'=/  { printf $3 }' ${INI} | tr -d ' '`
598		if [ "${ENABLED}" == "enabled" -o "${ENABLED}" == "okay" ]; then
599			NAME=`awk -F "," '/'${LOAD_IDX}'=/ { printf $1 }' ${INI} | tr -d ' ' | awk -F "=" '{ print $2 }'`
600			OFFS=`awk -F "," '/'${LOAD_IDX}'=/ { printf $2 }' ${INI} | tr -d ' '`
601			cp ${RKBIN}/${NAME} ${LOAD_BIN}
602			if [ -z ${OFFS} ]; then
603				echo "ERROR: No ${LOAD_BIN} address in ${INI}"
604				exit 1
605			fi
606			LOAD_ARG=${LOAD_ARG}" -l${i} ${OFFS}"
607		fi
608	done
609
610	# COMPRESSION
611	COMPRESSION=`awk -F"," '/COMPRESSION=/  { printf $1 }' ${INI} | tr -d ' ' | cut -c 13-`
612	if [ ! -z "${COMPRESSION}" -a "${COMPRESSION}" != "none" ]; then
613		COMPRESSION_ARG="-c ${COMPRESSION}"
614	fi
615
616	if [ -d ${REP_DIR} ]; then
617		mv ${REP_DIR}/* ./
618	fi
619
620	SPL_FIT_SOURCE=`sed -n "/CONFIG_SPL_FIT_SOURCE=/s/CONFIG_SPL_FIT_SOURCE=//p" .config | tr -d '""'`
621	if [ ! -z ${SPL_FIT_SOURCE} ]; then
622		cp ${SPL_FIT_SOURCE} u-boot.its
623	else
624		SPL_FIT_GENERATOR=`sed -n "/CONFIG_SPL_FIT_GENERATOR=/s/CONFIG_SPL_FIT_GENERATOR=//p" .config | tr -d '""'`
625		# *.py is the legacy one.
626		if [[ ${SPL_FIT_GENERATOR} == *.py ]]; then
627			${SPL_FIT_GENERATOR} u-boot.dtb > u-boot.its
628		else
629			${SPL_FIT_GENERATOR} ${TEE_ARG} ${COMPRESSION_ARG} ${MCU_ARG} ${LOAD_ARG} > u-boot.its
630		fi
631	fi
632
633	./tools/mkimage -f u-boot.its -E u-boot.itb >/dev/null 2>&1
634	echo "pack u-boot.itb okay! Input: ${INI}"
635	echo
636}
637
638function pack_spl_loader_image()
639{
640	rm *_loader_*.bin -f
641	cd ${RKBIN}
642	if [ ! -z "${ARG_SPL_BIN}" -a ! -z "${ARG_TPL_BIN}" ]; then
643		${SCRIPT_SPL} --ini ${INI_LOADER} --tpl ${SRCTREE}/${ARG_TPL_BIN} --spl ${SRCTREE}/${ARG_SPL_BIN}
644	elif [ ! -z "${ARG_TPL_BIN}" ]; then
645		${SCRIPT_SPL} --ini ${INI_LOADER} --tpl ${SRCTREE}/${ARG_TPL_BIN}
646	else
647		${SCRIPT_SPL} --ini ${INI_LOADER} --spl ${SRCTREE}/${ARG_SPL_BIN}
648	fi
649	cd -
650	if [ -f ${RKBIN}/*_loader_*.bin ]; then
651		mv ${RKBIN}/*_loader_*.bin ./
652	fi
653}
654
655function pack_uboot_image()
656{
657	rm u-boot.img u-boot-dtb.img -f
658	LOAD_ADDR=`sed -n "/CONFIG_SYS_TEXT_BASE=/s/CONFIG_SYS_TEXT_BASE=//p" include/autoconf.mk|tr -d '\r'`
659	if [ -z "${LOAD_ADDR}" ]; then
660		# upstream U-Boot
661		LOAD_ADDR=`grep CONFIG_SYS_TEXT_BASE include/generated/autoconf.h | awk '{ print $3 }' | tr -d '\r'`
662	fi
663
664	if [ -z "${LOAD_ADDR}" ]; then
665		echo "ERROR: No CONFIG_SYS_TEXT_BASE for u-boot";
666		exit 1
667	fi
668
669	${SCRIPT_UBOOT} --load ${LOAD_ADDR} ${PLAT_UBOOT_SIZE}
670}
671
672function pack_loader_image()
673{
674	rm *_loader_*.bin -f
675	cd ${RKBIN}
676	${SCRIPT_LOADER} --ini ${INI_LOADER}
677	cd -
678	if [ -f ${RKBIN}/*_loader_*.bin ]; then
679		mv ${RKBIN}/*_loader_*.bin ./
680	fi
681}
682
683function pack_trust_image()
684{
685	DRAM_BASE=`sed -n "/CONFIG_SYS_SDRAM_BASE=/s/CONFIG_SYS_SDRAM_BASE=//p" include/autoconf.mk|tr -d '\r'`
686
687	rm trust*.img -f
688	cd ${RKBIN}
689	if [ "${ARM64_TRUSTZONE}" == "y" ]; then
690		${SCRIPT_ATF} --ini ${INI_TRUST} ${PLAT_SHA} ${PLAT_RSA} ${PLAT_TRUST_SIZE}
691	else
692		${SCRIPT_TOS} --ini ${INI_TRUST} --base ${DRAM_BASE} ${PLAT_TRUST_SIZE}
693	fi
694	cd -
695	if [ -f ${RKBIN}/trust*.img ]; then
696		mv ${RKBIN}/trust*.img ./
697	fi
698}
699
700function pack_fit_image()
701{
702	# check host tools
703	if ! which dtc >/dev/null 2>&1 ; then
704		echo "ERROR: No 'dtc', please: apt-get install device-tree-compiler"
705		exit 1
706	fi
707
708	# If we don't plan to have uboot in uboot.img in case of: SPL => Trust => Kernel, creating empty files.
709	if [ "${ARG_NO_UBOOT}" == "y" ]; then
710		rm u-boot-nodtb.bin u-boot.dtb -f
711		touch u-boot-nodtb.bin u-boot.dtb
712	fi
713
714	rm uboot.img trust*.img -rf
715	${SCRIPT_FIT} ${ARG_LIST_FIT} --chip ${RKCHIP_LABEL}
716
717	rm ${REP_DIR} -rf
718	echo "pack uboot.img okay! Input: ${INI_TRUST}"
719}
720
721function handle_args_late()
722{
723	ARG_LIST_FIT="${ARG_LIST_FIT} --ini-trust ${INI_TRUST} --ini-loader ${INI_LOADER}"
724}
725
726function clean_files()
727{
728	rm spl/u-boot-spl.dtb tpl/u-boot-tpl.dtb u-boot.dtb -f
729	rm spl/u-boot-spl tpl/u-boot-tpl u-boot -f
730}
731
732function pack_images()
733{
734	if [ "${ARG_RAW_COMPILE}" != "y" ]; then
735		if [ "${PLAT_TYPE}" == "FIT" ]; then
736			pack_fit_image ${ARG_LIST_FIT}
737		elif [ "${PLAT_TYPE}" == "DECOMP" ]; then
738			${SCRIPT_DECOMP}
739		else
740			pack_uboot_image
741			pack_trust_image
742			pack_loader_image
743		fi
744	fi
745}
746
747function finish()
748{
749	echo
750	if [ "${ARG_BOARD}" == "" ]; then
751		echo "Platform ${RKCHIP_LABEL} is build OK, with exist .config"
752	else
753		echo "Platform ${RKCHIP_LABEL} is build OK, with new .config(${MAKE_CMD})"
754	fi
755}
756
757process_args $*
758prepare
759select_toolchain
760select_chip_info
761fixup_platform_configure
762select_ini_file
763handle_args_late
764sub_commands
765clean_files
766make PYTHON=python2 CROSS_COMPILE=${TOOLCHAIN} all --jobs=${JOB}
767pack_images
768finish
769echo ${TOOLCHAIN}
770date
771