#
# Copyright (C) 2025 Amlogic, Inc. All rights reserved.
#

#!/bin/bash

# include uboot pre-build macros
#declare CONFIG_FILE=("${buildtree}/.config")
#declare AUTOCFG_FILE=("${buildtree}/include/autoconf.mk")

function init_vari() {
	#source ${CONFIG_FILE} &> /dev/null # ignore warning/error
	#source ${AUTOCFG_FILE} &> /dev/null # ignore warning/error

	AML_BL2_NAME="bl2.bin"
	AML_KEY_BLOB_NAME="aml-user-key.sig"

	if [ "y" == "${CONFIG_AML_SECURE_BOOT_V3}" ]; then
		V3_PROCESS_FLAG="--level v3"
	fi

	if [ "y" == "${CONFIG_AML_CRYPTO_AES}" ]; then
		BOOT_SIG_FLAG="--aeskey enable"
		EFUSE_GEN_FLAG="--aeskey enable"
	fi

	if [ "y" == "${CONFIG_AML_EFUSE_GEN_AES_ONLY}" ]; then
		EFUSE_GEN_FLAG="--aeskey only"
	fi

	if [ "y" == "${CONFIG_AML_BL33_COMPRESS_ENABLE}" ]; then
		BL33_COMPRESS_FLAG="--compress lz4"
	fi

	if [ "y" == "${CONFIG_FIP_IMG_SUPPORT}" ]; then
		BL3X_SUFFIX="img"
	fi

	if [ -n "${CONFIG_DDRFW_TYPE}" ]; then
		DDRFW_TYPE="${CONFIG_DDRFW_TYPE}"
	else
		DDRFW_TYPE="ddr4"
	fi

	if [ -n "${BLX_BIN_SUB_CHIP}" ]; then
		CHIPSET_NAME=`echo ${BLX_BIN_SUB_CHIP} | tr 'A-Z' 'a-z'`
	fi

	# script can use chipset variant to override config variant
	if [ -n "${SCRIPT_ARG_CHIPSET_VARIANT}" ]; then
		CHIPSET_VARIANT="${SCRIPT_ARG_CHIPSET_VARIANT}"
		CHIPSET_VARIANT_SUFFIX=".${CHIPSET_VARIANT}"
	elif [ -n "${CONFIG_CHIPSET_VARIANT}" ]; then
		CHIPSET_VARIANT="${CONFIG_CHIPSET_VARIANT}"
		CHIPSET_VARIANT_SUFFIX=".${CHIPSET_VARIANT}"
	else
		if [ -n "${CONFIG_FORMER_SIGN}" ]; then
			CHIPSET_VARIANT="no_variant"
		else
			CHIPSET_VARIANT="general"
		fi
		CHIPSET_VARIANT_SUFFIX=""
	fi

	if [ -n "${CONFIG_AMLOGIC_KEY_TYPE}" ]; then
		AMLOGIC_KEY_TYPE="${CONFIG_AMLOGIC_KEY_TYPE}"
	fi

	echo "------------------------------------------------------"
	echo "DDRFW_TYPE: ${DDRFW_TYPE} CHIPSET_NAME: ${CHIPSET_NAME} CHIPSET_VARIANT: ${CHIPSET_VARIANT} AMLOGIC_KEY_TYPE: ${AMLOGIC_KEY_TYPE}"
	echo "------------------------------------------------------"
}

function align_4k() {
	local file=$1
	local raw_size=
	local file_size=
	local rem_val=

	raw_size=`stat -c %s ${file}`
	file_size=`expr ${raw_size} + 4095`
	rem_val=`expr ${file_size} % 4096`
	file_size=`expr ${file_size} - ${rem_val}`
	#echo ==== align file ${file} from ${raw_size} to ${file_size} ====
	dd if=/dev/zero of=${file}_tmp.bin bs=1 count=${file_size} &> /dev/null
	dd if=${file} of=${file}_tmp.bin bs=1 count=${file_size} conv=notrunc &> /dev/null
	mv ${file}_tmp.bin ${file}
}

