mirror of
				https://github.com/espressif/esp-idf.git
				synced 2025-11-03 14:01:53 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			109 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			109 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
// Copyright 2015-2016 Espressif Systems (Shanghai) PTE LTD
 | 
						|
//
 | 
						|
// Licensed under the Apache License, Version 2.0 (the "License");
 | 
						|
// you may not use this file except in compliance with the License.
 | 
						|
// You may obtain a copy of the License at
 | 
						|
 | 
						|
//     http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
//
 | 
						|
// Unless required by applicable law or agreed to in writing, software
 | 
						|
// distributed under the License is distributed on an "AS IS" BASIS,
 | 
						|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
						|
// See the License for the specific language governing permissions and
 | 
						|
// limitations under the License.
 | 
						|
#pragma once
 | 
						|
 | 
						|
#ifdef MULTI_HEAP_FREERTOS
 | 
						|
 | 
						|
#include "freertos/FreeRTOS.h"
 | 
						|
 | 
						|
#include "sdkconfig.h"
 | 
						|
#include "esp_rom_sys.h"
 | 
						|
#if CONFIG_IDF_TARGET_ESP32
 | 
						|
#include "esp32/rom/ets_sys.h" // will be removed in idf v5.0
 | 
						|
#elif CONFIG_IDF_TARGET_ESP32S2
 | 
						|
#include "esp32s2/rom/ets_sys.h"
 | 
						|
#endif
 | 
						|
#include <assert.h>
 | 
						|
 | 
						|
typedef portMUX_TYPE multi_heap_lock_t;
 | 
						|
 | 
						|
/* Because malloc/free can happen inside an ISR context,
 | 
						|
   we need to use portmux spinlocks here not RTOS mutexes */
 | 
						|
#define MULTI_HEAP_LOCK(PLOCK) do {                         \
 | 
						|
        if((PLOCK) != NULL) {                               \
 | 
						|
            portENTER_CRITICAL((PLOCK));                    \
 | 
						|
        }                                                   \
 | 
						|
    } while(0)
 | 
						|
 | 
						|
 | 
						|
#define MULTI_HEAP_UNLOCK(PLOCK) do {                       \
 | 
						|
        if ((PLOCK) != NULL) {                              \
 | 
						|
            portEXIT_CRITICAL((PLOCK));                     \
 | 
						|
        }                                                   \
 | 
						|
    } while(0)
 | 
						|
 | 
						|
#define MULTI_HEAP_LOCK_INIT(PLOCK) do {                    \
 | 
						|
        vPortCPUInitializeMutex((PLOCK));                   \
 | 
						|
    } while(0)
 | 
						|
 | 
						|
#define MULTI_HEAP_LOCK_STATIC_INITIALIZER     portMUX_INITIALIZER_UNLOCKED
 | 
						|
 | 
						|
/* Not safe to use std i/o while in a portmux critical section,
 | 
						|
   can deadlock, so we use the ROM equivalent functions. */
 | 
						|
 | 
						|
#define MULTI_HEAP_PRINTF esp_rom_printf
 | 
						|
#define MULTI_HEAP_STDERR_PRINTF(MSG, ...) esp_rom_printf(MSG, __VA_ARGS__)
 | 
						|
 | 
						|
inline static void multi_heap_assert(bool condition, const char *format, int line, intptr_t address)
 | 
						|
{
 | 
						|
    /* Can't use libc assert() here as it calls printf() which can cause another malloc() for a newlib lock.
 | 
						|
 | 
						|
       Also, it's useful to be able to print the memory address where corruption was detected.
 | 
						|
    */
 | 
						|
#ifndef NDEBUG
 | 
						|
    if(!condition) {
 | 
						|
#ifndef CONFIG_COMPILER_OPTIMIZATION_ASSERTIONS_SILENT
 | 
						|
        esp_rom_printf(format, line, address);
 | 
						|
#endif  // CONFIG_COMPILER_OPTIMIZATION_ASSERTIONS_SILENT
 | 
						|
        abort();
 | 
						|
    }
 | 
						|
#else // NDEBUG
 | 
						|
    (void) condition;
 | 
						|
#endif // NDEBUG
 | 
						|
}
 | 
						|
 | 
						|
#define MULTI_HEAP_ASSERT(CONDITION, ADDRESS) \
 | 
						|
    multi_heap_assert((CONDITION), "CORRUPT HEAP: multi_heap.c:%d detected at 0x%08x\n", \
 | 
						|
                      __LINE__, (intptr_t)(ADDRESS))
 | 
						|
 | 
						|
#ifdef CONFIG_HEAP_TASK_TRACKING
 | 
						|
#include <freertos/task.h>
 | 
						|
#define MULTI_HEAP_BLOCK_OWNER TaskHandle_t task;
 | 
						|
#define MULTI_HEAP_SET_BLOCK_OWNER(HEAD) (HEAD)->task = xTaskGetCurrentTaskHandle()
 | 
						|
#define MULTI_HEAP_GET_BLOCK_OWNER(HEAD) ((HEAD)->task)
 | 
						|
#else
 | 
						|
#define MULTI_HEAP_BLOCK_OWNER
 | 
						|
#define MULTI_HEAP_SET_BLOCK_OWNER(HEAD)
 | 
						|
#define MULTI_HEAP_GET_BLOCK_OWNER(HEAD) (NULL)
 | 
						|
#endif
 | 
						|
 | 
						|
#else // MULTI_HEAP_FREERTOS
 | 
						|
 | 
						|
#include <assert.h>
 | 
						|
 | 
						|
#define MULTI_HEAP_PRINTF printf
 | 
						|
#define MULTI_HEAP_STDERR_PRINTF(MSG, ...) fprintf(stderr, MSG, __VA_ARGS__)
 | 
						|
#define MULTI_HEAP_LOCK(PLOCK)  (void) (PLOCK)
 | 
						|
#define MULTI_HEAP_UNLOCK(PLOCK)  (void) (PLOCK)
 | 
						|
#define MULTI_HEAP_LOCK_INIT(PLOCK)  (void) (PLOCK)
 | 
						|
#define MULTI_HEAP_LOCK_STATIC_INITIALIZER  0
 | 
						|
 | 
						|
#define MULTI_HEAP_ASSERT(CONDITION, ADDRESS) assert((CONDITION) && "Heap corrupt")
 | 
						|
 | 
						|
#define MULTI_HEAP_BLOCK_OWNER
 | 
						|
#define MULTI_HEAP_SET_BLOCK_OWNER(HEAD)
 | 
						|
#define MULTI_HEAP_GET_BLOCK_OWNER(HEAD) (NULL)
 | 
						|
 | 
						|
#endif // MULTI_HEAP_FREERTOS
 |