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 
alloc_simple(size_t bytes,int align)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 
malloc_simple(size_t bytes)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 
memalign_simple(size_t align,size_t bytes)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)
calloc(size_t nmemb,size_t elem_size)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)
free_simple(void * ptr)82 void free_simple(void *ptr)
83 {
84 	VALGRIND_FREELIKE_BLOCK(ptr, 0);
85 }
86 #endif
87 #endif
88 
malloc_simple_info(void)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 }
94