takashi kadono / Mbed OS Nucleo446_SSD1331

Color Oled(SSD1331) connect to STMicroelectronics Nucleo-F466

Dependencies:   ssd1331

mbed-os/platform/mbed_alloc_wrappers.cpp@3:f3764f852aa8, 2018-10-11 (annotated)

Committer:
kadonotakashi
Date:
Thu Oct 11 02:27:46 2018 +0000
Revision:
3:f3764f852aa8
Parent:
0:8fdf9a60065b 
Nucreo 446 + SSD1331 test version;

Who changed what in which revision?

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