cmake_minimum_required(VERSION 3.22)
set(CMAKE_C_STANDARD 11)
set(CMAKE_C_STANDARD_REQUIRED ON)
set(CMAKE_C_EXTENSIONS ON)

# Define the build type
if(NOT CMAKE_BUILD_TYPE)
    set(CMAKE_BUILD_TYPE "Debug")
endif()

# Set the project name
set(CMAKE_PROJECT_NAME bootloader)

# Enable compile command to ease indexing with e.g. clangd
set(CMAKE_EXPORT_COMPILE_COMMANDS TRUE)

# Core project settings
project(${CMAKE_PROJECT_NAME})
message("Build type: " ${CMAKE_BUILD_TYPE})

# Enable CMake support for ASM and C languages
enable_language(C ASM)

# Create an executable object type
add_executable(${CMAKE_PROJECT_NAME})

target_sources(${CMAKE_PROJECT_NAME} PRIVATE
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/common/cpu_endian.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/common/os_port_none.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/common/date_time.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/common/str.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/common/path.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/common/resource_manager.c
        ${CMAKE_CURRENT_SOURCE_DIR}/
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/core/crc32.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/core/mailbox.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/drivers/mcu_core/stm32l4xx_mcu_driver.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/drivers/flash/internal/stm32l4xx_flash_driver.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/modules/image/image.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/modules/image/image_process.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/modules/image/image_utils.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/modules/memory/memory.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/modules/memory/memory_ex.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/core/verify.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/modules/security/verify_auth.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/modules/security/verify_sign.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/modules/security/cipher.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/bootloader/second_stage/boot.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/bootloader/second_stage/boot_common.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/bootloader/second_stage/boot_fallback.c

        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/hardware/stm32l4xx/stm32l4xx_crypto.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/hardware/stm32l4xx/stm32l4xx_crypto_trng.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/hash/md5.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/hash/sha1.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/hash/sha224.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/hash/sha256.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/hash/sha384.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/hash/sha512.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/mac/hmac.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/cipher/rc4.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/cipher/idea.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/cipher/des.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/cipher/des3.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/cipher/aes.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/cipher/camellia.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/cipher/seed.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/cipher/aria.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/cipher_modes/cbc.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/aead/ccm.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/aead/gcm.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/cipher/chacha.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/mac/poly1305.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/aead/chacha20_poly1305.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkc/dh.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkc/rsa.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkc/dsa.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/ecc/ec.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/ecc/ec_curves.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/ecc/ecdh.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/ecc/ecdsa.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/ecc/eddsa.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/ecc/curve25519.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/ecc/curve448.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/ecc/x25519.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/ecc/x448.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/ecc/ed25519.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/ecc/ed448.c
           ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/mpi/mpi.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/mpi/mpi_misc.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/encoding/base64.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/encoding/asn1.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/encoding/oid.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkix/pem_import.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkix/pem_key_import.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkix/pem_decrypt.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkix/pem_common.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkix/pkcs5_decrypt.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkix/pkcs5_common.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkix/pkcs8_key_parse.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkix/x509_key_parse.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkix/x509_cert_parse.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkix/x509_cert_validate.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkix/x509_cert_ext_parse.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkix/x509_common.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkix/x509_sign_parse.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkix/x509_sign_verify.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/kdf/hkdf.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/rng/yarrow.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/ecc/ec_misc.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto/pkc/rsa_misc.c

        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/boards/stm32l4xx_nucleo/stm32l4xx_nucleo.c

        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_rcc.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_rcc_ex.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_flash.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_flash_ex.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_flash_ramfunc.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_gpio.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_i2c.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_i2c_ex.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_dma.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_dma_ex.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_pwr.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_pwr_ex.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_cortex.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_exti.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/src/stm32l4xx_hal_uart.c

        ${CMAKE_CURRENT_SOURCE_DIR}/startup_stm32l476xx.s
        ${CMAKE_CURRENT_SOURCE_DIR}/syscalls.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../src/system_stm32l4xx.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../src/stm32l4xx_it.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../src/main.c
        ${CMAKE_CURRENT_SOURCE_DIR}/../src/debug.c
)

target_include_directories(${CMAKE_PROJECT_NAME} PRIVATE
        ${CMAKE_CURRENT_SOURCE_DIR}/../src
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/cmsis/include
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/devices/stm32l4xx
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/inc
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/drivers/stm32l4xx_hal_driver/inc/legacy
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../third_party/st/boards/stm32l4xx_nucleo
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/common
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_crypto
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/bootloader/
        ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../src/cyclone_boot/modules
)

target_compile_definitions(${CMAKE_PROJECT_NAME} PRIVATE
    STM32L476xx
    STM32L47xG
    USE_HAL_DRIVER
    USE_STM32L4XX_NUCLEO
    __USE_C99_MATH

    $<$<CONFIG:Debug>:DEBUG>
)

# Convert .elf to .bin
add_custom_command(TARGET ${CMAKE_PROJECT_NAME} POST_BUILD
        COMMAND ${CMAKE_OBJCOPY} -O binary
        ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_PROJECT_NAME}.elf
        ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_PROJECT_NAME}.bin
        COMMENT "Generating binary file from ELF"
)


# Copy the generated .bin file to ../
add_custom_command(TARGET ${CMAKE_PROJECT_NAME} POST_BUILD
    COMMAND ${CMAKE_COMMAND} -E copy
            ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_PROJECT_NAME}.bin
            ${CMAKE_CURRENT_SOURCE_DIR}/../
    COMMENT "Copying ${CMAKE_PROJECT_NAME}.bin to parent directory"
)

add_custom_command(TARGET ${CMAKE_PROJECT_NAME}  POST_BUILD
        COMMAND ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../tools/ImageBuilder/bin/image_builder_windows
        --input ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_PROJECT_NAME}.bin
        --output ${CMAKE_CURRENT_SOURCE_DIR}/../${CMAKE_PROJECT_NAME}.bootable.bin
        --vtor-align

        COMMAND ${CMAKE_CURRENT_SOURCE_DIR}/../../../../../../tools/ImageBuilder/bin/image_builder_windows
        --input ${CMAKE_CURRENT_BINARY_DIR}/${CMAKE_PROJECT_NAME}.bin
        --output ${CMAKE_CURRENT_SOURCE_DIR}/../${CMAKE_PROJECT_NAME}_update.img
        --enc-algo aes-cbc
        --enc-key aa3ff7d43cc015682c7dfd00de9379e7
        --sign-algo rsa-sha256
        --sign-key ${CMAKE_SOURCE_DIR}/../resources/keys/rsa_private_key.pem
        --update-type boot
        --vtor-align
)
