package.sh

#! /bin/bash
#
# Copyright (c) 2021-2022 Amlogic, Inc. All rights reserved.
#
# SPDX-License-Identifier: MIT
#

RTOS_BUILD_DIR=$(realpath $(dirname $(readlink -f ${BASH_SOURCE[0]:-$0}))/..)
BUILD_SYSTEM_DIR=$RTOS_BUILD_DIR/build_system

export RTOS_BUILD_DIR

# Parsing large package combination information
function package_target_verify() {

    ARCH_PREFIX=""
    PRODUCT_SUFFIX=""

    #Get the packed sequence selected by the user
    j=0
    while IFS= read -r LINE; do
        if [ $j == $CHOICE_PACKAGE ]; then
            PACKAGE_ARRY=($(echo $LINE | tr ' ' ' '))
        fi
        j=$((j + 1))
    done <$PACKAGE_COMBINATION

    #parameter check
    if [ -z "$PACKAGE_ARRY" ]; then
        echo -e "\033[41;33m package list is not set, please execute scripts/pkg_env.sh \033[0m"
        exit 1
    fi

    #parameter check
    j=0
    for ((loop = 0; loop < ${#PACKAGE_ARRY[@]}; loop += 4)); do
        pkg_arch[j]=${PACKAGE_ARRY[loop]}
        pkg_soc[j]=${PACKAGE_ARRY[loop + 1]}
        pkg_board[j]=${PACKAGE_ARRY[loop + 2]}
        pkg_product[j]=${PACKAGE_ARRY[loop + 3]}
        ARCH_PREFIX="$ARCH_PREFIX""${pkg_arch[j]}""-"
        PRODUCT_SUFFIX="$PRODUCT_SUFFIX""-""${pkg_product[j]}"
        j+=1
    done

    # set the board configuration path
    IMAGE_BOARD_CONFIG_DIR="$RTOS_BUILD_DIR/image_packer/${pkg_soc[0]}/"

    # mirror storage path
    AML_IMAGE_STORAGE_PATH=$RTOS_BUILD_DIR/output/packages/"${ARCH_PREFIX}""${pkg_soc[0]}"-"${pkg_board[0]}"

    # create mirror file directory
    rm -fr $AML_IMAGE_STORAGE_PATH
    mkdir -p $AML_IMAGE_STORAGE_PATH
}

function package_kernel_for_mcuboot() {
    MCUBOOT_LIB_DIR=${RTOS_BUILD_DIR}/lib/mcuboot
    MCUBOOT_SIGNTOOL_DIR=${MCUBOOT_LIB_DIR}/scripts
    MCUBOOT_SIGNTOOL=${MCUBOOT_SIGNTOOL_DIR}/imgtool.py
    MCUBOOT_OUT_DIR=${RTOS_BUILD_DIR}/output/$1-$2-mcuboot
    MCUBOOT_CFG=$MCUBOOT_OUT_DIR/$KERNEL/.config
    IMGTOOL_INPUT_FILE=$OUTPUT_PATH/$KERNEL/$KERNEL.bin
    IMGTOOL_OUTPUT_FILE=$OUTPUT_PATH/images/$KERNEL-signed.bin

    KERNEL_SLOT_SZ="$(grep -E \
    "^CONFIG_LIB_MCUBOOT_KERNEL_SIZE=" "$MCUBOOT_CFG" | cut -d '=' -f2)"

    python3 $MCUBOOT_SIGNTOOL sign --key $MCUBOOT_LIB_DIR/root-rsa-2048.pem \
            --header-size 0x1000 --align 4 --slot-size ${KERNEL_SLOT_SZ} \
            --pad  --version 1.0.0 --pad-header --load-addr 0x10000 \
            ${IMGTOOL_INPUT_FILE} \
            ${IMGTOOL_OUTPUT_FILE}

    cp ${IMGTOOL_OUTPUT_FILE} ${IMAGE_PATH}/rtos-uImage

    PRODUCT_CFG=${RTOS_BUILD_DIR}/output/$1-$2-$3/$KERNEL/.config
    BT_INPUT_FILE=${RTOS_BUILD_DIR}/boards/${pkg_arch[0]}/${pkg_board[0]}/bt_fw.bin
    BT_OUTPUT_FILE=$AML_IMAGE_STORAGE_PATH/bt-signed.bin
    BT_SLOT_SZ="$(grep -E "^CONFIG_LIB_MCUBOOT_BT_SIZE=" "$MCUBOOT_CFG" | cut -d '=' -f2)"

    python3 $MCUBOOT_SIGNTOOL sign --key $MCUBOOT_LIB_DIR/root-rsa-2048.pem \
            --header-size 0x1000 --align 4 --slot-size  ${BT_SLOT_SZ} \
            --version 1.0.0 \
            --pad-header \
            ${BT_INPUT_FILE} \
            ${BT_OUTPUT_FILE}
    cp $BT_OUTPUT_FILE ${IMAGE_PATH}/bt

}

function compile_rtos_for_arm() {
    # target file path
    OUTPUT_PATH=${RTOS_BUILD_DIR}/output/$1-$3-$4

    IMAGE_PATH=${OUTPUT_PATH}/images
    BINARY_FILE=${IMAGE_PATH}/${KERNEL}-signed.bin
    DEBUG_FILE_PREFIX=${OUTPUT_PATH}/${KERNEL}/${KERNEL}
    XIP_CONFIG_FILE=${RTOS_BUILD_DIR}/boards/$1/$3/lscript.h
    BUILD_LINK_FILE=${RTOS_BUILD_DIR}/boards/$1/$3/lscript.h

    # Clean up rtos compilation intermediate files
    rm -rf $OUTPUT_PATH

    # rtos load address
    LINE=$(grep -m 1 "configTEXT_BASE" $BUILD_LINK_FILE)
    RTOS_LOAD_ADDR=$(echo "$LINE" | grep -oP '0x[0-9a-fA-F]+')

    # determine whether to enable the xip function
    RTOS_XIP=$(grep -E "^#define\s+CONFIG_XIP\s+[01]$" "$XIP_CONFIG_FILE" | awk '{print $3}')

    # start compile flow
    pushd $RTOS_BUILD_DIR

    source scripts/env.sh $1 $2 $3 $4

    if [ "$BACKTRACE_ENABLE" = "1" ]; then
        make backtrace
    else
        make
    fi

    if [ $? -ne 0 ]; then
        echo "bulid rtos image faile error:$?"
        popd
        exit 1
    fi

    if [ "$RTOS_XIP" = "1" ]; then
        make -f ${BUILD_SYSTEM_DIR}/xip.mk xip
        cp ${OUTPUT_PATH}/${KERNEL}/${KERNEL}.bin ${BINARY_FILE}
    fi

	if [ -n "$BUILD_MCUBOOT" ]; then
        package_kernel_for_mcuboot $1 $3 $4
	else
    	mkimage -A ${ARCH} -O u-boot -T standalone -C none -a ${RTOS_LOAD_ADDR} \
        -e ${RTOS_LOAD_ADDR} -n rtos -d ${BINARY_FILE} ${IMAGE_PATH}/rtos-uImage
    fi

	if [ "$RTOS_XIP" = "1" ]; then
        cp ${OUTPUT_PATH}/freertos/freertos_b.bin ${IMAGE_PATH}/rtos-xipA
        cp ${IMAGE_PATH}/* $AML_IMAGE_STORAGE_PATH/
    else
        test -f ${IMAGE_PATH}/rtos-uImage && cp ${IMAGE_PATH}/rtos-uImage $AML_IMAGE_STORAGE_PATH/rtos-uImage
    fi

    test -f ${DEBUG_FILE_PREFIX}.lst && cp ${DEBUG_FILE_PREFIX}.lst $AML_IMAGE_STORAGE_PATH/$1-$3.lst
    test -f ${DEBUG_FILE_PREFIX}.map && cp ${DEBUG_FILE_PREFIX}.map $AML_IMAGE_STORAGE_PATH/$1-$3.map

    popd
}

function compile_rtos_for_other() {
    # target file path
    OUTPUT_PATH=${RTOS_BUILD_DIR}/output/$1-$3-$4

    IMAGE_PATH=${OUTPUT_PATH}/images
    BINARY_FILE=${IMAGE_PATH}/${KERNEL}-signed.bin
    DEBUG_FILE_PREFIX=${OUTPUT_PATH}/${KERNEL}/${KERNEL}

    # Clean up rtos compilation intermediate files
    rm -rf $OUTPUT_PATH

    # start compile flow
    pushd $RTOS_BUILD_DIR

    source scripts/env.sh $1 $2 $3 $4

    make

    if [ "$1" == "xtensa" ]; then
        test -f ${BINARY_FILE} && cp ${BINARY_FILE} $AML_IMAGE_STORAGE_PATH/dspboot.bin
    else
        test -f ${BINARY_FILE} && cp ${BINARY_FILE} $AML_IMAGE_STORAGE_PATH/${KERNEL}-signed.bin
    fi

    test -f ${DEBUG_FILE_PREFIX}.lst && cp ${DEBUG_FILE_PREFIX}.lst $AML_IMAGE_STORAGE_PATH/$1-$3.lst
    test -f ${DEBUG_FILE_PREFIX}.map && cp ${DEBUG_FILE_PREFIX}.map $AML_IMAGE_STORAGE_PATH/$1-$3.map

    popd
}

# Compile rtos for all architectures
function compile_rtos_for_all() {
    for ((loop = 0; loop < ${#pkg_board[@]}; loop += 1)); do
        if [[ ${pkg_arch[loop]} == *"arm"* ]]; then
            compile_rtos_for_arm ${pkg_arch[loop]} ${pkg_soc[loop]} ${pkg_board[loop]} ${pkg_product[loop]}
        else
            compile_rtos_for_other ${pkg_arch[loop]} ${pkg_soc[loop]} ${pkg_board[loop]} ${pkg_product[loop]}
        fi
    done
}

# Compile the bootloader
function build_bootloader() {
    echo "start compiling bootloader ..."
    echo "<-------------- ${pkg_arch[0]} ${pkg_soc[0]} ${pkg_board[0]} ${pkg_product[0]} -------------->"

    if [ "$1" == "mcuboot" ]; then
        source $RTOS_BUILD_DIR/scripts/package_mcuboot.sh
        compile_mcuboot ${pkg_arch[0]} ${pkg_soc[0]} ${pkg_board[0]} mcuboot $AML_IMAGE_STORAGE_PATH
        echo "Compilation of MCUBoot is successful"
    else
        #Select the compile parameters of the bootstrap
        case ${pkg_board[0]} in
        'ad401_a113l')
            uboot_type="a1_ad401_nand_rtos"
            ;;
        'ad403_a113l')
            uboot_type="a1_ad403_nor_rtos"
            ;;
        *) ;;
        esac

        if [ -z "$uboot_type" ]; then
            echo "Waring: Select board(${pkg_board[0]}) not support compile uboot"
            exit 1
        else
            pushd $RTOS_BUILD_DIR/boot/u-boot
            ./mk $uboot_type
            test -f build/u-boot.bin && cp -av build/u-boot.bin* $AML_IMAGE_STORAGE_PATH
            popd
        fi
    fi
}

function aml_image_package() {
    install $IMAGE_BOARD_CONFIG_DIR/platform.conf $AML_IMAGE_STORAGE_PATH/
    install $IMAGE_BOARD_CONFIG_DIR/usb_flow.aml $AML_IMAGE_STORAGE_PATH/
    install $IMAGE_BOARD_CONFIG_DIR/aml_sdc_burn.ini $AML_IMAGE_STORAGE_PATH/

    if [ -e "$AML_IMAGE_STORAGE_PATH/mcuboot.bin" ]; then
        cp $IMAGE_BOARD_CONFIG_DIR/aml_upgrade_package_mcuboot.conf \
        $AML_IMAGE_STORAGE_PATH/aml_upgrade_package.conf
    elif [ -e "$AML_IMAGE_STORAGE_PATH/u-boot.bin" ]; then
        if [ -e "$AML_IMAGE_STORAGE_PATH/dspboot.bin" ]; then
            cp $IMAGE_BOARD_CONFIG_DIR/aml_upgrade_package.conf \
            $AML_IMAGE_STORAGE_PATH/aml_upgrade_package.conf
        elif [ -e "$AML_IMAGE_STORAGE_PATH/rtos-xipA.bin" ]; then
            cp $IMAGE_BOARD_CONFIG_DIR/aml_upgrade_package_xip.conf \
            $AML_IMAGE_STORAGE_PATH/aml_upgrade_package.conf
        elif [ -e "$AML_IMAGE_STORAGE_PATH/u-boot.bin" ]; then
            cp $IMAGE_BOARD_CONFIG_DIR/aml_upgrade_package_ndsp.conf \
            $AML_IMAGE_STORAGE_PATH/aml_upgrade_package.conf
        fi
    fi

    $RTOS_BUILD_DIR/image_packer/aml_image_v2_packer -r $AML_IMAGE_STORAGE_PATH/aml_upgrade_package.conf $AML_IMAGE_STORAGE_PATH $AML_IMAGE_STORAGE_PATH/aml_upgrade_package.img

    cd $AML_IMAGE_STORAGE_PATH && rm $(ls | grep -v ".lst" | grep -v ".map" | grep -v ".img")
}

package_target_verify
build_bootloader mcuboot
compile_rtos_for_all
aml_image_package