xref: /rk3399_rockchip-uboot/lib/sysmem.c (revision c5449fd30c8bd2024331ddf912fa467e4487c20b)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Copyright (c) 2019 Fuzhou Rockchip Electronics Co., Ltd
4  */
5 
6 #include <common.h>
7 #include <sysmem.h>
8 #include <lmb.h>
9 #include <malloc.h>
10 #include <asm/io.h>
11 
12 DECLARE_GLOBAL_DATA_PTR;
13 
14 #define SYSMEM_MAGIC		0x4D454D53	/* "SMEM" */
15 #define SYSMEM_ALLOC_ANYWHERE	0
16 #define SYSMEM_ALLOC_NO_ALIGN	1
17 
18 #define SYSMEM_I(fmt, args...)	printf("Sysmem: "fmt, ##args)
19 #define SYSMEM_W(fmt, args...)	printf("Sysmem Warn: "fmt, ##args)
20 #define SYSMEM_E(fmt, args...)	printf("Sysmem Error: "fmt, ##args)
21 #define SYSMEM_D(fmt, args...)	 debug("Sysmem Debug: "fmt, ##args)
22 
23 struct memcheck {
24 	uint32_t magic;
25 };
26 
27 /* Global for platform, must in data section */
28 struct sysmem plat_sysmem __section(".data") = {
29 	.has_initf = false,
30 	.has_initr = false,
31 };
32 
33 bool sysmem_has_init(void)
34 {
35 	return gd->flags & GD_FLG_RELOC ?
36 	       plat_sysmem.has_initr : plat_sysmem.has_initf;
37 }
38 
39 void sysmem_dump(void)
40 {
41 	struct sysmem *sysmem = &plat_sysmem;
42 	struct lmb *lmb = &sysmem->lmb;
43 	struct memblock *mem;
44 	struct memcheck *check;
45 	struct list_head *node;
46 	ulong memory_size = 0;
47 	ulong reserved_size = 0;
48 	ulong allocated_size = 0;
49 	bool overflow = false;
50 	ulong i;
51 
52 	if (!sysmem_has_init())
53 		return;
54 
55 	printf("\nsysmem_dump_all:\n");
56 
57 	/* Memory pool */
58 	printf("    --------------------------------------------------------------------\n");
59 	for (i = 0; i < lmb->memory.cnt; i++) {
60 		memory_size += lmb->memory.region[i].size;
61 		printf("    memory.rgn[%ld].addr     = 0x%08lx - 0x%08lx (size: 0x%08lx)\n", i,
62 		       (ulong)lmb->memory.region[i].base,
63 		       (ulong)lmb->memory.region[i].base +
64 		       (ulong)lmb->memory.region[i].size,
65 		       (ulong)lmb->memory.region[i].size);
66 	}
67 	printf("\n    memory.total	   = 0x%08lx (%ld MiB. %ld KiB)\n",
68 	       (ulong)memory_size,
69 	       SIZE_MB((ulong)memory_size),
70 	       SIZE_KB((ulong)memory_size));
71 
72 	/* Allocated */
73 	i = 0;
74 	printf("    --------------------------------------------------------------------\n");
75 	list_for_each(node, &sysmem->allocated_head) {
76 		mem = list_entry(node, struct memblock, node);
77 		allocated_size += mem->size;
78 		if (mem->attr.flags & M_ATTR_OFC) {
79 			check = (struct memcheck *)
80 				(mem->base + mem->size - sizeof(*check));
81 			overflow = (check->magic != SYSMEM_MAGIC);
82 		} else if (mem->attr.flags & M_ATTR_HOFC) {
83 			check = (struct memcheck *)
84 				(mem->base - sizeof(*check));
85 			overflow = (check->magic != SYSMEM_MAGIC);
86 		} else {
87 			overflow = false;
88 		}
89 
90 		printf("    allocated.rgn[%ld].name  = \"%s\" %s\n",
91 		       i, mem->attr.name, overflow ? "	   <Overflow!>" : "");
92 		printf("		    .addr  = 0x%08lx - 0x%08lx (size: 0x%08lx)\n",
93 		       (ulong)mem->base, (ulong)(mem->base + mem->size),
94 		       (ulong)mem->size);
95 		i++;
96 	}
97 
98 	printf("\n    malloc_r: %d MiB, malloc_f: %d KiB\n",
99 	       SIZE_MB(CONFIG_SYS_MALLOC_LEN), SIZE_KB(CONFIG_SYS_MALLOC_F_LEN));
100 	printf("\n    allocated.total	   = 0x%08lx (%ld MiB. %ld KiB)\n",
101 	       (ulong)allocated_size,
102 	       SIZE_MB((ulong)allocated_size),
103 	       SIZE_KB((ulong)allocated_size));
104 
105 	/* LMB core reserved */
106 	printf("    --------------------------------------------------------------------\n");
107 	reserved_size = 0;
108 	for (i = 0; i < lmb->reserved.cnt; i++) {
109 		reserved_size += lmb->reserved.region[i].size;
110 		printf("    LMB.reserved[%ld].addr   = 0x%08lx - 0x%08lx (size: 0x%08lx)\n", i,
111 		       (ulong)lmb->reserved.region[i].base,
112 		       (ulong)lmb->reserved.region[i].base +
113 		       (ulong)lmb->reserved.region[i].size,
114 		       (ulong)lmb->reserved.region[i].size);
115 	}
116 
117 	printf("\n    reserved.core.total	   = 0x%08lx (%ld MiB. %ld KiB)\n",
118 	       (ulong)reserved_size,
119 	       SIZE_MB((ulong)reserved_size),
120 	       SIZE_KB((ulong)reserved_size));
121 	printf("    --------------------------------------------------------------------\n\n");
122 }
123 
124 static inline int sysmem_is_overlap(phys_addr_t base1, phys_size_t size1,
125 				    phys_addr_t base2, phys_size_t size2)
126 {
127 	return ((base1 < (base2 + size2)) && (base2 < (base1 + size1)));
128 }
129 
130 static int sysmem_add(phys_addr_t base, phys_size_t size)
131 {
132 	struct sysmem *sysmem = &plat_sysmem;
133 	int ret;
134 
135 	if (!size)
136 		return -EINVAL;
137 
138 	ret = lmb_add(&sysmem->lmb, base, size);
139 	if (ret < 0)
140 		SYSMEM_E("Failed to add sysmem at 0x%08lx for 0x%08lx size\n",
141 			 (ulong)base, (ulong)size);
142 
143 	return (ret >= 0) ? 0 : ret;
144 }
145 
146 static const char *sysmem_alias2name(const char *name, int *id)
147 {
148 	const char *alias;
149 	int n, i, j;
150 	int match = 0;
151 
152 	for (i = 0; i < MEMBLK_ID_MAX; i++) {
153 		/* Pirmary name */
154 		if (mem_attr[i].name && !strcasecmp(mem_attr[i].name, name)) {
155 			match = 1;
156 			goto finish;
157 		}
158 
159 		/* Alias name */
160 		alias = mem_attr[i].alias[0];
161 		if (!alias)
162 			continue;
163 
164 		n = ARRAY_SIZE(mem_attr[i].alias);
165 		for (j = 0; j < n; j++, alias++) {
166 			if (alias && !strcasecmp(alias, name)) {
167 				match = 1;
168 				goto finish;
169 			}
170 		}
171 	}
172 
173 finish:
174 	if (match) {
175 		*id = i;
176 		return mem_attr[i].name;
177 	}
178 
179 	return name;
180 }
181 
182 static void *sysmem_alloc_align_base(enum memblk_id id,
183 				     const char *mem_name,
184 				     phys_addr_t base,
185 				     phys_size_t size,
186 				     ulong align)
187 {
188 	struct sysmem *sysmem = &plat_sysmem;
189 	struct memblk_attr attr;
190 	struct memblock *mem;
191 	struct memcheck *check;
192 	struct list_head *node;
193 	const char *name;
194 	phys_addr_t paddr;
195 	phys_addr_t alloc_base;
196 	phys_size_t alloc_size;
197 	bool req_overlap = false;
198 
199 	if (!sysmem_has_init())
200 		return NULL;
201 
202 	if (id == MEMBLK_ID_BY_NAME || id == MEMBLK_ID_FDT_RESV) {
203 		if (!mem_name) {
204 			SYSMEM_E("NULL name for alloc sysmem\n");
205 			return NULL;
206 		} else if (id == MEMBLK_ID_FDT_RESV) {
207 			req_overlap = true;
208 			if (base >= gd->ram_top)
209 				return (void *)base;
210 		}
211 		name = sysmem_alias2name(mem_name, (int *)&id);
212 		attr = mem_attr[id];
213 		if (!attr.name)
214 			attr.name = strdup(name);
215 	} else if (id > MEMBLK_ID_UNK && id < MEMBLK_ID_MAX) {
216 		attr = mem_attr[id];
217 		name = attr.name;
218 	} else {
219 		SYSMEM_E("Unsupport memblk id %d for alloc sysmem\n", id);
220 		return NULL;
221 	}
222 
223 	if (!size) {
224 		SYSMEM_E("\"%s\" size is 0 for alloc sysmem\n", name);
225 		return NULL;
226 	}
227 
228 	if (!IS_ALIGNED(base, 4)) {
229 		SYSMEM_E("\"%s\" base=0x%08lx is not 4-byte aligned\n",
230 			 name, (ulong)base);
231 		return NULL;
232 	}
233 
234 	/* Must be 4-byte aligned */
235 	size = ALIGN(size, 4);
236 
237 	SYSMEM_D("Enter alloc: \"%s\" 0x%08lx - 0x%08lx\n",
238 		 name, (ulong)base, (ulong)(base + size));
239 
240 	/* Already allocated ? */
241 	list_for_each(node, &sysmem->allocated_head) {
242 		mem = list_entry(node, struct memblock, node);
243 		SYSMEM_D("Has allcated: %s, 0x%08lx - 0x%08lx\n",
244 			 mem->attr.name, (ulong)mem->base,
245 			 (ulong)(mem->base + mem->size));
246 		if (!strcmp(mem->attr.name, name)) {
247 			SYSMEM_E("Failed to double alloc for existence \"%s\"\n", name);
248 			return NULL;
249 		} else if (sysmem_is_overlap(mem->base, mem->size, base, size)) {
250 			if (req_overlap && mem->attr.flags & M_ATTR_OVERLAP) {
251 				if (lmb_reserve(&sysmem->lmb, base, size))
252 					SYSMEM_E("Failed to overlap alloc \"%s\" "
253 						 "at 0x%08lx - 0x%08lx\n",
254 						 name, (ulong)base,
255 						 (ulong)(base + size));
256 				return (void *)base;
257 			}
258 
259 			SYSMEM_E("\"%s\" (0x%08lx - 0x%08lx) alloc is "
260 				 "overlap with existence \"%s\" (0x%08lx - "
261 				 "0x%08lx)\n",
262 				 name, (ulong)base, (ulong)(base + size),
263 				 mem->attr.name, (ulong)mem->base,
264 				 (ulong)(mem->base + mem->size));
265 			return NULL;
266 		}
267 	}
268 
269 	/* Add overflow check magic ? */
270 	if (attr.flags & M_ATTR_OFC)
271 		alloc_size = size + sizeof(*check);
272 	else
273 		alloc_size = size;
274 
275 	/* Alloc anywhere ? */
276 	if (base == SYSMEM_ALLOC_ANYWHERE)
277 		alloc_base = base;
278 	else
279 		alloc_base = base + alloc_size;	/* LMB is align down alloc mechanism */
280 
281 	paddr = lmb_alloc_base(&sysmem->lmb, alloc_size, align, alloc_base);
282 	if (paddr) {
283 		if ((paddr == base) || (base == SYSMEM_ALLOC_ANYWHERE)) {
284 			mem = malloc(sizeof(*mem));
285 			if (!mem) {
286 				SYSMEM_E("No memory for \"%s\" alloc sysmem\n", name);
287 				return NULL;
288 			}
289 
290 			mem->base = paddr;
291 			mem->size = alloc_size;
292 			mem->attr = attr;
293 			sysmem->allocated_cnt++;
294 			list_add_tail(&mem->node, &sysmem->allocated_head);
295 
296 			if (mem->attr.flags & M_ATTR_OFC) {
297 				check = (struct memcheck *)(paddr + size);
298 				check->magic = SYSMEM_MAGIC;
299 			} else if (mem->attr.flags & M_ATTR_HOFC) {
300 				check = (struct memcheck *)(paddr - sizeof(*check));
301 				check->magic = SYSMEM_MAGIC;
302 			}
303 		} else {
304 			SYSMEM_E("Failed to alloc \"%s\" expect at 0x%08lx - 0x%08lx "
305 				 "but at 0x%08lx - x%08lx\n",
306 				 name, (ulong)base, (ulong)(base + size),
307 				 (ulong)paddr, (ulong)(paddr + size));
308 			if (lmb_free(&sysmem->lmb, paddr, alloc_size))
309 				SYSMEM_E("Failed to free \"%s\"\n", name);
310 
311 			return NULL;
312 		}
313 	} else {
314 		SYSMEM_E("Failed to alloc \"%s\" at 0x%08lx - 0x%08lx\n",
315 			 name, (ulong)base, (ulong)(base + size));
316 	}
317 
318 	SYSMEM_D("Exit alloc: \"%s\", paddr=0x%08lx, size=0x%08lx, align=0x%x, anywhere=%d\n",
319 		 name, (ulong)paddr, (ulong)size, (u32)align, !base);
320 
321 	return (void *)paddr;
322 }
323 
324 void *sysmem_alloc_base(enum memblk_id id, phys_addr_t base, phys_size_t size)
325 {
326 	void *paddr;
327 
328 	paddr = sysmem_alloc_align_base(id,
329 					NULL,
330 					base,
331 					size,
332 					SYSMEM_ALLOC_NO_ALIGN);
333 	if (!paddr)
334 		sysmem_dump();
335 
336 	return paddr;
337 }
338 
339 void *sysmem_alloc_base_by_name(const char *name,
340 				phys_addr_t base, phys_size_t size)
341 {
342 	void *paddr;
343 
344 	paddr = sysmem_alloc_align_base(MEMBLK_ID_BY_NAME,
345 					name,
346 					base,
347 					size,
348 					SYSMEM_ALLOC_NO_ALIGN);
349 	if (!paddr)
350 		sysmem_dump();
351 
352 	return paddr;
353 }
354 
355 void *sysmem_fdt_reserve_alloc_base(const char *name,
356 				    phys_addr_t base, phys_size_t size)
357 {
358 	void *paddr;
359 
360 	paddr = sysmem_alloc_align_base(MEMBLK_ID_FDT_RESV,
361 					name,
362 					base,
363 					size,
364 					SYSMEM_ALLOC_NO_ALIGN);
365 	if (!paddr)
366 		sysmem_dump();
367 
368 	return paddr;
369 }
370 
371 int sysmem_free(phys_addr_t base)
372 {
373 	struct sysmem *sysmem = &plat_sysmem;
374 	struct memblock *mem;
375 	struct list_head *node;
376 	int ret, found = 0;
377 
378 	if (!sysmem_has_init())
379 		return -ENOSYS;
380 
381 	/* Find existence */
382 	list_for_each(node, &sysmem->allocated_head) {
383 		mem = list_entry(node, struct memblock, node);
384 		if (mem->base == base) {
385 			found = 1;
386 			break;
387 		}
388 	}
389 
390 	if (!found) {
391 		SYSMEM_E("Failed to free no allocated sysmem at 0x%08lx\n",
392 			 (ulong)base);
393 		return -EINVAL;
394 	}
395 
396 	ret = lmb_free(&sysmem->lmb, mem->base, mem->size);
397 	if (ret >= 0) {
398 		SYSMEM_D("Free: \"%s\" 0x%08lx - 0x%08lx\n",
399 			 mem->attr.name, (ulong)mem->base,
400 			 (ulong)(mem->base + mem->size));
401 		sysmem->allocated_cnt--;
402 		list_del(&mem->node);
403 		free(mem);
404 	} else {
405 		SYSMEM_E("Failed to free \"%s\" at 0x%08lx\n",
406 			 mem->attr.name, (ulong)base);
407 	}
408 
409 	return (ret >= 0) ? 0 : ret;
410 }
411 
412 int sysmem_initr(void)
413 {
414 	return sysmem_init();
415 }
416 
417 int sysmem_init(void)
418 {
419 	struct sysmem *sysmem = &plat_sysmem;
420 	phys_addr_t mem_start;
421 	phys_size_t mem_size;
422 	int ret;
423 
424 	lmb_init(&sysmem->lmb);
425 	INIT_LIST_HEAD(&sysmem->allocated_head);
426 	sysmem->allocated_cnt = 0;
427 	if (gd->flags & GD_FLG_RELOC) {
428 		sysmem->has_initr = true;
429 	} else {
430 		SYSMEM_I("init\n");
431 		sysmem->has_initf = true;
432 	}
433 
434 	/* Add all available system memory */
435 #ifdef CONFIG_NR_DRAM_BANKS
436 	int i;
437 
438 	for (i = 0; i < CONFIG_NR_DRAM_BANKS; i++) {
439 		if (!gd->bd->bi_dram[i].size)
440 			continue;
441 
442 		ret = sysmem_add(gd->bd->bi_dram[i].start,
443 				 gd->bd->bi_dram[i].size);
444 		if (ret) {
445 			SYSMEM_E("Failed to add sysmem from bi_dram[%d]\n", i);
446 			goto fail;
447 		}
448 	}
449 #else
450 	mem_start = env_get_bootm_low();
451 	mem_size = env_get_bootm_size();
452 	ret = sysmem_add(mem_start, mem_size);
453 	if (ret) {
454 		SYSMEM_E("Failed to add sysmem from bootm_low/size\n");
455 		goto fail;
456 	}
457 #endif
458 	/* Reserved for board */
459 	ret = board_sysmem_reserve(sysmem);
460 	if (ret) {
461 		SYSMEM_E("Failed to reserve sysmem for board\n");
462 		goto fail;
463 	}
464 
465 	/* Reserved for U-boot framework: 'reserve_xxx()' */
466 	mem_start = gd->start_addr_sp;
467 	mem_size = gd->ram_top - mem_start;
468 	if (!sysmem_alloc_base(MEMBLK_ID_UBOOT, mem_start, mem_size)) {
469 		SYSMEM_E("Failed to reserve sysmem for U-Boot framework\n");
470 		ret = -ENOMEM;
471 		goto fail;
472 	}
473 
474 	/* Reserved for U-Boot stack */
475 	mem_start = gd->start_addr_sp - CONFIG_SYS_STACK_SIZE;
476 	mem_size = CONFIG_SYS_STACK_SIZE;
477 	if (!sysmem_alloc_base(MEMBLK_ID_STACK, mem_start, mem_size)) {
478 		SYSMEM_E("Failed to reserve sysmem for stack\n");
479 		ret = -ENOMEM;
480 		goto fail;
481 	}
482 
483 	return 0;
484 
485 fail:
486 	if (ret && !(gd->flags & GD_FLG_RELOC)) {
487 		sysmem_dump();
488 		SYSMEM_W("Maybe malloc size %d MiB is too large?\n\n",
489 			 SIZE_MB(CONFIG_SYS_MALLOC_LEN));
490 	}
491 
492 	return ret;
493 }
494 
495 __weak int board_sysmem_reserve(struct sysmem *sysmem)
496 {
497 	/* please define platform specific board_sysmem_reserve() */
498 	return 0;
499 }
500 
501 static int do_dump_sysmem(cmd_tbl_t *cmdtp, int flag,
502 			  int argc, char *const argv[])
503 {
504 	sysmem_dump();
505 	return 0;
506 }
507 
508 U_BOOT_CMD(
509 	dump_sysmem, 1, 1, do_dump_sysmem,
510 	"Dump sysmem layout",
511 	""
512 );
513