function amfc_compress() {
	local file=$1

	amfc_zstd_hdr=amfc_zstd_hdr.bin
	#echo ==== amfc compress file $file ====
	./fip/tools/aml_zstd ${file} -19 -o ${file}.tmp
	bin_org_size=`stat -c %s ${file}`
	bin_zstd_size=`stat -c %s ${file}.tmp`
	printf "%s" "@ZST" >  $amfc_zstd_hdr

	printf "%02x%02x%02x%02x" $[(bin_org_size) & 0xff] \
	$[((bin_org_size) >> 8) & 0xff] $[((bin_org_size) >> 16) & 0xff] \
	$[((bin_org_size) >> 24) & 0xff] | xxd -r -ps >>  $amfc_zstd_hdr

	printf "%02x%02x%02x%02x" $[(bin_zstd_size) & 0xff] \
	$[((bin_zstd_size) >> 8) & 0xff] $[((bin_zstd_size) >> 16) & 0xff] \
	$[((bin_zstd_size) >> 24) & 0xff] | xxd -r -ps >>  $amfc_zstd_hdr

	cat $amfc_zstd_hdr ${file}.tmp > ${file}.zstd
	rm $amfc_zstd_hdr ${file}.tmp -f
}

function mk_bl2() {
	output=$1
	payload=$2

	if [ ! -f ${output}/bl2.bin.sto ]	|| \
	   [ ! -f ${output}/bl2.bin.usb ]; then
		echo "Error: ${output}/bl2.bin does not all exist... abort"
		ls -la ${output}
		exit -1
	fi

	echo "================================================================="
	echo "image packing with acpu-imagetool for bl2"

	dd if=/dev/zero of=${payload}/bl2.bin.sto bs=119712 count=1
	dd if=${output}/bl2.bin.sto of=${payload}/bl2.bin.sto conv=notrunc

	dd if=/dev/zero of=${payload}/bl2.bin.usb bs=119712 count=1
	dd if=${output}/bl2.bin.usb of=${payload}/bl2.bin.usb conv=notrunc

	echo "================================================================"
	echo "------ process for device params ------"
	INPUT_PARAMS=${output}

	if [ ! -f ${INPUT_PARAMS}/device_acs.bin ]; then
		echo "dev acs params not exist !"
		exit -1
	fi

	dev_acs_size=`stat -c %s ${INPUT_PARAMS}/device_acs.bin`

	if [ $dev_acs_size -gt 4096 ]; then
		echo "dev acs size exceed limit 4096, $dev_acs_size"
		exit -1
	else
		dd if=/dev/zero of=${payload}/device_acs.bin bs=4096 count=1
		dd if=${INPUT_PARAMS}/device_acs.bin of=${payload}/device_acs.bin conv=notrunc
	fi

	./${FIP_FOLDER}${CUR_SOC}/binary-tool/acpu-imagetool create-boot-blobs \
			--infile-bl2-payload=${payload}/bl2.bin.sto \
			--infile-dvinit-params=${payload}/device_acs.bin \
			--scs-family=${CUR_SOC} \
			--outfile-bb1st=${output}/bb1st.sto.bin \

	./${FIP_FOLDER}${CUR_SOC}/binary-tool/acpu-imagetool create-boot-blobs \
			--infile-bl2-payload=${payload}/bl2.bin.usb \
			--infile-dvinit-params=${payload}/device_acs.bin \
			--scs-family=${CUR_SOC} \
			--outfile-bb1st=${output}/bb1st.usb.bin \

	if [ ! -f ${output}/bb1st.sto.bin ] || \
	   [ ! -f ${output}/bb1st.usb.bin ]; then
		echo "Error: ${output}/ bootblobs do not all exist... abort"
		ls -la ${output}/
		exit -1
	fi
	echo "done to generate bb1st.bin folder"
}

function mk_fake_rtos() {
	echo "make fake rtos mata header"
	./${FIP_FOLDER}${CUR_SOC}/binary-tool/acpu-imagetool create-boot-blobs \
		--scs-family=${CUR_SOC}  \
		--size-payload-bl2-2=0 \
		--size-payload-rtos-1=0 \
		--size-payload-rtos-2=0 \
		--outfile-blob-bl2-rtos=${BUILD_PATH}/fake-rtos-mata.bin
}

