/* Copyright 2020 NXP
 *
 * SPDX-License-Identifier: Apache-2.0
 */

#ifndef __PSA_ALT_SERIALIZE_H__
#define __PSA_ALT_SERIALIZE_H__

#include "psa/crypto.h"
#include "psa/crypto_se_driver.h"
#include "psa/crypto_types.h"
#include "psa/crypto_values.h"
#include "psa_crypto_storage.h"
#include "mbedtls/entropy.h"

typedef struct _psa_import_key_into_slot
{
    psa_key_slot_t *slot;
    const uint8_t *data;
    size_t data_length;
} psa_import_key_into_slot_struct_t;

typedef struct _psa_wipe_key_slot
{
    psa_key_slot_t *slot;
} psa_wipe_key_slot_struct_t;

typedef struct _psa_destroy_key
{
    psa_key_handle_t handle;
} psa_destroy_key_struct_t;

typedef struct _psa_set_key_domain_parameters
{
    psa_key_attributes_t *attributes;
    psa_key_type_t type;
    const uint8_t *data;
    size_t data_length;
} psa_set_key_domain_parameters_struct_t;

typedef struct _psa_get_key_domain_parameters
{
    const psa_key_attributes_t *attributes;
    uint8_t *data;
    size_t data_size;
    size_t *data_length;
} psa_get_key_domain_parameters_struct_t;

typedef struct _psa_get_key_attributes
{
    psa_key_handle_t handle;
    psa_key_attributes_t *attributes;
} psa_get_key_attributes_struct_t;

typedef struct _psa_get_key_slot_number
{
    const psa_key_attributes_t *attributes;
    psa_key_slot_number_t *slot_number;
} psa_get_key_slot_number_struct_t;

typedef struct _psa_export_key
{
    psa_key_handle_t handle;
    uint8_t *data;
    size_t data_size;
    size_t *data_length;
} psa_export_key_struct_t;

typedef struct _psa_export_public_key
{
    psa_key_handle_t handle;
    uint8_t *data;
    size_t data_size;
    size_t *data_length;
} psa_export_public_key_struct_t;

typedef struct _psa_import_key
{
    const psa_key_attributes_t *attributes;
    const uint8_t *data;
    size_t data_length;
    psa_key_handle_t *handle;
} psa_import_key_struct_t;

typedef struct _mbedtls_psa_register_se_key
{
    const psa_key_attributes_t *attributes;
} mbedtls_psa_register_se_key_struct_t;

typedef struct _psa_copy_key
{
    psa_key_handle_t source_handle;
    const psa_key_attributes_t *specified_attributes;
    psa_key_handle_t *target_handle;
} psa_copy_key_struct_t;

typedef struct _psa_hash_abort
{
    psa_hash_operation_t *operation;
} psa_hash_abort_struct_t;

typedef struct _psa_hash_setup
{
    psa_hash_operation_t *operation;
    psa_algorithm_t alg;
} psa_hash_setup_struct_t;

typedef struct _psa_hash_update
{
    psa_hash_operation_t *operation;
    const uint8_t *input;
    size_t input_length;
} psa_hash_update_struct_t;

typedef struct _psa_hash_finish
{
    psa_hash_operation_t *operation;
    uint8_t *hash;
    size_t hash_size;
    size_t *hash_length;
} psa_hash_finish_struct_t;

typedef struct _psa_hash_verify
{
    psa_hash_operation_t *operation;
    const uint8_t *hash;
    size_t hash_length;
} psa_hash_verify_struct_t;

typedef struct _psa_hash_clone
{
    const psa_hash_operation_t *source_operation;
    psa_hash_operation_t *target_operation;
} psa_hash_clone_struct_t;

typedef struct _psa_mac_abort
{
    psa_mac_operation_t *operation;
} psa_mac_abort_struct_t;

typedef struct _psa_mac_sign_setup
{
    psa_mac_operation_t *operation;
    psa_key_handle_t handle;
    psa_algorithm_t alg;
} psa_mac_sign_setup_struct_t;

typedef struct _psa_mac_verify_setup
{
    psa_mac_operation_t *operation;
    psa_key_handle_t handle;
    psa_algorithm_t alg;
} psa_mac_verify_setup_struct_t;

