blob: a12e49c6a24cf3a3f0a4fb0e51d073404567a517 [file] [log] [blame]
Juan Castilloa57a4d52015-04-02 15:44:20 +01001/*
Manish V Badarkheabf1b8d2022-01-27 13:50:23 +00002 * Copyright (c) 2015-2022, Arm Limited and Contributors. All rights reserved.
Juan Castilloa57a4d52015-04-02 15:44:20 +01003 *
dp-armfa3cf0b2017-05-03 09:38:09 +01004 * SPDX-License-Identifier: BSD-3-Clause
Juan Castilloa57a4d52015-04-02 15:44:20 +01005 */
6
John Tsichritzis30f89642018-06-07 16:31:34 +01007#include <assert.h>
Antonio Nino Diaze0f90632018-12-14 00:18:21 +00008#include <stddef.h>
9
Juan Castillobae6b2a2015-11-05 09:24:53 +000010/* mbed TLS headers */
11#include <mbedtls/memory_buffer_alloc.h>
Antonio Nino Diaz6b90f5e2017-05-19 11:37:22 +010012#include <mbedtls/platform.h>
Antonio Nino Diaze0f90632018-12-14 00:18:21 +000013
14#include <common/debug.h>
15#include <drivers/auth/mbedtls/mbedtls_common.h>
Manish V Badarkheabf1b8d2022-01-27 13:50:23 +000016#include MBEDTLS_CONFIG_FILE
Antonio Nino Diaze0f90632018-12-14 00:18:21 +000017#include <plat/common/platform.h>
Juan Castilloa57a4d52015-04-02 15:44:20 +010018
Roberto Vargas64d4de02018-05-24 13:34:53 +010019static void cleanup(void)
20{
21 ERROR("EXIT from BL2\n");
22 panic();
23}
24
Juan Castilloa57a4d52015-04-02 15:44:20 +010025/*
Juan Castillobae6b2a2015-11-05 09:24:53 +000026 * mbed TLS initialization function
Juan Castilloa57a4d52015-04-02 15:44:20 +010027 */
28void mbedtls_init(void)
29{
30 static int ready;
John Tsichritzis30f89642018-06-07 16:31:34 +010031 void *heap_addr;
32 size_t heap_size = 0;
33 int err;
Juan Castilloa57a4d52015-04-02 15:44:20 +010034
35 if (!ready) {
Roberto Vargas64d4de02018-05-24 13:34:53 +010036 if (atexit(cleanup))
37 panic();
John Tsichritzis30f89642018-06-07 16:31:34 +010038
39 err = plat_get_mbedtls_heap(&heap_addr, &heap_size);
40
41 /* Ensure heap setup is proper */
42 if (err < 0) {
43 ERROR("Mbed TLS failed to get a heap\n");
44 panic();
45 }
46 assert(heap_size >= TF_MBEDTLS_HEAP_SIZE);
Roberto Vargas64d4de02018-05-24 13:34:53 +010047
Juan Castillobae6b2a2015-11-05 09:24:53 +000048 /* Initialize the mbed TLS heap */
John Tsichritzis30f89642018-06-07 16:31:34 +010049 mbedtls_memory_buffer_alloc_init(heap_addr, heap_size);
Antonio Nino Diaz6b90f5e2017-05-19 11:37:22 +010050
Antonio Nino Diazf1481b22017-06-06 10:54:39 +010051#ifdef MBEDTLS_PLATFORM_SNPRINTF_ALT
Antonio Nino Diaz00086e32018-08-16 16:46:06 +010052 mbedtls_platform_set_snprintf(snprintf);
Antonio Nino Diazf1481b22017-06-06 10:54:39 +010053#endif
Juan Castillobae6b2a2015-11-05 09:24:53 +000054 ready = 1;
Juan Castilloa57a4d52015-04-02 15:44:20 +010055 }
56}
John Tsichritzis69c88d12019-02-28 11:14:03 +000057
58/*
Ambroise Vincentd207f562019-04-10 12:50:27 +010059 * The following helper function simply returns the default allocated heap.
60 * It can be used by platforms for their plat_get_mbedtls_heap() implementation.
John Tsichritzis69c88d12019-02-28 11:14:03 +000061 */
Ambroise Vincentd207f562019-04-10 12:50:27 +010062int get_mbedtls_heap_helper(void **heap_addr, size_t *heap_size)
John Tsichritzis69c88d12019-02-28 11:14:03 +000063{
64 static unsigned char heap[TF_MBEDTLS_HEAP_SIZE];
65
66 assert(heap_addr != NULL);
67 assert(heap_size != NULL);
68
69 *heap_addr = heap;
70 *heap_size = sizeof(heap);
71 return 0;
72}