function mk_bl3_ree() {
	output=$1
	postfix=$2

	if [ ! -f ${output}/bl33.bin ]; then
		echo "Error: ${output}/bl33.bin does not all exist... abort"
		ls -la ${output}
		exit -1
	fi

	echo "================================================================="
	echo "image packing with acpu-imagetool for bl3 ree"

	if [ "y" == "${CONFIG_AML_BL33_COMPRESS_ENABLE}" ]; then
		mv -f ${BUILD_PATH}/bl33.bin  ${BUILD_PATH}/bl33.bin.org
		amfc_compress ${BUILD_PATH}/bl33.bin.org
		mv ${BUILD_PATH}/bl33.bin.org.zstd ${BUILD_PATH}/bl33.bin
	fi

	blx_size=`stat -c %s ${output}/bl33.bin`
	echo "BL33 size: ${blx_size}"
	if [ $blx_size -gt 6291456 ]; then
		echo "Error: bl33 size exceed limit 6MB"
		exit -1
	elif [ $blx_size -lt 307200 ]; then
		dd if=/dev/zero of=${output}/bl33.bin_tmp bs=307200 count=1 &> /dev/null
		dd if=${output}/bl33.bin of=${payload}/bl33.bin.tmp conv=notrunc &> /dev/null
		mv =${payload}/bl33.bin.tmp =${payload}/bl33.bin
	else
		align_4k ${output}/bl33.bin
	fi

	blx_size=`stat -c %s ${output}/bl33.bin`
	mv ${output}/bl33.bin  ${output}/bl33-payload.bin

	./${FIP_FOLDER}${CUR_SOC}/binary-tool/acpu-imagetool create-boot-blobs \
			--infile-bl33-payload=${output}/bl33-payload.bin \
			--scs-family=${CUR_SOC} \
			--outfile-blob-bl3-ree=${output}/blob-bl3-ree.bin${postfix} \
			--size-payload-bl33=${blx_size}

	if [ ! -f ${output}/blob-bl3-ree.bin${postfix} ]; then
		echo "Error: ${output}/ bootblobs do not all exist... abort"
		ls -la ${output}/
		exit -1
	fi

	mk_fake_rtos
	if [ -f ${BUILD_PATH}/fake-rtos-mata.bin ]; then
		cat ${BUILD_PATH}/fake-rtos-mata.bin ${output}/blob-bl3-ree.bin${postfix} > ${output}/blob-bl3-ree.bin${postfix}.tmp
		mv ${output}/blob-bl3-ree.bin${postfix}.tmp ${output}/blob-bl3-ree.bin${postfix}
	fi

	blx_size=`stat -c %s ${output}/blob-bl3-ree.bin${postfix}`
	blob_ree_size=blob_ree_size.bin

	printf "%02x%02x%02x%02x" $[(blx_size) & 0xff] \
	$[((blx_size) >> 8) & 0xff] $[((blx_size) >> 16) & 0xff] \
	$[((blx_size) >> 24) & 0xff] | xxd -r -ps >>  $blob_ree_size
	dd if=$blob_ree_size of=${BUILD_PATH}/device_acs.bin bs=1 count=4 seek=1292 conv=notrunc
        rm -rf $blob_ree_size

	echo "done to generate blob-bl3-ree.bin folder"
}

function mk_bl3_tee() {
	output=$1
	payload=$2

	EXEC_ARGS="${EXEC_ARGS}"

	if [[ ! -f "${output}/bl32.bin" ]]; then
		cp ${BLX_BIN_FOLDER[2]}/${CUR_SOC}/blob-bl3-tee.bin ${BUILD_PATH}/blob-bl3-tee.bin
		return
	fi

	echo "================================================================="
	echo "image packing with acpu-imagetool for bl3 tee"

	if [ -f ${output}/bl32.bin ]; then
		echo "compress bl32.bin"
		mv -f ${output}/bl32.bin  ${output}/bl32.bin.org

		amfc_compress ${output}/bl32.bin.org
		bl32_size=`stat -c %s ${output}/bl32.bin.org.zstd`
		if [ $bl32_size -gt 524288 ]; then
			echo "Error: bl32 size exceed limit 524288"
			exit -1
		elif [ $bl32_size -lt 65536 ]; then
			dd if=/dev/zero of=${payload}/bl32.bin.tmp bs=65536 count=1 &> /dev/null
			dd if=${output}/bl32.bin.org.zstd of=${payload}/bl32.bin.tmp conv=notrunc &> /dev/null
			mv ${payload}/bl32.bin.tmp ${payload}/bl32.bin
		else
			align_4k ${output}/bl32.bin.org.zstd
			mv ${output}/bl32.bin.org.zstd ${payload}/bl32.bin
		fi

		bl32_size=`stat -c %s ${payload}/bl32.bin`
		EXEC_ARGS="${EXEC_ARGS} --infile-bl32-payload=${payload}/bl32.bin"
		EXEC_ARGS="${EXEC_ARGS} --size-payload-bl32=${bl32_size}"
	fi

	EXEC_ARGS="${EXEC_ARGS} --outfile-blob-bl3-tee=${output}/blob-bl3-tee.bin"
	EXEC_ARGS="${EXEC_ARGS} --scs-family=${CUR_SOC}"

	echo "${EXEC_ARGS}"
	./${FIP_FOLDER}${CUR_SOC}/binary-tool/acpu-imagetool create-boot-blobs \
		${EXEC_ARGS}

	if [ ! -f ${output}/blob-bl3-tee.bin ]; then
		echo "Error: ${output}/ bootblobs do not all exist... abort"
		ls -la ${output}/
		exit -1
	fi
	echo "done to generate blob-bl3-tee.bin folder"
}