typedef struct _psa_mac_update
{
    psa_mac_operation_t *operation;
    const uint8_t *input;
    size_t input_length;
} psa_mac_update_struct_t;

typedef struct _psa_mac_sign_finish
{
    psa_mac_operation_t *operation;
    uint8_t *mac;
    size_t mac_size;
    size_t *mac_length;
} psa_mac_sign_finish_struct_t;

typedef struct _psa_mac_verify_finish
{
    psa_mac_operation_t *operation;
    const uint8_t *mac;
    size_t mac_length;
} psa_mac_verify_finish_struct_t;

typedef struct _psa_sign_hash
{
    psa_key_handle_t handle;
    psa_algorithm_t alg;
    const uint8_t *hash;
    size_t hash_length;
    uint8_t *signature;
    size_t signature_size;
    size_t *signature_length;
} psa_sign_hash_struct_t;

typedef struct _psa_verify_hash
{
    psa_key_handle_t handle;
    psa_algorithm_t alg;
    const uint8_t *hash;
    size_t hash_length;
    const uint8_t *signature;
    size_t signature_length;
} psa_verify_hash_struct_t;

typedef struct _psa_asymmetric_encrypt
{
    psa_key_handle_t handle;
    psa_algorithm_t alg;
    const uint8_t *input;
    size_t input_length;
    const uint8_t *salt;
    size_t salt_length;
    uint8_t *output;
    size_t output_size;
    size_t *output_length;
} psa_asymmetric_encrypt_struct_t;

typedef struct _psa_asymmetric_decrypt
{
    psa_key_handle_t handle;
    psa_algorithm_t alg;
    const uint8_t *input;
    size_t input_length;
    const uint8_t *salt;
    size_t salt_length;
    uint8_t *output;
    size_t output_size;
    size_t *output_length;
} psa_asymmetric_decrypt_struct_t;

typedef struct _psa_cipher_encrypt_setup
{
    psa_cipher_operation_t *operation;
    psa_key_handle_t handle;
    psa_algorithm_t alg;
} psa_cipher_encrypt_setup_struct_t;

typedef struct _psa_cipher_decrypt_setup
{
    psa_cipher_operation_t *operation;
    psa_key_handle_t handle;
    psa_algorithm_t alg;
} psa_cipher_decrypt_setup_struct_t;

typedef struct _psa_cipher_generate_iv
{
    psa_cipher_operation_t *operation;
    uint8_t *iv;
    size_t iv_size;
    size_t *iv_length;
} psa_cipher_generate_iv_struct_t;

typedef struct _psa_cipher_set_iv
{
    psa_cipher_operation_t *operation;
    const uint8_t *iv;
    size_t iv_length;
} psa_cipher_set_iv_struct_t;

typedef struct _psa_cipher_update
{
    psa_cipher_operation_t *operation;
    const uint8_t *input;
    size_t input_length;
    uint8_t *output;
    size_t output_size;
    size_t *output_length;
} psa_cipher_update_struct_t;

typedef struct _psa_cipher_finish
{
    psa_cipher_operation_t *operation;
    uint8_t *output;
    size_t output_size;
    size_t *output_length;
} psa_cipher_finish_struct_t;

typedef struct _psa_cipher_abort
{
    psa_cipher_operation_t *operation;
} psa_cipher_abort_struct_t;

typedef struct _psa_aead_encrypt
{
    psa_key_handle_t handle;
    psa_algorithm_t alg;
    const uint8_t *nonce;
    size_t nonce_length;
    const uint8_t *additional_data;
    size_t additional_data_length;
    const uint8_t *plaintext;
    size_t plaintext_length;
    uint8_t *ciphertext;
    size_t ciphertext_size;
    size_t *ciphertext_length;
} psa_aead_encrypt_struct_t;

typedef struct _psa_aead_decrypt
{
    psa_key_handle_t handle;
    psa_algorithm_t alg;
    const uint8_t *nonce;
    size_t nonce_length;
    const uint8_t *additional_data;
    size_t additional_data_length;
    const uint8_t *ciphertext;
    size_t ciphertext_length;
    uint8_t *plaintext;
    size_t plaintext_size;
    size_t *plaintext_length;
} psa_aead_decrypt_struct_t;

