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