function mk_uboot() {
	output_images=$1
	input_payloads=$2
	postfix=$3
	storage_type_suffix=$4
	chipset_variant_suffix=$5

	#device_fip="${input_payloads}/device-fip.bin${postfix}"
	bl3_ree="${input_payloads}/blob-bl3-ree.bin${postfix}"
	bl3_tee="${input_payloads}/blob-bl3-tee.bin${postfix}"
	bb1st="${input_payloads}/bb1st${storage_type_suffix}${chipset_variant_suffix}.bin${postfix}"

	if [ ! -f ${bl3_ree} ] || \
		[ ! -f ${bl3_tee} ] || \
		[ ! -f ${bb1st} ]; then
		echo bl3_ree:${bl3_ree}
		echo bl3_tee:${bl3_tee}
		echo bb1st:${bb1st}
		echo "Error: ${input_payloads}/ bootblob does not all exist... abort"
		ls -la ${input_payloads}/
		exit -1
	fi

	file_info_cfg="${output_images}/aml-payload.cfg"
	file_info_cfg_temp=${temp_cfg}.temp

	bootloader="${output_images}/u-boot.bin${storage_type_suffix}${postfix}"
	sdcard_image="${output_images}/u-boot.bin.sd.bin${postfix}"

	#cat those together with 4K upper aligned for sdcard
	align_base=4096
	total_size=0
	for file in ${bb1st} ${bl3_ree} ${bl3_tee}; do
		size=`stat -c "%s" ${file}`
		upper=$[(size+align_base-1)/align_base*align_base]
		total_size=$[total_size+upper]
		echo ${file} ${size} ${upper}
	done

	echo ${total_size}
	rm -f ${bootloader}
	dd if=/dev/zero of=${bootloader} bs=${total_size} count=1 status=none

	sector=512
	seek=0
	seek_sector=0
	dateStamp=C4-${CHIPSET_NAME}-`date +%y%m%d%H%M%S`

	echo @AMLBOOT > ${file_info_cfg_temp}
	dd if=${file_info_cfg_temp} of=${file_info_cfg} bs=1 count=8 conv=notrunc &> /dev/null
	nItemNum=3
	nSizeHDR=$[64+nItemNum*16]
	printf "02 %02x %02x %02x" $[(nItemNum)&0xFF] $[(nSizeHDR)&0xFF] $[((nSizeHDR)>>8)&0xFF] \
		| xxd -r -ps > ${file_info_cfg_temp}
	cat ${file_info_cfg_temp} >> ${file_info_cfg}

	echo ${dateStamp} > ${file_info_cfg_temp}
	dd if=${file_info_cfg_temp} of=${file_info_cfg} bs=1 count=20 oflag=append conv=notrunc &> /dev/null

	index=0
	arrPayload=("BBST" "BL3E" "BL3T");
	nPayloadOffset=0
	nPayloadSize=0
	for file in ${bb1st} ${bl3_ree} ${bl3_tee}; do
		size=`stat -c "%s" ${file}`
		size_sector=$[(size+align_base-1)/align_base*align_base]
		nPayloadSize=$[size_sector]
		size_sector=$[size_sector/sector]
		seek_sector=$[seek/sector+seek_sector]
		#nPayloadOffset=$[sector*(seek_sector+1)]
		nPayloadOffset=$[sector*(seek_sector)]
		echo ${file} ${seek_sector} ${size_sector}
		dd if=${file} of=${bootloader} bs=${sector} seek=${seek_sector} conv=notrunc status=none

		echo ${arrPayload[$index]} > ${file_info_cfg_temp}.x
		index=$((index+1))
		dd if=${file_info_cfg_temp}.x of=${file_info_cfg_temp} bs=1 count=4 &> /dev/null
		rm -f ${file_info_cfg_temp}.x
		printf "%02x %02x %02x %02x %02x %02x %02x %02x 00 00 00 00" $[(nPayloadOffset)&0xFF] $[((nPayloadOffset)>>8)&0xFF] $[((nPayloadOffset)>>16)&0xFF] $[((nPayloadOffset)>>24)&0xFF] \
		$[(nPayloadSize)&0xFF] $[((nPayloadSize)>>8)&0xFF] $[((nPayloadSize)>>16)&0xFF] $[((nPayloadSize)>>24)&0xFF] | xxd -r -ps >> ${file_info_cfg_temp}
		dd if=${file_info_cfg_temp} of=${file_info_cfg} oflag=append conv=notrunc &> /dev/null
		rm -f ${file_info_cfg_temp}
		seek=$[(size+align_base-1)/align_base*align_base]
	done

	openssl dgst -sha256 -binary ${file_info_cfg} > ${file_info_cfg}.sha256
	cat ${file_info_cfg} >> ${file_info_cfg}.sha256
	#cat ${file_info_cfg}.sha256 >> ${file_info_cfg}
	rm -f ${file_info_cfg}
	mv -f ${file_info_cfg}.sha256 ${file_info_cfg}

	dd if=${file_info_cfg} of=${bootloader} bs=512 seek=260 conv=notrunc status=none

	if [ ${storage_type_suffix} == ".sto" ]; then
		echo "Image SDCARD"
		total_size=$[total_size+512]
		rm -f ${sdcard_image}
		dd if=/dev/zero of=${sdcard_image} bs=${total_size} count=1 status=none
		dd if=${file_info_cfg}   of=${sdcard_image} conv=notrunc status=none
		dd if=${bootloader} of=${sdcard_image} bs=512 seek=1 conv=notrunc status=none

		mv ${bootloader} ${output_images}/u-boot.bin${postfix}
	fi

	rm -f ${file_info_cfg}
}


