common/malloc_simple.c

   1 // SPDX-License-Identifier: GPL-2.0+
   2 /*
   3  * Simple malloc implementation
   4  *
   5  * Copyright (c) 2014 Google, Inc
   6  */
   7
   8 #define LOG_CATEGORY LOGC_ALLOC
   9
  10 #include <common.h>
  11 #include <log.h>
  12 #include <malloc.h>
  13 #include <mapmem.h>
  14 #include <asm/global_data.h>
  15 #include <asm/io.h>
  16 #include <valgrind/valgrind.h>
  17
  18 DECLARE_GLOBAL_DATA_PTR;
  19
  20 static void *alloc_simple(size_t bytes, int align)
  21 {
  22         ulong addr, new_ptr;
  23         void *ptr;
  24
  25         addr = ALIGN(gd->malloc_base + gd->malloc_ptr, align);
  26         new_ptr = addr + bytes - gd->malloc_base;
  27         log_debug("size=%lx, ptr=%lx, limit=%lx: ", (ulong)bytes, new_ptr,
  28                   gd->malloc_limit);
  29         if (new_ptr > gd->malloc_limit) {
  30                 log_err("alloc space exhausted\n");
  31                 return NULL;
  32         }
  33
  34         ptr = map_sysmem(addr, bytes);
  35         gd->malloc_ptr = ALIGN(new_ptr, sizeof(new_ptr));
  36
  37         return ptr;
  38 }
  39
  40 void *malloc_simple(size_t bytes)
  41 {
  42         void *ptr;
  43
  44         ptr = alloc_simple(bytes, 1);
  45         if (!ptr)
  46                 return ptr;
  47
  48         log_debug("%lx\n", (ulong)ptr);
  49         VALGRIND_MALLOCLIKE_BLOCK(ptr, bytes, 0, false);
  50
  51         return ptr;
  52 }
  53
  54 void *memalign_simple(size_t align, size_t bytes)
  55 {
  56         void *ptr;
  57
  58         ptr = alloc_simple(bytes, align);
  59         if (!ptr)
  60                 return ptr;
  61         log_debug("aligned to %lx\n", (ulong)ptr);
  62         VALGRIND_MALLOCLIKE_BLOCK(ptr, bytes, 0, false);
  63
  64         return ptr;
  65 }
  66
  67 #if CONFIG_IS_ENABLED(SYS_MALLOC_SIMPLE)
  68 void *calloc(size_t nmemb, size_t elem_size)
  69 {
  70         size_t size = nmemb * elem_size;
  71         void *ptr;
  72
  73         ptr = malloc(size);
  74         if (!ptr)
  75                 return ptr;
  76         memset(ptr, '\0', size);
  77
  78         return ptr;
  79 }
  80
  81 #if IS_ENABLED(CONFIG_VALGRIND)
  82 void free_simple(void *ptr)
  83 {
  84         VALGRIND_FREELIKE_BLOCK(ptr, 0);
  85 }
  86 #endif
  87 #endif
  88
  89 void malloc_simple_info(void)
  90 {
  91         log_info("malloc_simple: %lx bytes used, %lx remain\n", gd->malloc_ptr,
  92                  CONFIG_VAL(SYS_MALLOC_F_LEN) - gd->malloc_ptr);
  93 }