/* * libwebsockets - small server side websockets and web server implementation * * Copyright (C) 2010 - 2022 Andy Green * * Permission is hereby granted, free of charge, to any person obtaining a copy * of this software and associated documentation files (the "Software"), to * deal in the Software without restriction, including without limitation the * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or * sell copies of the Software, and to permit persons to whom the Software is * furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included in * all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS * IN THE SOFTWARE. */ #include "private-lib-core.h" #if defined(LWS_HAVE_MALLOC_USABLE_SIZE) #include /* the heap is processwide */ static size_t allocated; #endif #if defined(LWS_WITH_ALLOC_METADATA_LWS) static lws_dll2_owner_t active; #endif #if defined(LWS_PLAT_OPTEE) #define TEE_USER_MEM_HINT_NO_FILL_ZERO 0x80000000 #if defined (LWS_WITH_NETWORK) /* normal TA apis */ void *__attribute__((weak)) TEE_Malloc(uint32_t size, uint32_t hint) { return NULL; } void *__attribute__((weak)) TEE_Realloc(void *buffer, uint32_t newSize) { return NULL; } void __attribute__((weak)) TEE_Free(void *buffer) { } #else /* in-OP-TEE core apis */ void * TEE_Malloc(uint32_t size, uint32_t hint) { return malloc(size); } void * TEE_Realloc(void *buffer, uint32_t newSize) { return realloc(buffer, newSize); } void TEE_Free(void *buffer) { free(buffer); } #endif void *lws_realloc(void *ptr, size_t size, const char *reason) { return TEE_Realloc(ptr, size); } void *lws_malloc(size_t size, const char *reason) { return TEE_Malloc(size, TEE_USER_MEM_HINT_NO_FILL_ZERO); } void lws_free(void *p) { TEE_Free(p); } void *lws_zalloc(size_t size, const char *reason) { void *ptr = TEE_Malloc(size, TEE_USER_MEM_HINT_NO_FILL_ZERO); if (ptr) memset(ptr, 0, size); return ptr; } void lws_set_allocator(void *(*cb)(void *ptr, size_t size, const char *reason)) { (void)cb; } #else static void * _realloc(void *ptr, size_t size, const char *reason) { #if defined(LWS_WITH_ALLOC_METADATA_LWS) uint8_t comp[16 * LWS_ARRAY_SIZE(((lws_backtrace_info_t *)NULL)->st)]; size_t complen; size_t adj = 0; #endif void *v; if (size) { #if defined(LWS_WITH_ALLOC_METADATA_LWS) lws_alloc_metadata_gen(size, comp, sizeof(comp), &adj, &complen); size += adj; #endif #if defined(LWS_PLAT_FREERTOS) lwsl_debug("%s: size %lu: %s (free heap %d)\n", __func__, #if defined(LWS_AMAZON_RTOS) (unsigned long)size, reason, (unsigned int)xPortGetFreeHeapSize() - (int)size); #else (unsigned long)size, reason, (unsigned int)esp_get_free_heap_size() - (int)size); #endif #else lwsl_debug("%s: size %lu: %s\n", __func__, (unsigned long)size, reason); #endif #if defined(LWS_HAVE_MALLOC_USABLE_SIZE) if (ptr) allocated -= malloc_usable_size(ptr); #endif #if defined(LWS_WITH_ALLOC_METADATA_LWS) size += adj; #endif #if defined(LWS_PLAT_OPTEE) v = (void *)TEE_Realloc(ptr, size); #else v = (void *)realloc(ptr, size); #endif if (!v) return v; #if defined(LWS_HAVE_MALLOC_USABLE_SIZE) allocated += malloc_usable_size(v); #endif #if defined(LWS_WITH_ALLOC_METADATA_LWS) _lws_alloc_metadata_adjust(&active, &v, adj, comp, (unsigned int)complen); #endif return v; } /* * We are freeing it then... */ if (ptr) { #if defined(LWS_WITH_ALLOC_METADATA_LWS) _lws_alloc_metadata_trim(&ptr, NULL, NULL); #endif #if defined(LWS_HAVE_MALLOC_USABLE_SIZE) allocated -= malloc_usable_size(ptr); #endif free(ptr); #if defined(LWS_PLAT_FREERTOS) lwsl_debug("%s: free heap %d\n", __func__, #if defined(LWS_AMAZON_RTOS) (unsigned int)xPortGetFreeHeapSize() - (int)size); #else (unsigned int)esp_get_free_heap_size() - (int)size); #endif #endif } return NULL; } #if defined(LWS_WITH_ALLOC_METADATA_LWS) void _lws_alloc_metadata_dump_lws(lws_dll2_foreach_cb_t cb, void *arg) { lwsl_err("%s\n", __func__); _lws_alloc_metadata_dump(&active, cb, arg); } #endif void *(*_lws_realloc)(void *ptr, size_t size, const char *reason) = _realloc; void *lws_realloc(void *ptr, size_t size, const char *reason) { return _lws_realloc(ptr, size, reason); } void *lws_zalloc(size_t size, const char *reason) { void *ptr = _lws_realloc(NULL, size, reason); if (ptr) memset(ptr, 0, size); return ptr; } void lws_set_allocator(void *(*cb)(void *ptr, size_t size, const char *reason)) { _lws_realloc = cb; } size_t lws_get_allocated_heap(void) { #if defined(LWS_HAVE_MALLOC_USABLE_SIZE) return allocated; #else return 0; #endif } #endif