/**
 * @file sdrv_crypto_mailbox_ske_mac.c
 * @brief crypto ske mac cmd interface
 *
 * @copyright Copyright (c) 2021  Semidrive Semiconductor.
 *            All rights reserved.
 */

#include <sdrv_crypto_mailbox_ske_mac.h>

/**
 * @brief ske gen mac
 *
 * This function gen mac.
 *
 * @param[in] mode ske mode
 * @param[in] msg msg buff
 * @param[in] msg_bytes msg bytelen
 * @param[in] key_type enc key type
 * @param[in] key enc key buff
 * @param[in] key_id enc key id
 * @param[out] mac mac buff
 * @param[in] mac_bytes mac bytelen
 * @param[in] mac_buf mac buff
 * @return CMD_RETURN_SUCCESS or error code.
 */
sdrv_crypto_error_status_e cmd_ske_gen_mac(uint8_t mode, uint8_t *msg,
                                           uint32_t msg_bytes,
                                           cmd_key_type_e key_type,
                                           uint8_t *key, uint16_t key_id,
                                           uint8_t *mac, uint8_t mac_bytes,
                                           uint8_t *mac_buf)
{
    uint16_t key_id_array[2];
    uint32_t cmd_buf[8] = {0};
    uint32_t ret;
    cmd_generateMAC_t *cmd_generate_ske_mac = NULL;

    /*********** generate ske mac ***********/
    cmd_generate_ske_mac = (cmd_generateMAC_t *)cmd_buf;

    cmd_generate_ske_mac->cmd_id = GENERATE_MAC;
    cmd_generate_ske_mac->need_seed = 0;
    cmd_generate_ske_mac->mode = mode;
    cmd_generate_ske_mac->mac_len = mac_bytes;

    set_big_endian_4byte((uint8_t *)cmd_generate_ske_mac->aut_info_ptr,
                         (uint32_t)mac_buf);

    set_big_endian_4byte((uint8_t *)cmd_generate_ske_mac->data_ptr,
                         (uint32_t)msg);
    set_big_endian_4byte((uint8_t *)cmd_generate_ske_mac->data_len,
                         (uint32_t)msg_bytes);

    if (CMD_KEY_EXTERNAL_PLAINTEXT == key_type) {
        key_id_array[0] = 0;
        key_id_array[1] = 0;
        set_big_endian_4byte((uint8_t *)cmd_generate_ske_mac->key_ptr,
                             (uint32_t)key);
    } else if (CMD_KEY_EXTERNAL_CIPHERTEXT == key_type) {
        key_id_array[0] = 0;
        key_id_array[1] = key_id;
        set_big_endian_4byte((uint8_t *)cmd_generate_ske_mac->key_ptr,
                             (uint32_t)key);
    } else if (CMD_KEY_INTERNAL == key_type) {
        key_id_array[0] = key_id;
        key_id_array[1] = 0;
        set_big_endian_4byte((uint8_t *)cmd_generate_ske_mac->key_ptr,
                             (uint32_t)0);
    } else {
        return CMD_RETURN_FAIL;
    }

    set_big_endian_2byte((uint8_t *)cmd_generate_ske_mac->key_id0,
                         key_id_array[0]);
    set_big_endian_2byte((uint8_t *)cmd_generate_ske_mac->key_id1,
                         key_id_array[1]);
    set_big_endian_4byte((uint8_t *)cmd_generate_ske_mac->mac_ptr,
                         (uint32_t)mac);

    get_otp_key_aut_mac(key_id_array, 2, (uint8_t *)cmd_buf, mac_buf);

    ret = send_cmd_and_wait_response((uint32_t *)cmd_buf);

    if ((SEIP_SUCCESS == ret)) {

    } else {
        print_buf_U8((uint8_t *)msg, msg_bytes, "msg");
        print_buf_U8((uint8_t *)mac, (msg_bytes + 15) & (~0x0F), "output");
        printf("SKE generate mac failure, mode=%d, ret=0x%x \n", mode, ret);
        return CMD_RETURN_FAIL;
    }

    return CMD_RETURN_SUCCESS;
}