function cleanup() {
	cp ${FIP_BUILD_FOLDER}u-boot.bin* ${BUILD_FOLDER}
	# cp bootblobs for PXP
	#cp ${FIP_BUILD_FOLDER}device-fip.bin ${BUILD_FOLDER} -f
	#cp ${FIP_BUILD_FOLDER}bb1st.bin ${BUILD_FOLDER} -f
	#cp ${FIP_BUILD_FOLDER}blob-bl* ${BUILD_FOLDER} -f
	echo "output file are generated in ${BUILD_FOLDER} folder"
	#rm -f ${BUILD_PATH}/test-*
	#rm -rf ${BUILD_PAYLOAD}
	rm -f ${BUILD_PATH}/bl*.enc ${BUILD_PATH}/bl2*.sig
}

function build_fip() {

	# acs_tool process ddr timing and configurable parameters
	#python ${FIP_FOLDER}/acs_tool.pyc ${BUILD_PATH}/${AML_BL2_NAME} ${BUILD_PATH}/bl2_acs.bin ${BUILD_PATH}/acs.bin 0

	# fix bl2
	if [ -d ${BUILD_PAYLOAD} ]; then
		rm -rf ${BUILD_PAYLOAD}
	fi
	mkdir -p ${BUILD_PAYLOAD}/

		# make bl3 tee
	mk_bl3_ree ${BUILD_PATH}

	# make boot blobs
	mk_bl2 ${BUILD_PATH} ${BUILD_PAYLOAD}


	# make bl3 ree
	mk_bl3_tee ${BUILD_PATH} ${BUILD_PAYLOAD}

	# build final bootloader
	#mk_uboot ${BUILD_PATH} ${BUILD_PATH}
	mk_uboot ${BUILD_PATH} ${BUILD_PATH} "" .sto ${CHIPSET_VARIANT_SUFFIX}
	mk_uboot ${BUILD_PATH} ${BUILD_PATH} "" .usb ${CHIPSET_VARIANT_SUFFIX}

	cat ${BUILD_PATH}/blob-bl3-ree.bin ${BUILD_PATH}/blob-bl3-tee.bin > ${BUILD_PATH}/tpl.bin
	cp ${BUILD_PATH}/bb1st.sto.bin	${BUILD_FOLDER}/spl.bin
	cp ${BUILD_PATH}/tpl.bin	${BUILD_FOLDER}/tpl.bin

	return
}

