| /* SPDX-License-Identifier: BSD-2-Clause */ |
| /* |
| * Copyright (c) 2014, STMicroelectronics International N.V. |
| */ |
| #ifndef MALLOC_H |
| #define MALLOC_H |
| |
| #include <stddef.h> |
| #include <types_ext.h> |
| |
| void free(void *ptr); |
| |
| #ifdef ENABLE_MDBG |
| |
| void *mdbg_malloc(const char *fname, int lineno, size_t size); |
| void *mdbg_calloc(const char *fname, int lineno, size_t nmemb, size_t size); |
| void *mdbg_realloc(const char *fname, int lineno, void *ptr, size_t size); |
| |
| void mdbg_check(int bufdump); |
| |
| #define malloc(size) mdbg_malloc(__FILE__, __LINE__, (size)) |
| #define calloc(nmemb, size) \ |
| mdbg_calloc(__FILE__, __LINE__, (nmemb), (size)) |
| #define realloc(ptr, size) \ |
| mdbg_realloc(__FILE__, __LINE__, (ptr), (size)) |
| |
| #else |
| |
| void *malloc(size_t size); |
| void *calloc(size_t nmemb, size_t size); |
| void *realloc(void *ptr, size_t size); |
| |
| #define mdbg_check(x) do { } while (0) |
| |
| #endif |
| |
| |
| /* |
| * Returns true if the supplied memory area is within a buffer |
| * previously allocated (and not freed yet). |
| * |
| * Used internally by TAs |
| */ |
| bool malloc_buffer_is_within_alloced(void *buf, size_t len); |
| |
| /* |
| * Returns true if the supplied memory area is overlapping the area used |
| * for heap. |
| * |
| * Used internally by TAs |
| */ |
| bool malloc_buffer_overlaps_heap(void *buf, size_t len); |
| |
| /* |
| * Adds a pool of memory to allocate from. |
| */ |
| void malloc_add_pool(void *buf, size_t len); |
| |
| #ifdef CFG_WITH_STATS |
| /* |
| * Get/reset allocation statistics |
| */ |
| |
| #define TEE_ALLOCATOR_DESC_LENGTH 32 |
| struct malloc_stats { |
| char desc[TEE_ALLOCATOR_DESC_LENGTH]; |
| uint32_t allocated; /* Bytes currently allocated */ |
| uint32_t max_allocated; /* Tracks max value of allocated */ |
| uint32_t size; /* Total size for this allocator */ |
| uint32_t num_alloc_fail; /* Number of failed alloc requests */ |
| uint32_t biggest_alloc_fail; /* Size of biggest failed alloc */ |
| uint32_t biggest_alloc_fail_used; /* Alloc bytes when above occurred */ |
| }; |
| |
| void malloc_get_stats(struct malloc_stats *stats); |
| void malloc_reset_stats(void); |
| #endif /* CFG_WITH_STATS */ |
| |
| |
| #ifdef CFG_VIRTUALIZATION |
| |
| void nex_free(void *ptr); |
| |
| #ifdef ENABLE_MDBG |
| |
| void *nex_mdbg_malloc(const char *fname, int lineno, size_t size); |
| void *nex_mdbg_calloc(const char *fname, int lineno, size_t nmemb, size_t size); |
| void *nex_mdbg_realloc(const char *fname, int lineno, void *ptr, size_t size); |
| |
| void nex_mdbg_check(int bufdump); |
| |
| #define nex_malloc(size) nex_mdbg_malloc(__FILE__, __LINE__, (size)) |
| #define nex_calloc(nmemb, size) \ |
| nex_mdbg_calloc(__FILE__, __LINE__, (nmemb), (size)) |
| #define nex_realloc(ptr, size) \ |
| nex_mdbg_realloc(__FILE__, __LINE__, (ptr), (size)) |
| |
| #else /* ENABLE_MDBG */ |
| |
| void *nex_malloc(size_t size); |
| void *nex_calloc(size_t nmemb, size_t size); |
| void *nex_realloc(void *ptr, size_t size); |
| |
| #define nex_mdbg_check(x) do { } while (0) |
| |
| #endif /* ENABLE_MDBG */ |
| |
| bool nex_malloc_buffer_is_within_alloced(void *buf, size_t len); |
| bool nex_malloc_buffer_overlaps_heap(void *buf, size_t len); |
| void nex_malloc_add_pool(void *buf, size_t len); |
| |
| #ifdef CFG_WITH_STATS |
| /* |
| * Get/reset allocation statistics |
| */ |
| |
| void nex_malloc_get_stats(struct malloc_stats *stats); |
| void nex_malloc_reset_stats(void); |
| |
| #endif /* CFG_WITH_STATS */ |
| #else /* CFG_VIRTUALIZATION */ |
| |
| #define nex_free(ptr) free(ptr) |
| #define nex_malloc(size) malloc(size) |
| #define nex_calloc(nmemb, size) calloc(nmemb, size) |
| #define nex_realloc(ptr, size) realloc(ptr, size) |
| |
| #endif /* CFG_VIRTUALIZATION */ |
| |
| #endif /* MALLOC_H */ |