108 lines
2.1 KiB
C
108 lines
2.1 KiB
C
/*
|
|
* Copyright (c) 2023 O.S.Systems
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
#include <zephyr/logging/log.h>
|
|
LOG_MODULE_DECLARE(updatehub, CONFIG_UPDATEHUB_LOG_LEVEL);
|
|
|
|
#include "updatehub_integrity.h"
|
|
|
|
#if defined(CONFIG_PSA_CRYPTO_CLIENT)
|
|
#define SUCCESS_VALUE PSA_SUCCESS
|
|
#else
|
|
#define SUCCESS_VALUE 0
|
|
#endif
|
|
|
|
int updatehub_integrity_init(updatehub_crypto_context_t *ctx)
|
|
{
|
|
int ret;
|
|
|
|
if (ctx == NULL) {
|
|
LOG_DBG("Invalid integrity context");
|
|
return -EINVAL;
|
|
}
|
|
|
|
#if defined(CONFIG_PSA_CRYPTO_CLIENT)
|
|
*ctx = psa_hash_operation_init();
|
|
ret = psa_hash_setup(ctx, PSA_ALG_SHA_256);
|
|
#else
|
|
mbedtls_sha256_init(ctx);
|
|
ret = mbedtls_sha256_starts(ctx, false);
|
|
#endif
|
|
if (ret != SUCCESS_VALUE) {
|
|
LOG_DBG("Failed to %s SHA-256 operation. (%d)", "set up", ret);
|
|
return -EFAULT;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int updatehub_integrity_update(updatehub_crypto_context_t *ctx,
|
|
const uint8_t *buffer, const uint32_t len)
|
|
{
|
|
int ret;
|
|
|
|
if (ctx == NULL || buffer == NULL) {
|
|
return -EINVAL;
|
|
}
|
|
|
|
/* bypass */
|
|
if (len == 0) {
|
|
return 0;
|
|
}
|
|
|
|
#if defined(CONFIG_PSA_CRYPTO_CLIENT)
|
|
ret = psa_hash_update(ctx, buffer, len);
|
|
if (ret != PSA_SUCCESS) {
|
|
psa_hash_abort(ctx);
|
|
}
|
|
#else
|
|
ret = mbedtls_sha256_update(ctx, buffer, len);
|
|
if (ret != 0) {
|
|
mbedtls_sha256_free(ctx);
|
|
}
|
|
#endif
|
|
|
|
if (ret != SUCCESS_VALUE) {
|
|
LOG_DBG("Failed to %s SHA-256 operation. (%d)", "update", ret);
|
|
return -EFAULT;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int updatehub_integrity_finish(updatehub_crypto_context_t *ctx,
|
|
uint8_t *hash, const uint32_t size)
|
|
{
|
|
int ret;
|
|
|
|
if (ctx == NULL || hash == NULL) {
|
|
return -EINVAL;
|
|
}
|
|
|
|
if (size < SHA256_BIN_DIGEST_SIZE) {
|
|
LOG_DBG("HASH input buffer is to small to store the message digest");
|
|
return -EINVAL;
|
|
}
|
|
|
|
#if defined(CONFIG_PSA_CRYPTO_CLIENT)
|
|
size_t hash_len;
|
|
|
|
ret = psa_hash_finish(ctx, hash, size, &hash_len);
|
|
if (ret != PSA_SUCCESS) {
|
|
psa_hash_abort(ctx);
|
|
}
|
|
#else
|
|
ret = mbedtls_sha256_finish(ctx, hash);
|
|
mbedtls_sha256_free(ctx);
|
|
#endif
|
|
if (ret != SUCCESS_VALUE) {
|
|
LOG_DBG("Failed to %s SHA-256 operation. (%d)", "finish", ret);
|
|
return -EFAULT;
|
|
}
|
|
|
|
return 0;
|
|
}
|