#!/bin/bash
# operation string and platform string
OPS=$1
PLAT=$2

ops_array=("build" "debugserver")
plat_array=("qemu_max_smp" "fvp_cortex_a55x4_a75x2_smp" "roc_rk3568_pc_smp")

# 只有在用户没手动设时才设置path，避免覆盖外部配置
REPO_ROOT="$(git rev-parse --show-toplevel 2>/dev/null || pwd)"
export ZEPHYR_BASE="${ZEPHYR_BASE:-${REPO_ROOT}}"
export CMAKE_PREFIX_PATH="${ZEPHYR_BASE}:${CMAKE_PREFIX_PATH:-}"

ops_found=false
for i in "${ops_array[@]}"
do
    if [ "$1" = "$i" ]; then
        ops_found=true
        break
    fi
done

if [ "$ops_found" = false ]; then
    echo "Invalid operation. Please use one of the following:"
    for index in "${!ops_array[@]}"; do
        echo "Argument $(($index+1)): ${ops_array[$index]}"
    done
    echo "For example: ./auto_zvm.sh ${ops_array[0]} ${plat_array[0]}"
    exit 1
fi

plat_found=false
for i in "${plat_array[@]}"
do
    if [ "$2" = "$i" ]; then
        plat_found=true
        break
    fi
done

if [ "$plat_found" = false ]; then
    echo "Invalid platform. Please use one of the following:"
    for index in "${!plat_array[@]}"; do
        echo "Argument $(($index+1)): ${plat_array[$index]}"
    done
    echo "For example: ./auto_zvm.sh ${ops_array[0]} ${plat_array[0]}"
    exit 1
fi

# Build system
if [ "$OPS" = "${ops_array[0]}" ]; then
    rm -rf build/
    if [ "$PLAT" = "${plat_array[0]}" ]; then
        west build -b qemu_max_smp samples/subsys/zvm
    elif [ "$PLAT" = "${plat_array[1]}" ]; then
        west build -b fvp_cortex_a55x4_a75x2_smp samples/subsys/zvm \
        -DARMFVP_BL1_FILE=$(pwd)/zvm_config/fvp_platform/hub/bl1.bin \
        -DARMFVP_FIP_FILE=$(pwd)/zvm_config/fvp_platform/hub/fip.bin
    elif [ "$PLAT" = "${plat_array[2]}" ]; then
        west build -b roc_rk3568_pc_smp samples/subsys/zvm
    else
        echo "Error arguments for this auto.sh! \n Please input command like: ./auto_build.sh build qemu. "
    fi
# debug system
elif [ "$OPS" = "${ops_array[1]}" ]; then
    if [ "$PLAT" = "${plat_array[0]}" ]; then
        ./zvm_config/qemu_platform/hub/qemu-system-aarch64 \
        -cpu max -m 4G -nographic -machine virt,virtualization=on,gic-version=3 \
        -net none -pidfile qemu.pid -chardev stdio,id=con,mux=on \
        -serial chardev:con -mon chardev=con,mode=readline -smp cpus=8 \
        -device loader,file=$(pwd)/zvm_config/qemu_platform/hub/Image,addr=0x60000000,force-raw=on \
        -device loader,file=$(pwd)/zvm_config/qemu_platform/hub/zephyr_qemu_c1_m128.bin,addr=0x65800000,force-raw=on \
        -device loader,file=$(pwd)/zvm_config/qemu_platform/hub/zephyr_qemu_c2_m128.bin,addr=0x65a00000,force-raw=on \
        -device loader,file=$(pwd)/zvm_config/qemu_platform/hub/zephyr_qemu_c3_m128.bin,addr=0x65c00000,force-raw=on \
        -device loader,file=$(pwd)/zvm_config/qemu_platform/hub/zephyr_qemu_c4_m128.bin,addr=0x65e00000,force-raw=on \
        -device loader,file=$(pwd)/zvm_config/qemu_platform/hub/linux_qemu_c1_m512.dtb,addr=0x64000000,force-raw=on \
        -device loader,file=$(pwd)/zvm_config/qemu_platform/hub/linux_qemu_c2_m512.dtb,addr=0x64100000,force-raw=on \
        -device loader,file=$(pwd)/zvm_config/qemu_platform/hub/linux_qemu_c3_m512.dtb,addr=0x64200000,force-raw=on \
        -device loader,file=$(pwd)/zvm_config/qemu_platform/hub/linux_qemu_c4_m512.dtb,addr=0x64300000,force-raw=on \
        -device loader,file=$(pwd)/zvm_config/qemu_platform/hub/linux_qemu_c1_m1024.dtb,addr=0x64400000,force-raw=on \
        -device loader,file=$(pwd)/zvm_config/qemu_platform/hub/linux_qemu_c2_m1024.dtb,addr=0x64500000,force-raw=on \
        -device loader,file=$(pwd)/zvm_config/qemu_platform/hub/linux_qemu_c3_m1024.dtb,addr=0x64600000,force-raw=on \
        -device loader,file=$(pwd)/zvm_config/qemu_platform/hub/linux_qemu_c4_m1024.dtb,addr=0x64700000,force-raw=on \
        -device loader,file=$(pwd)/zvm_config/qemu_platform/hub/rootfs.cpio.gz,addr=0x67000000,force-raw=on \
        -kernel $(pwd)/build/zephyr/zvm_host.elf
### using gdb to connect it:
# gdb-multiarch -q -ex 'file ./build/zephyr/zvm_host.elf' -ex 'target remote localhost:1234'
### using trace to record qemu info when boot qemu
# strace -o qemu_bug.txt

    elif [ "$PLAT" = "${plat_array[1]}" ]; then
### export lib for Foundation_Platform, sometimes some lib may not be found
## export LD_LIBRARY_PATH=/usr/aarch64-linux-gnu/lib
        $(pwd)/zvm_config/fvp_platform/hub/Foundation_Platformpkg/models/Linux64_armv8l_GCC-9.3/Foundation_Platform \
        --no-secure-memory --cores 4 --arm-v8.1 --quantum=10000\
        --image $(pwd)/build/zephyr/zvm_host.elf \
        --nsdata $(pwd)/zvm_config/fvp_platform/hub/zephyr.elf@0xa0000000 \
        --cadi-server --print-port-number
    else
        echo "Error arguments for this auto.sh! \n Please input command like: ./z_auto.sh build qemu. "
    fi
fi