xref: /rk3399_rockchip-uboot/lib/sysmem.c (revision 94fbbf0f6db608f9349f507ed0734ea9befc8ec3)
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 <bidram.h>
8 #include <sysmem.h>
9 #include <lmb.h>
10 #include <malloc.h>
11 #include <asm/io.h>
12 
13 DECLARE_GLOBAL_DATA_PTR;
14 
15 #define SYSMEM_MAGIC		0x4D454D53	/* "SMEM" */
16 
17 #define LMB_ALLOC_ANYWHERE	0		/* sync with lmb.c */
18 #define SYSMEM_ALLOC_NO_ALIGN	1
19 #define SYSMEM_ALLOC_ANYWHERE	2
20 
21 #define SYSMEM_I(fmt, args...)	printf("Sysmem: "fmt, ##args)
22 #define SYSMEM_W(fmt, args...)	printf("Sysmem Warn: "fmt, ##args)
23 #define SYSMEM_E(fmt, args...)	printf("Sysmem Error: "fmt, ##args)
24 #define SYSMEM_D(fmt, args...)	 debug("Sysmem Debug: "fmt, ##args)
25 
26 struct memcheck {
27 	uint32_t magic;
28 };
29 
30 /* Global for platform, must in data section */
31 struct sysmem plat_sysmem __section(".data") = {
32 	.has_initf = false,
33 	.has_initr = false,
34 };
35 
36 bool sysmem_has_init(void)
37 {
38 	return gd->flags & GD_FLG_RELOC ?
39 	       plat_sysmem.has_initr : plat_sysmem.has_initf;
40 }
41 
42 static inline int sysmem_is_overlap(phys_addr_t base1, phys_size_t size1,
43 				    phys_addr_t base2, phys_size_t size2)
44 {
45 	return ((base1 < (base2 + size2)) && (base2 < (base1 + size1)));
46 }
47 
48 static inline int sysmem_is_sub_region(struct memblock *sub,
49 				       struct memblock *main)
50 {
51 	if (!sub || !main)
52 		return false;
53 
54 	return ((sub->base >= main->base) &&
55 		(sub->base + sub->size <= main->base + main->size));
56 }
57 
58 void sysmem_dump(void)
59 {
60 	struct sysmem *sysmem = &plat_sysmem;
61 	struct lmb *lmb = &sysmem->lmb;
62 	struct memblock *mem;
63 	struct memcheck *check;
64 	struct list_head *node;
65 	ulong memory_size = 0;
66 	ulong reserved_size = 0;
67 	ulong allocated_size = 0;
68 	bool overflow = false;
69 	ulong i;
70 
71 	if (!sysmem_has_init())
72 		return;
73 
74 	printf("\nsysmem_dump_all:\n");
75 
76 	/* Memory pool */
77 	printf("    --------------------------------------------------------------------\n");
78 	for (i = 0; i < lmb->memory.cnt; i++) {
79 		memory_size += lmb->memory.region[i].size;
80 		printf("    memory.rgn[%ld].addr     = 0x%08lx - 0x%08lx (size: 0x%08lx)\n", i,
81 		       (ulong)lmb->memory.region[i].base,
82 		       (ulong)lmb->memory.region[i].base +
83 		       (ulong)lmb->memory.region[i].size,
84 		       (ulong)lmb->memory.region[i].size);
85 	}
86 	printf("\n    memory.total	   = 0x%08lx (%ld MiB. %ld KiB)\n",
87 	       (ulong)memory_size,
88 	       SIZE_MB((ulong)memory_size),
89 	       SIZE_KB((ulong)memory_size));
90 
91 	/* Allocated */
92 	i = 0;
93 	printf("    --------------------------------------------------------------------\n");
94 	list_for_each(node, &sysmem->allocated_head) {
95 		mem = list_entry(node, struct memblock, node);
96 		allocated_size += mem->size;
97 		if (mem->attr.flags & M_ATTR_OFC) {
98 			check = (struct memcheck *)
99 				(mem->base + mem->size - sizeof(*check));
100 			overflow = (check->magic != SYSMEM_MAGIC);
101 		} else if (mem->attr.flags & M_ATTR_HOFC) {
102 			check = (struct memcheck *)
103 				(mem->base - sizeof(*check));
104 			overflow = (check->magic != SYSMEM_MAGIC);
105 		} else {
106 			overflow = false;
107 		}
108 
109 		printf("    allocated.rgn[%ld].name  = \"%s\" %s %s\n",
110 		       i, mem->attr.name, overflow ? "	   <Overflow!>" : "",
111 		       mem->orig_base != mem->base ? "<*>" : "");
112 		printf("		    .addr  = 0x%08lx - 0x%08lx (size: 0x%08lx)\n",
113 		       (ulong)mem->orig_base,
114 		       (ulong)(mem->orig_base + mem->size),
115 		       (ulong)mem->size);
116 		i++;
117 	}
118 
119 	/* Kernel 'reserved-memory' */
120 	i = 0;
121 	printf("\n");
122 	list_for_each(node, &sysmem->kmem_resv_head) {
123 		mem = list_entry(node, struct memblock, node);
124 		allocated_size += mem->size;
125 		printf("    kmem-resv.rgn[%ld].name  = \"%s\" %s\n",
126 		       i, mem->attr.name,
127 		       mem->orig_base != mem->base ? "<*>" : "");
128 		printf("		    .addr  = 0x%08lx - 0x%08lx (size: 0x%08lx)\n",
129 		       (ulong)mem->orig_base,
130 		       (ulong)(mem->orig_base + mem->size),
131 		       (ulong)mem->size);
132 		i++;
133 	}
134 
135 	printf("\n    framework malloc_r     = %3d MiB",
136 	       SIZE_MB(CONFIG_SYS_MALLOC_LEN));
137 	printf("\n    framework malloc_f     = %3d KiB\n",
138 	       SIZE_KB(CONFIG_SYS_MALLOC_F_LEN));
139 
140 	printf("\n    allocated.total	   = 0x%08lx (%ld MiB. %ld KiB)\n",
141 	       (ulong)allocated_size,
142 	       SIZE_MB((ulong)allocated_size),
143 	       SIZE_KB((ulong)allocated_size));
144 
145 	/* LMB core reserved */
146 	printf("    --------------------------------------------------------------------\n");
147 	reserved_size = 0;
148 	for (i = 0; i < lmb->reserved.cnt; i++) {
149 		reserved_size += lmb->reserved.region[i].size;
150 		printf("    LMB.allocated[%ld].addr  = 0x%08lx - 0x%08lx (size: 0x%08lx)\n", i,
151 		       (ulong)lmb->reserved.region[i].base,
152 		       (ulong)lmb->reserved.region[i].base +
153 		       (ulong)lmb->reserved.region[i].size,
154 		       (ulong)lmb->reserved.region[i].size);
155 	}
156 
157 	printf("\n    reserved.core.total	   = 0x%08lx (%ld MiB. %ld KiB)\n",
158 	       (ulong)reserved_size,
159 	       SIZE_MB((ulong)reserved_size),
160 	       SIZE_KB((ulong)reserved_size));
161 	printf("    --------------------------------------------------------------------\n\n");
162 }
163 
164 void sysmem_overflow_check(void)
165 {
166 	struct sysmem *sysmem = &plat_sysmem;
167 	struct list_head *node, *knode;
168 	struct memcheck *check;
169 	struct memblock *kmem;
170 	struct memblock *smem;
171 	struct memblock *rmem;
172 	int overflow = 0, overlap = 0;
173 
174 	if (!sysmem_has_init())
175 		return;
176 
177 #ifdef CONFIG_BIDRAM
178 	/*
179 	 * Check kernel 'reserved-memory' overlap with invisible regions
180 	 *
181 	 * Here, only print warning message when overlap with invisible region
182 	 */
183 	list_for_each(knode, &sysmem->kmem_resv_head) {
184 		kmem = list_entry(knode, struct memblock, node);
185 		rmem = bidram_reserved_is_overlap(kmem->base, kmem->size);
186 		if (rmem) {
187 			const char *alias;
188 			int i, dump = 1;
189 
190 			/*
191 			 * Ignore the sub region of invisible region.
192 			 * eg: ramoops of SHM.
193 			 */
194 			alias = rmem->attr.alias[0];
195 			if (alias && sysmem_is_sub_region(kmem, rmem)) {
196 				for (i = 0; i < ALIAS_COUNT_MAX; i++, alias++) {
197 					alias = rmem->attr.alias[i];
198 					if (!alias)
199 						continue;
200 					if (!strncasecmp(kmem->attr.name, alias,
201 							 strlen(alias))) {
202 						dump = 0;
203 						break;
204 					}
205 				}
206 			}
207 
208 			if (dump)
209 				SYSMEM_W("kernel 'reserved-memory' \"%s\"(0x%08lx - 0x%08lx) "
210 					 "is overlap with [invisible] \"%s\" (0x%08lx - 0x%08lx)\n",
211 					 kmem->attr.name, (ulong)kmem->base,
212 					 (ulong)(kmem->base + kmem->size),
213 					 rmem->attr.name, (ulong)rmem->base,
214 					 (ulong)(rmem->base + rmem->size));
215 		}
216 	}
217 #endif
218 
219 	list_for_each(node, &sysmem->allocated_head) {
220 		smem = list_entry(node, struct memblock, node);
221 		/*
222 		 * Check kernel 'reserved-memory' overlap with sysmem allocated regions
223 		 */
224 		list_for_each(knode, &sysmem->kmem_resv_head) {
225 			kmem = list_entry(knode, struct memblock, node);
226 			if (sysmem_is_overlap(smem->base, smem->size,
227 					      kmem->base, kmem->size)) {
228 				if (smem->attr.flags & M_ATTR_KMEM_CAN_OVERLAP)
229 					continue;
230 
231 				overlap = 1;
232 				SYSMEM_W("kernel 'reserved-memory' \"%s\"(0x%08lx - 0x%08lx) "
233 					 "is overlap with \"%s\" (0x%08lx - 0x%08lx)\n",
234 					 kmem->attr.name, (ulong)kmem->base,
235 					 (ulong)(kmem->base + kmem->size),
236 					 smem->attr.name, (ulong)smem->base,
237 					 (ulong)(smem->base + smem->size));
238 			}
239 		}
240 
241 		/*
242 		 * Check sysmem allocated regions overflow.
243 		 */
244 		if (smem->attr.flags & M_ATTR_OFC) {
245 			check = (struct memcheck *)
246 				(smem->base + smem->size - sizeof(*check));
247 			overflow = (check->magic != SYSMEM_MAGIC);
248 		} else if (smem->attr.flags & M_ATTR_HOFC) {
249 			check = (struct memcheck *)
250 				(smem->base - sizeof(*check));
251 			overflow = (check->magic != SYSMEM_MAGIC);
252 		} else {
253 			overflow = 0;
254 		}
255 
256 		if (overflow) {
257 			SYSMEM_E("Found there is region overflow!\n");
258 			break;
259 		}
260 	}
261 
262 	if (overflow || overlap)
263 		sysmem_dump();
264 }
265 
266 static int sysmem_add(phys_addr_t base, phys_size_t size)
267 {
268 	struct sysmem *sysmem = &plat_sysmem;
269 	int ret;
270 
271 	if (!size)
272 		return -EINVAL;
273 
274 	ret = lmb_add(&sysmem->lmb, base, size);
275 	if (ret < 0)
276 		SYSMEM_E("Failed to add sysmem at 0x%08lx for 0x%08lx size\n",
277 			 (ulong)base, (ulong)size);
278 
279 	return (ret >= 0) ? 0 : ret;
280 }
281 
282 static const char *sysmem_alias2name(const char *name, int *id)
283 {
284 	const char *alias;
285 	int i, j;
286 	int match = 0;
287 
288 	for (i = 0; i < MEMBLK_ID_MAX; i++) {
289 		/* Pirmary name */
290 		if (mem_attr[i].name && !strcasecmp(mem_attr[i].name, name)) {
291 			match = 1;
292 			goto finish;
293 		}
294 
295 		/* Alias name */
296 		alias = mem_attr[i].alias[0];
297 		if (!alias)
298 			continue;
299 
300 		for (j = 0; j < ALIAS_COUNT_MAX; j++) {
301 			alias = mem_attr[i].alias[j];
302 			if (alias && !strcasecmp(alias, name)) {
303 				match = 1;
304 				goto finish;
305 			}
306 		}
307 	}
308 
309 finish:
310 	if (match) {
311 		*id = i;
312 		return mem_attr[i].name;
313 	}
314 
315 	return name;
316 }
317 
318 static void *sysmem_alloc_align_base(enum memblk_id id,
319 				     const char *mem_name,
320 				     phys_addr_t base,
321 				     phys_size_t size,
322 				     ulong align)
323 {
324 	struct sysmem *sysmem = &plat_sysmem;
325 	struct memblk_attr attr;
326 	struct memblock *mem;
327 	struct memcheck *check;
328 	struct list_head *node;
329 	const char *name;
330 	phys_addr_t paddr;
331 	phys_addr_t alloc_base;
332 	phys_size_t alloc_size;
333 	phys_addr_t orig_base = base;
334 
335 	if (!sysmem_has_init())
336 		goto out;
337 
338 	if (id == MEMBLK_ID_BY_NAME || id == MEMBLK_ID_KMEM_RESERVED) {
339 		if (!mem_name) {
340 			SYSMEM_E("NULL name for alloc sysmem\n");
341 			goto out;
342 		}
343 
344 		/* Find: name, id and attr by outer mem_name & id */
345 		name = sysmem_alias2name(mem_name, (int *)&id);
346 		attr = mem_attr[id];
347 		if (!attr.name)
348 			attr.name = strdup(name);
349 
350 		/* Always make kernel 'reserved-memory' alloc successfully */
351 		if (id == MEMBLK_ID_KMEM_RESERVED) {
352 			struct memblock *mem;
353 
354 			mem = malloc(sizeof(*mem));
355 			if (!mem) {
356 				SYSMEM_E("No memory for \"%s\" alloc sysmem\n", name);
357 				return mem;
358 			}
359 
360 			attr.flags |= M_ATTR_KMEM_RESERVED;
361 			mem->orig_base = orig_base;
362 			mem->base = base;
363 			mem->size = size;
364 			mem->attr = attr;
365 			sysmem->kmem_resv_cnt++;
366 			list_add_tail(&mem->node, &sysmem->kmem_resv_head);
367 
368 			return (void *)base;
369 		}
370 	} else if (id > MEMBLK_ID_UNK && id < MEMBLK_ID_MAX) {
371 		attr = mem_attr[id];
372 		name = attr.name;
373 
374 		/*
375 		 * Special handle for Android AVB alloc(on any where)
376 		 *
377 		 * Fixup base and place right after U-Boot stack, adding a lot
378 		 * of space(4KB) maybe safer.
379 		 */
380 		if (attr.flags & M_ATTR_HIGHEST_MEM) {
381 			base = gd->start_addr_sp -
382 					CONFIG_SYS_STACK_SIZE - size - 0x1000;
383 
384 		/*
385 		 * The 0x0 address is usually allocated by 32-bit uncompressed
386 		 * kernel and this alloc action is just a peek.
387 		 *
388 		 * Due to LMB core doesn't support alloc at 0x0 address, we have
389 		 * to alloc the memblk backword a few bytes.
390 		 *
391 		 * ARCH_DMA_MINALIGN maybe a good choice.
392 		 */
393 		} else if (!base) {
394 			base += ARCH_DMA_MINALIGN;
395 		} else if (base <= gd->bd->bi_dram[0].start) {
396 			/*
397 			 * On Rockchip platform:
398 			 *
399 			 * So far, we use M_ATTR_IGNORE_INVISIBLE for uncompress
400 			 * kernel alloc, and for ARMv8 enabling AArch32 mode, the
401 			 * ATF is still AArch64 and ocuppies 0~1MB and shmem 1~2M.
402 			 * So let's ignore the region which overlap with them.
403 			 */
404 			if (attr.flags & M_ATTR_IGNORE_INVISIBLE) {
405 				base = gd->bd->bi_dram[0].start;
406 			} else {
407 				SYSMEM_E("Failed to alloc invisible sub region 0x%08lx - 0x%08lx "
408 					 "of \"%s\" at 0x%08lx - 0x%08lx\n",
409 					 (ulong)base, (ulong)gd->bd->bi_dram[0].start,
410 					 name, (ulong)base, (ulong)(base + size));
411 				goto out;
412 			}
413 		}
414 	} else {
415 		SYSMEM_E("Unsupport memblk id %d for alloc sysmem\n", id);
416 		goto out;
417 	}
418 
419 	if (!size) {
420 		SYSMEM_E("\"%s\" size is 0 for alloc sysmem\n", name);
421 		goto out;
422 	}
423 
424 	/*
425 	 * Some modules use "sysmem_alloc()" to alloc region for storage
426 	 * read/write buffer, it should be aligned to cacheline size. eg: AVB.
427 	 *
428 	 * Aligned down to cacheline size if not aligned, otherwise the tail
429 	 * of region maybe overflow.
430 	 */
431 	if (attr.flags & M_ATTR_CACHELINE_ALIGN &&
432 	    !IS_ALIGNED(base, ARCH_DMA_MINALIGN)) {
433 		base = ALIGN(base, ARCH_DMA_MINALIGN);
434 		base -= ARCH_DMA_MINALIGN;
435 	}
436 
437 	if (base != SYSMEM_ALLOC_ANYWHERE && !IS_ALIGNED(base, 4)) {
438 		SYSMEM_E("\"%s\" base=0x%08lx is not 4-byte aligned\n",
439 			 name, (ulong)base);
440 		goto out;
441 	}
442 
443 	/* Must be sizeof(long) byte aligned */
444 	size = ALIGN(size, sizeof(long));
445 
446 	SYSMEM_D("Enter alloc: \"%s\" 0x%08lx - 0x%08lx\n",
447 		 name, (ulong)base, (ulong)(base + size));
448 
449 	/* Already allocated ? */
450 	list_for_each(node, &sysmem->allocated_head) {
451 		mem = list_entry(node, struct memblock, node);
452 		SYSMEM_D("Has allcated: %s, 0x%08lx - 0x%08lx\n",
453 			 mem->attr.name, (ulong)mem->base,
454 			 (ulong)(mem->base + mem->size));
455 		if (!strcmp(mem->attr.name, name)) {
456 			/* Allow double alloc for same but smaller region */
457 			if (mem->base <= base && mem->size >= size)
458 				return (void *)base;
459 
460 			SYSMEM_E("Failed to double alloc for existence \"%s\"\n", name);
461 			goto out;
462 		} else if (sysmem_is_overlap(mem->base, mem->size, base, size)) {
463 			SYSMEM_E("\"%s\" (0x%08lx - 0x%08lx) alloc is "
464 				 "overlap with existence \"%s\" (0x%08lx - "
465 				 "0x%08lx)\n",
466 				 name, (ulong)base, (ulong)(base + size),
467 				 mem->attr.name, (ulong)mem->base,
468 				 (ulong)(mem->base + mem->size));
469 			goto out;
470 		}
471 	}
472 
473 	/* Add overflow check magic ? */
474 	if (attr.flags & M_ATTR_OFC)
475 		alloc_size = size + sizeof(*check);
476 	else
477 		alloc_size = size;
478 
479 	/* Alloc anywhere ? */
480 	if (base == SYSMEM_ALLOC_ANYWHERE)
481 		alloc_base = LMB_ALLOC_ANYWHERE;
482 	else
483 		alloc_base = base + alloc_size;	/* LMB is align down alloc mechanism */
484 
485 	SYSMEM_D("DO alloc... base: 0x%08lx\n", (ulong)alloc_base);
486 
487 	paddr = lmb_alloc_base(&sysmem->lmb, alloc_size, align, alloc_base);
488 	if (paddr) {
489 		if ((paddr == base) || (base == SYSMEM_ALLOC_ANYWHERE)) {
490 			mem = malloc(sizeof(*mem));
491 			if (!mem) {
492 				SYSMEM_E("No memory for \"%s\" alloc sysmem\n", name);
493 				goto out;
494 			}
495 			/* Record original base for dump */
496 			if (attr.flags & M_ATTR_HIGHEST_MEM)
497 				mem->orig_base = base;
498 			else
499 				mem->orig_base = orig_base;
500 
501 			mem->base = paddr;
502 			mem->size = alloc_size;
503 			mem->attr = attr;
504 			sysmem->allocated_cnt++;
505 			list_add_tail(&mem->node, &sysmem->allocated_head);
506 
507 			/* Add overflow check magic */
508 			if (mem->attr.flags & M_ATTR_OFC) {
509 				check = (struct memcheck *)(paddr + size);
510 				check->magic = SYSMEM_MAGIC;
511 			} else if (mem->attr.flags & M_ATTR_HOFC) {
512 				check = (struct memcheck *)(paddr - sizeof(*check));
513 				check->magic = SYSMEM_MAGIC;
514 			}
515 		} else {
516 			SYSMEM_E("Failed to alloc \"%s\" expect at 0x%08lx - 0x%08lx "
517 				 "but at 0x%08lx - x%08lx\n",
518 				 name, (ulong)base, (ulong)(base + size),
519 				 (ulong)paddr, (ulong)(paddr + size));
520 			/* Free what we don't want allocated region */
521 			if (lmb_free(&sysmem->lmb, paddr, alloc_size) < 0)
522 				SYSMEM_E("Failed to free \"%s\"\n", name);
523 
524 			goto out;
525 		}
526 	} else {
527 		SYSMEM_E("Failed to alloc \"%s\" at 0x%08lx - 0x%08lx\n",
528 			 name, (ulong)base, (ulong)(base + size));
529 		goto out;
530 	}
531 
532 	SYSMEM_D("Exit alloc: \"%s\", paddr=0x%08lx, size=0x%08lx, align=0x%x, anywhere=%d\n",
533 		 name, (ulong)paddr, (ulong)size, (u32)align, !base);
534 
535 	return (void *)paddr;
536 
537 out:
538 	/*
539 	 * Why: base + sizeof(ulong) ?
540 	 * It's not a standard way to handle the case: the input base is 0.
541 	 * Because 0 equals NULL, but we don't want to return NULL when alloc
542 	 * successfully, so just return a !NULL value is okay.
543 	 *
544 	 * When it happens ?
545 	 * Maybe 32-bit platform would alloc region for uncompress kernel
546 	 * at 0 address.
547 	 */
548 	if (base == 0)
549 		base = base + sizeof(ulong);
550 
551 	return (attr.flags & M_ATTR_IGNORE_INVISIBLE) ? (void *)base : NULL;
552 }
553 
554 void *sysmem_alloc(enum memblk_id id, phys_size_t size)
555 {
556 	void *paddr;
557 
558 	paddr = sysmem_alloc_align_base(id,
559 					NULL,
560 					SYSMEM_ALLOC_ANYWHERE,
561 					size,
562 					ARCH_DMA_MINALIGN);
563 	if (!paddr)
564 		sysmem_dump();
565 
566 	return paddr;
567 }
568 
569 void *sysmem_alloc_by_name(const char *name, phys_size_t size)
570 {
571 	void *paddr;
572 
573 	paddr = sysmem_alloc_align_base(MEMBLK_ID_BY_NAME,
574 					name,
575 					SYSMEM_ALLOC_ANYWHERE,
576 					size,
577 					ARCH_DMA_MINALIGN);
578 	if (!paddr)
579 		sysmem_dump();
580 
581 	return paddr;
582 }
583 
584 void *sysmem_alloc_base(enum memblk_id id, phys_addr_t base, phys_size_t size)
585 {
586 	void *paddr;
587 
588 	paddr = sysmem_alloc_align_base(id,
589 					NULL,
590 					base,
591 					size,
592 					SYSMEM_ALLOC_NO_ALIGN);
593 	if (!paddr)
594 		sysmem_dump();
595 
596 	return paddr;
597 }
598 
599 void *sysmem_alloc_base_by_name(const char *name,
600 				phys_addr_t base, phys_size_t size)
601 {
602 	void *paddr;
603 
604 	paddr = sysmem_alloc_align_base(MEMBLK_ID_BY_NAME,
605 					name,
606 					base,
607 					size,
608 					SYSMEM_ALLOC_NO_ALIGN);
609 	if (!paddr)
610 		sysmem_dump();
611 
612 	return paddr;
613 }
614 
615 void *sysmem_fdt_reserve_alloc_base(const char *name,
616 				    phys_addr_t base, phys_size_t size)
617 {
618 	void *paddr;
619 
620 	paddr = sysmem_alloc_align_base(MEMBLK_ID_KMEM_RESERVED,
621 					name,
622 					base,
623 					size,
624 					SYSMEM_ALLOC_NO_ALIGN);
625 	if (!paddr)
626 		sysmem_dump();
627 
628 	return paddr;
629 }
630 
631 bool sysmem_can_alloc(phys_size_t base, phys_size_t size)
632 {
633 	struct sysmem *sysmem = &plat_sysmem;
634 	phys_addr_t alloc_base;
635 	phys_addr_t paddr;
636 	int ret;
637 
638 	if (!sysmem_has_init())
639 		return false;
640 
641 	/* LMB is align down alloc mechanism */
642 	alloc_base = base + size;
643 	paddr = __lmb_alloc_base(&sysmem->lmb,
644 				 size,
645 				 SYSMEM_ALLOC_NO_ALIGN,
646 				 alloc_base);
647 	if (paddr) {
648 		/* If free failed, return false */
649 		ret = lmb_free(&sysmem->lmb, base, size);
650 		if (ret < 0) {
651 			SYSMEM_E("Can't free at 0x%08lx - 0x%08lx, ret=%d\n",
652 				 (ulong)base, (ulong)(base + size), ret);
653 			return false;
654 		}
655 	} else {
656 		SYSMEM_D("Can't alloc at 0x%08lx - 0x%08lx\n",
657 			 (ulong)base, (ulong)(base + size));
658 	}
659 
660 	return (paddr == base) ? true : false;
661 }
662 
663 int sysmem_free(phys_addr_t base)
664 {
665 	struct sysmem *sysmem = &plat_sysmem;
666 	struct memblock *mem;
667 	struct list_head *node;
668 	int ret, found = 0;
669 
670 	if (!sysmem_has_init())
671 		return -ENOSYS;
672 
673 	/* Find existence */
674 	list_for_each(node, &sysmem->allocated_head) {
675 		mem = list_entry(node, struct memblock, node);
676 		if (mem->base == base || mem->orig_base == base) {
677 			found = 1;
678 			break;
679 		}
680 	}
681 
682 	if (!found) {
683 		SYSMEM_E("Failed to free no allocated sysmem at 0x%08lx\n",
684 			 (ulong)base);
685 		return -EINVAL;
686 	}
687 
688 	ret = lmb_free(&sysmem->lmb, mem->base, mem->size);
689 	if (ret >= 0) {
690 		SYSMEM_D("Free: \"%s\" 0x%08lx - 0x%08lx\n",
691 			 mem->attr.name, (ulong)mem->base,
692 			 (ulong)(mem->base + mem->size));
693 		sysmem->allocated_cnt--;
694 		list_del(&mem->node);
695 		free(mem);
696 	} else {
697 		SYSMEM_E("Failed to free \"%s\" at 0x%08lx\n",
698 			 mem->attr.name, (ulong)base);
699 	}
700 
701 	return (ret >= 0) ? 0 : ret;
702 }
703 
704 int sysmem_initr(void)
705 {
706 	return sysmem_init();
707 }
708 
709 int sysmem_init(void)
710 {
711 	struct sysmem *sysmem = &plat_sysmem;
712 	phys_addr_t mem_start;
713 	phys_size_t mem_size;
714 	int ret;
715 
716 	lmb_init(&sysmem->lmb);
717 	INIT_LIST_HEAD(&sysmem->allocated_head);
718 	INIT_LIST_HEAD(&sysmem->kmem_resv_head);
719 	sysmem->allocated_cnt = 0;
720 	sysmem->kmem_resv_cnt = 0;
721 
722 	if (gd->flags & GD_FLG_RELOC) {
723 		sysmem->has_initr = true;
724 	} else {
725 		SYSMEM_I("init\n");
726 		sysmem->has_initf = true;
727 	}
728 
729 	/* Add all available system memory */
730 #ifdef CONFIG_NR_DRAM_BANKS
731 	int i;
732 
733 	for (i = 0; i < CONFIG_NR_DRAM_BANKS; i++) {
734 		if (!gd->bd->bi_dram[i].size)
735 			continue;
736 
737 		ret = sysmem_add(gd->bd->bi_dram[i].start,
738 				 gd->bd->bi_dram[i].size);
739 		if (ret) {
740 			SYSMEM_E("Failed to add sysmem from bi_dram[%d]\n", i);
741 			goto fail;
742 		}
743 	}
744 #else
745 	mem_start = env_get_bootm_low();
746 	mem_size = env_get_bootm_size();
747 	ret = sysmem_add(mem_start, mem_size);
748 	if (ret) {
749 		SYSMEM_E("Failed to add sysmem from bootm_low/size\n");
750 		goto fail;
751 	}
752 #endif
753 	/* Reserved for board */
754 	ret = board_sysmem_reserve(sysmem);
755 	if (ret) {
756 		SYSMEM_E("Failed to reserve sysmem for board\n");
757 		goto fail;
758 	}
759 
760 	/* Reserved for U-boot framework: 'reserve_xxx()' */
761 	mem_start = gd->start_addr_sp;
762 	mem_size = gd->ram_top - mem_start;
763 	if (!sysmem_alloc_base(MEMBLK_ID_UBOOT, mem_start, mem_size)) {
764 		SYSMEM_E("Failed to reserve sysmem for U-Boot framework\n");
765 		ret = -ENOMEM;
766 		goto fail;
767 	}
768 
769 	/* Reserved for U-Boot stack */
770 	mem_start = gd->start_addr_sp - CONFIG_SYS_STACK_SIZE;
771 	mem_size = CONFIG_SYS_STACK_SIZE;
772 	if (!sysmem_alloc_base(MEMBLK_ID_STACK, mem_start, mem_size)) {
773 		SYSMEM_E("Failed to reserve sysmem for stack\n");
774 		ret = -ENOMEM;
775 		goto fail;
776 	}
777 
778 	return 0;
779 
780 fail:
781 	if (ret && !(gd->flags & GD_FLG_RELOC)) {
782 		sysmem_dump();
783 		SYSMEM_W("Maybe malloc size %d MiB is too large?\n\n",
784 			 SIZE_MB(CONFIG_SYS_MALLOC_LEN));
785 	}
786 
787 	return ret;
788 }
789 
790 __weak int board_sysmem_reserve(struct sysmem *sysmem)
791 {
792 	/* please define platform specific board_sysmem_reserve() */
793 	return 0;
794 }
795 
796 static int do_dump_sysmem(cmd_tbl_t *cmdtp, int flag,
797 			  int argc, char *const argv[])
798 {
799 	sysmem_dump();
800 	return 0;
801 }
802 
803 U_BOOT_CMD(
804 	dump_sysmem, 1, 1, do_dump_sysmem,
805 	"Dump sysmem layout",
806 	""
807 );
808