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