mbedtls/tests/suites/test_suite_psa_crypto_hash.function

/* BEGIN_HEADER */

#include <stdint.h>
#include "psa/crypto.h"

/* END_HEADER */

/* BEGIN_DEPENDENCIES
 * depends_on:MBEDTLS_PSA_CRYPTO_C
 * END_DEPENDENCIES
 */

/* BEGIN_CASE */
void hash_finish(int alg_arg, data_t *input, data_t *expected_hash)
{
    psa_algorithm_t alg = alg_arg;
    unsigned char actual_hash[PSA_HASH_MAX_SIZE];
    size_t actual_hash_length;
    psa_hash_operation_t operation = PSA_HASH_OPERATION_INIT;

    PSA_ASSERT(psa_crypto_init());

    PSA_ASSERT(psa_hash_setup(&operation, alg));
    PSA_ASSERT(psa_hash_update(&operation,
                               input->x, input->len));
    PSA_ASSERT(psa_hash_finish(&operation,
                               actual_hash, sizeof(actual_hash),
                               &actual_hash_length));
    ASSERT_COMPARE(expected_hash->x, expected_hash->len,
                   actual_hash, actual_hash_length);

exit:
    psa_hash_abort(&operation);
    PSA_DONE();
}
/* END_CASE */

/* BEGIN_CASE */
void hmac(int alg_arg, char *input, data_t *expected_mac)
{
    psa_algorithm_t alg = PSA_ALG_HMAC(alg_arg);

    mbedtls_svc_key_id_t key = MBEDTLS_SVC_KEY_ID_INIT;
    psa_key_type_t key_type = PSA_KEY_TYPE_HMAC;
    const uint8_t key_data[] = { // 32 bytes of 0xaa
        0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa,
        0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa,
        0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa, 0xaa
    };
    psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT;

    PSA_ASSERT(psa_crypto_init());

    psa_set_key_usage_flags(&attributes, PSA_KEY_USAGE_SIGN_MESSAGE);
    psa_set_key_algorithm(&attributes, alg);
    psa_set_key_type(&attributes, key_type);
    PSA_ASSERT(psa_import_key(&attributes, key_data, sizeof(key_data), &key));

    uint8_t mac[PSA_MAC_MAX_SIZE + 10] = { 0 };
    size_t mac_length = 0;

    PSA_ASSERT(psa_mac_compute(key, alg, (uint8_t const *) input, strlen(input), mac, sizeof(mac), &mac_length));

    ASSERT_COMPARE(expected_mac->x, expected_mac->len, mac, mac_length);

    PSA_ASSERT(psa_destroy_key(key));
exit:
    PSA_DONE();
}
/* END_CASE */

/* BEGIN_CASE */
void hash_verify(int alg_arg, data_t *input, data_t *expected_hash)
{
    psa_algorithm_t alg = alg_arg;
    psa_hash_operation_t operation = PSA_HASH_OPERATION_INIT;

    PSA_ASSERT(psa_crypto_init());

    PSA_ASSERT(psa_hash_setup(&operation, alg));
    PSA_ASSERT(psa_hash_update(&operation,
                               input->x,
                               input->len));
    PSA_ASSERT(psa_hash_verify(&operation,
                               expected_hash->x,
                               expected_hash->len));

exit:
    psa_hash_abort(&operation);
    PSA_DONE();
}
/* END_CASE */

/* BEGIN_CASE */
void hash_multi_part(int alg_arg, data_t *input, data_t *expected_hash)
{
    psa_algorithm_t alg = alg_arg;
    unsigned char actual_hash[PSA_HASH_MAX_SIZE];
    size_t actual_hash_length;
    psa_hash_operation_t operation = PSA_HASH_OPERATION_INIT;
    psa_hash_operation_t operation2 = PSA_HASH_OPERATION_INIT;
    uint32_t len = 0;

    PSA_ASSERT(psa_crypto_init());

    do {
        memset(actual_hash, 0, sizeof(actual_hash));
        PSA_ASSERT(psa_hash_setup(&operation, alg));

        PSA_ASSERT(psa_hash_update(&operation,
                                   input->x, len));
        PSA_ASSERT(psa_hash_clone(&operation, &operation2));
        PSA_ASSERT(psa_hash_update(&operation,
                                   input->x + len, input->len - len));
        PSA_ASSERT(psa_hash_update(&operation2,
                                   input->x + len, input->len - len));

        PSA_ASSERT(psa_hash_finish(&operation,
                                   actual_hash, sizeof(actual_hash),
                                   &actual_hash_length));
        ASSERT_COMPARE(expected_hash->x, expected_hash->len,
                       actual_hash, actual_hash_length);

        PSA_ASSERT(psa_hash_finish(&operation2,
                                   actual_hash, sizeof(actual_hash),
                                   &actual_hash_length));
        ASSERT_COMPARE(expected_hash->x, expected_hash->len,
                       actual_hash, actual_hash_length);
    } while (len++ != input->len);

exit:
    psa_hash_abort(&operation);
    psa_hash_abort(&operation2);
    PSA_DONE();
}
/* END_CASE */