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