erkin yucel / mbed-os

mbed os with nrf51 internal bandgap enabled to read battery level

Dependents:   BLE_file_test BLE_Blink ExternalEncoder

features/mbedtls/src/memory_buffer_alloc.c@0:f269e3021894, 2016-10-23 (annotated)

Committer:
elessair
Date:
Sun Oct 23 15:10:02 2016 +0000
Revision:
0:f269e3021894
Initial commit

Who changed what in which revision?

UserRevisionLine numberNew contents of line
elessair 0:f269e3021894 1 /*
elessair 0:f269e3021894 2 * Buffer-based memory allocator
elessair 0:f269e3021894 3 *
elessair 0:f269e3021894 4 * Copyright (C) 2006-2015, ARM Limited, All Rights Reserved
elessair 0:f269e3021894 5 * SPDX-License-Identifier: Apache-2.0
elessair 0:f269e3021894 6 *
elessair 0:f269e3021894 7 * Licensed under the Apache License, Version 2.0 (the "License"); you may
elessair 0:f269e3021894 8 * not use this file except in compliance with the License.
elessair 0:f269e3021894 9 * You may obtain a copy of the License at
elessair 0:f269e3021894 10 *
elessair 0:f269e3021894 11 * http://www.apache.org/licenses/LICENSE-2.0
elessair 0:f269e3021894 12 *
elessair 0:f269e3021894 13 * Unless required by applicable law or agreed to in writing, software
elessair 0:f269e3021894 14 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
elessair 0:f269e3021894 15 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
elessair 0:f269e3021894 16 * See the License for the specific language governing permissions and
elessair 0:f269e3021894 17 * limitations under the License.
elessair 0:f269e3021894 18 *
elessair 0:f269e3021894 19 * This file is part of mbed TLS (https://tls.mbed.org)
elessair 0:f269e3021894 20 */
elessair 0:f269e3021894 21
elessair 0:f269e3021894 22 #if !defined(MBEDTLS_CONFIG_FILE)
elessair 0:f269e3021894 23 #include "mbedtls/config.h"
elessair 0:f269e3021894 24 #else
elessair 0:f269e3021894 25 #include MBEDTLS_CONFIG_FILE
elessair 0:f269e3021894 26 #endif
elessair 0:f269e3021894 27
elessair 0:f269e3021894 28 #if defined(MBEDTLS_MEMORY_BUFFER_ALLOC_C)
elessair 0:f269e3021894 29 #include "mbedtls/memory_buffer_alloc.h"
elessair 0:f269e3021894 30
elessair 0:f269e3021894 31 /* No need for the header guard as MBEDTLS_MEMORY_BUFFER_ALLOC_C
elessair 0:f269e3021894 32 is dependent upon MBEDTLS_PLATFORM_C */
elessair 0:f269e3021894 33 #include "mbedtls/platform.h"
elessair 0:f269e3021894 34
elessair 0:f269e3021894 35 #include <string.h>
elessair 0:f269e3021894 36
elessair 0:f269e3021894 37 #if defined(MBEDTLS_MEMORY_BACKTRACE)
elessair 0:f269e3021894 38 #include <execinfo.h>
elessair 0:f269e3021894 39 #endif
elessair 0:f269e3021894 40
elessair 0:f269e3021894 41 #if defined(MBEDTLS_THREADING_C)
elessair 0:f269e3021894 42 #include "mbedtls/threading.h"
elessair 0:f269e3021894 43 #endif
elessair 0:f269e3021894 44
elessair 0:f269e3021894 45 /* Implementation that should never be optimized out by the compiler */
elessair 0:f269e3021894 46 static void mbedtls_zeroize( void *v, size_t n ) {
elessair 0:f269e3021894 47 volatile unsigned char *p = v; while( n-- ) *p++ = 0;
elessair 0:f269e3021894 48 }
elessair 0:f269e3021894 49
elessair 0:f269e3021894 50 #define MAGIC1 0xFF00AA55
elessair 0:f269e3021894 51 #define MAGIC2 0xEE119966
elessair 0:f269e3021894 52 #define MAX_BT 20
elessair 0:f269e3021894 53
elessair 0:f269e3021894 54 typedef struct _memory_header memory_header;
elessair 0:f269e3021894 55 struct _memory_header
elessair 0:f269e3021894 56 {
elessair 0:f269e3021894 57 size_t magic1;
elessair 0:f269e3021894 58 size_t size;
elessair 0:f269e3021894 59 size_t alloc;
elessair 0:f269e3021894 60 memory_header *prev;
elessair 0:f269e3021894 61 memory_header *next;
elessair 0:f269e3021894 62 memory_header *prev_free;
elessair 0:f269e3021894 63 memory_header *next_free;
elessair 0:f269e3021894 64 #if defined(MBEDTLS_MEMORY_BACKTRACE)
elessair 0:f269e3021894 65 char **trace;
elessair 0:f269e3021894 66 size_t trace_count;
elessair 0:f269e3021894 67 #endif
elessair 0:f269e3021894 68 size_t magic2;
elessair 0:f269e3021894 69 };
elessair 0:f269e3021894 70
elessair 0:f269e3021894 71 typedef struct
elessair 0:f269e3021894 72 {
elessair 0:f269e3021894 73 unsigned char *buf;
elessair 0:f269e3021894 74 size_t len;
elessair 0:f269e3021894 75 memory_header *first;
elessair 0:f269e3021894 76 memory_header *first_free;
elessair 0:f269e3021894 77 int verify;
elessair 0:f269e3021894 78 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 79 size_t alloc_count;
elessair 0:f269e3021894 80 size_t free_count;
elessair 0:f269e3021894 81 size_t total_used;
elessair 0:f269e3021894 82 size_t maximum_used;
elessair 0:f269e3021894 83 size_t header_count;
elessair 0:f269e3021894 84 size_t maximum_header_count;
elessair 0:f269e3021894 85 #endif
elessair 0:f269e3021894 86 #if defined(MBEDTLS_THREADING_C)
elessair 0:f269e3021894 87 mbedtls_threading_mutex_t mutex;
elessair 0:f269e3021894 88 #endif
elessair 0:f269e3021894 89 }
elessair 0:f269e3021894 90 buffer_alloc_ctx;
elessair 0:f269e3021894 91
elessair 0:f269e3021894 92 static buffer_alloc_ctx heap;
elessair 0:f269e3021894 93
elessair 0:f269e3021894 94 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 95 static void debug_header( memory_header *hdr )
elessair 0:f269e3021894 96 {
elessair 0:f269e3021894 97 #if defined(MBEDTLS_MEMORY_BACKTRACE)
elessair 0:f269e3021894 98 size_t i;
elessair 0:f269e3021894 99 #endif
elessair 0:f269e3021894 100
elessair 0:f269e3021894 101 mbedtls_fprintf( stderr, "HDR: PTR(%10zu), PREV(%10zu), NEXT(%10zu), "
elessair 0:f269e3021894 102 "ALLOC(%zu), SIZE(%10zu)\n",
elessair 0:f269e3021894 103 (size_t) hdr, (size_t) hdr->prev, (size_t) hdr->next,
elessair 0:f269e3021894 104 hdr->alloc, hdr->size );
elessair 0:f269e3021894 105 mbedtls_fprintf( stderr, " FPREV(%10zu), FNEXT(%10zu)\n",
elessair 0:f269e3021894 106 (size_t) hdr->prev_free, (size_t) hdr->next_free );
elessair 0:f269e3021894 107
elessair 0:f269e3021894 108 #if defined(MBEDTLS_MEMORY_BACKTRACE)
elessair 0:f269e3021894 109 mbedtls_fprintf( stderr, "TRACE: \n" );
elessair 0:f269e3021894 110 for( i = 0; i < hdr->trace_count; i++ )
elessair 0:f269e3021894 111 mbedtls_fprintf( stderr, "%s\n", hdr->trace[i] );
elessair 0:f269e3021894 112 mbedtls_fprintf( stderr, "\n" );
elessair 0:f269e3021894 113 #endif
elessair 0:f269e3021894 114 }
elessair 0:f269e3021894 115
elessair 0:f269e3021894 116 static void debug_chain()
elessair 0:f269e3021894 117 {
elessair 0:f269e3021894 118 memory_header *cur = heap.first;
elessair 0:f269e3021894 119
elessair 0:f269e3021894 120 mbedtls_fprintf( stderr, "\nBlock list\n" );
elessair 0:f269e3021894 121 while( cur != NULL )
elessair 0:f269e3021894 122 {
elessair 0:f269e3021894 123 debug_header( cur );
elessair 0:f269e3021894 124 cur = cur->next;
elessair 0:f269e3021894 125 }
elessair 0:f269e3021894 126
elessair 0:f269e3021894 127 mbedtls_fprintf( stderr, "Free list\n" );
elessair 0:f269e3021894 128 cur = heap.first_free;
elessair 0:f269e3021894 129
elessair 0:f269e3021894 130 while( cur != NULL )
elessair 0:f269e3021894 131 {
elessair 0:f269e3021894 132 debug_header( cur );
elessair 0:f269e3021894 133 cur = cur->next_free;
elessair 0:f269e3021894 134 }
elessair 0:f269e3021894 135 }
elessair 0:f269e3021894 136 #endif /* MBEDTLS_MEMORY_DEBUG */
elessair 0:f269e3021894 137
elessair 0:f269e3021894 138 static int verify_header( memory_header *hdr )
elessair 0:f269e3021894 139 {
elessair 0:f269e3021894 140 if( hdr->magic1 != MAGIC1 )
elessair 0:f269e3021894 141 {
elessair 0:f269e3021894 142 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 143 mbedtls_fprintf( stderr, "FATAL: MAGIC1 mismatch\n" );
elessair 0:f269e3021894 144 #endif
elessair 0:f269e3021894 145 return( 1 );
elessair 0:f269e3021894 146 }
elessair 0:f269e3021894 147
elessair 0:f269e3021894 148 if( hdr->magic2 != MAGIC2 )
elessair 0:f269e3021894 149 {
elessair 0:f269e3021894 150 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 151 mbedtls_fprintf( stderr, "FATAL: MAGIC2 mismatch\n" );
elessair 0:f269e3021894 152 #endif
elessair 0:f269e3021894 153 return( 1 );
elessair 0:f269e3021894 154 }
elessair 0:f269e3021894 155
elessair 0:f269e3021894 156 if( hdr->alloc > 1 )
elessair 0:f269e3021894 157 {
elessair 0:f269e3021894 158 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 159 mbedtls_fprintf( stderr, "FATAL: alloc has illegal value\n" );
elessair 0:f269e3021894 160 #endif
elessair 0:f269e3021894 161 return( 1 );
elessair 0:f269e3021894 162 }
elessair 0:f269e3021894 163
elessair 0:f269e3021894 164 if( hdr->prev != NULL && hdr->prev == hdr->next )
elessair 0:f269e3021894 165 {
elessair 0:f269e3021894 166 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 167 mbedtls_fprintf( stderr, "FATAL: prev == next\n" );
elessair 0:f269e3021894 168 #endif
elessair 0:f269e3021894 169 return( 1 );
elessair 0:f269e3021894 170 }
elessair 0:f269e3021894 171
elessair 0:f269e3021894 172 if( hdr->prev_free != NULL && hdr->prev_free == hdr->next_free )
elessair 0:f269e3021894 173 {
elessair 0:f269e3021894 174 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 175 mbedtls_fprintf( stderr, "FATAL: prev_free == next_free\n" );
elessair 0:f269e3021894 176 #endif
elessair 0:f269e3021894 177 return( 1 );
elessair 0:f269e3021894 178 }
elessair 0:f269e3021894 179
elessair 0:f269e3021894 180 return( 0 );
elessair 0:f269e3021894 181 }
elessair 0:f269e3021894 182
elessair 0:f269e3021894 183 static int verify_chain()
elessair 0:f269e3021894 184 {
elessair 0:f269e3021894 185 memory_header *prv = heap.first, *cur = heap.first->next;
elessair 0:f269e3021894 186
elessair 0:f269e3021894 187 if( verify_header( heap.first ) != 0 )
elessair 0:f269e3021894 188 {
elessair 0:f269e3021894 189 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 190 mbedtls_fprintf( stderr, "FATAL: verification of first header "
elessair 0:f269e3021894 191 "failed\n" );
elessair 0:f269e3021894 192 #endif
elessair 0:f269e3021894 193 return( 1 );
elessair 0:f269e3021894 194 }
elessair 0:f269e3021894 195
elessair 0:f269e3021894 196 if( heap.first->prev != NULL )
elessair 0:f269e3021894 197 {
elessair 0:f269e3021894 198 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 199 mbedtls_fprintf( stderr, "FATAL: verification failed: "
elessair 0:f269e3021894 200 "first->prev != NULL\n" );
elessair 0:f269e3021894 201 #endif
elessair 0:f269e3021894 202 return( 1 );
elessair 0:f269e3021894 203 }
elessair 0:f269e3021894 204
elessair 0:f269e3021894 205 while( cur != NULL )
elessair 0:f269e3021894 206 {
elessair 0:f269e3021894 207 if( verify_header( cur ) != 0 )
elessair 0:f269e3021894 208 {
elessair 0:f269e3021894 209 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 210 mbedtls_fprintf( stderr, "FATAL: verification of header "
elessair 0:f269e3021894 211 "failed\n" );
elessair 0:f269e3021894 212 #endif
elessair 0:f269e3021894 213 return( 1 );
elessair 0:f269e3021894 214 }
elessair 0:f269e3021894 215
elessair 0:f269e3021894 216 if( cur->prev != prv )
elessair 0:f269e3021894 217 {
elessair 0:f269e3021894 218 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 219 mbedtls_fprintf( stderr, "FATAL: verification failed: "
elessair 0:f269e3021894 220 "cur->prev != prv\n" );
elessair 0:f269e3021894 221 #endif
elessair 0:f269e3021894 222 return( 1 );
elessair 0:f269e3021894 223 }
elessair 0:f269e3021894 224
elessair 0:f269e3021894 225 prv = cur;
elessair 0:f269e3021894 226 cur = cur->next;
elessair 0:f269e3021894 227 }
elessair 0:f269e3021894 228
elessair 0:f269e3021894 229 return( 0 );
elessair 0:f269e3021894 230 }
elessair 0:f269e3021894 231
elessair 0:f269e3021894 232 static void *buffer_alloc_calloc( size_t n, size_t size )
elessair 0:f269e3021894 233 {
elessair 0:f269e3021894 234 memory_header *new, *cur = heap.first_free;
elessair 0:f269e3021894 235 unsigned char *p;
elessair 0:f269e3021894 236 void *ret;
elessair 0:f269e3021894 237 size_t original_len, len;
elessair 0:f269e3021894 238 #if defined(MBEDTLS_MEMORY_BACKTRACE)
elessair 0:f269e3021894 239 void *trace_buffer[MAX_BT];
elessair 0:f269e3021894 240 size_t trace_cnt;
elessair 0:f269e3021894 241 #endif
elessair 0:f269e3021894 242
elessair 0:f269e3021894 243 if( heap.buf == NULL || heap.first == NULL )
elessair 0:f269e3021894 244 return( NULL );
elessair 0:f269e3021894 245
elessair 0:f269e3021894 246 original_len = len = n * size;
elessair 0:f269e3021894 247
elessair 0:f269e3021894 248 if( n != 0 && len / n != size )
elessair 0:f269e3021894 249 return( NULL );
elessair 0:f269e3021894 250
elessair 0:f269e3021894 251 if( len % MBEDTLS_MEMORY_ALIGN_MULTIPLE )
elessair 0:f269e3021894 252 {
elessair 0:f269e3021894 253 len -= len % MBEDTLS_MEMORY_ALIGN_MULTIPLE;
elessair 0:f269e3021894 254 len += MBEDTLS_MEMORY_ALIGN_MULTIPLE;
elessair 0:f269e3021894 255 }
elessair 0:f269e3021894 256
elessair 0:f269e3021894 257 // Find block that fits
elessair 0:f269e3021894 258 //
elessair 0:f269e3021894 259 while( cur != NULL )
elessair 0:f269e3021894 260 {
elessair 0:f269e3021894 261 if( cur->size >= len )
elessair 0:f269e3021894 262 break;
elessair 0:f269e3021894 263
elessair 0:f269e3021894 264 cur = cur->next_free;
elessair 0:f269e3021894 265 }
elessair 0:f269e3021894 266
elessair 0:f269e3021894 267 if( cur == NULL )
elessair 0:f269e3021894 268 return( NULL );
elessair 0:f269e3021894 269
elessair 0:f269e3021894 270 if( cur->alloc != 0 )
elessair 0:f269e3021894 271 {
elessair 0:f269e3021894 272 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 273 mbedtls_fprintf( stderr, "FATAL: block in free_list but allocated "
elessair 0:f269e3021894 274 "data\n" );
elessair 0:f269e3021894 275 #endif
elessair 0:f269e3021894 276 mbedtls_exit( 1 );
elessair 0:f269e3021894 277 }
elessair 0:f269e3021894 278
elessair 0:f269e3021894 279 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 280 heap.alloc_count++;
elessair 0:f269e3021894 281 #endif
elessair 0:f269e3021894 282
elessair 0:f269e3021894 283 // Found location, split block if > memory_header + 4 room left
elessair 0:f269e3021894 284 //
elessair 0:f269e3021894 285 if( cur->size - len < sizeof(memory_header) +
elessair 0:f269e3021894 286 MBEDTLS_MEMORY_ALIGN_MULTIPLE )
elessair 0:f269e3021894 287 {
elessair 0:f269e3021894 288 cur->alloc = 1;
elessair 0:f269e3021894 289
elessair 0:f269e3021894 290 // Remove from free_list
elessair 0:f269e3021894 291 //
elessair 0:f269e3021894 292 if( cur->prev_free != NULL )
elessair 0:f269e3021894 293 cur->prev_free->next_free = cur->next_free;
elessair 0:f269e3021894 294 else
elessair 0:f269e3021894 295 heap.first_free = cur->next_free;
elessair 0:f269e3021894 296
elessair 0:f269e3021894 297 if( cur->next_free != NULL )
elessair 0:f269e3021894 298 cur->next_free->prev_free = cur->prev_free;
elessair 0:f269e3021894 299
elessair 0:f269e3021894 300 cur->prev_free = NULL;
elessair 0:f269e3021894 301 cur->next_free = NULL;
elessair 0:f269e3021894 302
elessair 0:f269e3021894 303 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 304 heap.total_used += cur->size;
elessair 0:f269e3021894 305 if( heap.total_used > heap.maximum_used )
elessair 0:f269e3021894 306 heap.maximum_used = heap.total_used;
elessair 0:f269e3021894 307 #endif
elessair 0:f269e3021894 308 #if defined(MBEDTLS_MEMORY_BACKTRACE)
elessair 0:f269e3021894 309 trace_cnt = backtrace( trace_buffer, MAX_BT );
elessair 0:f269e3021894 310 cur->trace = backtrace_symbols( trace_buffer, trace_cnt );
elessair 0:f269e3021894 311 cur->trace_count = trace_cnt;
elessair 0:f269e3021894 312 #endif
elessair 0:f269e3021894 313
elessair 0:f269e3021894 314 if( ( heap.verify & MBEDTLS_MEMORY_VERIFY_ALLOC ) && verify_chain() != 0 )
elessair 0:f269e3021894 315 mbedtls_exit( 1 );
elessair 0:f269e3021894 316
elessair 0:f269e3021894 317 ret = (unsigned char *) cur + sizeof( memory_header );
elessair 0:f269e3021894 318 memset( ret, 0, original_len );
elessair 0:f269e3021894 319
elessair 0:f269e3021894 320 return( ret );
elessair 0:f269e3021894 321 }
elessair 0:f269e3021894 322
elessair 0:f269e3021894 323 p = ( (unsigned char *) cur ) + sizeof(memory_header) + len;
elessair 0:f269e3021894 324 new = (memory_header *) p;
elessair 0:f269e3021894 325
elessair 0:f269e3021894 326 new->size = cur->size - len - sizeof(memory_header);
elessair 0:f269e3021894 327 new->alloc = 0;
elessair 0:f269e3021894 328 new->prev = cur;
elessair 0:f269e3021894 329 new->next = cur->next;
elessair 0:f269e3021894 330 #if defined(MBEDTLS_MEMORY_BACKTRACE)
elessair 0:f269e3021894 331 new->trace = NULL;
elessair 0:f269e3021894 332 new->trace_count = 0;
elessair 0:f269e3021894 333 #endif
elessair 0:f269e3021894 334 new->magic1 = MAGIC1;
elessair 0:f269e3021894 335 new->magic2 = MAGIC2;
elessair 0:f269e3021894 336
elessair 0:f269e3021894 337 if( new->next != NULL )
elessair 0:f269e3021894 338 new->next->prev = new;
elessair 0:f269e3021894 339
elessair 0:f269e3021894 340 // Replace cur with new in free_list
elessair 0:f269e3021894 341 //
elessair 0:f269e3021894 342 new->prev_free = cur->prev_free;
elessair 0:f269e3021894 343 new->next_free = cur->next_free;
elessair 0:f269e3021894 344 if( new->prev_free != NULL )
elessair 0:f269e3021894 345 new->prev_free->next_free = new;
elessair 0:f269e3021894 346 else
elessair 0:f269e3021894 347 heap.first_free = new;
elessair 0:f269e3021894 348
elessair 0:f269e3021894 349 if( new->next_free != NULL )
elessair 0:f269e3021894 350 new->next_free->prev_free = new;
elessair 0:f269e3021894 351
elessair 0:f269e3021894 352 cur->alloc = 1;
elessair 0:f269e3021894 353 cur->size = len;
elessair 0:f269e3021894 354 cur->next = new;
elessair 0:f269e3021894 355 cur->prev_free = NULL;
elessair 0:f269e3021894 356 cur->next_free = NULL;
elessair 0:f269e3021894 357
elessair 0:f269e3021894 358 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 359 heap.header_count++;
elessair 0:f269e3021894 360 if( heap.header_count > heap.maximum_header_count )
elessair 0:f269e3021894 361 heap.maximum_header_count = heap.header_count;
elessair 0:f269e3021894 362 heap.total_used += cur->size;
elessair 0:f269e3021894 363 if( heap.total_used > heap.maximum_used )
elessair 0:f269e3021894 364 heap.maximum_used = heap.total_used;
elessair 0:f269e3021894 365 #endif
elessair 0:f269e3021894 366 #if defined(MBEDTLS_MEMORY_BACKTRACE)
elessair 0:f269e3021894 367 trace_cnt = backtrace( trace_buffer, MAX_BT );
elessair 0:f269e3021894 368 cur->trace = backtrace_symbols( trace_buffer, trace_cnt );
elessair 0:f269e3021894 369 cur->trace_count = trace_cnt;
elessair 0:f269e3021894 370 #endif
elessair 0:f269e3021894 371
elessair 0:f269e3021894 372 if( ( heap.verify & MBEDTLS_MEMORY_VERIFY_ALLOC ) && verify_chain() != 0 )
elessair 0:f269e3021894 373 mbedtls_exit( 1 );
elessair 0:f269e3021894 374
elessair 0:f269e3021894 375 ret = (unsigned char *) cur + sizeof( memory_header );
elessair 0:f269e3021894 376 memset( ret, 0, original_len );
elessair 0:f269e3021894 377
elessair 0:f269e3021894 378 return( ret );
elessair 0:f269e3021894 379 }
elessair 0:f269e3021894 380
elessair 0:f269e3021894 381 static void buffer_alloc_free( void *ptr )
elessair 0:f269e3021894 382 {
elessair 0:f269e3021894 383 memory_header *hdr, *old = NULL;
elessair 0:f269e3021894 384 unsigned char *p = (unsigned char *) ptr;
elessair 0:f269e3021894 385
elessair 0:f269e3021894 386 if( ptr == NULL || heap.buf == NULL || heap.first == NULL )
elessair 0:f269e3021894 387 return;
elessair 0:f269e3021894 388
elessair 0:f269e3021894 389 if( p < heap.buf || p > heap.buf + heap.len )
elessair 0:f269e3021894 390 {
elessair 0:f269e3021894 391 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 392 mbedtls_fprintf( stderr, "FATAL: mbedtls_free() outside of managed "
elessair 0:f269e3021894 393 "space\n" );
elessair 0:f269e3021894 394 #endif
elessair 0:f269e3021894 395 mbedtls_exit( 1 );
elessair 0:f269e3021894 396 }
elessair 0:f269e3021894 397
elessair 0:f269e3021894 398 p -= sizeof(memory_header);
elessair 0:f269e3021894 399 hdr = (memory_header *) p;
elessair 0:f269e3021894 400
elessair 0:f269e3021894 401 if( verify_header( hdr ) != 0 )
elessair 0:f269e3021894 402 mbedtls_exit( 1 );
elessair 0:f269e3021894 403
elessair 0:f269e3021894 404 if( hdr->alloc != 1 )
elessair 0:f269e3021894 405 {
elessair 0:f269e3021894 406 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 407 mbedtls_fprintf( stderr, "FATAL: mbedtls_free() on unallocated "
elessair 0:f269e3021894 408 "data\n" );
elessair 0:f269e3021894 409 #endif
elessair 0:f269e3021894 410 mbedtls_exit( 1 );
elessair 0:f269e3021894 411 }
elessair 0:f269e3021894 412
elessair 0:f269e3021894 413 hdr->alloc = 0;
elessair 0:f269e3021894 414
elessair 0:f269e3021894 415 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 416 heap.free_count++;
elessair 0:f269e3021894 417 heap.total_used -= hdr->size;
elessair 0:f269e3021894 418 #endif
elessair 0:f269e3021894 419
elessair 0:f269e3021894 420 #if defined(MBEDTLS_MEMORY_BACKTRACE)
elessair 0:f269e3021894 421 free( hdr->trace );
elessair 0:f269e3021894 422 hdr->trace = NULL;
elessair 0:f269e3021894 423 hdr->trace_count = 0;
elessair 0:f269e3021894 424 #endif
elessair 0:f269e3021894 425
elessair 0:f269e3021894 426 // Regroup with block before
elessair 0:f269e3021894 427 //
elessair 0:f269e3021894 428 if( hdr->prev != NULL && hdr->prev->alloc == 0 )
elessair 0:f269e3021894 429 {
elessair 0:f269e3021894 430 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 431 heap.header_count--;
elessair 0:f269e3021894 432 #endif
elessair 0:f269e3021894 433 hdr->prev->size += sizeof(memory_header) + hdr->size;
elessair 0:f269e3021894 434 hdr->prev->next = hdr->next;
elessair 0:f269e3021894 435 old = hdr;
elessair 0:f269e3021894 436 hdr = hdr->prev;
elessair 0:f269e3021894 437
elessair 0:f269e3021894 438 if( hdr->next != NULL )
elessair 0:f269e3021894 439 hdr->next->prev = hdr;
elessair 0:f269e3021894 440
elessair 0:f269e3021894 441 memset( old, 0, sizeof(memory_header) );
elessair 0:f269e3021894 442 }
elessair 0:f269e3021894 443
elessair 0:f269e3021894 444 // Regroup with block after
elessair 0:f269e3021894 445 //
elessair 0:f269e3021894 446 if( hdr->next != NULL && hdr->next->alloc == 0 )
elessair 0:f269e3021894 447 {
elessair 0:f269e3021894 448 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 449 heap.header_count--;
elessair 0:f269e3021894 450 #endif
elessair 0:f269e3021894 451 hdr->size += sizeof(memory_header) + hdr->next->size;
elessair 0:f269e3021894 452 old = hdr->next;
elessair 0:f269e3021894 453 hdr->next = hdr->next->next;
elessair 0:f269e3021894 454
elessair 0:f269e3021894 455 if( hdr->prev_free != NULL || hdr->next_free != NULL )
elessair 0:f269e3021894 456 {
elessair 0:f269e3021894 457 if( hdr->prev_free != NULL )
elessair 0:f269e3021894 458 hdr->prev_free->next_free = hdr->next_free;
elessair 0:f269e3021894 459 else
elessair 0:f269e3021894 460 heap.first_free = hdr->next_free;
elessair 0:f269e3021894 461
elessair 0:f269e3021894 462 if( hdr->next_free != NULL )
elessair 0:f269e3021894 463 hdr->next_free->prev_free = hdr->prev_free;
elessair 0:f269e3021894 464 }
elessair 0:f269e3021894 465
elessair 0:f269e3021894 466 hdr->prev_free = old->prev_free;
elessair 0:f269e3021894 467 hdr->next_free = old->next_free;
elessair 0:f269e3021894 468
elessair 0:f269e3021894 469 if( hdr->prev_free != NULL )
elessair 0:f269e3021894 470 hdr->prev_free->next_free = hdr;
elessair 0:f269e3021894 471 else
elessair 0:f269e3021894 472 heap.first_free = hdr;
elessair 0:f269e3021894 473
elessair 0:f269e3021894 474 if( hdr->next_free != NULL )
elessair 0:f269e3021894 475 hdr->next_free->prev_free = hdr;
elessair 0:f269e3021894 476
elessair 0:f269e3021894 477 if( hdr->next != NULL )
elessair 0:f269e3021894 478 hdr->next->prev = hdr;
elessair 0:f269e3021894 479
elessair 0:f269e3021894 480 memset( old, 0, sizeof(memory_header) );
elessair 0:f269e3021894 481 }
elessair 0:f269e3021894 482
elessair 0:f269e3021894 483 // Prepend to free_list if we have not merged
elessair 0:f269e3021894 484 // (Does not have to stay in same order as prev / next list)
elessair 0:f269e3021894 485 //
elessair 0:f269e3021894 486 if( old == NULL )
elessair 0:f269e3021894 487 {
elessair 0:f269e3021894 488 hdr->next_free = heap.first_free;
elessair 0:f269e3021894 489 if( heap.first_free != NULL )
elessair 0:f269e3021894 490 heap.first_free->prev_free = hdr;
elessair 0:f269e3021894 491 heap.first_free = hdr;
elessair 0:f269e3021894 492 }
elessair 0:f269e3021894 493
elessair 0:f269e3021894 494 if( ( heap.verify & MBEDTLS_MEMORY_VERIFY_FREE ) && verify_chain() != 0 )
elessair 0:f269e3021894 495 mbedtls_exit( 1 );
elessair 0:f269e3021894 496 }
elessair 0:f269e3021894 497
elessair 0:f269e3021894 498 void mbedtls_memory_buffer_set_verify( int verify )
elessair 0:f269e3021894 499 {
elessair 0:f269e3021894 500 heap.verify = verify;
elessair 0:f269e3021894 501 }
elessair 0:f269e3021894 502
elessair 0:f269e3021894 503 int mbedtls_memory_buffer_alloc_verify()
elessair 0:f269e3021894 504 {
elessair 0:f269e3021894 505 return verify_chain();
elessair 0:f269e3021894 506 }
elessair 0:f269e3021894 507
elessair 0:f269e3021894 508 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 509 void mbedtls_memory_buffer_alloc_status()
elessair 0:f269e3021894 510 {
elessair 0:f269e3021894 511 mbedtls_fprintf( stderr,
elessair 0:f269e3021894 512 "Current use: %zu blocks / %zu bytes, max: %zu blocks / "
elessair 0:f269e3021894 513 "%zu bytes (total %zu bytes), alloc / free: %zu / %zu\n",
elessair 0:f269e3021894 514 heap.header_count, heap.total_used,
elessair 0:f269e3021894 515 heap.maximum_header_count, heap.maximum_used,
elessair 0:f269e3021894 516 heap.maximum_header_count * sizeof( memory_header )
elessair 0:f269e3021894 517 + heap.maximum_used,
elessair 0:f269e3021894 518 heap.alloc_count, heap.free_count );
elessair 0:f269e3021894 519
elessair 0:f269e3021894 520 if( heap.first->next == NULL )
elessair 0:f269e3021894 521 mbedtls_fprintf( stderr, "All memory de-allocated in stack buffer\n" );
elessair 0:f269e3021894 522 else
elessair 0:f269e3021894 523 {
elessair 0:f269e3021894 524 mbedtls_fprintf( stderr, "Memory currently allocated:\n" );
elessair 0:f269e3021894 525 debug_chain();
elessair 0:f269e3021894 526 }
elessair 0:f269e3021894 527 }
elessair 0:f269e3021894 528
elessair 0:f269e3021894 529 void mbedtls_memory_buffer_alloc_max_get( size_t *max_used, size_t *max_blocks )
elessair 0:f269e3021894 530 {
elessair 0:f269e3021894 531 *max_used = heap.maximum_used;
elessair 0:f269e3021894 532 *max_blocks = heap.maximum_header_count;
elessair 0:f269e3021894 533 }
elessair 0:f269e3021894 534
elessair 0:f269e3021894 535 void mbedtls_memory_buffer_alloc_max_reset( void )
elessair 0:f269e3021894 536 {
elessair 0:f269e3021894 537 heap.maximum_used = 0;
elessair 0:f269e3021894 538 heap.maximum_header_count = 0;
elessair 0:f269e3021894 539 }
elessair 0:f269e3021894 540
elessair 0:f269e3021894 541 void mbedtls_memory_buffer_alloc_cur_get( size_t *cur_used, size_t *cur_blocks )
elessair 0:f269e3021894 542 {
elessair 0:f269e3021894 543 *cur_used = heap.total_used;
elessair 0:f269e3021894 544 *cur_blocks = heap.header_count;
elessair 0:f269e3021894 545 }
elessair 0:f269e3021894 546 #endif /* MBEDTLS_MEMORY_DEBUG */
elessair 0:f269e3021894 547
elessair 0:f269e3021894 548 #if defined(MBEDTLS_THREADING_C)
elessair 0:f269e3021894 549 static void *buffer_alloc_calloc_mutexed( size_t n, size_t size )
elessair 0:f269e3021894 550 {
elessair 0:f269e3021894 551 void *buf;
elessair 0:f269e3021894 552 if( mbedtls_mutex_lock( &heap.mutex ) != 0 )
elessair 0:f269e3021894 553 return( NULL );
elessair 0:f269e3021894 554 buf = buffer_alloc_calloc( n, size );
elessair 0:f269e3021894 555 if( mbedtls_mutex_unlock( &heap.mutex ) )
elessair 0:f269e3021894 556 return( NULL );
elessair 0:f269e3021894 557 return( buf );
elessair 0:f269e3021894 558 }
elessair 0:f269e3021894 559
elessair 0:f269e3021894 560 static void buffer_alloc_free_mutexed( void *ptr )
elessair 0:f269e3021894 561 {
elessair 0:f269e3021894 562 /* We have to good option here, but corrupting the heap seems
elessair 0:f269e3021894 563 * worse than loosing memory. */
elessair 0:f269e3021894 564 if( mbedtls_mutex_lock( &heap.mutex ) )
elessair 0:f269e3021894 565 return;
elessair 0:f269e3021894 566 buffer_alloc_free( ptr );
elessair 0:f269e3021894 567 (void) mbedtls_mutex_unlock( &heap.mutex );
elessair 0:f269e3021894 568 }
elessair 0:f269e3021894 569 #endif /* MBEDTLS_THREADING_C */
elessair 0:f269e3021894 570
elessair 0:f269e3021894 571 void mbedtls_memory_buffer_alloc_init( unsigned char *buf, size_t len )
elessair 0:f269e3021894 572 {
elessair 0:f269e3021894 573 memset( &heap, 0, sizeof(buffer_alloc_ctx) );
elessair 0:f269e3021894 574 memset( buf, 0, len );
elessair 0:f269e3021894 575
elessair 0:f269e3021894 576 #if defined(MBEDTLS_THREADING_C)
elessair 0:f269e3021894 577 mbedtls_mutex_init( &heap.mutex );
elessair 0:f269e3021894 578 mbedtls_platform_set_calloc_free( buffer_alloc_calloc_mutexed,
elessair 0:f269e3021894 579 buffer_alloc_free_mutexed );
elessair 0:f269e3021894 580 #else
elessair 0:f269e3021894 581 mbedtls_platform_set_calloc_free( buffer_alloc_calloc, buffer_alloc_free );
elessair 0:f269e3021894 582 #endif
elessair 0:f269e3021894 583
elessair 0:f269e3021894 584 if( (size_t) buf % MBEDTLS_MEMORY_ALIGN_MULTIPLE )
elessair 0:f269e3021894 585 {
elessair 0:f269e3021894 586 /* Adjust len first since buf is used in the computation */
elessair 0:f269e3021894 587 len -= MBEDTLS_MEMORY_ALIGN_MULTIPLE
elessair 0:f269e3021894 588 - (size_t) buf % MBEDTLS_MEMORY_ALIGN_MULTIPLE;
elessair 0:f269e3021894 589 buf += MBEDTLS_MEMORY_ALIGN_MULTIPLE
elessair 0:f269e3021894 590 - (size_t) buf % MBEDTLS_MEMORY_ALIGN_MULTIPLE;
elessair 0:f269e3021894 591 }
elessair 0:f269e3021894 592
elessair 0:f269e3021894 593 heap.buf = buf;
elessair 0:f269e3021894 594 heap.len = len;
elessair 0:f269e3021894 595
elessair 0:f269e3021894 596 heap.first = (memory_header *) buf;
elessair 0:f269e3021894 597 heap.first->size = len - sizeof(memory_header);
elessair 0:f269e3021894 598 heap.first->magic1 = MAGIC1;
elessair 0:f269e3021894 599 heap.first->magic2 = MAGIC2;
elessair 0:f269e3021894 600 heap.first_free = heap.first;
elessair 0:f269e3021894 601 }
elessair 0:f269e3021894 602
elessair 0:f269e3021894 603 void mbedtls_memory_buffer_alloc_free()
elessair 0:f269e3021894 604 {
elessair 0:f269e3021894 605 #if defined(MBEDTLS_THREADING_C)
elessair 0:f269e3021894 606 mbedtls_mutex_free( &heap.mutex );
elessair 0:f269e3021894 607 #endif
elessair 0:f269e3021894 608 mbedtls_zeroize( &heap, sizeof(buffer_alloc_ctx) );
elessair 0:f269e3021894 609 }
elessair 0:f269e3021894 610
elessair 0:f269e3021894 611 #if defined(MBEDTLS_SELF_TEST)
elessair 0:f269e3021894 612 static int check_pointer( void *p )
elessair 0:f269e3021894 613 {
elessair 0:f269e3021894 614 if( p == NULL )
elessair 0:f269e3021894 615 return( -1 );
elessair 0:f269e3021894 616
elessair 0:f269e3021894 617 if( (size_t) p % MBEDTLS_MEMORY_ALIGN_MULTIPLE != 0 )
elessair 0:f269e3021894 618 return( -1 );
elessair 0:f269e3021894 619
elessair 0:f269e3021894 620 return( 0 );
elessair 0:f269e3021894 621 }
elessair 0:f269e3021894 622
elessair 0:f269e3021894 623 static int check_all_free( )
elessair 0:f269e3021894 624 {
elessair 0:f269e3021894 625 if(
elessair 0:f269e3021894 626 #if defined(MBEDTLS_MEMORY_DEBUG)
elessair 0:f269e3021894 627 heap.total_used != 0 ||
elessair 0:f269e3021894 628 #endif
elessair 0:f269e3021894 629 heap.first != heap.first_free ||
elessair 0:f269e3021894 630 (void *) heap.first != (void *) heap.buf )
elessair 0:f269e3021894 631 {
elessair 0:f269e3021894 632 return( -1 );
elessair 0:f269e3021894 633 }
elessair 0:f269e3021894 634
elessair 0:f269e3021894 635 return( 0 );
elessair 0:f269e3021894 636 }
elessair 0:f269e3021894 637
elessair 0:f269e3021894 638 #define TEST_ASSERT( condition ) \
elessair 0:f269e3021894 639 if( ! (condition) ) \
elessair 0:f269e3021894 640 { \
elessair 0:f269e3021894 641 if( verbose != 0 ) \
elessair 0:f269e3021894 642 mbedtls_printf( "failed\n" ); \
elessair 0:f269e3021894 643 \
elessair 0:f269e3021894 644 ret = 1; \
elessair 0:f269e3021894 645 goto cleanup; \
elessair 0:f269e3021894 646 }
elessair 0:f269e3021894 647
elessair 0:f269e3021894 648 int mbedtls_memory_buffer_alloc_self_test( int verbose )
elessair 0:f269e3021894 649 {
elessair 0:f269e3021894 650 unsigned char buf[1024];
elessair 0:f269e3021894 651 unsigned char *p, *q, *r, *end;
elessair 0:f269e3021894 652 int ret = 0;
elessair 0:f269e3021894 653
elessair 0:f269e3021894 654 if( verbose != 0 )
elessair 0:f269e3021894 655 mbedtls_printf( " MBA test #1 (basic alloc-free cycle): " );
elessair 0:f269e3021894 656
elessair 0:f269e3021894 657 mbedtls_memory_buffer_alloc_init( buf, sizeof( buf ) );
elessair 0:f269e3021894 658
elessair 0:f269e3021894 659 p = mbedtls_calloc( 1, 1 );
elessair 0:f269e3021894 660 q = mbedtls_calloc( 1, 128 );
elessair 0:f269e3021894 661 r = mbedtls_calloc( 1, 16 );
elessair 0:f269e3021894 662
elessair 0:f269e3021894 663 TEST_ASSERT( check_pointer( p ) == 0 &&
elessair 0:f269e3021894 664 check_pointer( q ) == 0 &&
elessair 0:f269e3021894 665 check_pointer( r ) == 0 );
elessair 0:f269e3021894 666
elessair 0:f269e3021894 667 mbedtls_free( r );
elessair 0:f269e3021894 668 mbedtls_free( q );
elessair 0:f269e3021894 669 mbedtls_free( p );
elessair 0:f269e3021894 670
elessair 0:f269e3021894 671 TEST_ASSERT( check_all_free( ) == 0 );
elessair 0:f269e3021894 672
elessair 0:f269e3021894 673 /* Memorize end to compare with the next test */
elessair 0:f269e3021894 674 end = heap.buf + heap.len;
elessair 0:f269e3021894 675
elessair 0:f269e3021894 676 mbedtls_memory_buffer_alloc_free( );
elessair 0:f269e3021894 677
elessair 0:f269e3021894 678 if( verbose != 0 )
elessair 0:f269e3021894 679 mbedtls_printf( "passed\n" );
elessair 0:f269e3021894 680
elessair 0:f269e3021894 681 if( verbose != 0 )
elessair 0:f269e3021894 682 mbedtls_printf( " MBA test #2 (buf not aligned): " );
elessair 0:f269e3021894 683
elessair 0:f269e3021894 684 mbedtls_memory_buffer_alloc_init( buf + 1, sizeof( buf ) - 1 );
elessair 0:f269e3021894 685
elessair 0:f269e3021894 686 TEST_ASSERT( heap.buf + heap.len == end );
elessair 0:f269e3021894 687
elessair 0:f269e3021894 688 p = mbedtls_calloc( 1, 1 );
elessair 0:f269e3021894 689 q = mbedtls_calloc( 1, 128 );
elessair 0:f269e3021894 690 r = mbedtls_calloc( 1, 16 );
elessair 0:f269e3021894 691
elessair 0:f269e3021894 692 TEST_ASSERT( check_pointer( p ) == 0 &&
elessair 0:f269e3021894 693 check_pointer( q ) == 0 &&
elessair 0:f269e3021894 694 check_pointer( r ) == 0 );
elessair 0:f269e3021894 695
elessair 0:f269e3021894 696 mbedtls_free( r );
elessair 0:f269e3021894 697 mbedtls_free( q );
elessair 0:f269e3021894 698 mbedtls_free( p );
elessair 0:f269e3021894 699
elessair 0:f269e3021894 700 TEST_ASSERT( check_all_free( ) == 0 );
elessair 0:f269e3021894 701
elessair 0:f269e3021894 702 mbedtls_memory_buffer_alloc_free( );
elessair 0:f269e3021894 703
elessair 0:f269e3021894 704 if( verbose != 0 )
elessair 0:f269e3021894 705 mbedtls_printf( "passed\n" );
elessair 0:f269e3021894 706
elessair 0:f269e3021894 707 if( verbose != 0 )
elessair 0:f269e3021894 708 mbedtls_printf( " MBA test #3 (full): " );
elessair 0:f269e3021894 709
elessair 0:f269e3021894 710 mbedtls_memory_buffer_alloc_init( buf, sizeof( buf ) );
elessair 0:f269e3021894 711
elessair 0:f269e3021894 712 p = mbedtls_calloc( 1, sizeof( buf ) - sizeof( memory_header ) );
elessair 0:f269e3021894 713
elessair 0:f269e3021894 714 TEST_ASSERT( check_pointer( p ) == 0 );
elessair 0:f269e3021894 715 TEST_ASSERT( mbedtls_calloc( 1, 1 ) == NULL );
elessair 0:f269e3021894 716
elessair 0:f269e3021894 717 mbedtls_free( p );
elessair 0:f269e3021894 718
elessair 0:f269e3021894 719 p = mbedtls_calloc( 1, sizeof( buf ) - 2 * sizeof( memory_header ) - 16 );
elessair 0:f269e3021894 720 q = mbedtls_calloc( 1, 16 );
elessair 0:f269e3021894 721
elessair 0:f269e3021894 722 TEST_ASSERT( check_pointer( p ) == 0 && check_pointer( q ) == 0 );
elessair 0:f269e3021894 723 TEST_ASSERT( mbedtls_calloc( 1, 1 ) == NULL );
elessair 0:f269e3021894 724
elessair 0:f269e3021894 725 mbedtls_free( q );
elessair 0:f269e3021894 726
elessair 0:f269e3021894 727 TEST_ASSERT( mbedtls_calloc( 1, 17 ) == NULL );
elessair 0:f269e3021894 728
elessair 0:f269e3021894 729 mbedtls_free( p );
elessair 0:f269e3021894 730
elessair 0:f269e3021894 731 TEST_ASSERT( check_all_free( ) == 0 );
elessair 0:f269e3021894 732
elessair 0:f269e3021894 733 mbedtls_memory_buffer_alloc_free( );
elessair 0:f269e3021894 734
elessair 0:f269e3021894 735 if( verbose != 0 )
elessair 0:f269e3021894 736 mbedtls_printf( "passed\n" );
elessair 0:f269e3021894 737
elessair 0:f269e3021894 738 cleanup:
elessair 0:f269e3021894 739 mbedtls_memory_buffer_alloc_free( );
elessair 0:f269e3021894 740
elessair 0:f269e3021894 741 return( ret );
elessair 0:f269e3021894 742 }
elessair 0:f269e3021894 743 #endif /* MBEDTLS_SELF_TEST */
elessair 0:f269e3021894 744
elessair 0:f269e3021894 745 #endif /* MBEDTLS_MEMORY_BUFFER_ALLOC_C */