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

#include "otp.h"

#include <stdint.h>

#include "region_defs.h"
#include "tfm_plat_otp.h"
#include "tfm_plat_nv_counters.h"
#include "util.h"

#ifdef TEST_BL1_1
extern uint8_t tfm_bl1_key_test_1_buf[];
extern uint8_t tfm_bl1_key_test_2_buf[];
#endif /* TEST_BL1_1 */

fih_int bl1_otp_read(uint8_t *dst, uint8_t *src, size_t size);
fih_int bl1_otp_write(uint8_t *dst, uint8_t *src, size_t size);

fih_int bl1_otp_read_key(enum tfm_bl1_key_id_t key_id, uint8_t *key_buf)
{
    fih_int fih_rc;
    enum tfm_plat_err_t plat_err;

    switch (key_id) {
    case TFM_BL1_KEY_HUK:
        plat_err = tfm_plat_otp_read(PLAT_OTP_ID_HUK, 32, key_buf);
        fih_rc = fih_int_encode_zero_equality(plat_err);
        break;
    case TFM_BL1_KEY_GUK:
        plat_err = tfm_plat_otp_read(PLAT_OTP_ID_GUK, GUK_SIZE, key_buf);
        fih_rc = fih_int_encode_zero_equality(plat_err);
        break;
    case TFM_BL1_KEY_BL2_ENCRYPTION:
        plat_err = tfm_plat_otp_read(PLAT_OTP_ID_KEY_BL2_ENCRYPTION, 32,
                                     key_buf);
        fih_rc = fih_int_encode_zero_equality(plat_err);
        break;
    case TFM_BL1_KEY_ROTPK_0:
        plat_err = tfm_plat_otp_read(PLAT_OTP_ID_BL1_ROTPK_0, 56, key_buf);
        fih_rc = fih_int_encode_zero_equality(plat_err);
        break;
    default:
        FIH_RET(FIH_FAILURE);
    }

    FIH_RET(fih_rc);
}