Gan likun / The prosthetic control

The prosthetic control（MIT）

platform/mbed_mem_trace.c@0:20e0c61e0684, 2022-06-23 (annotated)

Committer:
ganlikun
Date:
Thu Jun 23 05:23:34 2022 +0000
Revision:
0:20e0c61e0684
01

Who changed what in which revision?

UserRevisionLine numberNew contents of line
ganlikun 0:20e0c61e0684 1 /* mbed Microcontroller Library
ganlikun 0:20e0c61e0684 2 * Copyright (c) 2006-2016 ARM Limited
ganlikun 0:20e0c61e0684 3 *
ganlikun 0:20e0c61e0684 4 * Licensed under the Apache License, Version 2.0 (the "License");
ganlikun 0:20e0c61e0684 5 * you may not use this file except in compliance with the License.
ganlikun 0:20e0c61e0684 6 * You may obtain a copy of the License at
ganlikun 0:20e0c61e0684 7 *
ganlikun 0:20e0c61e0684 8 * http://www.apache.org/licenses/LICENSE-2.0
ganlikun 0:20e0c61e0684 9 *
ganlikun 0:20e0c61e0684 10 * Unless required by applicable law or agreed to in writing, software
ganlikun 0:20e0c61e0684 11 * distributed under the License is distributed on an "AS IS" BASIS,
ganlikun 0:20e0c61e0684 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
ganlikun 0:20e0c61e0684 13 * See the License for the specific language governing permissions and
ganlikun 0:20e0c61e0684 14 * limitations under the License.
ganlikun 0:20e0c61e0684 15 */
ganlikun 0:20e0c61e0684 16
ganlikun 0:20e0c61e0684 17 #include <stdlib.h>
ganlikun 0:20e0c61e0684 18 #include <stdarg.h>
ganlikun 0:20e0c61e0684 19 #include <stdio.h>
ganlikun 0:20e0c61e0684 20 #include "platform/mbed_mem_trace.h"
ganlikun 0:20e0c61e0684 21 #include "platform/mbed_critical.h"
ganlikun 0:20e0c61e0684 22
ganlikun 0:20e0c61e0684 23 /******************************************************************************
ganlikun 0:20e0c61e0684 24 * Internal variables, functions and helpers
ganlikun 0:20e0c61e0684 25 *****************************************************************************/
ganlikun 0:20e0c61e0684 26
ganlikun 0:20e0c61e0684 27 /* The callback function that will be called after a traced memory operations finishes. */
ganlikun 0:20e0c61e0684 28 static mbed_mem_trace_cb_t mem_trace_cb;
ganlikun 0:20e0c61e0684 29 /* 'trave_level' guards "trace inside trace" situations (for example, the implementation
ganlikun 0:20e0c61e0684 30 * of realloc() might call malloc() internally, and since malloc() is also traced, this could
ganlikun 0:20e0c61e0684 31 * result in two calls to the callback function instead of one. */
ganlikun 0:20e0c61e0684 32 static uint8_t trace_level;
ganlikun 0:20e0c61e0684 33
ganlikun 0:20e0c61e0684 34 /******************************************************************************
ganlikun 0:20e0c61e0684 35 * Public interface
ganlikun 0:20e0c61e0684 36 *****************************************************************************/
ganlikun 0:20e0c61e0684 37
ganlikun 0:20e0c61e0684 38 void mbed_mem_trace_set_callback(mbed_mem_trace_cb_t cb) {
ganlikun 0:20e0c61e0684 39 mem_trace_cb = cb;
ganlikun 0:20e0c61e0684 40 }
ganlikun 0:20e0c61e0684 41
ganlikun 0:20e0c61e0684 42 void *mbed_mem_trace_malloc(void *res, size_t size, void *caller) {
ganlikun 0:20e0c61e0684 43 if (mem_trace_cb) {
ganlikun 0:20e0c61e0684 44 if (core_util_atomic_incr_u8(&trace_level, 1) == 1) {
ganlikun 0:20e0c61e0684 45 mem_trace_cb(MBED_MEM_TRACE_MALLOC, res, caller, size);
ganlikun 0:20e0c61e0684 46 }
ganlikun 0:20e0c61e0684 47 core_util_atomic_decr_u8(&trace_level, 1);
ganlikun 0:20e0c61e0684 48 }
ganlikun 0:20e0c61e0684 49 return res;
ganlikun 0:20e0c61e0684 50 }
ganlikun 0:20e0c61e0684 51
ganlikun 0:20e0c61e0684 52 void *mbed_mem_trace_realloc(void *res, void *ptr, size_t size, void *caller) {
ganlikun 0:20e0c61e0684 53 if (mem_trace_cb) {
ganlikun 0:20e0c61e0684 54 if (core_util_atomic_incr_u8(&trace_level, 1) == 1) {
ganlikun 0:20e0c61e0684 55 mem_trace_cb(MBED_MEM_TRACE_REALLOC, res, caller, ptr, size);
ganlikun 0:20e0c61e0684 56 }
ganlikun 0:20e0c61e0684 57 core_util_atomic_decr_u8(&trace_level, 1);
ganlikun 0:20e0c61e0684 58 }
ganlikun 0:20e0c61e0684 59 return res;
ganlikun 0:20e0c61e0684 60 }
ganlikun 0:20e0c61e0684 61
ganlikun 0:20e0c61e0684 62 void *mbed_mem_trace_calloc(void *res, size_t num, size_t size, void *caller) {
ganlikun 0:20e0c61e0684 63 if (mem_trace_cb) {
ganlikun 0:20e0c61e0684 64 if (core_util_atomic_incr_u8(&trace_level, 1) == 1) {
ganlikun 0:20e0c61e0684 65 mem_trace_cb(MBED_MEM_TRACE_CALLOC, res, caller, num, size);
ganlikun 0:20e0c61e0684 66 }
ganlikun 0:20e0c61e0684 67 core_util_atomic_decr_u8(&trace_level, 1);
ganlikun 0:20e0c61e0684 68 }
ganlikun 0:20e0c61e0684 69 return res;
ganlikun 0:20e0c61e0684 70 }
ganlikun 0:20e0c61e0684 71
ganlikun 0:20e0c61e0684 72 void mbed_mem_trace_free(void *ptr, void *caller) {
ganlikun 0:20e0c61e0684 73 if (mem_trace_cb) {
ganlikun 0:20e0c61e0684 74 if (core_util_atomic_incr_u8(&trace_level, 1) == 1) {
ganlikun 0:20e0c61e0684 75 mem_trace_cb(MBED_MEM_TRACE_FREE, NULL, caller, ptr);
ganlikun 0:20e0c61e0684 76 }
ganlikun 0:20e0c61e0684 77 core_util_atomic_decr_u8(&trace_level, 1);
ganlikun 0:20e0c61e0684 78 }
ganlikun 0:20e0c61e0684 79 }
ganlikun 0:20e0c61e0684 80
ganlikun 0:20e0c61e0684 81 void mbed_mem_trace_default_callback(uint8_t op, void *res, void *caller, ...) {
ganlikun 0:20e0c61e0684 82 va_list va;
ganlikun 0:20e0c61e0684 83 size_t temp_s1, temp_s2;
ganlikun 0:20e0c61e0684 84 void *temp_ptr;
ganlikun 0:20e0c61e0684 85
ganlikun 0:20e0c61e0684 86 va_start(va, caller);
ganlikun 0:20e0c61e0684 87 switch(op) {
ganlikun 0:20e0c61e0684 88 case MBED_MEM_TRACE_MALLOC:
ganlikun 0:20e0c61e0684 89 temp_s1 = va_arg(va, size_t);
ganlikun 0:20e0c61e0684 90 printf(MBED_MEM_DEFAULT_TRACER_PREFIX "m:%p;%p-%u\n", res, caller, temp_s1);
ganlikun 0:20e0c61e0684 91 break;
ganlikun 0:20e0c61e0684 92
ganlikun 0:20e0c61e0684 93 case MBED_MEM_TRACE_REALLOC:
ganlikun 0:20e0c61e0684 94 temp_ptr = va_arg(va, void*);
ganlikun 0:20e0c61e0684 95 temp_s1 = va_arg(va, size_t);
ganlikun 0:20e0c61e0684 96 printf(MBED_MEM_DEFAULT_TRACER_PREFIX "r:%p;%p-%p;%u\n", res, caller, temp_ptr, temp_s1);
ganlikun 0:20e0c61e0684 97 break;
ganlikun 0:20e0c61e0684 98
ganlikun 0:20e0c61e0684 99 case MBED_MEM_TRACE_CALLOC:
ganlikun 0:20e0c61e0684 100 temp_s1 = va_arg(va, size_t);
ganlikun 0:20e0c61e0684 101 temp_s2 = va_arg(va, size_t);
ganlikun 0:20e0c61e0684 102 printf(MBED_MEM_DEFAULT_TRACER_PREFIX "c:%p;%p-%u;%u\n", res, caller, temp_s1, temp_s2);
ganlikun 0:20e0c61e0684 103 break;
ganlikun 0:20e0c61e0684 104
ganlikun 0:20e0c61e0684 105 case MBED_MEM_TRACE_FREE:
ganlikun 0:20e0c61e0684 106 temp_ptr = va_arg(va, void*);
ganlikun 0:20e0c61e0684 107 printf(MBED_MEM_DEFAULT_TRACER_PREFIX "f:%p;%p-%p\n", res, caller, temp_ptr);
ganlikun 0:20e0c61e0684 108 break;
ganlikun 0:20e0c61e0684 109
ganlikun 0:20e0c61e0684 110 default:
ganlikun 0:20e0c61e0684 111 printf("?\n");
ganlikun 0:20e0c61e0684 112 }
ganlikun 0:20e0c61e0684 113 va_end(va);
ganlikun 0:20e0c61e0684 114 }
ganlikun 0:20e0c61e0684 115
ganlikun 0:20e0c61e0684 116