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 phys_addr_t bank_base; 198 phys_size_t bank_size; 199 bool req_overlap = false; /* Only for kernel reserved-memory */ 200 int i; 201 202 if (!sysmem_has_init()) 203 goto out; 204 205 if (id == MEMBLK_ID_BY_NAME || id == MEMBLK_ID_FDT_RESV) { 206 if (!mem_name) { 207 SYSMEM_E("NULL name for alloc sysmem\n"); 208 goto out; 209 } else if (id == MEMBLK_ID_FDT_RESV) { 210 211 /* 212 * Allow fdt reserved memory to overlap with the region 213 * only used in U-Boot, like: stack, fastboot, u-boot... 214 * these regions are marked as M_ATTR_OVERLAP in flags. 215 * 216 * Here we check whether it overlaps with others, if 217 * so, set req_overlap as true. 218 */ 219 for (i = 0; i < CONFIG_NR_DRAM_BANKS; i++) { 220 if (!gd->bd->bi_dram[i].size) 221 continue; 222 223 bank_base = gd->bd->bi_dram[i].start; 224 bank_size = gd->bd->bi_dram[i].size; 225 if (sysmem_is_overlap(base, size, 226 bank_base, bank_size)) { 227 req_overlap = true; 228 break; 229 } 230 } 231 232 /* 233 * If this request region is out size of all available 234 * region, ignore and return success. 235 */ 236 if (!req_overlap) 237 return (void *)base; 238 } 239 240 /* Find name, id and attr by outer mem_name */ 241 name = sysmem_alias2name(mem_name, (int *)&id); 242 attr = mem_attr[id]; 243 if (!attr.name) 244 attr.name = strdup(name); 245 } else if (id > MEMBLK_ID_UNK && id < MEMBLK_ID_MAX) { 246 attr = mem_attr[id]; 247 name = attr.name; 248 249 /* 250 * Fixup base and place right after U-Boot stack, adding a lot 251 * of space(4KB) maybe safer. 252 */ 253 if ((id == MEMBLK_ID_AVB_ANDROID) && 254 (base == SYSMEM_ALLOC_ANYWHERE)) 255 base = gd->start_addr_sp - 256 CONFIG_SYS_STACK_SIZE - size - 0x1000; 257 } else { 258 SYSMEM_E("Unsupport memblk id %d for alloc sysmem\n", id); 259 goto out; 260 } 261 262 if (!size) { 263 SYSMEM_E("\"%s\" size is 0 for alloc sysmem\n", name); 264 goto out; 265 } 266 267 if (!IS_ALIGNED(base, 4)) { 268 SYSMEM_E("\"%s\" base=0x%08lx is not 4-byte aligned\n", 269 name, (ulong)base); 270 goto out; 271 } 272 273 /* Must be 4-byte aligned */ 274 size = ALIGN(size, 4); 275 276 SYSMEM_D("Enter alloc: \"%s\" 0x%08lx - 0x%08lx\n", 277 name, (ulong)base, (ulong)(base + size)); 278 279 /* Already allocated ? */ 280 list_for_each(node, &sysmem->allocated_head) { 281 mem = list_entry(node, struct memblock, node); 282 SYSMEM_D("Has allcated: %s, 0x%08lx - 0x%08lx\n", 283 mem->attr.name, (ulong)mem->base, 284 (ulong)(mem->base + mem->size)); 285 if (!strcmp(mem->attr.name, name)) { 286 /* Allow double alloc for same but smaller region */ 287 if (mem->base <= base && mem->size >= size) 288 return (void *)base; 289 290 SYSMEM_E("Failed to double alloc for existence \"%s\"\n", name); 291 goto out; 292 } else if (sysmem_is_overlap(mem->base, mem->size, base, size)) { 293 /* 294 * If this new alloc region expects overlap and the old 295 * region is also allowed to be overlap, just do reserve. 296 */ 297 if (req_overlap && mem->attr.flags & M_ATTR_OVERLAP) { 298 if (lmb_reserve(&sysmem->lmb, base, size)) 299 SYSMEM_E("Failed to overlap alloc \"%s\" " 300 "at 0x%08lx - 0x%08lx\n", 301 name, (ulong)base, 302 (ulong)(base + size)); 303 return (void *)base; 304 } 305 306 SYSMEM_E("\"%s\" (0x%08lx - 0x%08lx) alloc is " 307 "overlap with existence \"%s\" (0x%08lx - " 308 "0x%08lx)\n", 309 name, (ulong)base, (ulong)(base + size), 310 mem->attr.name, (ulong)mem->base, 311 (ulong)(mem->base + mem->size)); 312 goto out; 313 } 314 } 315 316 /* Add overflow check magic ? */ 317 if (attr.flags & M_ATTR_OFC) 318 alloc_size = size + sizeof(*check); 319 else 320 alloc_size = size; 321 322 /* Alloc anywhere ? */ 323 if (base == SYSMEM_ALLOC_ANYWHERE) 324 alloc_base = base; 325 else 326 alloc_base = base + alloc_size; /* LMB is align down alloc mechanism */ 327 328 paddr = lmb_alloc_base(&sysmem->lmb, alloc_size, align, alloc_base); 329 if (paddr) { 330 if ((paddr == base) || (base == SYSMEM_ALLOC_ANYWHERE)) { 331 mem = malloc(sizeof(*mem)); 332 if (!mem) { 333 SYSMEM_E("No memory for \"%s\" alloc sysmem\n", name); 334 goto out; 335 } 336 337 mem->base = paddr; 338 mem->size = alloc_size; 339 mem->attr = attr; 340 sysmem->allocated_cnt++; 341 list_add_tail(&mem->node, &sysmem->allocated_head); 342 343 /* Add overflow check magic */ 344 if (mem->attr.flags & M_ATTR_OFC) { 345 check = (struct memcheck *)(paddr + size); 346 check->magic = SYSMEM_MAGIC; 347 } else if (mem->attr.flags & M_ATTR_HOFC) { 348 check = (struct memcheck *)(paddr - sizeof(*check)); 349 check->magic = SYSMEM_MAGIC; 350 } 351 } else { 352 SYSMEM_E("Failed to alloc \"%s\" expect at 0x%08lx - 0x%08lx " 353 "but at 0x%08lx - x%08lx\n", 354 name, (ulong)base, (ulong)(base + size), 355 (ulong)paddr, (ulong)(paddr + size)); 356 /* Free what we don't want allocated region */ 357 if (lmb_free(&sysmem->lmb, paddr, alloc_size) < 0) 358 SYSMEM_E("Failed to free \"%s\"\n", name); 359 360 goto out; 361 } 362 } else { 363 SYSMEM_E("Failed to alloc \"%s\" at 0x%08lx - 0x%08lx\n", 364 name, (ulong)base, (ulong)(base + size)); 365 } 366 367 SYSMEM_D("Exit alloc: \"%s\", paddr=0x%08lx, size=0x%08lx, align=0x%x, anywhere=%d\n", 368 name, (ulong)paddr, (ulong)size, (u32)align, !base); 369 370 return (void *)paddr; 371 372 out: 373 return (attr.flags & M_ATTR_PEEK) ? (void *)base : NULL; 374 } 375 376 void *sysmem_alloc(enum memblk_id id, phys_size_t size) 377 { 378 void *paddr; 379 380 paddr = sysmem_alloc_align_base(id, 381 NULL, 382 SYSMEM_ALLOC_ANYWHERE, 383 size, 384 SYSMEM_ALLOC_NO_ALIGN); 385 if (!paddr) 386 sysmem_dump(); 387 388 return paddr; 389 } 390 391 void *sysmem_alloc_base(enum memblk_id id, phys_addr_t base, phys_size_t size) 392 { 393 void *paddr; 394 395 paddr = sysmem_alloc_align_base(id, 396 NULL, 397 base, 398 size, 399 SYSMEM_ALLOC_NO_ALIGN); 400 if (!paddr) 401 sysmem_dump(); 402 403 return paddr; 404 } 405 406 void *sysmem_alloc_base_by_name(const char *name, 407 phys_addr_t base, phys_size_t size) 408 { 409 void *paddr; 410 411 paddr = sysmem_alloc_align_base(MEMBLK_ID_BY_NAME, 412 name, 413 base, 414 size, 415 SYSMEM_ALLOC_NO_ALIGN); 416 if (!paddr) 417 sysmem_dump(); 418 419 return paddr; 420 } 421 422 void *sysmem_fdt_reserve_alloc_base(const char *name, 423 phys_addr_t base, phys_size_t size) 424 { 425 void *paddr; 426 427 paddr = sysmem_alloc_align_base(MEMBLK_ID_FDT_RESV, 428 name, 429 base, 430 size, 431 SYSMEM_ALLOC_NO_ALIGN); 432 if (!paddr) 433 sysmem_dump(); 434 435 return paddr; 436 } 437 438 bool sysmem_can_alloc(phys_size_t base, phys_size_t size) 439 { 440 struct sysmem *sysmem = &plat_sysmem; 441 phys_addr_t alloc_base; 442 phys_addr_t paddr; 443 int ret; 444 445 if (!sysmem_has_init()) 446 return false; 447 448 /* LMB is align down alloc mechanism */ 449 alloc_base = base + size; 450 paddr = __lmb_alloc_base(&sysmem->lmb, 451 size, 452 SYSMEM_ALLOC_NO_ALIGN, 453 alloc_base); 454 if (paddr) { 455 /* If free failed, return false */ 456 ret = lmb_free(&sysmem->lmb, base, size); 457 if (ret < 0) { 458 SYSMEM_E("Can't free at 0x%08lx - 0x%08lx, ret=%d\n", 459 (ulong)base, (ulong)(base + size), ret); 460 return false; 461 } 462 } else { 463 SYSMEM_D("Can't alloc at 0x%08lx - 0x%08lx\n", 464 (ulong)base, (ulong)(base + size)); 465 } 466 467 return (paddr == base) ? true : false; 468 } 469 470 int sysmem_free(phys_addr_t base) 471 { 472 struct sysmem *sysmem = &plat_sysmem; 473 struct memblock *mem; 474 struct list_head *node; 475 int ret, found = 0; 476 477 if (!sysmem_has_init()) 478 return -ENOSYS; 479 480 /* Find existence */ 481 list_for_each(node, &sysmem->allocated_head) { 482 mem = list_entry(node, struct memblock, node); 483 if (mem->base == base) { 484 found = 1; 485 break; 486 } 487 } 488 489 if (!found) { 490 SYSMEM_E("Failed to free no allocated sysmem at 0x%08lx\n", 491 (ulong)base); 492 return -EINVAL; 493 } 494 495 ret = lmb_free(&sysmem->lmb, mem->base, mem->size); 496 if (ret >= 0) { 497 SYSMEM_D("Free: \"%s\" 0x%08lx - 0x%08lx\n", 498 mem->attr.name, (ulong)mem->base, 499 (ulong)(mem->base + mem->size)); 500 sysmem->allocated_cnt--; 501 list_del(&mem->node); 502 free(mem); 503 } else { 504 SYSMEM_E("Failed to free \"%s\" at 0x%08lx\n", 505 mem->attr.name, (ulong)base); 506 } 507 508 return (ret >= 0) ? 0 : ret; 509 } 510 511 int sysmem_initr(void) 512 { 513 return sysmem_init(); 514 } 515 516 int sysmem_init(void) 517 { 518 struct sysmem *sysmem = &plat_sysmem; 519 phys_addr_t mem_start; 520 phys_size_t mem_size; 521 int ret; 522 523 lmb_init(&sysmem->lmb); 524 INIT_LIST_HEAD(&sysmem->allocated_head); 525 sysmem->allocated_cnt = 0; 526 if (gd->flags & GD_FLG_RELOC) { 527 sysmem->has_initr = true; 528 } else { 529 SYSMEM_I("init\n"); 530 sysmem->has_initf = true; 531 } 532 533 /* Add all available system memory */ 534 #ifdef CONFIG_NR_DRAM_BANKS 535 int i; 536 537 for (i = 0; i < CONFIG_NR_DRAM_BANKS; i++) { 538 if (!gd->bd->bi_dram[i].size) 539 continue; 540 541 ret = sysmem_add(gd->bd->bi_dram[i].start, 542 gd->bd->bi_dram[i].size); 543 if (ret) { 544 SYSMEM_E("Failed to add sysmem from bi_dram[%d]\n", i); 545 goto fail; 546 } 547 } 548 #else 549 mem_start = env_get_bootm_low(); 550 mem_size = env_get_bootm_size(); 551 ret = sysmem_add(mem_start, mem_size); 552 if (ret) { 553 SYSMEM_E("Failed to add sysmem from bootm_low/size\n"); 554 goto fail; 555 } 556 #endif 557 /* Reserved for board */ 558 ret = board_sysmem_reserve(sysmem); 559 if (ret) { 560 SYSMEM_E("Failed to reserve sysmem for board\n"); 561 goto fail; 562 } 563 564 /* Reserved for U-boot framework: 'reserve_xxx()' */ 565 mem_start = gd->start_addr_sp; 566 mem_size = gd->ram_top - mem_start; 567 if (!sysmem_alloc_base(MEMBLK_ID_UBOOT, mem_start, mem_size)) { 568 SYSMEM_E("Failed to reserve sysmem for U-Boot framework\n"); 569 ret = -ENOMEM; 570 goto fail; 571 } 572 573 /* Reserved for U-Boot stack */ 574 mem_start = gd->start_addr_sp - CONFIG_SYS_STACK_SIZE; 575 mem_size = CONFIG_SYS_STACK_SIZE; 576 if (!sysmem_alloc_base(MEMBLK_ID_STACK, mem_start, mem_size)) { 577 SYSMEM_E("Failed to reserve sysmem for stack\n"); 578 ret = -ENOMEM; 579 goto fail; 580 } 581 582 return 0; 583 584 fail: 585 if (ret && !(gd->flags & GD_FLG_RELOC)) { 586 sysmem_dump(); 587 SYSMEM_W("Maybe malloc size %d MiB is too large?\n\n", 588 SIZE_MB(CONFIG_SYS_MALLOC_LEN)); 589 } 590 591 return ret; 592 } 593 594 __weak int board_sysmem_reserve(struct sysmem *sysmem) 595 { 596 /* please define platform specific board_sysmem_reserve() */ 597 return 0; 598 } 599 600 static int do_dump_sysmem(cmd_tbl_t *cmdtp, int flag, 601 int argc, char *const argv[]) 602 { 603 sysmem_dump(); 604 return 0; 605 } 606 607 U_BOOT_CMD( 608 dump_sysmem, 1, 1, do_dump_sysmem, 609 "Dump sysmem layout", 610 "" 611 ); 612