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