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