/**
 * @brief ske verify mac
 *
 * This function verify mac
 *
 * @param[in] mode ske mode
 * @param[in] msg msg buff
 * @param[in] msg_bytes msg bytelen
 * @param[in] key_type enc key type
 * @param[in] key enc key buff
 * @param[in] key_id enc key id
 * @param[in] mac mac buff
 * @param[in] mac_bytes mac bytelen
 * @param[in] mac_buf mac buff
 * @return CMD_RETURN_SUCCESS or error code.
 */
sdrv_crypto_error_status_e cmd_ske_verify_mac(uint8_t mode, uint8_t *msg,
                                              uint32_t msg_bytes,
                                              cmd_key_type_e key_type,
                                              uint8_t *key, uint16_t key_id,
                                              uint8_t *mac, uint8_t mac_bytes,
                                              uint8_t *mac_buf)
{
    uint16_t key_id_array[2];
    uint32_t cmd_buf[8] = {0};
    uint32_t ret;
    cmd_verifyMAC_t *cmd_verify_ske_mac = NULL;

    /*********** generate ske mac ***********/
    cmd_verify_ske_mac = (cmd_verifyMAC_t *)cmd_buf;

    cmd_verify_ske_mac->cmd_id = VERIFY_MAC;
    cmd_verify_ske_mac->need_seed = 0;
    cmd_verify_ske_mac->mode = mode;
    cmd_verify_ske_mac->mac_len = mac_bytes;
    set_big_endian_4byte((uint8_t *)cmd_verify_ske_mac->aut_info_ptr,
                         (uint32_t)mac_buf);

    set_big_endian_4byte((uint8_t *)cmd_verify_ske_mac->data_ptr,
                         (uint32_t)msg);
    set_big_endian_4byte((uint8_t *)cmd_verify_ske_mac->data_len,
                         (uint32_t)msg_bytes);

    if (CMD_KEY_EXTERNAL_PLAINTEXT == key_type) {
        key_id_array[0] = 0;
        key_id_array[1] = 0;
        set_big_endian_4byte((uint8_t *)cmd_verify_ske_mac->key_ptr,
                             (uint32_t)key);
    } else if (CMD_KEY_EXTERNAL_CIPHERTEXT == key_type) {
        key_id_array[0] = 0;
        key_id_array[1] = key_id;
        set_big_endian_4byte((uint8_t *)cmd_verify_ske_mac->key_ptr,
                             (uint32_t)key);
    } else if (CMD_KEY_INTERNAL == key_type) {
        key_id_array[0] = key_id;
        key_id_array[1] = 0;
        set_big_endian_4byte((uint8_t *)cmd_verify_ske_mac->key_ptr,
                             (uint32_t)0);
    } else {
        return CMD_RETURN_FAIL;
    }

    set_big_endian_2byte((uint8_t *)cmd_verify_ske_mac->key_id0,
                         key_id_array[0]);
    set_big_endian_2byte((uint8_t *)cmd_verify_ske_mac->key_id1,
                         key_id_array[1]);
    set_big_endian_4byte((uint8_t *)cmd_verify_ske_mac->mac_ptr, (uint32_t)mac);

    get_otp_key_aut_mac(key_id_array, 2, (uint8_t *)cmd_buf, mac_buf);

    ret = send_cmd_and_wait_response((uint32_t *)cmd_buf);

    if ((SEIP_SUCCESS == ret)) {

    } else {
        print_buf_U8((uint8_t *)msg, msg_bytes, "msg");
        print_buf_U8((uint8_t *)mac, (msg_bytes + 15) & (~0x0F), "output");
        printf("SKE generate mac failure, mode=%d, ret=0x%x \n", mode, ret);
        return CMD_RETURN_FAIL;
    }

    return CMD_RETURN_SUCCESS;
}