declare CHIPACS_SIZE="8192"
declare DDRFW_SIZE="212992"
function process_blx() {
	# process loop
	for loop in ${!BLX_NAME[@]}; do
		if [ "NULL" != "${BLX_RAWBIN_NAME[$loop]}" ] && \
			[ -n "${BLX_RAWBIN_NAME[$loop]}" ] && \
			[ -f ${BUILD_PATH}/${BLX_RAWBIN_NAME[$loop]} ]; then
			if [ -n "${CONFIG_FORMER_SIGN}" ]; then
				if [ ${BLX_NAME[$loop]} == "bl2" ]; then
					./${FIP_FOLDER}${CUR_SOC}/bin/gen-merge-bin.sh --input0 ${BUILD_PATH}/chip_acs.bin --size0 ${CHIPACS_SIZE} \
						--input1 ${BUILD_PATH}/ddrfw_data.bin --size1 ${DDRFW_SIZE} --output ${BUILD_PATH}/chip_acs.bin

					./${FIP_FOLDER}${CUR_SOC}/bin/sign-blx.sh --blxname ${BLX_NAME[$loop]} --input ${BUILD_PATH}/${BLX_RAWBIN_NAME[$loop]} \
						-output ${BUILD_PATH}/${BLX_BIN_NAME[$loop]} --chipset_name ${CHIPSET_NAME} --chipset_variant ${CHIPSET_VARIANT} \
						--key_type ${AMLOGIC_KEY_TYPE} --soc ${CUR_SOC} --chip_acs ${BUILD_PATH}/chip_acs.bin --ddr_type ${DDRFW_TYPE}
				fi
			else
				if [ -n "${CONFIG_JENKINS_SIGN}" ]; then
					if [ ${BLX_NAME[$loop]} == "bl2" ]; then
						./${FIP_FOLDER}${CUR_SOC}/bin/gen-merge-bin.sh --input0 ${BUILD_PATH}/chip_acs.bin --size0 ${CHIPACS_SIZE} \
							--input1 ${BUILD_PATH}/ddrfw_data.bin --size1 ${DDRFW_SIZE} --output ${BUILD_PATH}/chip_acs.bin
						/usr/bin/python3 ./sign.py --type ${BLX_NAME[$loop]} --in ${BUILD_PATH}/${BLX_RAWBIN_NAME[$loop]} \
							--out ${BUILD_PATH}/${BLX_BIN_NAME[$loop]} --chip ${CHIPSET_NAME}  --chipVariant ${CHIPSET_VARIANT} \
							--keyType ${AMLOGIC_KEY_TYPE}  --chipAcsFile ${BUILD_PATH}/chip_acs.bin --ddrType ${DDRFW_TYPE}
					fi
				else
					if [ ${BLX_NAME[$loop]} == "bl2" ]; then
						./${FIP_FOLDER}${CUR_SOC}/bin/gen-merge-bin.sh --input0 ${BUILD_PATH}/chip_acs.bin --size0 ${CHIPACS_SIZE} \
							--input1 ${BUILD_PATH}/ddrfw_data.bin --size1 ${DDRFW_SIZE} --output ${BUILD_PATH}/chip_acs.bin
					fi
					/usr/bin/python3 ./${FIP_FOLDER}/jenkins_sign.py --type ${BLX_NAME[$loop]} --in ${BUILD_PATH}/${BLX_RAWBIN_NAME[$loop]} \
						--out ${BUILD_PATH}/${BLX_BIN_NAME[$loop]} --chip ${CHIPSET_NAME} --chipVariant ${CHIPSET_VARIANT} --keyType ${AMLOGIC_KEY_TYPE} \
						--chipAcsFile ${BUILD_PATH}/chip_acs.bin --ddrType ${DDRFW_TYPE}
				fi
			fi
		fi
		if [ "NULL" != "${BLX_BIN_SIZE[$loop]}" ] && \
		    [ "NULL" != "${BLX_BIN_NAME[$loop]}" ] && \
			[ -n "${BLX_BIN_NAME[$loop]}" ] && \
			[ -f ${BUILD_PATH}/${BLX_BIN_NAME[$loop]} ]; then
			blx_size=`stat -c %s ${BUILD_PATH}/${BLX_BIN_NAME[$loop]}`
			if [ $blx_size -gt ${BLX_BIN_SIZE[$loop]} ]; then
				echo "Error: ${BUILD_PATH}/${BLX_BIN_NAME[$loop]} size not match"
				exit -1
			fi
		fi
	done

	if [ ! -f ${BUILD_PATH}/device_acs.bin ]; then
		echo "dev acs params not exist !"
		exit -1
	fi

	dev_acs_size=`stat -c %s ${BUILD_PATH}/device_acs.bin`

	if [ $dev_acs_size -gt ${DEV_ACS_BIN_SIZE} ]; then
		echo "chip acs size exceed limit ${DEV_ACS_BIN_SIZE}, $dev_acs_size"
		exit -1
	else
		dd if=/dev/zero of=${BUILD_PATH}/dvinit-params.bin bs=${DEV_ACS_BIN_SIZE} count=1 &> /dev/null
		dd if=${BUILD_PATH}/device_acs.bin of=${BUILD_PATH}/dvinit-params.bin conv=notrunc &> /dev/null
	fi

	./${FIP_FOLDER}${CUR_SOC}/bin/add-dvinit-params.sh ${BUILD_PATH}/bb1st.sto${CHIPSET_VARIANT_SUFFIX}.bin.signed ${BUILD_PATH}/dvinit-params.bin ${BUILD_PATH}/bb1st.sto${CHIPSET_VARIANT_SUFFIX}.bin.signed ${CUR_SOC}
	./${FIP_FOLDER}${CUR_SOC}/bin/add-dvinit-params.sh ${BUILD_PATH}/bb1st.usb${CHIPSET_VARIANT_SUFFIX}.bin.signed ${BUILD_PATH}/dvinit-params.bin ${BUILD_PATH}/bb1st.usb${CHIPSET_VARIANT_SUFFIX}.bin.signed ${CUR_SOC}

	#./${FIP_FOLDER}${CUR_SOC}/bin/gen-bl.sh ${BUILD_PATH} ${BUILD_PATH} ${BUILD_PATH}

	return
}

