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