/*
 * Copyright (c) 2021-2022, Arm Limited. All rights reserved.
 *
 * SPDX-License-Identifier: BSD-3-Clause
 *
 */

#include "its_flash.h"

#include "config_tfm.h"
#include "flash_fs/its_flash_fs.h"

#if ITS_RAM_FS
#ifndef ITS_RAM_FS_SIZE
#error "ITS_RAM_FS_SIZE must be defined by the target in flash_layout.h"
#endif
uint8_t its_block_data[ITS_RAM_FS_SIZE];

#elif (TFM_HAL_ITS_PROGRAM_UNIT > 16)
#ifndef ITS_FLASH_NAND_BUF_SIZE
#error "ITS_FLASH_NAND_BUF_SIZE must be defined by the target in flash_layout.h"
#endif
static uint8_t its_write_buf_0[ITS_FLASH_NAND_BUF_SIZE];
static uint8_t its_write_buf_1[ITS_FLASH_NAND_BUF_SIZE];
struct its_flash_nand_dev_t its_flash_nand_dev = {
    .driver = &TFM_HAL_ITS_FLASH_DRIVER,
    .buf_block_id_0 = ITS_BLOCK_INVALID_ID,
    .buf_block_id_1 = ITS_BLOCK_INVALID_ID,
    .write_buf_0 = its_write_buf_0,
    .write_buf_1 = its_write_buf_1,
    .buf_size = sizeof(its_write_buf_0),
};
#endif

#ifdef TFM_PARTITION_PROTECTED_STORAGE
#if PS_RAM_FS
#ifndef PS_RAM_FS_SIZE
#error "PS_RAM_FS_SIZE must be defined by the target in flash_layout.h"
#endif
uint8_t ps_block_data[PS_RAM_FS_SIZE];

#elif (TFM_HAL_PS_PROGRAM_UNIT > 16)
#ifndef PS_FLASH_NAND_BUF_SIZE
#error "PS_FLASH_NAND_BUF_SIZE must be defined by the target in flash_layout.h"
#endif
static uint8_t ps_write_buf_0[PS_FLASH_NAND_BUF_SIZE];
static uint8_t ps_write_buf_1[PS_FLASH_NAND_BUF_SIZE];
struct its_flash_nand_dev_t ps_flash_nand_dev = {
    .driver = &TFM_HAL_PS_FLASH_DRIVER,
    .buf_block_id_0 = ITS_BLOCK_INVALID_ID,
    .buf_block_id_1 = ITS_BLOCK_INVALID_ID,
    .write_buf_0 = ps_write_buf_0,
    .write_buf_1 = ps_write_buf_1,
    .buf_size = sizeof(ps_write_buf_0),
};
#endif
#endif /* TFM_PARTITION_PROTECTED_STORAGE */