xref: /rk3399_ARM-atf/drivers/auth/mbedtls/mbedtls_common.c (revision 84f95bed549eab4ca40fbd0505e0e3720384880c)
1 /*
2  * Copyright (c) 2015, ARM Limited and Contributors. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions are met:
6  *
7  * Redistributions of source code must retain the above copyright notice, this
8  * list of conditions and the following disclaimer.
9  *
10  * Redistributions in binary form must reproduce the above copyright notice,
11  * this list of conditions and the following disclaimer in the documentation
12  * and/or other materials provided with the distribution.
13  *
14  * Neither the name of ARM nor the names of its contributors may be used
15  * to endorse or promote products derived from this software without specific
16  * prior written permission.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
19  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
22  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28  * POSSIBILITY OF SUCH DAMAGE.
29  */
30 
31 #include <assert.h>
32 
33 /* mbedTLS headers */
34 #include <polarssl/memory_buffer_alloc.h>
35 
36 /*
37  * mbedTLS heap
38  */
39 #if (MBEDTLS_KEY_ALG_ID == MBEDTLS_ECDSA)
40 #define MBEDTLS_HEAP_SIZE		(14*1024)
41 #elif (MBEDTLS_KEY_ALG_ID == MBEDTLS_RSA)
42 #define MBEDTLS_HEAP_SIZE		(8*1024)
43 #endif
44 static unsigned char heap[MBEDTLS_HEAP_SIZE];
45 
46 /*
47  * mbedTLS initialization function
48  *
49  * Return: 0 = success, Otherwise = error
50  */
51 void mbedtls_init(void)
52 {
53 	static int ready;
54 	int rc;
55 
56 	if (!ready) {
57 		/* Initialize the mbedTLS heap */
58 		rc = memory_buffer_alloc_init(heap, MBEDTLS_HEAP_SIZE);
59 		if (rc == 0) {
60 			ready = 1;
61 		} else {
62 			assert(0);
63 		}
64 	}
65 }
66