process_tee() {
	if [ ! -f ${BUILD_PATH}/bl32.bin ]; then
		if [ -f ${BLX_BIN_FOLDER[2]}/${CUR_SOC}/${CONFIG_CHIPSET_NAME}/${BLX_BIN_NAME[2]} ]; then
			echo "fip: process_tee using existing ${BLX_BIN_NAME[2]}"
			mv ${BUILD_PATH}/${BLX_BIN_NAME[2]} ${BUILD_PATH}/blob-bl3-tee.bin.signed
			return
		fi
	fi

	if [ -n "${CONFIG_FORMER_SIGN}" ]; then
		echo "compress bl32.bin"
		mv -f ${BUILD_PATH}/bl32.bin  ${BUILD_PATH}/bl32.bin.org

		amfc_compress ${BUILD_PATH}/bl32.bin.org
		mv ${BUILD_PATH}/bl32.bin.org.zstd ${BUILD_PATH}/bl32.bin

		./${FIP_FOLDER}${CUR_SOC}/bin/sign-tee.sh \
			--bl32 ${BUILD_PATH}/bl32.bin \
			--output ${BUILD_PATH}/blob-bl3-tee.bin.signed --chipset_name ${CHIPSET_NAME} \
			--chipset_variant ${CHIPSET_VARIANT} --key_type ${AMLOGIC_KEY_TYPE} --soc ${CUR_SOC}
	elif [ -n "${CONFIG_JENKINS_SIGN}" ]; then
		/usr/bin/python3 ./sign_bl3tee.py --type bl3tee \
			--bl32 ${BUILD_PATH}/bl32.bin \
			--out ${BUILD_PATH}/blob-bl3-tee.bin.signed --chip ${CHIPSET_NAME}  --chipVariant ${CHIPSET_VARIANT} \
			--keyType ${AMLOGIC_KEY_TYPE}
	else
		/usr/bin/python3 ./${FIP_FOLDER}/jenkins_sign.py --type bl3tee \
			--bl32 ${BUILD_PATH}/bl32.bin \
			--out ${BUILD_PATH}/blob-bl3-tee.bin.signed --chip ${CHIPSET_NAME} --chipVariant ${CHIPSET_VARIANT} \
			--keyType ${AMLOGIC_KEY_TYPE}
	fi
	ret=$?
	if [ $ret -ne 0 ]; then
		echo "fip: process_tee failed ret=$ret"
		exit 1
	fi
}

