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