Important changes to repositories hosted on mbed.com
Mbed hosted mercurial repositories are deprecated and are due to be permanently deleted in July 2026.
To keep a copy of this software download the repository Zip archive or clone locally using Mercurial.
It is also possible to export all your personal repositories from the account settings page.
mbed-os/platform/mbed_alloc_wrappers.cpp@0:fb8047b156bb, 2020-01-15 (annotated)
- Committer:
- rahul_dahiya
- Date:
- Wed Jan 15 15:57:15 2020 +0530
- Revision:
- 0:fb8047b156bb
STM32F7 LWIP
Who changed what in which revision?
User | Revision | Line number | New contents of line |
---|---|---|---|
rahul_dahiya |
0:fb8047b156bb | 1 | /* mbed Microcontroller Library |
rahul_dahiya |
0:fb8047b156bb | 2 | * Copyright (c) 2006-2016 ARM Limited |
rahul_dahiya |
0:fb8047b156bb | 3 | * |
rahul_dahiya |
0:fb8047b156bb | 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
rahul_dahiya |
0:fb8047b156bb | 5 | * you may not use this file except in compliance with the License. |
rahul_dahiya |
0:fb8047b156bb | 6 | * You may obtain a copy of the License at |
rahul_dahiya |
0:fb8047b156bb | 7 | * |
rahul_dahiya |
0:fb8047b156bb | 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
rahul_dahiya |
0:fb8047b156bb | 9 | * |
rahul_dahiya |
0:fb8047b156bb | 10 | * Unless required by applicable law or agreed to in writing, software |
rahul_dahiya |
0:fb8047b156bb | 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
rahul_dahiya |
0:fb8047b156bb | 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
rahul_dahiya |
0:fb8047b156bb | 13 | * See the License for the specific language governing permissions and |
rahul_dahiya |
0:fb8047b156bb | 14 | * limitations under the License. |
rahul_dahiya |
0:fb8047b156bb | 15 | */ |
rahul_dahiya |
0:fb8047b156bb | 16 | |
rahul_dahiya |
0:fb8047b156bb | 17 | #include "platform/mbed_mem_trace.h" |
rahul_dahiya |
0:fb8047b156bb | 18 | #include "platform/mbed_stats.h" |
rahul_dahiya |
0:fb8047b156bb | 19 | #include "platform/mbed_toolchain.h" |
rahul_dahiya |
0:fb8047b156bb | 20 | #include "platform/SingletonPtr.h" |
rahul_dahiya |
0:fb8047b156bb | 21 | #include "platform/PlatformMutex.h" |
rahul_dahiya |
0:fb8047b156bb | 22 | #include <stddef.h> |
rahul_dahiya |
0:fb8047b156bb | 23 | #include <stdio.h> |
rahul_dahiya |
0:fb8047b156bb | 24 | #include <string.h> |
rahul_dahiya |
0:fb8047b156bb | 25 | #include <stdlib.h> |
rahul_dahiya |
0:fb8047b156bb | 26 | |
rahul_dahiya |
0:fb8047b156bb | 27 | /* There are two memory tracers in mbed OS: |
rahul_dahiya |
0:fb8047b156bb | 28 | |
rahul_dahiya |
0:fb8047b156bb | 29 | - the first can be used to detect the maximum heap usage at runtime. It is |
rahul_dahiya |
0:fb8047b156bb | 30 | activated by defining the MBED_HEAP_STATS_ENABLED macro. |
rahul_dahiya |
0:fb8047b156bb | 31 | - the second can be used to trace each memory call by automatically invoking |
rahul_dahiya |
0:fb8047b156bb | 32 | a callback on each memory operation (see hal/api/mbed_mem_trace.h). It is |
rahul_dahiya |
0:fb8047b156bb | 33 | activated by defining the MBED_MEM_TRACING_ENABLED macro. |
rahul_dahiya |
0:fb8047b156bb | 34 | |
rahul_dahiya |
0:fb8047b156bb | 35 | Both tracers can be activated and deactivated in any combination. If both tracers |
rahul_dahiya |
0:fb8047b156bb | 36 | are active, the second one (MBED_MEM_TRACING_ENABLED) will trace the first one's |
rahul_dahiya |
0:fb8047b156bb | 37 | (MBED_HEAP_STATS_ENABLED) memory calls.*/ |
rahul_dahiya |
0:fb8047b156bb | 38 | |
rahul_dahiya |
0:fb8047b156bb | 39 | /******************************************************************************/ |
rahul_dahiya |
0:fb8047b156bb | 40 | /* Implementation of the runtime max heap usage checker */ |
rahul_dahiya |
0:fb8047b156bb | 41 | /******************************************************************************/ |
rahul_dahiya |
0:fb8047b156bb | 42 | |
rahul_dahiya |
0:fb8047b156bb | 43 | /* Size must be a multiple of 8 to keep alignment */ |
rahul_dahiya |
0:fb8047b156bb | 44 | typedef struct { |
rahul_dahiya |
0:fb8047b156bb | 45 | uint32_t size; |
rahul_dahiya |
0:fb8047b156bb | 46 | uint32_t pad; |
rahul_dahiya |
0:fb8047b156bb | 47 | } alloc_info_t; |
rahul_dahiya |
0:fb8047b156bb | 48 | |
rahul_dahiya |
0:fb8047b156bb | 49 | #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 50 | static SingletonPtr<PlatformMutex> mem_trace_mutex; |
rahul_dahiya |
0:fb8047b156bb | 51 | #endif |
rahul_dahiya |
0:fb8047b156bb | 52 | #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 53 | static SingletonPtr<PlatformMutex> malloc_stats_mutex; |
rahul_dahiya |
0:fb8047b156bb | 54 | static mbed_stats_heap_t heap_stats = {0, 0, 0, 0, 0}; |
rahul_dahiya |
0:fb8047b156bb | 55 | #endif |
rahul_dahiya |
0:fb8047b156bb | 56 | |
rahul_dahiya |
0:fb8047b156bb | 57 | void mbed_stats_heap_get(mbed_stats_heap_t *stats) |
rahul_dahiya |
0:fb8047b156bb | 58 | { |
rahul_dahiya |
0:fb8047b156bb | 59 | #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 60 | extern uint32_t mbed_heap_size; |
rahul_dahiya |
0:fb8047b156bb | 61 | heap_stats.reserved_size = mbed_heap_size; |
rahul_dahiya |
0:fb8047b156bb | 62 | |
rahul_dahiya |
0:fb8047b156bb | 63 | malloc_stats_mutex->lock(); |
rahul_dahiya |
0:fb8047b156bb | 64 | memcpy(stats, &heap_stats, sizeof(mbed_stats_heap_t)); |
rahul_dahiya |
0:fb8047b156bb | 65 | malloc_stats_mutex->unlock(); |
rahul_dahiya |
0:fb8047b156bb | 66 | #else |
rahul_dahiya |
0:fb8047b156bb | 67 | memset(stats, 0, sizeof(mbed_stats_heap_t)); |
rahul_dahiya |
0:fb8047b156bb | 68 | #endif |
rahul_dahiya |
0:fb8047b156bb | 69 | } |
rahul_dahiya |
0:fb8047b156bb | 70 | |
rahul_dahiya |
0:fb8047b156bb | 71 | /******************************************************************************/ |
rahul_dahiya |
0:fb8047b156bb | 72 | /* GCC memory allocation wrappers */ |
rahul_dahiya |
0:fb8047b156bb | 73 | /******************************************************************************/ |
rahul_dahiya |
0:fb8047b156bb | 74 | |
rahul_dahiya |
0:fb8047b156bb | 75 | #if defined(TOOLCHAIN_GCC) |
rahul_dahiya |
0:fb8047b156bb | 76 | |
rahul_dahiya |
0:fb8047b156bb | 77 | #ifdef FEATURE_UVISOR |
rahul_dahiya |
0:fb8047b156bb | 78 | #include "uvisor-lib/uvisor-lib.h" |
rahul_dahiya |
0:fb8047b156bb | 79 | #endif/* FEATURE_UVISOR */ |
rahul_dahiya |
0:fb8047b156bb | 80 | |
rahul_dahiya |
0:fb8047b156bb | 81 | extern "C" { |
rahul_dahiya |
0:fb8047b156bb | 82 | void * __real__malloc_r(struct _reent * r, size_t size); |
rahul_dahiya |
0:fb8047b156bb | 83 | void * __real__memalign_r(struct _reent * r, size_t alignment, size_t bytes); |
rahul_dahiya |
0:fb8047b156bb | 84 | void * __real__realloc_r(struct _reent * r, void * ptr, size_t size); |
rahul_dahiya |
0:fb8047b156bb | 85 | void __real__free_r(struct _reent * r, void * ptr); |
rahul_dahiya |
0:fb8047b156bb | 86 | void* __real__calloc_r(struct _reent * r, size_t nmemb, size_t size); |
rahul_dahiya |
0:fb8047b156bb | 87 | } |
rahul_dahiya |
0:fb8047b156bb | 88 | |
rahul_dahiya |
0:fb8047b156bb | 89 | // TODO: memory tracing doesn't work with uVisor enabled. |
rahul_dahiya |
0:fb8047b156bb | 90 | #if !defined(FEATURE_UVISOR) |
rahul_dahiya |
0:fb8047b156bb | 91 | |
rahul_dahiya |
0:fb8047b156bb | 92 | extern "C" void * __wrap__malloc_r(struct _reent * r, size_t size) { |
rahul_dahiya |
0:fb8047b156bb | 93 | void *ptr = NULL; |
rahul_dahiya |
0:fb8047b156bb | 94 | #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 95 | malloc_stats_mutex->lock(); |
rahul_dahiya |
0:fb8047b156bb | 96 | alloc_info_t *alloc_info = (alloc_info_t*)__real__malloc_r(r, size + sizeof(alloc_info_t)); |
rahul_dahiya |
0:fb8047b156bb | 97 | if (alloc_info != NULL) { |
rahul_dahiya |
0:fb8047b156bb | 98 | alloc_info->size = size; |
rahul_dahiya |
0:fb8047b156bb | 99 | ptr = (void*)(alloc_info + 1); |
rahul_dahiya |
0:fb8047b156bb | 100 | heap_stats.current_size += size; |
rahul_dahiya |
0:fb8047b156bb | 101 | heap_stats.total_size += size; |
rahul_dahiya |
0:fb8047b156bb | 102 | heap_stats.alloc_cnt += 1; |
rahul_dahiya |
0:fb8047b156bb | 103 | if (heap_stats.current_size > heap_stats.max_size) { |
rahul_dahiya |
0:fb8047b156bb | 104 | heap_stats.max_size = heap_stats.current_size; |
rahul_dahiya |
0:fb8047b156bb | 105 | } |
rahul_dahiya |
0:fb8047b156bb | 106 | } else { |
rahul_dahiya |
0:fb8047b156bb | 107 | heap_stats.alloc_fail_cnt += 1; |
rahul_dahiya |
0:fb8047b156bb | 108 | } |
rahul_dahiya |
0:fb8047b156bb | 109 | malloc_stats_mutex->unlock(); |
rahul_dahiya |
0:fb8047b156bb | 110 | #else // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 111 | ptr = __real__malloc_r(r, size); |
rahul_dahiya |
0:fb8047b156bb | 112 | #endif // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 113 | #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 114 | mem_trace_mutex->lock(); |
rahul_dahiya |
0:fb8047b156bb | 115 | mbed_mem_trace_malloc(ptr, size, MBED_CALLER_ADDR()); |
rahul_dahiya |
0:fb8047b156bb | 116 | mem_trace_mutex->unlock(); |
rahul_dahiya |
0:fb8047b156bb | 117 | #endif // #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 118 | return ptr; |
rahul_dahiya |
0:fb8047b156bb | 119 | } |
rahul_dahiya |
0:fb8047b156bb | 120 | |
rahul_dahiya |
0:fb8047b156bb | 121 | extern "C" void * __wrap__realloc_r(struct _reent * r, void * ptr, size_t size) { |
rahul_dahiya |
0:fb8047b156bb | 122 | void *new_ptr = NULL; |
rahul_dahiya |
0:fb8047b156bb | 123 | #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 124 | // Implement realloc_r with malloc and free. |
rahul_dahiya |
0:fb8047b156bb | 125 | // The function realloc_r can't be used here directly since |
rahul_dahiya |
0:fb8047b156bb | 126 | // it can call into __wrap__malloc_r (returns ptr + 4) or |
rahul_dahiya |
0:fb8047b156bb | 127 | // resize memory directly (returns ptr + 0). |
rahul_dahiya |
0:fb8047b156bb | 128 | |
rahul_dahiya |
0:fb8047b156bb | 129 | // Note - no lock needed since malloc and free are thread safe |
rahul_dahiya |
0:fb8047b156bb | 130 | |
rahul_dahiya |
0:fb8047b156bb | 131 | // Get old size |
rahul_dahiya |
0:fb8047b156bb | 132 | uint32_t old_size = 0; |
rahul_dahiya |
0:fb8047b156bb | 133 | if (ptr != NULL) { |
rahul_dahiya |
0:fb8047b156bb | 134 | alloc_info_t *alloc_info = ((alloc_info_t*)ptr) - 1; |
rahul_dahiya |
0:fb8047b156bb | 135 | old_size = alloc_info->size; |
rahul_dahiya |
0:fb8047b156bb | 136 | } |
rahul_dahiya |
0:fb8047b156bb | 137 | |
rahul_dahiya |
0:fb8047b156bb | 138 | // Allocate space |
rahul_dahiya |
0:fb8047b156bb | 139 | if (size != 0) { |
rahul_dahiya |
0:fb8047b156bb | 140 | new_ptr = malloc(size); |
rahul_dahiya |
0:fb8047b156bb | 141 | } |
rahul_dahiya |
0:fb8047b156bb | 142 | |
rahul_dahiya |
0:fb8047b156bb | 143 | // If the new buffer has been allocated copy the data to it |
rahul_dahiya |
0:fb8047b156bb | 144 | // and free the old buffer |
rahul_dahiya |
0:fb8047b156bb | 145 | if (new_ptr != NULL) { |
rahul_dahiya |
0:fb8047b156bb | 146 | uint32_t copy_size = (old_size < size) ? old_size : size; |
rahul_dahiya |
0:fb8047b156bb | 147 | memcpy(new_ptr, (void*)ptr, copy_size); |
rahul_dahiya |
0:fb8047b156bb | 148 | free(ptr); |
rahul_dahiya |
0:fb8047b156bb | 149 | } |
rahul_dahiya |
0:fb8047b156bb | 150 | #else // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 151 | new_ptr = __real__realloc_r(r, ptr, size); |
rahul_dahiya |
0:fb8047b156bb | 152 | #endif // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 153 | #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 154 | mem_trace_mutex->lock(); |
rahul_dahiya |
0:fb8047b156bb | 155 | mbed_mem_trace_realloc(new_ptr, ptr, size, MBED_CALLER_ADDR()); |
rahul_dahiya |
0:fb8047b156bb | 156 | mem_trace_mutex->unlock(); |
rahul_dahiya |
0:fb8047b156bb | 157 | #endif // #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 158 | return new_ptr; |
rahul_dahiya |
0:fb8047b156bb | 159 | } |
rahul_dahiya |
0:fb8047b156bb | 160 | |
rahul_dahiya |
0:fb8047b156bb | 161 | extern "C" void __wrap__free_r(struct _reent * r, void * ptr) { |
rahul_dahiya |
0:fb8047b156bb | 162 | #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 163 | malloc_stats_mutex->lock(); |
rahul_dahiya |
0:fb8047b156bb | 164 | alloc_info_t *alloc_info = NULL; |
rahul_dahiya |
0:fb8047b156bb | 165 | if (ptr != NULL) { |
rahul_dahiya |
0:fb8047b156bb | 166 | alloc_info = ((alloc_info_t*)ptr) - 1; |
rahul_dahiya |
0:fb8047b156bb | 167 | heap_stats.current_size -= alloc_info->size; |
rahul_dahiya |
0:fb8047b156bb | 168 | heap_stats.alloc_cnt -= 1; |
rahul_dahiya |
0:fb8047b156bb | 169 | } |
rahul_dahiya |
0:fb8047b156bb | 170 | __real__free_r(r, (void*)alloc_info); |
rahul_dahiya |
0:fb8047b156bb | 171 | malloc_stats_mutex->unlock(); |
rahul_dahiya |
0:fb8047b156bb | 172 | #else // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 173 | __real__free_r(r, ptr); |
rahul_dahiya |
0:fb8047b156bb | 174 | #endif // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 175 | #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 176 | mem_trace_mutex->lock(); |
rahul_dahiya |
0:fb8047b156bb | 177 | mbed_mem_trace_free(ptr, MBED_CALLER_ADDR()); |
rahul_dahiya |
0:fb8047b156bb | 178 | mem_trace_mutex->unlock(); |
rahul_dahiya |
0:fb8047b156bb | 179 | #endif // #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 180 | } |
rahul_dahiya |
0:fb8047b156bb | 181 | |
rahul_dahiya |
0:fb8047b156bb | 182 | extern "C" void * __wrap__calloc_r(struct _reent * r, size_t nmemb, size_t size) { |
rahul_dahiya |
0:fb8047b156bb | 183 | void *ptr = NULL; |
rahul_dahiya |
0:fb8047b156bb | 184 | #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 185 | // Note - no lock needed since malloc is thread safe |
rahul_dahiya |
0:fb8047b156bb | 186 | |
rahul_dahiya |
0:fb8047b156bb | 187 | ptr = malloc(nmemb * size); |
rahul_dahiya |
0:fb8047b156bb | 188 | if (ptr != NULL) { |
rahul_dahiya |
0:fb8047b156bb | 189 | memset(ptr, 0, nmemb * size); |
rahul_dahiya |
0:fb8047b156bb | 190 | } |
rahul_dahiya |
0:fb8047b156bb | 191 | #else // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 192 | ptr = __real__calloc_r(r, nmemb, size); |
rahul_dahiya |
0:fb8047b156bb | 193 | #endif // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 194 | #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 195 | mem_trace_mutex->lock(); |
rahul_dahiya |
0:fb8047b156bb | 196 | mbed_mem_trace_calloc(ptr, nmemb, size, MBED_CALLER_ADDR()); |
rahul_dahiya |
0:fb8047b156bb | 197 | mem_trace_mutex->unlock(); |
rahul_dahiya |
0:fb8047b156bb | 198 | #endif // #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 199 | return ptr; |
rahul_dahiya |
0:fb8047b156bb | 200 | } |
rahul_dahiya |
0:fb8047b156bb | 201 | |
rahul_dahiya |
0:fb8047b156bb | 202 | extern "C" void * __wrap__memalign_r(struct _reent * r, size_t alignment, size_t bytes) { |
rahul_dahiya |
0:fb8047b156bb | 203 | return __real__memalign_r(r, alignment, bytes); |
rahul_dahiya |
0:fb8047b156bb | 204 | } |
rahul_dahiya |
0:fb8047b156bb | 205 | |
rahul_dahiya |
0:fb8047b156bb | 206 | #endif // if !defined(FEATURE_UVISOR) |
rahul_dahiya |
0:fb8047b156bb | 207 | |
rahul_dahiya |
0:fb8047b156bb | 208 | |
rahul_dahiya |
0:fb8047b156bb | 209 | /******************************************************************************/ |
rahul_dahiya |
0:fb8047b156bb | 210 | /* ARMCC / IAR memory allocation wrappers */ |
rahul_dahiya |
0:fb8047b156bb | 211 | /******************************************************************************/ |
rahul_dahiya |
0:fb8047b156bb | 212 | |
rahul_dahiya |
0:fb8047b156bb | 213 | #elif defined(TOOLCHAIN_ARM) || defined(__ICCARM__) |
rahul_dahiya |
0:fb8047b156bb | 214 | |
rahul_dahiya |
0:fb8047b156bb | 215 | #if defined(TOOLCHAIN_ARM) |
rahul_dahiya |
0:fb8047b156bb | 216 | #define SUPER_MALLOC $Super$$malloc |
rahul_dahiya |
0:fb8047b156bb | 217 | #define SUB_MALLOC $Sub$$malloc |
rahul_dahiya |
0:fb8047b156bb | 218 | #define SUPER_REALLOC $Super$$realloc |
rahul_dahiya |
0:fb8047b156bb | 219 | #define SUB_REALLOC $Sub$$realloc |
rahul_dahiya |
0:fb8047b156bb | 220 | #define SUPER_CALLOC $Super$$calloc |
rahul_dahiya |
0:fb8047b156bb | 221 | #define SUB_CALLOC $Sub$$calloc |
rahul_dahiya |
0:fb8047b156bb | 222 | #define SUPER_FREE $Super$$free |
rahul_dahiya |
0:fb8047b156bb | 223 | #define SUB_FREE $Sub$$free |
rahul_dahiya |
0:fb8047b156bb | 224 | #elif defined(__ICCARM__) |
rahul_dahiya |
0:fb8047b156bb | 225 | #define SUPER_MALLOC $Super$$__iar_dlmalloc |
rahul_dahiya |
0:fb8047b156bb | 226 | #define SUB_MALLOC $Sub$$__iar_dlmalloc |
rahul_dahiya |
0:fb8047b156bb | 227 | #define SUPER_REALLOC $Super$$__iar_dlrealloc |
rahul_dahiya |
0:fb8047b156bb | 228 | #define SUB_REALLOC $Sub$$__iar_dlrealloc |
rahul_dahiya |
0:fb8047b156bb | 229 | #define SUPER_CALLOC $Super$$__iar_dlcalloc |
rahul_dahiya |
0:fb8047b156bb | 230 | #define SUB_CALLOC $Sub$$__iar_dlcalloc |
rahul_dahiya |
0:fb8047b156bb | 231 | #define SUPER_FREE $Super$$__iar_dlfree |
rahul_dahiya |
0:fb8047b156bb | 232 | #define SUB_FREE $Sub$$__iar_dlfree |
rahul_dahiya |
0:fb8047b156bb | 233 | #endif |
rahul_dahiya |
0:fb8047b156bb | 234 | |
rahul_dahiya |
0:fb8047b156bb | 235 | /* Enable hooking of memory function only if tracing is also enabled */ |
rahul_dahiya |
0:fb8047b156bb | 236 | #if defined(MBED_MEM_TRACING_ENABLED) || defined(MBED_HEAP_STATS_ENABLED) |
rahul_dahiya |
0:fb8047b156bb | 237 | |
rahul_dahiya |
0:fb8047b156bb | 238 | extern "C" { |
rahul_dahiya |
0:fb8047b156bb | 239 | void *SUPER_MALLOC(size_t size); |
rahul_dahiya |
0:fb8047b156bb | 240 | void *SUPER_REALLOC(void *ptr, size_t size); |
rahul_dahiya |
0:fb8047b156bb | 241 | void *SUPER_CALLOC(size_t nmemb, size_t size); |
rahul_dahiya |
0:fb8047b156bb | 242 | void SUPER_FREE(void *ptr); |
rahul_dahiya |
0:fb8047b156bb | 243 | } |
rahul_dahiya |
0:fb8047b156bb | 244 | |
rahul_dahiya |
0:fb8047b156bb | 245 | extern "C" void* SUB_MALLOC(size_t size) { |
rahul_dahiya |
0:fb8047b156bb | 246 | void *ptr = NULL; |
rahul_dahiya |
0:fb8047b156bb | 247 | #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 248 | malloc_stats_mutex->lock(); |
rahul_dahiya |
0:fb8047b156bb | 249 | alloc_info_t *alloc_info = (alloc_info_t*)SUPER_MALLOC(size + sizeof(alloc_info_t)); |
rahul_dahiya |
0:fb8047b156bb | 250 | if (alloc_info != NULL) { |
rahul_dahiya |
0:fb8047b156bb | 251 | alloc_info->size = size; |
rahul_dahiya |
0:fb8047b156bb | 252 | ptr = (void*)(alloc_info + 1); |
rahul_dahiya |
0:fb8047b156bb | 253 | heap_stats.current_size += size; |
rahul_dahiya |
0:fb8047b156bb | 254 | heap_stats.total_size += size; |
rahul_dahiya |
0:fb8047b156bb | 255 | heap_stats.alloc_cnt += 1; |
rahul_dahiya |
0:fb8047b156bb | 256 | if (heap_stats.current_size > heap_stats.max_size) { |
rahul_dahiya |
0:fb8047b156bb | 257 | heap_stats.max_size = heap_stats.current_size; |
rahul_dahiya |
0:fb8047b156bb | 258 | } |
rahul_dahiya |
0:fb8047b156bb | 259 | } else { |
rahul_dahiya |
0:fb8047b156bb | 260 | heap_stats.alloc_fail_cnt += 1; |
rahul_dahiya |
0:fb8047b156bb | 261 | } |
rahul_dahiya |
0:fb8047b156bb | 262 | malloc_stats_mutex->unlock(); |
rahul_dahiya |
0:fb8047b156bb | 263 | #else // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 264 | ptr = SUPER_MALLOC(size); |
rahul_dahiya |
0:fb8047b156bb | 265 | #endif // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 266 | #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 267 | mem_trace_mutex->lock(); |
rahul_dahiya |
0:fb8047b156bb | 268 | mbed_mem_trace_malloc(ptr, size, MBED_CALLER_ADDR()); |
rahul_dahiya |
0:fb8047b156bb | 269 | mem_trace_mutex->unlock(); |
rahul_dahiya |
0:fb8047b156bb | 270 | #endif // #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 271 | return ptr; |
rahul_dahiya |
0:fb8047b156bb | 272 | } |
rahul_dahiya |
0:fb8047b156bb | 273 | |
rahul_dahiya |
0:fb8047b156bb | 274 | extern "C" void* SUB_REALLOC(void *ptr, size_t size) { |
rahul_dahiya |
0:fb8047b156bb | 275 | void *new_ptr = NULL; |
rahul_dahiya |
0:fb8047b156bb | 276 | #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 277 | // Note - no lock needed since malloc and free are thread safe |
rahul_dahiya |
0:fb8047b156bb | 278 | |
rahul_dahiya |
0:fb8047b156bb | 279 | // Get old size |
rahul_dahiya |
0:fb8047b156bb | 280 | uint32_t old_size = 0; |
rahul_dahiya |
0:fb8047b156bb | 281 | if (ptr != NULL) { |
rahul_dahiya |
0:fb8047b156bb | 282 | alloc_info_t *alloc_info = ((alloc_info_t*)ptr) - 1; |
rahul_dahiya |
0:fb8047b156bb | 283 | old_size = alloc_info->size; |
rahul_dahiya |
0:fb8047b156bb | 284 | } |
rahul_dahiya |
0:fb8047b156bb | 285 | |
rahul_dahiya |
0:fb8047b156bb | 286 | // Allocate space |
rahul_dahiya |
0:fb8047b156bb | 287 | if (size != 0) { |
rahul_dahiya |
0:fb8047b156bb | 288 | new_ptr = malloc(size); |
rahul_dahiya |
0:fb8047b156bb | 289 | } |
rahul_dahiya |
0:fb8047b156bb | 290 | |
rahul_dahiya |
0:fb8047b156bb | 291 | // If the new buffer has been allocated copy the data to it |
rahul_dahiya |
0:fb8047b156bb | 292 | // and free the old buffer |
rahul_dahiya |
0:fb8047b156bb | 293 | if (new_ptr != NULL) { |
rahul_dahiya |
0:fb8047b156bb | 294 | uint32_t copy_size = (old_size < size) ? old_size : size; |
rahul_dahiya |
0:fb8047b156bb | 295 | memcpy(new_ptr, (void*)ptr, copy_size); |
rahul_dahiya |
0:fb8047b156bb | 296 | free(ptr); |
rahul_dahiya |
0:fb8047b156bb | 297 | } |
rahul_dahiya |
0:fb8047b156bb | 298 | #else // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 299 | new_ptr = SUPER_REALLOC(ptr, size); |
rahul_dahiya |
0:fb8047b156bb | 300 | #endif // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 301 | #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 302 | mem_trace_mutex->lock(); |
rahul_dahiya |
0:fb8047b156bb | 303 | mbed_mem_trace_realloc(new_ptr, ptr, size, MBED_CALLER_ADDR()); |
rahul_dahiya |
0:fb8047b156bb | 304 | mem_trace_mutex->unlock(); |
rahul_dahiya |
0:fb8047b156bb | 305 | #endif // #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 306 | return new_ptr; |
rahul_dahiya |
0:fb8047b156bb | 307 | } |
rahul_dahiya |
0:fb8047b156bb | 308 | |
rahul_dahiya |
0:fb8047b156bb | 309 | extern "C" void *SUB_CALLOC(size_t nmemb, size_t size) { |
rahul_dahiya |
0:fb8047b156bb | 310 | void *ptr = NULL; |
rahul_dahiya |
0:fb8047b156bb | 311 | #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 312 | // Note - no lock needed since malloc is thread safe |
rahul_dahiya |
0:fb8047b156bb | 313 | ptr = malloc(nmemb * size); |
rahul_dahiya |
0:fb8047b156bb | 314 | if (ptr != NULL) { |
rahul_dahiya |
0:fb8047b156bb | 315 | memset(ptr, 0, nmemb * size); |
rahul_dahiya |
0:fb8047b156bb | 316 | } |
rahul_dahiya |
0:fb8047b156bb | 317 | #else // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 318 | ptr = SUPER_CALLOC(nmemb, size); |
rahul_dahiya |
0:fb8047b156bb | 319 | #endif // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 320 | #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 321 | mem_trace_mutex->lock(); |
rahul_dahiya |
0:fb8047b156bb | 322 | mbed_mem_trace_calloc(ptr, nmemb, size, MBED_CALLER_ADDR()); |
rahul_dahiya |
0:fb8047b156bb | 323 | mem_trace_mutex->unlock(); |
rahul_dahiya |
0:fb8047b156bb | 324 | #endif // #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 325 | return ptr; |
rahul_dahiya |
0:fb8047b156bb | 326 | } |
rahul_dahiya |
0:fb8047b156bb | 327 | |
rahul_dahiya |
0:fb8047b156bb | 328 | extern "C" void SUB_FREE(void *ptr) { |
rahul_dahiya |
0:fb8047b156bb | 329 | #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 330 | malloc_stats_mutex->lock(); |
rahul_dahiya |
0:fb8047b156bb | 331 | alloc_info_t *alloc_info = NULL; |
rahul_dahiya |
0:fb8047b156bb | 332 | if (ptr != NULL) { |
rahul_dahiya |
0:fb8047b156bb | 333 | alloc_info = ((alloc_info_t*)ptr) - 1; |
rahul_dahiya |
0:fb8047b156bb | 334 | heap_stats.current_size -= alloc_info->size; |
rahul_dahiya |
0:fb8047b156bb | 335 | heap_stats.alloc_cnt -= 1; |
rahul_dahiya |
0:fb8047b156bb | 336 | } |
rahul_dahiya |
0:fb8047b156bb | 337 | SUPER_FREE((void*)alloc_info); |
rahul_dahiya |
0:fb8047b156bb | 338 | malloc_stats_mutex->unlock(); |
rahul_dahiya |
0:fb8047b156bb | 339 | #else // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 340 | SUPER_FREE(ptr); |
rahul_dahiya |
0:fb8047b156bb | 341 | #endif // #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 342 | #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 343 | mem_trace_mutex->lock(); |
rahul_dahiya |
0:fb8047b156bb | 344 | mbed_mem_trace_free(ptr, MBED_CALLER_ADDR()); |
rahul_dahiya |
0:fb8047b156bb | 345 | mem_trace_mutex->unlock(); |
rahul_dahiya |
0:fb8047b156bb | 346 | #endif // #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 347 | } |
rahul_dahiya |
0:fb8047b156bb | 348 | |
rahul_dahiya |
0:fb8047b156bb | 349 | #endif // #if defined(MBED_MEM_TRACING_ENABLED) || defined(MBED_HEAP_STATS_ENABLED) |
rahul_dahiya |
0:fb8047b156bb | 350 | |
rahul_dahiya |
0:fb8047b156bb | 351 | /******************************************************************************/ |
rahul_dahiya |
0:fb8047b156bb | 352 | /* Allocation wrappers for other toolchains are not supported yet */ |
rahul_dahiya |
0:fb8047b156bb | 353 | /******************************************************************************/ |
rahul_dahiya |
0:fb8047b156bb | 354 | |
rahul_dahiya |
0:fb8047b156bb | 355 | #else |
rahul_dahiya |
0:fb8047b156bb | 356 | |
rahul_dahiya |
0:fb8047b156bb | 357 | #ifdef MBED_MEM_TRACING_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 358 | #error Memory tracing is not supported with the current toolchain. |
rahul_dahiya |
0:fb8047b156bb | 359 | #endif |
rahul_dahiya |
0:fb8047b156bb | 360 | |
rahul_dahiya |
0:fb8047b156bb | 361 | #ifdef MBED_HEAP_STATS_ENABLED |
rahul_dahiya |
0:fb8047b156bb | 362 | #error Heap statistics are not supported with the current toolchain. |
rahul_dahiya |
0:fb8047b156bb | 363 | #endif |
rahul_dahiya |
0:fb8047b156bb | 364 | |
rahul_dahiya |
0:fb8047b156bb | 365 | #endif // #if defined(TOOLCHAIN_GCC) |