#!/usr/bin/env bash
#
# Copyright(c) 2011-2026 Intel Corporation
#
# SPDX-License-Identifier: BSD-3-Clause
#

set -e

[[ $# -eq 1 ]] || [[ $# -eq 2 ]] || {
    echo "Usage : ./build-installpkg.sh sdk | psw "
    exit 1
}

[[ "$1" == "sdk" ]] || [[ "$1" == "psw" ]] ||  {
    echo "Usage : ./build-installpkg.sh sdk | psw "
    exit 1
}

INSTALLER_TYPE="$1"

SCRIPT_DIR=$(dirname "$0")
ROOT_DIR="${SCRIPT_DIR}/../../.."
LINUX_INSTALLER_COMMON_DIR="${ROOT_DIR}/linux/installer/common"
LINUX_INSTALLER_COMMON_SDK_DIR="${LINUX_INSTALLER_COMMON_DIR}/sdk"
LINUX_INSTALLER_COMMON_PSW_DIR="${LINUX_INSTALLER_COMMON_DIR}/psw"

# The result dir of the build
BUILD_DIR=${ROOT_DIR}/build/linux

# Get the architecture of the build from generated binary
get_arch()
{
    case "$INSTALLER_TYPE" in
        psw)
            local a=$(readelf -h $BUILD_DIR/aesm_service | sed -n '2p' | awk '{print $6}')
        ;;
        sdk)
            local a=$(readelf -h $BUILD_DIR/sgx_sign | sed -n '2p' | awk '{print $6}')
        ;;
    esac
    test $a = 01 && echo 'x86' || echo 'x64'
}
 

ARCH=$(get_arch)
PACKAGE_SUFFIX="$ARCH"

case "$INSTALLER_TYPE" in
    psw)
        source ${LINUX_INSTALLER_COMMON_PSW_DIR}/installConfig.${PACKAGE_SUFFIX}
        ${LINUX_INSTALLER_COMMON_PSW_DIR}/createTarball.sh
        cp  ${LINUX_INSTALLER_COMMON_PSW_DIR}/output/${TARBALL_NAME} ${SCRIPT_DIR}
    ;;
    sdk)
        source ${LINUX_INSTALLER_COMMON_SDK_DIR}/installConfig.${PACKAGE_SUFFIX}
        ${LINUX_INSTALLER_COMMON_SDK_DIR}/createTarball.sh ${2:-}
        cp  ${LINUX_INSTALLER_COMMON_SDK_DIR}/output/${TARBALL_NAME} ${SCRIPT_DIR}
    ;;
esac

trap "rm -f ${SCRIPT_DIR}/$TARBALL_NAME 2>/dev/null" 0

# Create the tarball and compute its MD5 check sum.
m=$(md5sum ${SCRIPT_DIR}/$TARBALL_NAME | awk '{print $1}')
v=$(awk '/STRFILEVER/ {print $3}' ${ROOT_DIR}/common/inc/internal/se_version.h|sed 's/^\"\(.*\)\"$/\1/')
TEMPLATE_FILE=$SCRIPT_DIR/install-sgx-"$INSTALLER_TYPE".bin.tmpl
INSTALLER_NAME=$SCRIPT_DIR/sgx_linux_"${PACKAGE_SUFFIX}"_"$INSTALLER_TYPE"_"$v".bin
l=$(wc -l $TEMPLATE_FILE | awk '{print $1}')
l=$(($l+1))

sed -e "s:@linenum@:$l:" \
    -e "s:@md5sum@:$m:"  \
    -e "s:@arch@:$ARCH:"  \
    $TEMPLATE_FILE > $INSTALLER_NAME

cat ${SCRIPT_DIR}/${TARBALL_NAME} >> $INSTALLER_NAME
chmod +x $INSTALLER_NAME
echo "Generated ${INSTALLER_TYPE} installer: $INSTALLER_NAME"
exit 0