1 /*
2  * SPDX-FileCopyrightText: 2018-2022 Espressif Systems (Shanghai) CO LTD
3  *
4  * SPDX-License-Identifier: Apache-2.0
5  */
6 
7 #include <esp_attr.h>
8 #include <esp_heap_caps.h>
9 #include <sdkconfig.h>
10 #include "esp_mem.h"
11 
12 #ifndef CONFIG_MBEDTLS_CUSTOM_MEM_ALLOC
13 
esp_mbedtls_mem_calloc(size_t n,size_t size)14 IRAM_ATTR void *esp_mbedtls_mem_calloc(size_t n, size_t size)
15 {
16 #ifdef CONFIG_MBEDTLS_INTERNAL_MEM_ALLOC
17     return heap_caps_calloc(n, size, MALLOC_CAP_INTERNAL|MALLOC_CAP_8BIT);
18 #elif CONFIG_MBEDTLS_EXTERNAL_MEM_ALLOC
19     return heap_caps_calloc(n, size, MALLOC_CAP_SPIRAM|MALLOC_CAP_8BIT);
20 #elif CONFIG_MBEDTLS_IRAM_8BIT_MEM_ALLOC
21 #ifdef CONFIG_MBEDTLS_ASYMMETRIC_CONTENT_LEN
22     if ((n*size) >= CONFIG_MBEDTLS_SSL_IN_CONTENT_LEN || (n*size) >= CONFIG_MBEDTLS_SSL_OUT_CONTENT_LEN) {
23 #else
24     if ((n*size) >= CONFIG_MBEDTLS_SSL_MAX_CONTENT_LEN) {
25 #endif
26         return heap_caps_calloc_prefer(n, size, 2, MALLOC_CAP_INTERNAL|MALLOC_CAP_IRAM_8BIT, MALLOC_CAP_INTERNAL|MALLOC_CAP_8BIT);
27     } else {
28         return heap_caps_calloc(n, size, MALLOC_CAP_INTERNAL|MALLOC_CAP_8BIT);
29     }
30 
31 #else
32     return calloc(n, size);
33 #endif
34 }
35 
36 IRAM_ATTR void esp_mbedtls_mem_free(void *ptr)
37 {
38     return heap_caps_free(ptr);
39 }
40 
41 #endif /* !CONFIG_MBEDTLS_CUSTOM_MEM_ALLOC */
42