function build_signed() {

	process_tee $@

	#./${FIP_FOLDER}${CUR_SOC}/bin/gen-bl.sh ${BUILD_PATH} ${BUILD_PATH} ${BUILD_PATH} ${BUILD_PATH} ${CHIPSET_VARIANT_SUFFIX}

	postfix=.signed

	mk_bl3_ree ${BUILD_PATH} ${postfix}

	process_blx $@

	mk_uboot ${BUILD_PATH} ${BUILD_PATH} ${postfix} .sto ${CHIPSET_VARIANT_SUFFIX}
	mk_uboot ${BUILD_PATH} ${BUILD_PATH} ${postfix} .usb ${CHIPSET_VARIANT_SUFFIX}

	cat ${BUILD_PATH}/blob-bl3-ree.bin${postfix} ${BUILD_PATH}/blob-bl3-tee.bin${postfix} > ${BUILD_PATH}/tpl.bin
	cp ${BUILD_PATH}/bb1st.sto${CHIPSET_VARIANT_SUFFIX}.bin${postfix}	${BUILD_FOLDER}/spl.bin
	cp ${BUILD_PATH}/tpl.bin ${BUILD_FOLDER}/tpl.bin

	list_pack="${BUILD_PATH}/bb1st.sto${CHIPSET_VARIANT_SUFFIX}.bin.signed ${BUILD_PATH}/bb1st.usb${CHIPSET_VARIANT_SUFFIX}.bin.signed"
	list_pack="$list_pack ${BUILD_PATH}/blob-bl3-tee.bin.signed ${BUILD_PATH}/bl33-payload.bin ${BUILD_PATH}/dvinit-params.bin ${BUILD_PATH}/bl33.bin.org"
	u_pack=${BUILD_FOLDER}/"$(basename ${BOARD_DIR})"-u-boot.aml.zip
	zip -j $u_pack ${list_pack} >& /dev/null

	if [ "y" == "${CONFIG_AML_SIGNED_UBOOT}" ]; then
		if [ ! -d "${UBOOT_SRC_FOLDER}/${BOARD_DIR}/device-keys" ]; then
			./${FIP_FOLDER}${CUR_SOC}/bin/download-keys.sh ${AMLOGIC_KEY_TYPE} ${CUR_SOC} device ${UBOOT_SRC_FOLDER}/${BOARD_DIR}/device-keys/
		fi

		if [ "y" == "${CONFIG_DEVICE_ROOTRSA_INDEX}" ]; then
			DEVICE_ROOTRSA_INDEX=1
		elif [ -n "${CONFIG_DEVICE_ROOTRSA_INDEX}" ]; then
			DEVICE_ROOTRSA_INDEX=${CONFIG_DEVICE_ROOTRSA_INDEX}
		else
			DEVICE_ROOTRSA_INDEX=0
		fi

		if [ "y" == "${CONFIG_DEVICE_NO_CRYPTO_AES}" ]; then
			DEVICE_NO_CRYPTO_AES="true"
		else
			DEVICE_NO_CRYPTO_AES="false"
		fi

		./${FIP_FOLDER}${CUR_SOC}/bin/device-vendor-scs-signing.sh --key-dir ${UBOOT_SRC_FOLDER}/${BOARD_DIR}/device-keys/ --project ${CHIPSET_NAME} \
			--input-dir ${BUILD_PATH} --rootkey-index ${DEVICE_ROOTRSA_INDEX} --chipset-variant ${CHIPSET_VARIANT} \
			--arb-config ${UBOOT_SRC_FOLDER}/${BOARD_DIR}/fw_arb.cfg --disable-device-aes-crypto ${DEVICE_NO_CRYPTO_AES} --out-dir ${BUILD_PATH}
	fi

	return
}

function copy_other_soc() {
	cp ${BL33_BUILD_FOLDER}${BOARD_DIR}/firmware/acs.bin ${BUILD_PATH}/device_acs.bin -f

	if [ ! -f ${BUILD_PATH}/chip_acs.bin ]; then
		cp ./${FIP_FOLDER}${CUR_SOC}/chip_acs.bin ${BUILD_PATH}/chip_acs.bin -f
	fi

	# device acs params parse for ddr timing
	#./${FIP_FOLDER}parse ${BUILD_PATH}/device_acs.bin
}

function package() {
	# BUILD_PATH without "/"
	x=$((${#BUILD_PATH}-1))
	if [ "\\" == "${BUILD_PATH:$x:1}" ] || [ "/" == "${BUILD_PATH:$x:1}" ]; then
		BUILD_PATH=${BUILD_PATH:0:$x}
	fi

	init_vari $@
	# Enable Clear Image Packing for PXP
	if [ -n "${CONFIG_BUILD_UNSIGN}" ]; then
		build_fip $@
	else
	# Bypass Sign Process for PXP
		build_signed $@
	fi
	#copy_file
	cleanup
	echo "Bootloader build done!"
}