typedef struct _psa_key_derivation_abort
{
    psa_key_derivation_operation_t *operation;
} psa_key_derivation_abort_struct_t;

typedef struct _psa_key_derivation_get_capacity
{
    const psa_key_derivation_operation_t *operation;
    size_t *capacity;
} psa_key_derivation_get_capacity_struct_t;

typedef struct _psa_key_derivation_set_capacity
{
    psa_key_derivation_operation_t *operation;
    size_t capacity;
} psa_key_derivation_set_capacity_struct_t;

typedef struct _psa_key_derivation_output_bytes
{
    psa_key_derivation_operation_t *operation;
    uint8_t *output;
    size_t output_length;
} psa_key_derivation_output_bytes_struct_t;

typedef struct _psa_key_derivation_output_key
{
    const psa_key_attributes_t *attributes;
    psa_key_derivation_operation_t *operation;
    psa_key_handle_t *handle;
} psa_key_derivation_output_key_struct_t;

typedef struct _psa_key_derivation_setup
{
    psa_key_derivation_operation_t *operation;
    psa_algorithm_t alg;
} psa_key_derivation_setup_struct_t;

typedef struct _psa_key_derivation_input_bytes
{
    psa_key_derivation_operation_t *operation;
    psa_key_derivation_step_t step;
    const uint8_t *data;
    size_t data_length;
} psa_key_derivation_input_bytes_struct_t;

typedef struct _psa_key_derivation_input_key
{
    psa_key_derivation_operation_t *operation;
    psa_key_derivation_step_t step;
    psa_key_handle_t handle;
} psa_key_derivation_input_key_struct_t;

typedef struct _psa_key_derivation_key_agreement
{
    psa_key_derivation_operation_t *operation;
    psa_key_derivation_step_t step;
    psa_key_handle_t private_key;
    const uint8_t *peer_key;
    size_t peer_key_length;
} psa_key_derivation_key_agreement_struct_t;

typedef struct _psa_raw_key_agreement
{
    psa_algorithm_t alg;
    psa_key_handle_t private_key;
    const uint8_t *peer_key;
    size_t peer_key_length;
    uint8_t *output;
    size_t output_size;
    size_t *output_length;
} psa_raw_key_agreement_struct_t;

typedef struct _psa_generate_random
{
    uint8_t *output;
    size_t output_size;
} psa_generate_random_struct_t;

typedef struct _mbedtls_psa_inject_entropy
{
    const uint8_t *seed;
    size_t seed_size;
} mbedtls_psa_inject_entropy_struct_t;

typedef struct _psa_generate_key
{
    const psa_key_attributes_t *attributes;
    psa_key_handle_t *handle;
} psa_generate_key_struct_t;

typedef struct _mbedtls_psa_crypto_configure_entropy_sources
{
    void (*entropy_init)(mbedtls_entropy_context *ctx);
    void (*entropy_free)(mbedtls_entropy_context *ctx);
} mbedtls_psa_crypto_configure_entropy_sources_struct_t;

// psa_status_t psa_crypto_init(void);

/* PSA CRYPTO SLOT MANAGEMENT */

typedef struct _psa_close_key
{
    psa_key_handle_t handle;
} psa_close_key_struct_t;

typedef struct _psa_open_key
{
    psa_key_file_id_t id;
    psa_key_handle_t *handle;
} psa_open_key_struct_t;

typedef struct _psa_validate_persistent_key_parameters
{
    psa_key_lifetime_t lifetime;
    psa_key_file_id_t id;
    psa_se_drv_table_entry_t **p_drv;
    int creating;
} psa_validate_persistent_key_parameters_struct_t;

typedef struct _psa_get_empty_key_slot
{
    psa_key_handle_t *handle;
    psa_key_slot_t **p_slot;
} psa_get_empty_key_slot_struct_t;

// void psa_wipe_all_key_slots(void);

// psa_status_t psa_initialize_key_slots(void);

typedef struct _psa_get_key_slot
{
    psa_key_handle_t handle;
    psa_key_slot_t **p_slot;
} psa_get_key_slot_struct_t;

#endif // __PSA_ALT_SERIALIZE_H__