|  | /* | 
|  | *  PSA hashing layer on top of Mbed TLS software crypto | 
|  | */ | 
|  | /* | 
|  | *  Copyright The Mbed TLS Contributors | 
|  | *  SPDX-License-Identifier: Apache-2.0 | 
|  | * | 
|  | *  Licensed under the Apache License, Version 2.0 (the "License"); you may | 
|  | *  not use this file except in compliance with the License. | 
|  | *  You may obtain a copy of the License at | 
|  | * | 
|  | *  http://www.apache.org/licenses/LICENSE-2.0 | 
|  | * | 
|  | *  Unless required by applicable law or agreed to in writing, software | 
|  | *  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT | 
|  | *  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | 
|  | *  See the License for the specific language governing permissions and | 
|  | *  limitations under the License. | 
|  | */ | 
|  |  | 
|  | #include "common.h" | 
|  |  | 
|  | #if defined(MBEDTLS_PSA_CRYPTO_C) | 
|  |  | 
|  | #include <psa/crypto.h> | 
|  | #include "psa_crypto_core.h" | 
|  | #include "psa_crypto_hash.h" | 
|  |  | 
|  | #include <mbedtls/error.h> | 
|  | #include <string.h> | 
|  |  | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_HASH) | 
|  | psa_status_t mbedtls_psa_hash_abort( | 
|  | mbedtls_psa_hash_operation_t *operation ) | 
|  | { | 
|  | switch( operation->alg ) | 
|  | { | 
|  | case 0: | 
|  | /* The object has (apparently) been initialized but it is not | 
|  | * in use. It's ok to call abort on such an object, and there's | 
|  | * nothing to do. */ | 
|  | break; | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_MD5) | 
|  | case PSA_ALG_MD5: | 
|  | mbedtls_md5_free( &operation->ctx.md5 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_RIPEMD160) | 
|  | case PSA_ALG_RIPEMD160: | 
|  | mbedtls_ripemd160_free( &operation->ctx.ripemd160 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_1) | 
|  | case PSA_ALG_SHA_1: | 
|  | mbedtls_sha1_free( &operation->ctx.sha1 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_224) | 
|  | case PSA_ALG_SHA_224: | 
|  | mbedtls_sha256_free( &operation->ctx.sha256 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_256) | 
|  | case PSA_ALG_SHA_256: | 
|  | mbedtls_sha256_free( &operation->ctx.sha256 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_384) | 
|  | case PSA_ALG_SHA_384: | 
|  | mbedtls_sha512_free( &operation->ctx.sha512 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_512) | 
|  | case PSA_ALG_SHA_512: | 
|  | mbedtls_sha512_free( &operation->ctx.sha512 ); | 
|  | break; | 
|  | #endif | 
|  | default: | 
|  | return( PSA_ERROR_BAD_STATE ); | 
|  | } | 
|  | operation->alg = 0; | 
|  | return( PSA_SUCCESS ); | 
|  | } | 
|  |  | 
|  | psa_status_t mbedtls_psa_hash_setup( | 
|  | mbedtls_psa_hash_operation_t *operation, | 
|  | psa_algorithm_t alg ) | 
|  | { | 
|  | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
|  |  | 
|  | /* A context must be freshly initialized before it can be set up. */ | 
|  | if( operation->alg != 0 ) | 
|  | { | 
|  | return( PSA_ERROR_BAD_STATE ); | 
|  | } | 
|  |  | 
|  | switch( alg ) | 
|  | { | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_MD5) | 
|  | case PSA_ALG_MD5: | 
|  | mbedtls_md5_init( &operation->ctx.md5 ); | 
|  | ret = mbedtls_md5_starts( &operation->ctx.md5 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_RIPEMD160) | 
|  | case PSA_ALG_RIPEMD160: | 
|  | mbedtls_ripemd160_init( &operation->ctx.ripemd160 ); | 
|  | ret = mbedtls_ripemd160_starts( &operation->ctx.ripemd160 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_1) | 
|  | case PSA_ALG_SHA_1: | 
|  | mbedtls_sha1_init( &operation->ctx.sha1 ); | 
|  | ret = mbedtls_sha1_starts( &operation->ctx.sha1 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_224) | 
|  | case PSA_ALG_SHA_224: | 
|  | mbedtls_sha256_init( &operation->ctx.sha256 ); | 
|  | ret = mbedtls_sha256_starts( &operation->ctx.sha256, 1 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_256) | 
|  | case PSA_ALG_SHA_256: | 
|  | mbedtls_sha256_init( &operation->ctx.sha256 ); | 
|  | ret = mbedtls_sha256_starts( &operation->ctx.sha256, 0 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_384) | 
|  | case PSA_ALG_SHA_384: | 
|  | mbedtls_sha512_init( &operation->ctx.sha512 ); | 
|  | ret = mbedtls_sha512_starts( &operation->ctx.sha512, 1 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_512) | 
|  | case PSA_ALG_SHA_512: | 
|  | mbedtls_sha512_init( &operation->ctx.sha512 ); | 
|  | ret = mbedtls_sha512_starts( &operation->ctx.sha512, 0 ); | 
|  | break; | 
|  | #endif | 
|  | default: | 
|  | return( PSA_ALG_IS_HASH( alg ) ? | 
|  | PSA_ERROR_NOT_SUPPORTED : | 
|  | PSA_ERROR_INVALID_ARGUMENT ); | 
|  | } | 
|  | if( ret == 0 ) | 
|  | operation->alg = alg; | 
|  | else | 
|  | mbedtls_psa_hash_abort( operation ); | 
|  | return( mbedtls_to_psa_error( ret ) ); | 
|  | } | 
|  |  | 
|  | psa_status_t mbedtls_psa_hash_clone( | 
|  | const mbedtls_psa_hash_operation_t *source_operation, | 
|  | mbedtls_psa_hash_operation_t *target_operation ) | 
|  | { | 
|  | switch( source_operation->alg ) | 
|  | { | 
|  | case 0: | 
|  | return( PSA_ERROR_BAD_STATE ); | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_MD5) | 
|  | case PSA_ALG_MD5: | 
|  | mbedtls_md5_clone( &target_operation->ctx.md5, | 
|  | &source_operation->ctx.md5 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_RIPEMD160) | 
|  | case PSA_ALG_RIPEMD160: | 
|  | mbedtls_ripemd160_clone( &target_operation->ctx.ripemd160, | 
|  | &source_operation->ctx.ripemd160 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_1) | 
|  | case PSA_ALG_SHA_1: | 
|  | mbedtls_sha1_clone( &target_operation->ctx.sha1, | 
|  | &source_operation->ctx.sha1 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_224) | 
|  | case PSA_ALG_SHA_224: | 
|  | mbedtls_sha256_clone( &target_operation->ctx.sha256, | 
|  | &source_operation->ctx.sha256 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_256) | 
|  | case PSA_ALG_SHA_256: | 
|  | mbedtls_sha256_clone( &target_operation->ctx.sha256, | 
|  | &source_operation->ctx.sha256 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_384) | 
|  | case PSA_ALG_SHA_384: | 
|  | mbedtls_sha512_clone( &target_operation->ctx.sha512, | 
|  | &source_operation->ctx.sha512 ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_512) | 
|  | case PSA_ALG_SHA_512: | 
|  | mbedtls_sha512_clone( &target_operation->ctx.sha512, | 
|  | &source_operation->ctx.sha512 ); | 
|  | break; | 
|  | #endif | 
|  | default: | 
|  | (void) source_operation; | 
|  | (void) target_operation; | 
|  | return( PSA_ERROR_NOT_SUPPORTED ); | 
|  | } | 
|  |  | 
|  | target_operation->alg = source_operation->alg; | 
|  | return( PSA_SUCCESS ); | 
|  | } | 
|  |  | 
|  | psa_status_t mbedtls_psa_hash_update( | 
|  | mbedtls_psa_hash_operation_t *operation, | 
|  | const uint8_t *input, | 
|  | size_t input_length ) | 
|  | { | 
|  | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
|  |  | 
|  | switch( operation->alg ) | 
|  | { | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_MD5) | 
|  | case PSA_ALG_MD5: | 
|  | ret = mbedtls_md5_update( &operation->ctx.md5, | 
|  | input, input_length ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_RIPEMD160) | 
|  | case PSA_ALG_RIPEMD160: | 
|  | ret = mbedtls_ripemd160_update( &operation->ctx.ripemd160, | 
|  | input, input_length ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_1) | 
|  | case PSA_ALG_SHA_1: | 
|  | ret = mbedtls_sha1_update( &operation->ctx.sha1, | 
|  | input, input_length ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_224) | 
|  | case PSA_ALG_SHA_224: | 
|  | ret = mbedtls_sha256_update( &operation->ctx.sha256, | 
|  | input, input_length ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_256) | 
|  | case PSA_ALG_SHA_256: | 
|  | ret = mbedtls_sha256_update( &operation->ctx.sha256, | 
|  | input, input_length ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_384) | 
|  | case PSA_ALG_SHA_384: | 
|  | ret = mbedtls_sha512_update( &operation->ctx.sha512, | 
|  | input, input_length ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_512) | 
|  | case PSA_ALG_SHA_512: | 
|  | ret = mbedtls_sha512_update( &operation->ctx.sha512, | 
|  | input, input_length ); | 
|  | break; | 
|  | #endif | 
|  | default: | 
|  | (void) input; | 
|  | (void) input_length; | 
|  | return( PSA_ERROR_BAD_STATE ); | 
|  | } | 
|  |  | 
|  | return( mbedtls_to_psa_error( ret ) ); | 
|  | } | 
|  |  | 
|  | psa_status_t mbedtls_psa_hash_finish( | 
|  | mbedtls_psa_hash_operation_t *operation, | 
|  | uint8_t *hash, | 
|  | size_t hash_size, | 
|  | size_t *hash_length ) | 
|  | { | 
|  | psa_status_t status; | 
|  | int ret = MBEDTLS_ERR_ERROR_CORRUPTION_DETECTED; | 
|  | size_t actual_hash_length = PSA_HASH_LENGTH( operation->alg ); | 
|  |  | 
|  | /* Fill the output buffer with something that isn't a valid hash | 
|  | * (barring an attack on the hash and deliberately-crafted input), | 
|  | * in case the caller doesn't check the return status properly. */ | 
|  | *hash_length = hash_size; | 
|  | /* If hash_size is 0 then hash may be NULL and then the | 
|  | * call to memset would have undefined behavior. */ | 
|  | if( hash_size != 0 ) | 
|  | memset( hash, '!', hash_size ); | 
|  |  | 
|  | if( hash_size < actual_hash_length ) | 
|  | { | 
|  | status = PSA_ERROR_BUFFER_TOO_SMALL; | 
|  | goto exit; | 
|  | } | 
|  |  | 
|  | switch( operation->alg ) | 
|  | { | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_MD5) | 
|  | case PSA_ALG_MD5: | 
|  | ret = mbedtls_md5_finish( &operation->ctx.md5, hash ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_RIPEMD160) | 
|  | case PSA_ALG_RIPEMD160: | 
|  | ret = mbedtls_ripemd160_finish( &operation->ctx.ripemd160, hash ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_1) | 
|  | case PSA_ALG_SHA_1: | 
|  | ret = mbedtls_sha1_finish( &operation->ctx.sha1, hash ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_224) | 
|  | case PSA_ALG_SHA_224: | 
|  | ret = mbedtls_sha256_finish( &operation->ctx.sha256, hash ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_256) | 
|  | case PSA_ALG_SHA_256: | 
|  | ret = mbedtls_sha256_finish( &operation->ctx.sha256, hash ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_384) | 
|  | case PSA_ALG_SHA_384: | 
|  | ret = mbedtls_sha512_finish( &operation->ctx.sha512, hash ); | 
|  | break; | 
|  | #endif | 
|  | #if defined(MBEDTLS_PSA_BUILTIN_ALG_SHA_512) | 
|  | case PSA_ALG_SHA_512: | 
|  | ret = mbedtls_sha512_finish( &operation->ctx.sha512, hash ); | 
|  | break; | 
|  | #endif | 
|  | default: | 
|  | (void) hash; | 
|  | return( PSA_ERROR_BAD_STATE ); | 
|  | } | 
|  | status = mbedtls_to_psa_error( ret ); | 
|  |  | 
|  | exit: | 
|  | if( status == PSA_SUCCESS ) | 
|  | *hash_length = actual_hash_length; | 
|  | return( status ); | 
|  | } | 
|  |  | 
|  | psa_status_t mbedtls_psa_hash_compute( | 
|  | psa_algorithm_t alg, | 
|  | const uint8_t *input, | 
|  | size_t input_length, | 
|  | uint8_t *hash, | 
|  | size_t hash_size, | 
|  | size_t *hash_length) | 
|  | { | 
|  | mbedtls_psa_hash_operation_t operation = MBEDTLS_PSA_HASH_OPERATION_INIT; | 
|  | psa_status_t status = PSA_ERROR_CORRUPTION_DETECTED; | 
|  | psa_status_t abort_status = PSA_ERROR_CORRUPTION_DETECTED; | 
|  |  | 
|  | *hash_length = hash_size; | 
|  | status = mbedtls_psa_hash_setup( &operation, alg ); | 
|  | if( status != PSA_SUCCESS ) | 
|  | goto exit; | 
|  | status = mbedtls_psa_hash_update( &operation, input, input_length ); | 
|  | if( status != PSA_SUCCESS ) | 
|  | goto exit; | 
|  | status = mbedtls_psa_hash_finish( &operation, hash, hash_size, hash_length ); | 
|  | if( status != PSA_SUCCESS ) | 
|  | goto exit; | 
|  |  | 
|  | exit: | 
|  | abort_status = mbedtls_psa_hash_abort( &operation ); | 
|  | if( status == PSA_SUCCESS ) | 
|  | return( abort_status ); | 
|  | else | 
|  | return( status ); | 
|  |  | 
|  | } | 
|  | #endif /* MBEDTLS_PSA_BUILTIN_HASH */ | 
|  |  | 
|  | #endif /* MBEDTLS_PSA_CRYPTO_C */ |