#!/bin/bash
set -e
BIN_PATH_FIXUP="--replace tools/rk_tools/ ./"

pack_loader_image()
{
	local files ini

	files=`ls ./RKBOOT/*MINIALL*.ini`
	for ini in $files
	do
		if [ -f "$ini" ]; then
			# Ignore unused
			if [ "$ini" = "./RKBOOT/RK302AMINIALL.ini" -o \
				 "$ini" = "./RKBOOT/RK30BMINIALL.ini" -o \
				 "$ini" = "./RKBOOT/RK30MINIALL.ini" -o \
				 "$ini" = "./RKBOOT/RK310BMINIALL.ini" ]; then
				continue;
			fi

			echo "pack Input: $ini"
			./tools/boot_merger ${BIN_PATH_FIXUP} $ini
			rm *loader*.bin
			echo
		fi
	done
}

pack_trust_image()
{
	local files ini TOS TOS_TA

# Pack 32-bit trust
	files=`ls ./RKTRUST/*TOS*.ini`
	for ini in $files
	do
		if [ -f "$ini" ]; then
			echo "pack Input: $ini"

			# Parse orignal path
			TOS=`sed -n "/TOS=/s/TOS=//p" $ini|tr -d '\r'`
			TOS_TA=`sed -n "/TOSTA=/s/TOSTA=//p" $ini|tr -d '\r'`

			# replace "./tools/rk_tools/" with "./" to compatible legacy ini content of rkdevelop branch
			TOS=$(echo ${TOS} | sed "s/tools\/rk_tools\//\.\//g")
			TOS_TA=$(echo ${TOS_TA} | sed "s/tools\/rk_tools\//\.\//g")

			if [ x$TOS_TA != x -a x$TOS != x ]; then
				./tools/loaderimage --pack --trustos ${TOS} ./trust.img 0x68400000
				./tools/loaderimage --pack --trustos ${TOS_TA} ./trust_with_ta.img 0x68400000
			elif [ $TOS ]; then
				./tools/loaderimage --pack --trustos ${TOS} ./trust.img 0x68400000
			elif [ $TOS_TA ]; then
				./tools/loaderimage --pack --trustos ${TOS_TA} ./trust.img 0x68400000
			else
				exit 1
			fi
			rm trust*.img
			echo
		fi
	done

# Pack 64-bit trust
	files=`ls ./RKTRUST/*TRUST*.ini`
	for ini in $files
	do
		if [ -f "$ini" ]; then
			echo "pack Input: $ini"
			./tools/trust_merger ${BIN_PATH_FIXUP} $ini
			rm trust*.img
			echo
		fi
	done
}

finish()
{
	echo "Packing loader and trust successfully."
	echo
}

pack_loader_image
pack_trust_image
finish