/* * Copyright (c) 2018 Intel Corporation * * SPDX-License-Identifier: Apache-2.0 */ #include #include #include #include #include #include #include #include #include #include #define LOG_LEVEL CONFIG_KERNEL_LOG_LEVEL #include LOG_MODULE_DECLARE(os, CONFIG_KERNEL_LOG_LEVEL); #ifdef CONFIG_MINIMAL_LIBC_MALLOC #if (CONFIG_MINIMAL_LIBC_MALLOC_ARENA_SIZE > 0) #ifdef CONFIG_USERSPACE K_APPMEM_PARTITION_DEFINE(z_malloc_partition); #define POOL_SECTION K_APP_DMEM_SECTION(z_malloc_partition) #else #define POOL_SECTION .bss #endif /* CONFIG_USERSPACE */ #define HEAP_BYTES CONFIG_MINIMAL_LIBC_MALLOC_ARENA_SIZE Z_GENERIC_SECTION(POOL_SECTION) static struct sys_heap z_malloc_heap; Z_GENERIC_SECTION(POOL_SECTION) struct sys_mutex z_malloc_heap_mutex; Z_GENERIC_SECTION(POOL_SECTION) static char z_malloc_heap_mem[HEAP_BYTES]; void *malloc(size_t size) { int lock_ret; lock_ret = sys_mutex_lock(&z_malloc_heap_mutex, K_FOREVER); __ASSERT_NO_MSG(lock_ret == 0); void *ret = sys_heap_aligned_alloc(&z_malloc_heap, __alignof__(z_max_align_t), size); if (ret == NULL && size != 0) { errno = ENOMEM; } (void) sys_mutex_unlock(&z_malloc_heap_mutex); return ret; } static int malloc_prepare(const struct device *unused) { ARG_UNUSED(unused); sys_heap_init(&z_malloc_heap, z_malloc_heap_mem, HEAP_BYTES); sys_mutex_init(&z_malloc_heap_mutex); return 0; } void *realloc(void *ptr, size_t requested_size) { int lock_ret; lock_ret = sys_mutex_lock(&z_malloc_heap_mutex, K_FOREVER); __ASSERT_NO_MSG(lock_ret == 0); void *ret = sys_heap_aligned_realloc(&z_malloc_heap, ptr, __alignof__(z_max_align_t), requested_size); if (ret == NULL && requested_size != 0) { errno = ENOMEM; } (void) sys_mutex_unlock(&z_malloc_heap_mutex); return ret; } void free(void *ptr) { int lock_ret; lock_ret = sys_mutex_lock(&z_malloc_heap_mutex, K_FOREVER); __ASSERT_NO_MSG(lock_ret == 0); sys_heap_free(&z_malloc_heap, ptr); (void) sys_mutex_unlock(&z_malloc_heap_mutex); } SYS_INIT(malloc_prepare, APPLICATION, CONFIG_KERNEL_INIT_PRIORITY_DEFAULT); #else /* No malloc arena */ void *malloc(size_t size) { ARG_UNUSED(size); LOG_ERR("CONFIG_MINIMAL_LIBC_MALLOC_ARENA_SIZE is 0"); errno = ENOMEM; return NULL; } void free(void *ptr) { ARG_UNUSED(ptr); } void *realloc(void *ptr, size_t size) { ARG_UNUSED(ptr); return malloc(size); } #endif #endif /* CONFIG_MINIMAL_LIBC_MALLOC */ #ifdef CONFIG_MINIMAL_LIBC_CALLOC void *calloc(size_t nmemb, size_t size) { void *ret; if (size_mul_overflow(nmemb, size, &size)) { errno = ENOMEM; return NULL; } ret = malloc(size); if (ret != NULL) { (void)memset(ret, 0, size); } return ret; } #endif /* CONFIG_MINIMAL_LIBC_CALLOC */ #ifdef CONFIG_MINIMAL_LIBC_REALLOCARRAY void *reallocarray(void *ptr, size_t nmemb, size_t size) { #if (CONFIG_MINIMAL_LIBC_MALLOC_ARENA_SIZE > 0) if (size_mul_overflow(nmemb, size, &size)) { errno = ENOMEM; return NULL; } return realloc(ptr, size); #else return NULL; #endif } #endif /* CONFIG_MINIMAL_LIBC_REALLOCARRAY */