195f72d1eSYinghai Lu /* 295f72d1eSYinghai Lu * Procedures for maintaining information about logical memory blocks. 395f72d1eSYinghai Lu * 495f72d1eSYinghai Lu * Peter Bergner, IBM Corp. June 2001. 595f72d1eSYinghai Lu * Copyright (C) 2001 Peter Bergner. 695f72d1eSYinghai Lu * 795f72d1eSYinghai Lu * This program is free software; you can redistribute it and/or 895f72d1eSYinghai Lu * modify it under the terms of the GNU General Public License 995f72d1eSYinghai Lu * as published by the Free Software Foundation; either version 1095f72d1eSYinghai Lu * 2 of the License, or (at your option) any later version. 1195f72d1eSYinghai Lu */ 1295f72d1eSYinghai Lu 1395f72d1eSYinghai Lu #include <linux/kernel.h> 14142b45a7SBenjamin Herrenschmidt #include <linux/slab.h> 1595f72d1eSYinghai Lu #include <linux/init.h> 1695f72d1eSYinghai Lu #include <linux/bitops.h> 17449e8df3SBenjamin Herrenschmidt #include <linux/poison.h> 18c196f76fSBenjamin Herrenschmidt #include <linux/pfn.h> 196d03b885SBenjamin Herrenschmidt #include <linux/debugfs.h> 206d03b885SBenjamin Herrenschmidt #include <linux/seq_file.h> 2195f72d1eSYinghai Lu #include <linux/memblock.h> 2295f72d1eSYinghai Lu 23c4c5ad6bSChristoph Hellwig #include <asm/sections.h> 2426f09e9bSSantosh Shilimkar #include <linux/io.h> 2526f09e9bSSantosh Shilimkar 2626f09e9bSSantosh Shilimkar #include "internal.h" 2779442ed1STang Chen 28fe091c20STejun Heo static struct memblock_region memblock_memory_init_regions[INIT_MEMBLOCK_REGIONS] __initdata_memblock; 29fe091c20STejun Heo static struct memblock_region memblock_reserved_init_regions[INIT_MEMBLOCK_REGIONS] __initdata_memblock; 3070210ed9SPhilipp Hachtmann #ifdef CONFIG_HAVE_MEMBLOCK_PHYS_MAP 3170210ed9SPhilipp Hachtmann static struct memblock_region memblock_physmem_init_regions[INIT_PHYSMEM_REGIONS] __initdata_memblock; 3270210ed9SPhilipp Hachtmann #endif 33fe091c20STejun Heo 34fe091c20STejun Heo struct memblock memblock __initdata_memblock = { 35fe091c20STejun Heo .memory.regions = memblock_memory_init_regions, 36fe091c20STejun Heo .memory.cnt = 1, /* empty dummy entry */ 37fe091c20STejun Heo .memory.max = INIT_MEMBLOCK_REGIONS, 38fe091c20STejun Heo 39fe091c20STejun Heo .reserved.regions = memblock_reserved_init_regions, 40fe091c20STejun Heo .reserved.cnt = 1, /* empty dummy entry */ 41fe091c20STejun Heo .reserved.max = INIT_MEMBLOCK_REGIONS, 42fe091c20STejun Heo 4370210ed9SPhilipp Hachtmann #ifdef CONFIG_HAVE_MEMBLOCK_PHYS_MAP 4470210ed9SPhilipp Hachtmann .physmem.regions = memblock_physmem_init_regions, 4570210ed9SPhilipp Hachtmann .physmem.cnt = 1, /* empty dummy entry */ 4670210ed9SPhilipp Hachtmann .physmem.max = INIT_PHYSMEM_REGIONS, 4770210ed9SPhilipp Hachtmann #endif 4870210ed9SPhilipp Hachtmann 4979442ed1STang Chen .bottom_up = false, 50fe091c20STejun Heo .current_limit = MEMBLOCK_ALLOC_ANYWHERE, 51fe091c20STejun Heo }; 5295f72d1eSYinghai Lu 5310d06439SYinghai Lu int memblock_debug __initdata_memblock; 5455ac590cSTang Chen #ifdef CONFIG_MOVABLE_NODE 5555ac590cSTang Chen bool movable_node_enabled __initdata_memblock = false; 5655ac590cSTang Chen #endif 57a3f5bafcSTony Luck static bool system_has_some_mirror __initdata_memblock = false; 581aadc056STejun Heo static int memblock_can_resize __initdata_memblock; 59181eb394SGavin Shan static int memblock_memory_in_slab __initdata_memblock = 0; 60181eb394SGavin Shan static int memblock_reserved_in_slab __initdata_memblock = 0; 6195f72d1eSYinghai Lu 62a3f5bafcSTony Luck ulong __init_memblock choose_memblock_flags(void) 63a3f5bafcSTony Luck { 64a3f5bafcSTony Luck return system_has_some_mirror ? MEMBLOCK_MIRROR : MEMBLOCK_NONE; 65a3f5bafcSTony Luck } 66a3f5bafcSTony Luck 67142b45a7SBenjamin Herrenschmidt /* inline so we don't get a warning when pr_debug is compiled out */ 68c2233116SRaghavendra D Prabhu static __init_memblock const char * 69c2233116SRaghavendra D Prabhu memblock_type_name(struct memblock_type *type) 70142b45a7SBenjamin Herrenschmidt { 71142b45a7SBenjamin Herrenschmidt if (type == &memblock.memory) 72142b45a7SBenjamin Herrenschmidt return "memory"; 73142b45a7SBenjamin Herrenschmidt else if (type == &memblock.reserved) 74142b45a7SBenjamin Herrenschmidt return "reserved"; 757409c5f7SHeiko Carstens #ifdef CONFIG_HAVE_MEMBLOCK_PHYS_MAP 767409c5f7SHeiko Carstens else if (type == &memblock.physmem) 777409c5f7SHeiko Carstens return "physmem"; 787409c5f7SHeiko Carstens #endif 79142b45a7SBenjamin Herrenschmidt else 80142b45a7SBenjamin Herrenschmidt return "unknown"; 81142b45a7SBenjamin Herrenschmidt } 82142b45a7SBenjamin Herrenschmidt 83eb18f1b5STejun Heo /* adjust *@size so that (@base + *@size) doesn't overflow, return new size */ 84eb18f1b5STejun Heo static inline phys_addr_t memblock_cap_size(phys_addr_t base, phys_addr_t *size) 85eb18f1b5STejun Heo { 86eb18f1b5STejun Heo return *size = min(*size, (phys_addr_t)ULLONG_MAX - base); 87eb18f1b5STejun Heo } 88eb18f1b5STejun Heo 896ed311b2SBenjamin Herrenschmidt /* 906ed311b2SBenjamin Herrenschmidt * Address comparison utilities 916ed311b2SBenjamin Herrenschmidt */ 9210d06439SYinghai Lu static unsigned long __init_memblock memblock_addrs_overlap(phys_addr_t base1, phys_addr_t size1, 932898cc4cSBenjamin Herrenschmidt phys_addr_t base2, phys_addr_t size2) 9495f72d1eSYinghai Lu { 9595f72d1eSYinghai Lu return ((base1 < (base2 + size2)) && (base2 < (base1 + size1))); 9695f72d1eSYinghai Lu } 9795f72d1eSYinghai Lu 9895cf82ecSTang Chen bool __init_memblock memblock_overlaps_region(struct memblock_type *type, 992d7d3eb2SH Hartley Sweeten phys_addr_t base, phys_addr_t size) 1006ed311b2SBenjamin Herrenschmidt { 1016ed311b2SBenjamin Herrenschmidt unsigned long i; 1026ed311b2SBenjamin Herrenschmidt 103f14516fbSAlexander Kuleshov for (i = 0; i < type->cnt; i++) 104f14516fbSAlexander Kuleshov if (memblock_addrs_overlap(base, size, type->regions[i].base, 105f14516fbSAlexander Kuleshov type->regions[i].size)) 1066ed311b2SBenjamin Herrenschmidt break; 107c5c5c9d1STang Chen return i < type->cnt; 1086ed311b2SBenjamin Herrenschmidt } 1096ed311b2SBenjamin Herrenschmidt 11079442ed1STang Chen /* 11179442ed1STang Chen * __memblock_find_range_bottom_up - find free area utility in bottom-up 11279442ed1STang Chen * @start: start of candidate range 11379442ed1STang Chen * @end: end of candidate range, can be %MEMBLOCK_ALLOC_{ANYWHERE|ACCESSIBLE} 11479442ed1STang Chen * @size: size of free area to find 11579442ed1STang Chen * @align: alignment of free area to find 116b1154233SGrygorii Strashko * @nid: nid of the free area to find, %NUMA_NO_NODE for any node 117fc6daaf9STony Luck * @flags: pick from blocks based on memory attributes 11879442ed1STang Chen * 11979442ed1STang Chen * Utility called from memblock_find_in_range_node(), find free area bottom-up. 12079442ed1STang Chen * 12179442ed1STang Chen * RETURNS: 12279442ed1STang Chen * Found address on success, 0 on failure. 12379442ed1STang Chen */ 12479442ed1STang Chen static phys_addr_t __init_memblock 12579442ed1STang Chen __memblock_find_range_bottom_up(phys_addr_t start, phys_addr_t end, 126fc6daaf9STony Luck phys_addr_t size, phys_addr_t align, int nid, 127fc6daaf9STony Luck ulong flags) 12879442ed1STang Chen { 12979442ed1STang Chen phys_addr_t this_start, this_end, cand; 13079442ed1STang Chen u64 i; 13179442ed1STang Chen 132fc6daaf9STony Luck for_each_free_mem_range(i, nid, flags, &this_start, &this_end, NULL) { 13379442ed1STang Chen this_start = clamp(this_start, start, end); 13479442ed1STang Chen this_end = clamp(this_end, start, end); 13579442ed1STang Chen 13679442ed1STang Chen cand = round_up(this_start, align); 13779442ed1STang Chen if (cand < this_end && this_end - cand >= size) 13879442ed1STang Chen return cand; 13979442ed1STang Chen } 14079442ed1STang Chen 14179442ed1STang Chen return 0; 14279442ed1STang Chen } 14379442ed1STang Chen 1447bd0b0f0STejun Heo /** 1451402899eSTang Chen * __memblock_find_range_top_down - find free area utility, in top-down 1461402899eSTang Chen * @start: start of candidate range 1471402899eSTang Chen * @end: end of candidate range, can be %MEMBLOCK_ALLOC_{ANYWHERE|ACCESSIBLE} 1481402899eSTang Chen * @size: size of free area to find 1491402899eSTang Chen * @align: alignment of free area to find 150b1154233SGrygorii Strashko * @nid: nid of the free area to find, %NUMA_NO_NODE for any node 151fc6daaf9STony Luck * @flags: pick from blocks based on memory attributes 1521402899eSTang Chen * 1531402899eSTang Chen * Utility called from memblock_find_in_range_node(), find free area top-down. 1541402899eSTang Chen * 1551402899eSTang Chen * RETURNS: 15679442ed1STang Chen * Found address on success, 0 on failure. 1571402899eSTang Chen */ 1581402899eSTang Chen static phys_addr_t __init_memblock 1591402899eSTang Chen __memblock_find_range_top_down(phys_addr_t start, phys_addr_t end, 160fc6daaf9STony Luck phys_addr_t size, phys_addr_t align, int nid, 161fc6daaf9STony Luck ulong flags) 1621402899eSTang Chen { 1631402899eSTang Chen phys_addr_t this_start, this_end, cand; 1641402899eSTang Chen u64 i; 1651402899eSTang Chen 166fc6daaf9STony Luck for_each_free_mem_range_reverse(i, nid, flags, &this_start, &this_end, 167fc6daaf9STony Luck NULL) { 1681402899eSTang Chen this_start = clamp(this_start, start, end); 1691402899eSTang Chen this_end = clamp(this_end, start, end); 1701402899eSTang Chen 1711402899eSTang Chen if (this_end < size) 1721402899eSTang Chen continue; 1731402899eSTang Chen 1741402899eSTang Chen cand = round_down(this_end - size, align); 1751402899eSTang Chen if (cand >= this_start) 1761402899eSTang Chen return cand; 1771402899eSTang Chen } 1781402899eSTang Chen 1791402899eSTang Chen return 0; 1801402899eSTang Chen } 1811402899eSTang Chen 1821402899eSTang Chen /** 1837bd0b0f0STejun Heo * memblock_find_in_range_node - find free area in given range and node 1847bd0b0f0STejun Heo * @size: size of free area to find 1857bd0b0f0STejun Heo * @align: alignment of free area to find 18687029ee9SGrygorii Strashko * @start: start of candidate range 18787029ee9SGrygorii Strashko * @end: end of candidate range, can be %MEMBLOCK_ALLOC_{ANYWHERE|ACCESSIBLE} 188b1154233SGrygorii Strashko * @nid: nid of the free area to find, %NUMA_NO_NODE for any node 189fc6daaf9STony Luck * @flags: pick from blocks based on memory attributes 1907bd0b0f0STejun Heo * 1917bd0b0f0STejun Heo * Find @size free area aligned to @align in the specified range and node. 1927bd0b0f0STejun Heo * 19379442ed1STang Chen * When allocation direction is bottom-up, the @start should be greater 19479442ed1STang Chen * than the end of the kernel image. Otherwise, it will be trimmed. The 19579442ed1STang Chen * reason is that we want the bottom-up allocation just near the kernel 19679442ed1STang Chen * image so it is highly likely that the allocated memory and the kernel 19779442ed1STang Chen * will reside in the same node. 19879442ed1STang Chen * 19979442ed1STang Chen * If bottom-up allocation failed, will try to allocate memory top-down. 20079442ed1STang Chen * 2017bd0b0f0STejun Heo * RETURNS: 20279442ed1STang Chen * Found address on success, 0 on failure. 2036ed311b2SBenjamin Herrenschmidt */ 20487029ee9SGrygorii Strashko phys_addr_t __init_memblock memblock_find_in_range_node(phys_addr_t size, 20587029ee9SGrygorii Strashko phys_addr_t align, phys_addr_t start, 206fc6daaf9STony Luck phys_addr_t end, int nid, ulong flags) 207f7210e6cSTang Chen { 2080cfb8f0cSTang Chen phys_addr_t kernel_end, ret; 20979442ed1STang Chen 210f7210e6cSTang Chen /* pump up @end */ 211f7210e6cSTang Chen if (end == MEMBLOCK_ALLOC_ACCESSIBLE) 212f7210e6cSTang Chen end = memblock.current_limit; 213f7210e6cSTang Chen 214f7210e6cSTang Chen /* avoid allocating the first page */ 215f7210e6cSTang Chen start = max_t(phys_addr_t, start, PAGE_SIZE); 216f7210e6cSTang Chen end = max(start, end); 21779442ed1STang Chen kernel_end = __pa_symbol(_end); 21879442ed1STang Chen 21979442ed1STang Chen /* 22079442ed1STang Chen * try bottom-up allocation only when bottom-up mode 22179442ed1STang Chen * is set and @end is above the kernel image. 22279442ed1STang Chen */ 22379442ed1STang Chen if (memblock_bottom_up() && end > kernel_end) { 22479442ed1STang Chen phys_addr_t bottom_up_start; 22579442ed1STang Chen 22679442ed1STang Chen /* make sure we will allocate above the kernel */ 22779442ed1STang Chen bottom_up_start = max(start, kernel_end); 22879442ed1STang Chen 22979442ed1STang Chen /* ok, try bottom-up allocation first */ 23079442ed1STang Chen ret = __memblock_find_range_bottom_up(bottom_up_start, end, 231fc6daaf9STony Luck size, align, nid, flags); 23279442ed1STang Chen if (ret) 23379442ed1STang Chen return ret; 23479442ed1STang Chen 23579442ed1STang Chen /* 23679442ed1STang Chen * we always limit bottom-up allocation above the kernel, 23779442ed1STang Chen * but top-down allocation doesn't have the limit, so 23879442ed1STang Chen * retrying top-down allocation may succeed when bottom-up 23979442ed1STang Chen * allocation failed. 24079442ed1STang Chen * 24179442ed1STang Chen * bottom-up allocation is expected to be fail very rarely, 24279442ed1STang Chen * so we use WARN_ONCE() here to see the stack trace if 24379442ed1STang Chen * fail happens. 24479442ed1STang Chen */ 245756a025fSJoe Perches WARN_ONCE(1, "memblock: bottom-up allocation failed, memory hotunplug may be affected\n"); 24679442ed1STang Chen } 247f7210e6cSTang Chen 248fc6daaf9STony Luck return __memblock_find_range_top_down(start, end, size, align, nid, 249fc6daaf9STony Luck flags); 250f7210e6cSTang Chen } 2516ed311b2SBenjamin Herrenschmidt 2527bd0b0f0STejun Heo /** 2537bd0b0f0STejun Heo * memblock_find_in_range - find free area in given range 2547bd0b0f0STejun Heo * @start: start of candidate range 2557bd0b0f0STejun Heo * @end: end of candidate range, can be %MEMBLOCK_ALLOC_{ANYWHERE|ACCESSIBLE} 2567bd0b0f0STejun Heo * @size: size of free area to find 2577bd0b0f0STejun Heo * @align: alignment of free area to find 2587bd0b0f0STejun Heo * 2597bd0b0f0STejun Heo * Find @size free area aligned to @align in the specified range. 2607bd0b0f0STejun Heo * 2617bd0b0f0STejun Heo * RETURNS: 26279442ed1STang Chen * Found address on success, 0 on failure. 2637bd0b0f0STejun Heo */ 2647bd0b0f0STejun Heo phys_addr_t __init_memblock memblock_find_in_range(phys_addr_t start, 2657bd0b0f0STejun Heo phys_addr_t end, phys_addr_t size, 2667bd0b0f0STejun Heo phys_addr_t align) 2677bd0b0f0STejun Heo { 268a3f5bafcSTony Luck phys_addr_t ret; 269a3f5bafcSTony Luck ulong flags = choose_memblock_flags(); 270a3f5bafcSTony Luck 271a3f5bafcSTony Luck again: 272a3f5bafcSTony Luck ret = memblock_find_in_range_node(size, align, start, end, 273a3f5bafcSTony Luck NUMA_NO_NODE, flags); 274a3f5bafcSTony Luck 275a3f5bafcSTony Luck if (!ret && (flags & MEMBLOCK_MIRROR)) { 276a3f5bafcSTony Luck pr_warn("Could not allocate %pap bytes of mirrored memory\n", 277a3f5bafcSTony Luck &size); 278a3f5bafcSTony Luck flags &= ~MEMBLOCK_MIRROR; 279a3f5bafcSTony Luck goto again; 280a3f5bafcSTony Luck } 281a3f5bafcSTony Luck 282a3f5bafcSTony Luck return ret; 2837bd0b0f0STejun Heo } 2847bd0b0f0STejun Heo 28510d06439SYinghai Lu static void __init_memblock memblock_remove_region(struct memblock_type *type, unsigned long r) 28695f72d1eSYinghai Lu { 2871440c4e2STejun Heo type->total_size -= type->regions[r].size; 2887c0caeb8STejun Heo memmove(&type->regions[r], &type->regions[r + 1], 2897c0caeb8STejun Heo (type->cnt - (r + 1)) * sizeof(type->regions[r])); 290e3239ff9SBenjamin Herrenschmidt type->cnt--; 29195f72d1eSYinghai Lu 2928f7a6605SBenjamin Herrenschmidt /* Special case for empty arrays */ 2938f7a6605SBenjamin Herrenschmidt if (type->cnt == 0) { 2941440c4e2STejun Heo WARN_ON(type->total_size != 0); 2958f7a6605SBenjamin Herrenschmidt type->cnt = 1; 2968f7a6605SBenjamin Herrenschmidt type->regions[0].base = 0; 2978f7a6605SBenjamin Herrenschmidt type->regions[0].size = 0; 29866a20757STang Chen type->regions[0].flags = 0; 2997c0caeb8STejun Heo memblock_set_region_node(&type->regions[0], MAX_NUMNODES); 3008f7a6605SBenjamin Herrenschmidt } 30195f72d1eSYinghai Lu } 30295f72d1eSYinghai Lu 303354f17e1SPhilipp Hachtmann #ifdef CONFIG_ARCH_DISCARD_MEMBLOCK 304354f17e1SPhilipp Hachtmann 30529f67386SYinghai Lu phys_addr_t __init_memblock get_allocated_memblock_reserved_regions_info( 30629f67386SYinghai Lu phys_addr_t *addr) 30729f67386SYinghai Lu { 30829f67386SYinghai Lu if (memblock.reserved.regions == memblock_reserved_init_regions) 30929f67386SYinghai Lu return 0; 31029f67386SYinghai Lu 31129f67386SYinghai Lu *addr = __pa(memblock.reserved.regions); 31229f67386SYinghai Lu 31329f67386SYinghai Lu return PAGE_ALIGN(sizeof(struct memblock_region) * 31429f67386SYinghai Lu memblock.reserved.max); 31529f67386SYinghai Lu } 31629f67386SYinghai Lu 3175e270e25SPhilipp Hachtmann phys_addr_t __init_memblock get_allocated_memblock_memory_regions_info( 3185e270e25SPhilipp Hachtmann phys_addr_t *addr) 3195e270e25SPhilipp Hachtmann { 3205e270e25SPhilipp Hachtmann if (memblock.memory.regions == memblock_memory_init_regions) 3215e270e25SPhilipp Hachtmann return 0; 3225e270e25SPhilipp Hachtmann 3235e270e25SPhilipp Hachtmann *addr = __pa(memblock.memory.regions); 3245e270e25SPhilipp Hachtmann 3255e270e25SPhilipp Hachtmann return PAGE_ALIGN(sizeof(struct memblock_region) * 3265e270e25SPhilipp Hachtmann memblock.memory.max); 3275e270e25SPhilipp Hachtmann } 3285e270e25SPhilipp Hachtmann 3295e270e25SPhilipp Hachtmann #endif 3305e270e25SPhilipp Hachtmann 33148c3b583SGreg Pearson /** 33248c3b583SGreg Pearson * memblock_double_array - double the size of the memblock regions array 33348c3b583SGreg Pearson * @type: memblock type of the regions array being doubled 33448c3b583SGreg Pearson * @new_area_start: starting address of memory range to avoid overlap with 33548c3b583SGreg Pearson * @new_area_size: size of memory range to avoid overlap with 33648c3b583SGreg Pearson * 33748c3b583SGreg Pearson * Double the size of the @type regions array. If memblock is being used to 33848c3b583SGreg Pearson * allocate memory for a new reserved regions array and there is a previously 33948c3b583SGreg Pearson * allocated memory range [@new_area_start,@new_area_start+@new_area_size] 34048c3b583SGreg Pearson * waiting to be reserved, ensure the memory used by the new array does 34148c3b583SGreg Pearson * not overlap. 34248c3b583SGreg Pearson * 34348c3b583SGreg Pearson * RETURNS: 34448c3b583SGreg Pearson * 0 on success, -1 on failure. 34548c3b583SGreg Pearson */ 34648c3b583SGreg Pearson static int __init_memblock memblock_double_array(struct memblock_type *type, 34748c3b583SGreg Pearson phys_addr_t new_area_start, 34848c3b583SGreg Pearson phys_addr_t new_area_size) 349142b45a7SBenjamin Herrenschmidt { 350142b45a7SBenjamin Herrenschmidt struct memblock_region *new_array, *old_array; 35129f67386SYinghai Lu phys_addr_t old_alloc_size, new_alloc_size; 352142b45a7SBenjamin Herrenschmidt phys_addr_t old_size, new_size, addr; 353142b45a7SBenjamin Herrenschmidt int use_slab = slab_is_available(); 354181eb394SGavin Shan int *in_slab; 355142b45a7SBenjamin Herrenschmidt 356142b45a7SBenjamin Herrenschmidt /* We don't allow resizing until we know about the reserved regions 357142b45a7SBenjamin Herrenschmidt * of memory that aren't suitable for allocation 358142b45a7SBenjamin Herrenschmidt */ 359142b45a7SBenjamin Herrenschmidt if (!memblock_can_resize) 360142b45a7SBenjamin Herrenschmidt return -1; 361142b45a7SBenjamin Herrenschmidt 362142b45a7SBenjamin Herrenschmidt /* Calculate new doubled size */ 363142b45a7SBenjamin Herrenschmidt old_size = type->max * sizeof(struct memblock_region); 364142b45a7SBenjamin Herrenschmidt new_size = old_size << 1; 36529f67386SYinghai Lu /* 36629f67386SYinghai Lu * We need to allocated new one align to PAGE_SIZE, 36729f67386SYinghai Lu * so we can free them completely later. 36829f67386SYinghai Lu */ 36929f67386SYinghai Lu old_alloc_size = PAGE_ALIGN(old_size); 37029f67386SYinghai Lu new_alloc_size = PAGE_ALIGN(new_size); 371142b45a7SBenjamin Herrenschmidt 372181eb394SGavin Shan /* Retrieve the slab flag */ 373181eb394SGavin Shan if (type == &memblock.memory) 374181eb394SGavin Shan in_slab = &memblock_memory_in_slab; 375181eb394SGavin Shan else 376181eb394SGavin Shan in_slab = &memblock_reserved_in_slab; 377181eb394SGavin Shan 378142b45a7SBenjamin Herrenschmidt /* Try to find some space for it. 379142b45a7SBenjamin Herrenschmidt * 380142b45a7SBenjamin Herrenschmidt * WARNING: We assume that either slab_is_available() and we use it or 381fd07383bSAndrew Morton * we use MEMBLOCK for allocations. That means that this is unsafe to 382fd07383bSAndrew Morton * use when bootmem is currently active (unless bootmem itself is 383fd07383bSAndrew Morton * implemented on top of MEMBLOCK which isn't the case yet) 384142b45a7SBenjamin Herrenschmidt * 385142b45a7SBenjamin Herrenschmidt * This should however not be an issue for now, as we currently only 386fd07383bSAndrew Morton * call into MEMBLOCK while it's still active, or much later when slab 387fd07383bSAndrew Morton * is active for memory hotplug operations 388142b45a7SBenjamin Herrenschmidt */ 389142b45a7SBenjamin Herrenschmidt if (use_slab) { 390142b45a7SBenjamin Herrenschmidt new_array = kmalloc(new_size, GFP_KERNEL); 3911f5026a7STejun Heo addr = new_array ? __pa(new_array) : 0; 3924e2f0775SGavin Shan } else { 39348c3b583SGreg Pearson /* only exclude range when trying to double reserved.regions */ 39448c3b583SGreg Pearson if (type != &memblock.reserved) 39548c3b583SGreg Pearson new_area_start = new_area_size = 0; 39648c3b583SGreg Pearson 39748c3b583SGreg Pearson addr = memblock_find_in_range(new_area_start + new_area_size, 39848c3b583SGreg Pearson memblock.current_limit, 39929f67386SYinghai Lu new_alloc_size, PAGE_SIZE); 40048c3b583SGreg Pearson if (!addr && new_area_size) 40148c3b583SGreg Pearson addr = memblock_find_in_range(0, 40248c3b583SGreg Pearson min(new_area_start, memblock.current_limit), 40329f67386SYinghai Lu new_alloc_size, PAGE_SIZE); 40448c3b583SGreg Pearson 40515674868SSachin Kamat new_array = addr ? __va(addr) : NULL; 4064e2f0775SGavin Shan } 4071f5026a7STejun Heo if (!addr) { 408142b45a7SBenjamin Herrenschmidt pr_err("memblock: Failed to double %s array from %ld to %ld entries !\n", 409142b45a7SBenjamin Herrenschmidt memblock_type_name(type), type->max, type->max * 2); 410142b45a7SBenjamin Herrenschmidt return -1; 411142b45a7SBenjamin Herrenschmidt } 412142b45a7SBenjamin Herrenschmidt 413fd07383bSAndrew Morton memblock_dbg("memblock: %s is doubled to %ld at [%#010llx-%#010llx]", 414fd07383bSAndrew Morton memblock_type_name(type), type->max * 2, (u64)addr, 415fd07383bSAndrew Morton (u64)addr + new_size - 1); 416ea9e4376SYinghai Lu 417fd07383bSAndrew Morton /* 418fd07383bSAndrew Morton * Found space, we now need to move the array over before we add the 419fd07383bSAndrew Morton * reserved region since it may be our reserved array itself that is 420fd07383bSAndrew Morton * full. 421142b45a7SBenjamin Herrenschmidt */ 422142b45a7SBenjamin Herrenschmidt memcpy(new_array, type->regions, old_size); 423142b45a7SBenjamin Herrenschmidt memset(new_array + type->max, 0, old_size); 424142b45a7SBenjamin Herrenschmidt old_array = type->regions; 425142b45a7SBenjamin Herrenschmidt type->regions = new_array; 426142b45a7SBenjamin Herrenschmidt type->max <<= 1; 427142b45a7SBenjamin Herrenschmidt 428fd07383bSAndrew Morton /* Free old array. We needn't free it if the array is the static one */ 429181eb394SGavin Shan if (*in_slab) 430181eb394SGavin Shan kfree(old_array); 431181eb394SGavin Shan else if (old_array != memblock_memory_init_regions && 432142b45a7SBenjamin Herrenschmidt old_array != memblock_reserved_init_regions) 43329f67386SYinghai Lu memblock_free(__pa(old_array), old_alloc_size); 434142b45a7SBenjamin Herrenschmidt 435fd07383bSAndrew Morton /* 436fd07383bSAndrew Morton * Reserve the new array if that comes from the memblock. Otherwise, we 437fd07383bSAndrew Morton * needn't do it 438181eb394SGavin Shan */ 439181eb394SGavin Shan if (!use_slab) 44029f67386SYinghai Lu BUG_ON(memblock_reserve(addr, new_alloc_size)); 441181eb394SGavin Shan 442181eb394SGavin Shan /* Update slab flag */ 443181eb394SGavin Shan *in_slab = use_slab; 444181eb394SGavin Shan 445142b45a7SBenjamin Herrenschmidt return 0; 446142b45a7SBenjamin Herrenschmidt } 447142b45a7SBenjamin Herrenschmidt 448784656f9STejun Heo /** 449784656f9STejun Heo * memblock_merge_regions - merge neighboring compatible regions 450784656f9STejun Heo * @type: memblock type to scan 451784656f9STejun Heo * 452784656f9STejun Heo * Scan @type and merge neighboring compatible regions. 453784656f9STejun Heo */ 454784656f9STejun Heo static void __init_memblock memblock_merge_regions(struct memblock_type *type) 455784656f9STejun Heo { 456784656f9STejun Heo int i = 0; 457784656f9STejun Heo 458784656f9STejun Heo /* cnt never goes below 1 */ 459784656f9STejun Heo while (i < type->cnt - 1) { 460784656f9STejun Heo struct memblock_region *this = &type->regions[i]; 461784656f9STejun Heo struct memblock_region *next = &type->regions[i + 1]; 462784656f9STejun Heo 4637c0caeb8STejun Heo if (this->base + this->size != next->base || 4647c0caeb8STejun Heo memblock_get_region_node(this) != 46566a20757STang Chen memblock_get_region_node(next) || 46666a20757STang Chen this->flags != next->flags) { 467784656f9STejun Heo BUG_ON(this->base + this->size > next->base); 468784656f9STejun Heo i++; 469784656f9STejun Heo continue; 470784656f9STejun Heo } 471784656f9STejun Heo 472784656f9STejun Heo this->size += next->size; 473c0232ae8SLin Feng /* move forward from next + 1, index of which is i + 2 */ 474c0232ae8SLin Feng memmove(next, next + 1, (type->cnt - (i + 2)) * sizeof(*next)); 475784656f9STejun Heo type->cnt--; 476784656f9STejun Heo } 477784656f9STejun Heo } 478784656f9STejun Heo 479784656f9STejun Heo /** 480784656f9STejun Heo * memblock_insert_region - insert new memblock region 481784656f9STejun Heo * @type: memblock type to insert into 482784656f9STejun Heo * @idx: index for the insertion point 483784656f9STejun Heo * @base: base address of the new region 484784656f9STejun Heo * @size: size of the new region 485209ff86dSTang Chen * @nid: node id of the new region 48666a20757STang Chen * @flags: flags of the new region 487784656f9STejun Heo * 488784656f9STejun Heo * Insert new memblock region [@base,@base+@size) into @type at @idx. 489412d0008SAlexander Kuleshov * @type must already have extra room to accommodate the new region. 490784656f9STejun Heo */ 491784656f9STejun Heo static void __init_memblock memblock_insert_region(struct memblock_type *type, 492784656f9STejun Heo int idx, phys_addr_t base, 49366a20757STang Chen phys_addr_t size, 49466a20757STang Chen int nid, unsigned long flags) 495784656f9STejun Heo { 496784656f9STejun Heo struct memblock_region *rgn = &type->regions[idx]; 497784656f9STejun Heo 498784656f9STejun Heo BUG_ON(type->cnt >= type->max); 499784656f9STejun Heo memmove(rgn + 1, rgn, (type->cnt - idx) * sizeof(*rgn)); 500784656f9STejun Heo rgn->base = base; 501784656f9STejun Heo rgn->size = size; 50266a20757STang Chen rgn->flags = flags; 5037c0caeb8STejun Heo memblock_set_region_node(rgn, nid); 504784656f9STejun Heo type->cnt++; 5051440c4e2STejun Heo type->total_size += size; 506784656f9STejun Heo } 507784656f9STejun Heo 508784656f9STejun Heo /** 509f1af9d3aSPhilipp Hachtmann * memblock_add_range - add new memblock region 510784656f9STejun Heo * @type: memblock type to add new region into 511784656f9STejun Heo * @base: base address of the new region 512784656f9STejun Heo * @size: size of the new region 5137fb0bc3fSTejun Heo * @nid: nid of the new region 51466a20757STang Chen * @flags: flags of the new region 515784656f9STejun Heo * 516784656f9STejun Heo * Add new memblock region [@base,@base+@size) into @type. The new region 517784656f9STejun Heo * is allowed to overlap with existing ones - overlaps don't affect already 518784656f9STejun Heo * existing regions. @type is guaranteed to be minimal (all neighbouring 519784656f9STejun Heo * compatible regions are merged) after the addition. 520784656f9STejun Heo * 521784656f9STejun Heo * RETURNS: 522784656f9STejun Heo * 0 on success, -errno on failure. 523784656f9STejun Heo */ 524f1af9d3aSPhilipp Hachtmann int __init_memblock memblock_add_range(struct memblock_type *type, 52566a20757STang Chen phys_addr_t base, phys_addr_t size, 52666a20757STang Chen int nid, unsigned long flags) 52795f72d1eSYinghai Lu { 528784656f9STejun Heo bool insert = false; 529eb18f1b5STejun Heo phys_addr_t obase = base; 530eb18f1b5STejun Heo phys_addr_t end = base + memblock_cap_size(base, &size); 5318c9c1701SAlexander Kuleshov int idx, nr_new; 5328c9c1701SAlexander Kuleshov struct memblock_region *rgn; 53395f72d1eSYinghai Lu 534b3dc627cSTejun Heo if (!size) 535b3dc627cSTejun Heo return 0; 536b3dc627cSTejun Heo 537784656f9STejun Heo /* special case for empty array */ 538784656f9STejun Heo if (type->regions[0].size == 0) { 5391440c4e2STejun Heo WARN_ON(type->cnt != 1 || type->total_size); 540784656f9STejun Heo type->regions[0].base = base; 541784656f9STejun Heo type->regions[0].size = size; 54266a20757STang Chen type->regions[0].flags = flags; 5437fb0bc3fSTejun Heo memblock_set_region_node(&type->regions[0], nid); 5441440c4e2STejun Heo type->total_size = size; 545784656f9STejun Heo return 0; 546784656f9STejun Heo } 547784656f9STejun Heo repeat: 548784656f9STejun Heo /* 549784656f9STejun Heo * The following is executed twice. Once with %false @insert and 550784656f9STejun Heo * then with %true. The first counts the number of regions needed 551412d0008SAlexander Kuleshov * to accommodate the new area. The second actually inserts them. 552784656f9STejun Heo */ 553784656f9STejun Heo base = obase; 554784656f9STejun Heo nr_new = 0; 555784656f9STejun Heo 5568c9c1701SAlexander Kuleshov for_each_memblock_type(type, rgn) { 557784656f9STejun Heo phys_addr_t rbase = rgn->base; 558784656f9STejun Heo phys_addr_t rend = rbase + rgn->size; 5598f7a6605SBenjamin Herrenschmidt 560784656f9STejun Heo if (rbase >= end) 5618f7a6605SBenjamin Herrenschmidt break; 562784656f9STejun Heo if (rend <= base) 563784656f9STejun Heo continue; 564784656f9STejun Heo /* 565784656f9STejun Heo * @rgn overlaps. If it separates the lower part of new 566784656f9STejun Heo * area, insert that portion. 5678f7a6605SBenjamin Herrenschmidt */ 568784656f9STejun Heo if (rbase > base) { 569c0a29498SWei Yang #ifdef CONFIG_HAVE_MEMBLOCK_NODE_MAP 570c0a29498SWei Yang WARN_ON(nid != memblock_get_region_node(rgn)); 571c0a29498SWei Yang #endif 5724fcab5f4SWei Yang WARN_ON(flags != rgn->flags); 573784656f9STejun Heo nr_new++; 574784656f9STejun Heo if (insert) 5758c9c1701SAlexander Kuleshov memblock_insert_region(type, idx++, base, 57666a20757STang Chen rbase - base, nid, 57766a20757STang Chen flags); 578784656f9STejun Heo } 579784656f9STejun Heo /* area below @rend is dealt with, forget about it */ 580784656f9STejun Heo base = min(rend, end); 5818f7a6605SBenjamin Herrenschmidt } 5828f7a6605SBenjamin Herrenschmidt 583784656f9STejun Heo /* insert the remaining portion */ 584784656f9STejun Heo if (base < end) { 585784656f9STejun Heo nr_new++; 586784656f9STejun Heo if (insert) 5878c9c1701SAlexander Kuleshov memblock_insert_region(type, idx, base, end - base, 58866a20757STang Chen nid, flags); 5898f7a6605SBenjamin Herrenschmidt } 5908f7a6605SBenjamin Herrenschmidt 591ef3cc4dbSnimisolo if (!nr_new) 592ef3cc4dbSnimisolo return 0; 593ef3cc4dbSnimisolo 594784656f9STejun Heo /* 595784656f9STejun Heo * If this was the first round, resize array and repeat for actual 596784656f9STejun Heo * insertions; otherwise, merge and return. 5978f7a6605SBenjamin Herrenschmidt */ 598784656f9STejun Heo if (!insert) { 599784656f9STejun Heo while (type->cnt + nr_new > type->max) 60048c3b583SGreg Pearson if (memblock_double_array(type, obase, size) < 0) 601784656f9STejun Heo return -ENOMEM; 602784656f9STejun Heo insert = true; 603784656f9STejun Heo goto repeat; 60495f72d1eSYinghai Lu } else { 605784656f9STejun Heo memblock_merge_regions(type); 60695f72d1eSYinghai Lu return 0; 60795f72d1eSYinghai Lu } 608784656f9STejun Heo } 60995f72d1eSYinghai Lu 6107fb0bc3fSTejun Heo int __init_memblock memblock_add_node(phys_addr_t base, phys_addr_t size, 6117fb0bc3fSTejun Heo int nid) 6127fb0bc3fSTejun Heo { 613f1af9d3aSPhilipp Hachtmann return memblock_add_range(&memblock.memory, base, size, nid, 0); 6147fb0bc3fSTejun Heo } 6157fb0bc3fSTejun Heo 616f705ac4bSAlexander Kuleshov int __init_memblock memblock_add(phys_addr_t base, phys_addr_t size) 6176a4055bcSAlexander Kuleshov { 6185d63f81cSMiles Chen phys_addr_t end = base + size - 1; 6195d63f81cSMiles Chen 6205d63f81cSMiles Chen memblock_dbg("memblock_add: [%pa-%pa] %pF\n", 6215d63f81cSMiles Chen &base, &end, (void *)_RET_IP_); 6226a4055bcSAlexander Kuleshov 623f705ac4bSAlexander Kuleshov return memblock_add_range(&memblock.memory, base, size, MAX_NUMNODES, 0); 62495f72d1eSYinghai Lu } 62595f72d1eSYinghai Lu 6266a9ceb31STejun Heo /** 6276a9ceb31STejun Heo * memblock_isolate_range - isolate given range into disjoint memblocks 6286a9ceb31STejun Heo * @type: memblock type to isolate range for 6296a9ceb31STejun Heo * @base: base of range to isolate 6306a9ceb31STejun Heo * @size: size of range to isolate 6316a9ceb31STejun Heo * @start_rgn: out parameter for the start of isolated region 6326a9ceb31STejun Heo * @end_rgn: out parameter for the end of isolated region 6336a9ceb31STejun Heo * 6346a9ceb31STejun Heo * Walk @type and ensure that regions don't cross the boundaries defined by 6356a9ceb31STejun Heo * [@base,@base+@size). Crossing regions are split at the boundaries, 6366a9ceb31STejun Heo * which may create at most two more regions. The index of the first 6376a9ceb31STejun Heo * region inside the range is returned in *@start_rgn and end in *@end_rgn. 6386a9ceb31STejun Heo * 6396a9ceb31STejun Heo * RETURNS: 6406a9ceb31STejun Heo * 0 on success, -errno on failure. 6416a9ceb31STejun Heo */ 6426a9ceb31STejun Heo static int __init_memblock memblock_isolate_range(struct memblock_type *type, 6436a9ceb31STejun Heo phys_addr_t base, phys_addr_t size, 6446a9ceb31STejun Heo int *start_rgn, int *end_rgn) 6456a9ceb31STejun Heo { 646eb18f1b5STejun Heo phys_addr_t end = base + memblock_cap_size(base, &size); 6478c9c1701SAlexander Kuleshov int idx; 6488c9c1701SAlexander Kuleshov struct memblock_region *rgn; 6496a9ceb31STejun Heo 6506a9ceb31STejun Heo *start_rgn = *end_rgn = 0; 6516a9ceb31STejun Heo 652b3dc627cSTejun Heo if (!size) 653b3dc627cSTejun Heo return 0; 654b3dc627cSTejun Heo 6556a9ceb31STejun Heo /* we'll create at most two more regions */ 6566a9ceb31STejun Heo while (type->cnt + 2 > type->max) 65748c3b583SGreg Pearson if (memblock_double_array(type, base, size) < 0) 6586a9ceb31STejun Heo return -ENOMEM; 6596a9ceb31STejun Heo 6608c9c1701SAlexander Kuleshov for_each_memblock_type(type, rgn) { 6616a9ceb31STejun Heo phys_addr_t rbase = rgn->base; 6626a9ceb31STejun Heo phys_addr_t rend = rbase + rgn->size; 6636a9ceb31STejun Heo 6646a9ceb31STejun Heo if (rbase >= end) 6656a9ceb31STejun Heo break; 6666a9ceb31STejun Heo if (rend <= base) 6676a9ceb31STejun Heo continue; 6686a9ceb31STejun Heo 6696a9ceb31STejun Heo if (rbase < base) { 6706a9ceb31STejun Heo /* 6716a9ceb31STejun Heo * @rgn intersects from below. Split and continue 6726a9ceb31STejun Heo * to process the next region - the new top half. 6736a9ceb31STejun Heo */ 6746a9ceb31STejun Heo rgn->base = base; 6751440c4e2STejun Heo rgn->size -= base - rbase; 6761440c4e2STejun Heo type->total_size -= base - rbase; 6778c9c1701SAlexander Kuleshov memblock_insert_region(type, idx, rbase, base - rbase, 67866a20757STang Chen memblock_get_region_node(rgn), 67966a20757STang Chen rgn->flags); 6806a9ceb31STejun Heo } else if (rend > end) { 6816a9ceb31STejun Heo /* 6826a9ceb31STejun Heo * @rgn intersects from above. Split and redo the 6836a9ceb31STejun Heo * current region - the new bottom half. 6846a9ceb31STejun Heo */ 6856a9ceb31STejun Heo rgn->base = end; 6861440c4e2STejun Heo rgn->size -= end - rbase; 6871440c4e2STejun Heo type->total_size -= end - rbase; 6888c9c1701SAlexander Kuleshov memblock_insert_region(type, idx--, rbase, end - rbase, 68966a20757STang Chen memblock_get_region_node(rgn), 69066a20757STang Chen rgn->flags); 6916a9ceb31STejun Heo } else { 6926a9ceb31STejun Heo /* @rgn is fully contained, record it */ 6936a9ceb31STejun Heo if (!*end_rgn) 6948c9c1701SAlexander Kuleshov *start_rgn = idx; 6958c9c1701SAlexander Kuleshov *end_rgn = idx + 1; 6966a9ceb31STejun Heo } 6976a9ceb31STejun Heo } 6986a9ceb31STejun Heo 6996a9ceb31STejun Heo return 0; 7006a9ceb31STejun Heo } 7016a9ceb31STejun Heo 70235bd16a2SAlexander Kuleshov static int __init_memblock memblock_remove_range(struct memblock_type *type, 7038f7a6605SBenjamin Herrenschmidt phys_addr_t base, phys_addr_t size) 70495f72d1eSYinghai Lu { 70571936180STejun Heo int start_rgn, end_rgn; 70671936180STejun Heo int i, ret; 70795f72d1eSYinghai Lu 70871936180STejun Heo ret = memblock_isolate_range(type, base, size, &start_rgn, &end_rgn); 70971936180STejun Heo if (ret) 71071936180STejun Heo return ret; 71195f72d1eSYinghai Lu 71271936180STejun Heo for (i = end_rgn - 1; i >= start_rgn; i--) 71371936180STejun Heo memblock_remove_region(type, i); 71495f72d1eSYinghai Lu return 0; 71595f72d1eSYinghai Lu } 71695f72d1eSYinghai Lu 717581adcbeSTejun Heo int __init_memblock memblock_remove(phys_addr_t base, phys_addr_t size) 71895f72d1eSYinghai Lu { 719f1af9d3aSPhilipp Hachtmann return memblock_remove_range(&memblock.memory, base, size); 72095f72d1eSYinghai Lu } 72195f72d1eSYinghai Lu 722f1af9d3aSPhilipp Hachtmann 723581adcbeSTejun Heo int __init_memblock memblock_free(phys_addr_t base, phys_addr_t size) 72495f72d1eSYinghai Lu { 7255d63f81cSMiles Chen phys_addr_t end = base + size - 1; 7265d63f81cSMiles Chen 7275d63f81cSMiles Chen memblock_dbg(" memblock_free: [%pa-%pa] %pF\n", 7285d63f81cSMiles Chen &base, &end, (void *)_RET_IP_); 72924aa0788STejun Heo 7309099daedSCatalin Marinas kmemleak_free_part_phys(base, size); 731f1af9d3aSPhilipp Hachtmann return memblock_remove_range(&memblock.reserved, base, size); 73295f72d1eSYinghai Lu } 73395f72d1eSYinghai Lu 734f705ac4bSAlexander Kuleshov int __init_memblock memblock_reserve(phys_addr_t base, phys_addr_t size) 73595f72d1eSYinghai Lu { 7365d63f81cSMiles Chen phys_addr_t end = base + size - 1; 7375d63f81cSMiles Chen 7385d63f81cSMiles Chen memblock_dbg("memblock_reserve: [%pa-%pa] %pF\n", 7395d63f81cSMiles Chen &base, &end, (void *)_RET_IP_); 74095f72d1eSYinghai Lu 741f705ac4bSAlexander Kuleshov return memblock_add_range(&memblock.reserved, base, size, MAX_NUMNODES, 0); 74295f72d1eSYinghai Lu } 74395f72d1eSYinghai Lu 74435fd0808STejun Heo /** 74566b16edfSTang Chen * 7464308ce17STony Luck * This function isolates region [@base, @base + @size), and sets/clears flag 74766b16edfSTang Chen * 748c1153931SAlexander Kuleshov * Return 0 on success, -errno on failure. 74966b16edfSTang Chen */ 7504308ce17STony Luck static int __init_memblock memblock_setclr_flag(phys_addr_t base, 7514308ce17STony Luck phys_addr_t size, int set, int flag) 75266b16edfSTang Chen { 75366b16edfSTang Chen struct memblock_type *type = &memblock.memory; 75466b16edfSTang Chen int i, ret, start_rgn, end_rgn; 75566b16edfSTang Chen 75666b16edfSTang Chen ret = memblock_isolate_range(type, base, size, &start_rgn, &end_rgn); 75766b16edfSTang Chen if (ret) 75866b16edfSTang Chen return ret; 75966b16edfSTang Chen 76066b16edfSTang Chen for (i = start_rgn; i < end_rgn; i++) 7614308ce17STony Luck if (set) 7624308ce17STony Luck memblock_set_region_flags(&type->regions[i], flag); 7634308ce17STony Luck else 7644308ce17STony Luck memblock_clear_region_flags(&type->regions[i], flag); 76566b16edfSTang Chen 76666b16edfSTang Chen memblock_merge_regions(type); 76766b16edfSTang Chen return 0; 76866b16edfSTang Chen } 76966b16edfSTang Chen 77066b16edfSTang Chen /** 7714308ce17STony Luck * memblock_mark_hotplug - Mark hotpluggable memory with flag MEMBLOCK_HOTPLUG. 7724308ce17STony Luck * @base: the base phys addr of the region 7734308ce17STony Luck * @size: the size of the region 7744308ce17STony Luck * 775c1153931SAlexander Kuleshov * Return 0 on success, -errno on failure. 7764308ce17STony Luck */ 7774308ce17STony Luck int __init_memblock memblock_mark_hotplug(phys_addr_t base, phys_addr_t size) 7784308ce17STony Luck { 7794308ce17STony Luck return memblock_setclr_flag(base, size, 1, MEMBLOCK_HOTPLUG); 7804308ce17STony Luck } 7814308ce17STony Luck 7824308ce17STony Luck /** 78366b16edfSTang Chen * memblock_clear_hotplug - Clear flag MEMBLOCK_HOTPLUG for a specified region. 78466b16edfSTang Chen * @base: the base phys addr of the region 78566b16edfSTang Chen * @size: the size of the region 78666b16edfSTang Chen * 787c1153931SAlexander Kuleshov * Return 0 on success, -errno on failure. 78866b16edfSTang Chen */ 78966b16edfSTang Chen int __init_memblock memblock_clear_hotplug(phys_addr_t base, phys_addr_t size) 79066b16edfSTang Chen { 7914308ce17STony Luck return memblock_setclr_flag(base, size, 0, MEMBLOCK_HOTPLUG); 79266b16edfSTang Chen } 79366b16edfSTang Chen 79466b16edfSTang Chen /** 795a3f5bafcSTony Luck * memblock_mark_mirror - Mark mirrored memory with flag MEMBLOCK_MIRROR. 796a3f5bafcSTony Luck * @base: the base phys addr of the region 797a3f5bafcSTony Luck * @size: the size of the region 798a3f5bafcSTony Luck * 799c1153931SAlexander Kuleshov * Return 0 on success, -errno on failure. 800a3f5bafcSTony Luck */ 801a3f5bafcSTony Luck int __init_memblock memblock_mark_mirror(phys_addr_t base, phys_addr_t size) 802a3f5bafcSTony Luck { 803a3f5bafcSTony Luck system_has_some_mirror = true; 804a3f5bafcSTony Luck 805a3f5bafcSTony Luck return memblock_setclr_flag(base, size, 1, MEMBLOCK_MIRROR); 806a3f5bafcSTony Luck } 807a3f5bafcSTony Luck 808bf3d3cc5SArd Biesheuvel /** 809bf3d3cc5SArd Biesheuvel * memblock_mark_nomap - Mark a memory region with flag MEMBLOCK_NOMAP. 810bf3d3cc5SArd Biesheuvel * @base: the base phys addr of the region 811bf3d3cc5SArd Biesheuvel * @size: the size of the region 812bf3d3cc5SArd Biesheuvel * 813bf3d3cc5SArd Biesheuvel * Return 0 on success, -errno on failure. 814bf3d3cc5SArd Biesheuvel */ 815bf3d3cc5SArd Biesheuvel int __init_memblock memblock_mark_nomap(phys_addr_t base, phys_addr_t size) 816bf3d3cc5SArd Biesheuvel { 817bf3d3cc5SArd Biesheuvel return memblock_setclr_flag(base, size, 1, MEMBLOCK_NOMAP); 818bf3d3cc5SArd Biesheuvel } 819a3f5bafcSTony Luck 820a3f5bafcSTony Luck /** 8218e7a7f86SRobin Holt * __next_reserved_mem_region - next function for for_each_reserved_region() 8228e7a7f86SRobin Holt * @idx: pointer to u64 loop variable 8238e7a7f86SRobin Holt * @out_start: ptr to phys_addr_t for start address of the region, can be %NULL 8248e7a7f86SRobin Holt * @out_end: ptr to phys_addr_t for end address of the region, can be %NULL 8258e7a7f86SRobin Holt * 8268e7a7f86SRobin Holt * Iterate over all reserved memory regions. 8278e7a7f86SRobin Holt */ 8288e7a7f86SRobin Holt void __init_memblock __next_reserved_mem_region(u64 *idx, 8298e7a7f86SRobin Holt phys_addr_t *out_start, 8308e7a7f86SRobin Holt phys_addr_t *out_end) 8318e7a7f86SRobin Holt { 832567d117bSAlexander Kuleshov struct memblock_type *type = &memblock.reserved; 8338e7a7f86SRobin Holt 834cd33a76bSRichard Leitner if (*idx < type->cnt) { 835567d117bSAlexander Kuleshov struct memblock_region *r = &type->regions[*idx]; 8368e7a7f86SRobin Holt phys_addr_t base = r->base; 8378e7a7f86SRobin Holt phys_addr_t size = r->size; 8388e7a7f86SRobin Holt 8398e7a7f86SRobin Holt if (out_start) 8408e7a7f86SRobin Holt *out_start = base; 8418e7a7f86SRobin Holt if (out_end) 8428e7a7f86SRobin Holt *out_end = base + size - 1; 8438e7a7f86SRobin Holt 8448e7a7f86SRobin Holt *idx += 1; 8458e7a7f86SRobin Holt return; 8468e7a7f86SRobin Holt } 8478e7a7f86SRobin Holt 8488e7a7f86SRobin Holt /* signal end of iteration */ 8498e7a7f86SRobin Holt *idx = ULLONG_MAX; 8508e7a7f86SRobin Holt } 8518e7a7f86SRobin Holt 8528e7a7f86SRobin Holt /** 853f1af9d3aSPhilipp Hachtmann * __next__mem_range - next function for for_each_free_mem_range() etc. 85435fd0808STejun Heo * @idx: pointer to u64 loop variable 855b1154233SGrygorii Strashko * @nid: node selector, %NUMA_NO_NODE for all nodes 856fc6daaf9STony Luck * @flags: pick from blocks based on memory attributes 857f1af9d3aSPhilipp Hachtmann * @type_a: pointer to memblock_type from where the range is taken 858f1af9d3aSPhilipp Hachtmann * @type_b: pointer to memblock_type which excludes memory from being taken 859dad7557eSWanpeng Li * @out_start: ptr to phys_addr_t for start address of the range, can be %NULL 860dad7557eSWanpeng Li * @out_end: ptr to phys_addr_t for end address of the range, can be %NULL 861dad7557eSWanpeng Li * @out_nid: ptr to int for nid of the range, can be %NULL 86235fd0808STejun Heo * 863f1af9d3aSPhilipp Hachtmann * Find the first area from *@idx which matches @nid, fill the out 86435fd0808STejun Heo * parameters, and update *@idx for the next iteration. The lower 32bit of 865f1af9d3aSPhilipp Hachtmann * *@idx contains index into type_a and the upper 32bit indexes the 866f1af9d3aSPhilipp Hachtmann * areas before each region in type_b. For example, if type_b regions 86735fd0808STejun Heo * look like the following, 86835fd0808STejun Heo * 86935fd0808STejun Heo * 0:[0-16), 1:[32-48), 2:[128-130) 87035fd0808STejun Heo * 87135fd0808STejun Heo * The upper 32bit indexes the following regions. 87235fd0808STejun Heo * 87335fd0808STejun Heo * 0:[0-0), 1:[16-32), 2:[48-128), 3:[130-MAX) 87435fd0808STejun Heo * 87535fd0808STejun Heo * As both region arrays are sorted, the function advances the two indices 87635fd0808STejun Heo * in lockstep and returns each intersection. 87735fd0808STejun Heo */ 878fc6daaf9STony Luck void __init_memblock __next_mem_range(u64 *idx, int nid, ulong flags, 879f1af9d3aSPhilipp Hachtmann struct memblock_type *type_a, 880f1af9d3aSPhilipp Hachtmann struct memblock_type *type_b, 88135fd0808STejun Heo phys_addr_t *out_start, 88235fd0808STejun Heo phys_addr_t *out_end, int *out_nid) 88335fd0808STejun Heo { 884f1af9d3aSPhilipp Hachtmann int idx_a = *idx & 0xffffffff; 885f1af9d3aSPhilipp Hachtmann int idx_b = *idx >> 32; 886b1154233SGrygorii Strashko 887f1af9d3aSPhilipp Hachtmann if (WARN_ONCE(nid == MAX_NUMNODES, 888f1af9d3aSPhilipp Hachtmann "Usage of MAX_NUMNODES is deprecated. Use NUMA_NO_NODE instead\n")) 889560dca27SGrygorii Strashko nid = NUMA_NO_NODE; 89035fd0808STejun Heo 891f1af9d3aSPhilipp Hachtmann for (; idx_a < type_a->cnt; idx_a++) { 892f1af9d3aSPhilipp Hachtmann struct memblock_region *m = &type_a->regions[idx_a]; 893f1af9d3aSPhilipp Hachtmann 89435fd0808STejun Heo phys_addr_t m_start = m->base; 89535fd0808STejun Heo phys_addr_t m_end = m->base + m->size; 896f1af9d3aSPhilipp Hachtmann int m_nid = memblock_get_region_node(m); 89735fd0808STejun Heo 89835fd0808STejun Heo /* only memory regions are associated with nodes, check it */ 899f1af9d3aSPhilipp Hachtmann if (nid != NUMA_NO_NODE && nid != m_nid) 90035fd0808STejun Heo continue; 90135fd0808STejun Heo 9020a313a99SXishi Qiu /* skip hotpluggable memory regions if needed */ 9030a313a99SXishi Qiu if (movable_node_is_enabled() && memblock_is_hotpluggable(m)) 9040a313a99SXishi Qiu continue; 9050a313a99SXishi Qiu 906a3f5bafcSTony Luck /* if we want mirror memory skip non-mirror memory regions */ 907a3f5bafcSTony Luck if ((flags & MEMBLOCK_MIRROR) && !memblock_is_mirror(m)) 908a3f5bafcSTony Luck continue; 909a3f5bafcSTony Luck 910bf3d3cc5SArd Biesheuvel /* skip nomap memory unless we were asked for it explicitly */ 911bf3d3cc5SArd Biesheuvel if (!(flags & MEMBLOCK_NOMAP) && memblock_is_nomap(m)) 912bf3d3cc5SArd Biesheuvel continue; 913bf3d3cc5SArd Biesheuvel 914f1af9d3aSPhilipp Hachtmann if (!type_b) { 915f1af9d3aSPhilipp Hachtmann if (out_start) 916f1af9d3aSPhilipp Hachtmann *out_start = m_start; 917f1af9d3aSPhilipp Hachtmann if (out_end) 918f1af9d3aSPhilipp Hachtmann *out_end = m_end; 919f1af9d3aSPhilipp Hachtmann if (out_nid) 920f1af9d3aSPhilipp Hachtmann *out_nid = m_nid; 921f1af9d3aSPhilipp Hachtmann idx_a++; 922f1af9d3aSPhilipp Hachtmann *idx = (u32)idx_a | (u64)idx_b << 32; 923f1af9d3aSPhilipp Hachtmann return; 924f1af9d3aSPhilipp Hachtmann } 92535fd0808STejun Heo 926f1af9d3aSPhilipp Hachtmann /* scan areas before each reservation */ 927f1af9d3aSPhilipp Hachtmann for (; idx_b < type_b->cnt + 1; idx_b++) { 928f1af9d3aSPhilipp Hachtmann struct memblock_region *r; 929f1af9d3aSPhilipp Hachtmann phys_addr_t r_start; 930f1af9d3aSPhilipp Hachtmann phys_addr_t r_end; 931f1af9d3aSPhilipp Hachtmann 932f1af9d3aSPhilipp Hachtmann r = &type_b->regions[idx_b]; 933f1af9d3aSPhilipp Hachtmann r_start = idx_b ? r[-1].base + r[-1].size : 0; 934f1af9d3aSPhilipp Hachtmann r_end = idx_b < type_b->cnt ? 935f1af9d3aSPhilipp Hachtmann r->base : ULLONG_MAX; 936f1af9d3aSPhilipp Hachtmann 937f1af9d3aSPhilipp Hachtmann /* 938f1af9d3aSPhilipp Hachtmann * if idx_b advanced past idx_a, 939f1af9d3aSPhilipp Hachtmann * break out to advance idx_a 940f1af9d3aSPhilipp Hachtmann */ 94135fd0808STejun Heo if (r_start >= m_end) 94235fd0808STejun Heo break; 94335fd0808STejun Heo /* if the two regions intersect, we're done */ 94435fd0808STejun Heo if (m_start < r_end) { 94535fd0808STejun Heo if (out_start) 946f1af9d3aSPhilipp Hachtmann *out_start = 947f1af9d3aSPhilipp Hachtmann max(m_start, r_start); 94835fd0808STejun Heo if (out_end) 94935fd0808STejun Heo *out_end = min(m_end, r_end); 95035fd0808STejun Heo if (out_nid) 951f1af9d3aSPhilipp Hachtmann *out_nid = m_nid; 95235fd0808STejun Heo /* 953f1af9d3aSPhilipp Hachtmann * The region which ends first is 954f1af9d3aSPhilipp Hachtmann * advanced for the next iteration. 95535fd0808STejun Heo */ 95635fd0808STejun Heo if (m_end <= r_end) 957f1af9d3aSPhilipp Hachtmann idx_a++; 95835fd0808STejun Heo else 959f1af9d3aSPhilipp Hachtmann idx_b++; 960f1af9d3aSPhilipp Hachtmann *idx = (u32)idx_a | (u64)idx_b << 32; 96135fd0808STejun Heo return; 96235fd0808STejun Heo } 96335fd0808STejun Heo } 96435fd0808STejun Heo } 96535fd0808STejun Heo 96635fd0808STejun Heo /* signal end of iteration */ 96735fd0808STejun Heo *idx = ULLONG_MAX; 96835fd0808STejun Heo } 96935fd0808STejun Heo 9707bd0b0f0STejun Heo /** 971f1af9d3aSPhilipp Hachtmann * __next_mem_range_rev - generic next function for for_each_*_range_rev() 972f1af9d3aSPhilipp Hachtmann * 973f1af9d3aSPhilipp Hachtmann * Finds the next range from type_a which is not marked as unsuitable 974f1af9d3aSPhilipp Hachtmann * in type_b. 975f1af9d3aSPhilipp Hachtmann * 9767bd0b0f0STejun Heo * @idx: pointer to u64 loop variable 977ad5ea8cdSAlexander Kuleshov * @nid: node selector, %NUMA_NO_NODE for all nodes 978fc6daaf9STony Luck * @flags: pick from blocks based on memory attributes 979f1af9d3aSPhilipp Hachtmann * @type_a: pointer to memblock_type from where the range is taken 980f1af9d3aSPhilipp Hachtmann * @type_b: pointer to memblock_type which excludes memory from being taken 981dad7557eSWanpeng Li * @out_start: ptr to phys_addr_t for start address of the range, can be %NULL 982dad7557eSWanpeng Li * @out_end: ptr to phys_addr_t for end address of the range, can be %NULL 983dad7557eSWanpeng Li * @out_nid: ptr to int for nid of the range, can be %NULL 9847bd0b0f0STejun Heo * 985f1af9d3aSPhilipp Hachtmann * Reverse of __next_mem_range(). 9867bd0b0f0STejun Heo */ 987fc6daaf9STony Luck void __init_memblock __next_mem_range_rev(u64 *idx, int nid, ulong flags, 988f1af9d3aSPhilipp Hachtmann struct memblock_type *type_a, 989f1af9d3aSPhilipp Hachtmann struct memblock_type *type_b, 9907bd0b0f0STejun Heo phys_addr_t *out_start, 9917bd0b0f0STejun Heo phys_addr_t *out_end, int *out_nid) 9927bd0b0f0STejun Heo { 993f1af9d3aSPhilipp Hachtmann int idx_a = *idx & 0xffffffff; 994f1af9d3aSPhilipp Hachtmann int idx_b = *idx >> 32; 995b1154233SGrygorii Strashko 996560dca27SGrygorii Strashko if (WARN_ONCE(nid == MAX_NUMNODES, "Usage of MAX_NUMNODES is deprecated. Use NUMA_NO_NODE instead\n")) 997560dca27SGrygorii Strashko nid = NUMA_NO_NODE; 9987bd0b0f0STejun Heo 9997bd0b0f0STejun Heo if (*idx == (u64)ULLONG_MAX) { 1000f1af9d3aSPhilipp Hachtmann idx_a = type_a->cnt - 1; 1001e47608abSzijun_hu if (type_b != NULL) 1002f1af9d3aSPhilipp Hachtmann idx_b = type_b->cnt; 1003e47608abSzijun_hu else 1004e47608abSzijun_hu idx_b = 0; 10057bd0b0f0STejun Heo } 10067bd0b0f0STejun Heo 1007f1af9d3aSPhilipp Hachtmann for (; idx_a >= 0; idx_a--) { 1008f1af9d3aSPhilipp Hachtmann struct memblock_region *m = &type_a->regions[idx_a]; 1009f1af9d3aSPhilipp Hachtmann 10107bd0b0f0STejun Heo phys_addr_t m_start = m->base; 10117bd0b0f0STejun Heo phys_addr_t m_end = m->base + m->size; 1012f1af9d3aSPhilipp Hachtmann int m_nid = memblock_get_region_node(m); 10137bd0b0f0STejun Heo 10147bd0b0f0STejun Heo /* only memory regions are associated with nodes, check it */ 1015f1af9d3aSPhilipp Hachtmann if (nid != NUMA_NO_NODE && nid != m_nid) 10167bd0b0f0STejun Heo continue; 10177bd0b0f0STejun Heo 101855ac590cSTang Chen /* skip hotpluggable memory regions if needed */ 101955ac590cSTang Chen if (movable_node_is_enabled() && memblock_is_hotpluggable(m)) 102055ac590cSTang Chen continue; 102155ac590cSTang Chen 1022a3f5bafcSTony Luck /* if we want mirror memory skip non-mirror memory regions */ 1023a3f5bafcSTony Luck if ((flags & MEMBLOCK_MIRROR) && !memblock_is_mirror(m)) 1024a3f5bafcSTony Luck continue; 1025a3f5bafcSTony Luck 1026bf3d3cc5SArd Biesheuvel /* skip nomap memory unless we were asked for it explicitly */ 1027bf3d3cc5SArd Biesheuvel if (!(flags & MEMBLOCK_NOMAP) && memblock_is_nomap(m)) 1028bf3d3cc5SArd Biesheuvel continue; 1029bf3d3cc5SArd Biesheuvel 1030f1af9d3aSPhilipp Hachtmann if (!type_b) { 1031f1af9d3aSPhilipp Hachtmann if (out_start) 1032f1af9d3aSPhilipp Hachtmann *out_start = m_start; 1033f1af9d3aSPhilipp Hachtmann if (out_end) 1034f1af9d3aSPhilipp Hachtmann *out_end = m_end; 1035f1af9d3aSPhilipp Hachtmann if (out_nid) 1036f1af9d3aSPhilipp Hachtmann *out_nid = m_nid; 1037fb399b48Szijun_hu idx_a--; 1038f1af9d3aSPhilipp Hachtmann *idx = (u32)idx_a | (u64)idx_b << 32; 1039f1af9d3aSPhilipp Hachtmann return; 1040f1af9d3aSPhilipp Hachtmann } 10417bd0b0f0STejun Heo 1042f1af9d3aSPhilipp Hachtmann /* scan areas before each reservation */ 1043f1af9d3aSPhilipp Hachtmann for (; idx_b >= 0; idx_b--) { 1044f1af9d3aSPhilipp Hachtmann struct memblock_region *r; 1045f1af9d3aSPhilipp Hachtmann phys_addr_t r_start; 1046f1af9d3aSPhilipp Hachtmann phys_addr_t r_end; 1047f1af9d3aSPhilipp Hachtmann 1048f1af9d3aSPhilipp Hachtmann r = &type_b->regions[idx_b]; 1049f1af9d3aSPhilipp Hachtmann r_start = idx_b ? r[-1].base + r[-1].size : 0; 1050f1af9d3aSPhilipp Hachtmann r_end = idx_b < type_b->cnt ? 1051f1af9d3aSPhilipp Hachtmann r->base : ULLONG_MAX; 1052f1af9d3aSPhilipp Hachtmann /* 1053f1af9d3aSPhilipp Hachtmann * if idx_b advanced past idx_a, 1054f1af9d3aSPhilipp Hachtmann * break out to advance idx_a 1055f1af9d3aSPhilipp Hachtmann */ 1056f1af9d3aSPhilipp Hachtmann 10577bd0b0f0STejun Heo if (r_end <= m_start) 10587bd0b0f0STejun Heo break; 10597bd0b0f0STejun Heo /* if the two regions intersect, we're done */ 10607bd0b0f0STejun Heo if (m_end > r_start) { 10617bd0b0f0STejun Heo if (out_start) 10627bd0b0f0STejun Heo *out_start = max(m_start, r_start); 10637bd0b0f0STejun Heo if (out_end) 10647bd0b0f0STejun Heo *out_end = min(m_end, r_end); 10657bd0b0f0STejun Heo if (out_nid) 1066f1af9d3aSPhilipp Hachtmann *out_nid = m_nid; 10677bd0b0f0STejun Heo if (m_start >= r_start) 1068f1af9d3aSPhilipp Hachtmann idx_a--; 10697bd0b0f0STejun Heo else 1070f1af9d3aSPhilipp Hachtmann idx_b--; 1071f1af9d3aSPhilipp Hachtmann *idx = (u32)idx_a | (u64)idx_b << 32; 10727bd0b0f0STejun Heo return; 10737bd0b0f0STejun Heo } 10747bd0b0f0STejun Heo } 10757bd0b0f0STejun Heo } 1076f1af9d3aSPhilipp Hachtmann /* signal end of iteration */ 10777bd0b0f0STejun Heo *idx = ULLONG_MAX; 10787bd0b0f0STejun Heo } 10797bd0b0f0STejun Heo 10807c0caeb8STejun Heo #ifdef CONFIG_HAVE_MEMBLOCK_NODE_MAP 10817c0caeb8STejun Heo /* 10827c0caeb8STejun Heo * Common iterator interface used to define for_each_mem_range(). 10837c0caeb8STejun Heo */ 10847c0caeb8STejun Heo void __init_memblock __next_mem_pfn_range(int *idx, int nid, 10857c0caeb8STejun Heo unsigned long *out_start_pfn, 10867c0caeb8STejun Heo unsigned long *out_end_pfn, int *out_nid) 10877c0caeb8STejun Heo { 10887c0caeb8STejun Heo struct memblock_type *type = &memblock.memory; 10897c0caeb8STejun Heo struct memblock_region *r; 10907c0caeb8STejun Heo 10917c0caeb8STejun Heo while (++*idx < type->cnt) { 10927c0caeb8STejun Heo r = &type->regions[*idx]; 10937c0caeb8STejun Heo 10947c0caeb8STejun Heo if (PFN_UP(r->base) >= PFN_DOWN(r->base + r->size)) 10957c0caeb8STejun Heo continue; 10967c0caeb8STejun Heo if (nid == MAX_NUMNODES || nid == r->nid) 10977c0caeb8STejun Heo break; 10987c0caeb8STejun Heo } 10997c0caeb8STejun Heo if (*idx >= type->cnt) { 11007c0caeb8STejun Heo *idx = -1; 11017c0caeb8STejun Heo return; 11027c0caeb8STejun Heo } 11037c0caeb8STejun Heo 11047c0caeb8STejun Heo if (out_start_pfn) 11057c0caeb8STejun Heo *out_start_pfn = PFN_UP(r->base); 11067c0caeb8STejun Heo if (out_end_pfn) 11077c0caeb8STejun Heo *out_end_pfn = PFN_DOWN(r->base + r->size); 11087c0caeb8STejun Heo if (out_nid) 11097c0caeb8STejun Heo *out_nid = r->nid; 11107c0caeb8STejun Heo } 11117c0caeb8STejun Heo 1112b92df1deSPaul Burton unsigned long __init_memblock memblock_next_valid_pfn(unsigned long pfn, 1113b92df1deSPaul Burton unsigned long max_pfn) 1114b92df1deSPaul Burton { 1115b92df1deSPaul Burton struct memblock_type *type = &memblock.memory; 1116b92df1deSPaul Burton unsigned int right = type->cnt; 1117b92df1deSPaul Burton unsigned int mid, left = 0; 1118b92df1deSPaul Burton phys_addr_t addr = PFN_PHYS(pfn + 1); 1119b92df1deSPaul Burton 1120b92df1deSPaul Burton do { 1121b92df1deSPaul Burton mid = (right + left) / 2; 1122b92df1deSPaul Burton 1123b92df1deSPaul Burton if (addr < type->regions[mid].base) 1124b92df1deSPaul Burton right = mid; 1125b92df1deSPaul Burton else if (addr >= (type->regions[mid].base + 1126b92df1deSPaul Burton type->regions[mid].size)) 1127b92df1deSPaul Burton left = mid + 1; 1128b92df1deSPaul Burton else { 1129b92df1deSPaul Burton /* addr is within the region, so pfn + 1 is valid */ 1130b92df1deSPaul Burton return min(pfn + 1, max_pfn); 1131b92df1deSPaul Burton } 1132b92df1deSPaul Burton } while (left < right); 1133b92df1deSPaul Burton 1134b92df1deSPaul Burton return min(PHYS_PFN(type->regions[right].base), max_pfn); 1135b92df1deSPaul Burton } 1136b92df1deSPaul Burton 11377c0caeb8STejun Heo /** 11387c0caeb8STejun Heo * memblock_set_node - set node ID on memblock regions 11397c0caeb8STejun Heo * @base: base of area to set node ID for 11407c0caeb8STejun Heo * @size: size of area to set node ID for 1141e7e8de59STang Chen * @type: memblock type to set node ID for 11427c0caeb8STejun Heo * @nid: node ID to set 11437c0caeb8STejun Heo * 1144e7e8de59STang Chen * Set the nid of memblock @type regions in [@base,@base+@size) to @nid. 11457c0caeb8STejun Heo * Regions which cross the area boundaries are split as necessary. 11467c0caeb8STejun Heo * 11477c0caeb8STejun Heo * RETURNS: 11487c0caeb8STejun Heo * 0 on success, -errno on failure. 11497c0caeb8STejun Heo */ 11507c0caeb8STejun Heo int __init_memblock memblock_set_node(phys_addr_t base, phys_addr_t size, 1151e7e8de59STang Chen struct memblock_type *type, int nid) 11527c0caeb8STejun Heo { 11536a9ceb31STejun Heo int start_rgn, end_rgn; 11546a9ceb31STejun Heo int i, ret; 11557c0caeb8STejun Heo 11566a9ceb31STejun Heo ret = memblock_isolate_range(type, base, size, &start_rgn, &end_rgn); 11576a9ceb31STejun Heo if (ret) 11586a9ceb31STejun Heo return ret; 11597c0caeb8STejun Heo 11606a9ceb31STejun Heo for (i = start_rgn; i < end_rgn; i++) 1161e9d24ad3SWanpeng Li memblock_set_region_node(&type->regions[i], nid); 11627c0caeb8STejun Heo 11637c0caeb8STejun Heo memblock_merge_regions(type); 11647c0caeb8STejun Heo return 0; 11657c0caeb8STejun Heo } 11667c0caeb8STejun Heo #endif /* CONFIG_HAVE_MEMBLOCK_NODE_MAP */ 11677c0caeb8STejun Heo 11682bfc2862SAkinobu Mita static phys_addr_t __init memblock_alloc_range_nid(phys_addr_t size, 11692bfc2862SAkinobu Mita phys_addr_t align, phys_addr_t start, 1170fc6daaf9STony Luck phys_addr_t end, int nid, ulong flags) 117195f72d1eSYinghai Lu { 11726ed311b2SBenjamin Herrenschmidt phys_addr_t found; 117395f72d1eSYinghai Lu 117479f40fabSGrygorii Strashko if (!align) 117579f40fabSGrygorii Strashko align = SMP_CACHE_BYTES; 117694f3d3afSVineet Gupta 1177fc6daaf9STony Luck found = memblock_find_in_range_node(size, align, start, end, nid, 1178fc6daaf9STony Luck flags); 1179aedf95eaSCatalin Marinas if (found && !memblock_reserve(found, size)) { 1180aedf95eaSCatalin Marinas /* 1181aedf95eaSCatalin Marinas * The min_count is set to 0 so that memblock allocations are 1182aedf95eaSCatalin Marinas * never reported as leaks. 1183aedf95eaSCatalin Marinas */ 11849099daedSCatalin Marinas kmemleak_alloc_phys(found, size, 0, 0); 11856ed311b2SBenjamin Herrenschmidt return found; 1186aedf95eaSCatalin Marinas } 11876ed311b2SBenjamin Herrenschmidt return 0; 118895f72d1eSYinghai Lu } 118995f72d1eSYinghai Lu 11902bfc2862SAkinobu Mita phys_addr_t __init memblock_alloc_range(phys_addr_t size, phys_addr_t align, 1191fc6daaf9STony Luck phys_addr_t start, phys_addr_t end, 1192fc6daaf9STony Luck ulong flags) 11932bfc2862SAkinobu Mita { 1194fc6daaf9STony Luck return memblock_alloc_range_nid(size, align, start, end, NUMA_NO_NODE, 1195fc6daaf9STony Luck flags); 11962bfc2862SAkinobu Mita } 11972bfc2862SAkinobu Mita 11982bfc2862SAkinobu Mita static phys_addr_t __init memblock_alloc_base_nid(phys_addr_t size, 11992bfc2862SAkinobu Mita phys_addr_t align, phys_addr_t max_addr, 1200fc6daaf9STony Luck int nid, ulong flags) 12012bfc2862SAkinobu Mita { 1202fc6daaf9STony Luck return memblock_alloc_range_nid(size, align, 0, max_addr, nid, flags); 12032bfc2862SAkinobu Mita } 12042bfc2862SAkinobu Mita 12057bd0b0f0STejun Heo phys_addr_t __init memblock_alloc_nid(phys_addr_t size, phys_addr_t align, int nid) 12067bd0b0f0STejun Heo { 1207a3f5bafcSTony Luck ulong flags = choose_memblock_flags(); 1208a3f5bafcSTony Luck phys_addr_t ret; 1209a3f5bafcSTony Luck 1210a3f5bafcSTony Luck again: 1211a3f5bafcSTony Luck ret = memblock_alloc_base_nid(size, align, MEMBLOCK_ALLOC_ACCESSIBLE, 1212a3f5bafcSTony Luck nid, flags); 1213a3f5bafcSTony Luck 1214a3f5bafcSTony Luck if (!ret && (flags & MEMBLOCK_MIRROR)) { 1215a3f5bafcSTony Luck flags &= ~MEMBLOCK_MIRROR; 1216a3f5bafcSTony Luck goto again; 1217a3f5bafcSTony Luck } 1218a3f5bafcSTony Luck return ret; 12197bd0b0f0STejun Heo } 12207bd0b0f0STejun Heo 12217bd0b0f0STejun Heo phys_addr_t __init __memblock_alloc_base(phys_addr_t size, phys_addr_t align, phys_addr_t max_addr) 12227bd0b0f0STejun Heo { 1223fc6daaf9STony Luck return memblock_alloc_base_nid(size, align, max_addr, NUMA_NO_NODE, 1224fc6daaf9STony Luck MEMBLOCK_NONE); 12257bd0b0f0STejun Heo } 12267bd0b0f0STejun Heo 12276ed311b2SBenjamin Herrenschmidt phys_addr_t __init memblock_alloc_base(phys_addr_t size, phys_addr_t align, phys_addr_t max_addr) 122895f72d1eSYinghai Lu { 12296ed311b2SBenjamin Herrenschmidt phys_addr_t alloc; 12306ed311b2SBenjamin Herrenschmidt 12316ed311b2SBenjamin Herrenschmidt alloc = __memblock_alloc_base(size, align, max_addr); 12326ed311b2SBenjamin Herrenschmidt 12336ed311b2SBenjamin Herrenschmidt if (alloc == 0) 12345d63f81cSMiles Chen panic("ERROR: Failed to allocate %pa bytes below %pa.\n", 12355d63f81cSMiles Chen &size, &max_addr); 12366ed311b2SBenjamin Herrenschmidt 12376ed311b2SBenjamin Herrenschmidt return alloc; 123895f72d1eSYinghai Lu } 123995f72d1eSYinghai Lu 12406ed311b2SBenjamin Herrenschmidt phys_addr_t __init memblock_alloc(phys_addr_t size, phys_addr_t align) 124195f72d1eSYinghai Lu { 12426ed311b2SBenjamin Herrenschmidt return memblock_alloc_base(size, align, MEMBLOCK_ALLOC_ACCESSIBLE); 124395f72d1eSYinghai Lu } 124495f72d1eSYinghai Lu 12459d1e2492SBenjamin Herrenschmidt phys_addr_t __init memblock_alloc_try_nid(phys_addr_t size, phys_addr_t align, int nid) 12469d1e2492SBenjamin Herrenschmidt { 12479d1e2492SBenjamin Herrenschmidt phys_addr_t res = memblock_alloc_nid(size, align, nid); 12489d1e2492SBenjamin Herrenschmidt 12499d1e2492SBenjamin Herrenschmidt if (res) 12509d1e2492SBenjamin Herrenschmidt return res; 125115fb0972STejun Heo return memblock_alloc_base(size, align, MEMBLOCK_ALLOC_ACCESSIBLE); 125295f72d1eSYinghai Lu } 125395f72d1eSYinghai Lu 125426f09e9bSSantosh Shilimkar /** 125526f09e9bSSantosh Shilimkar * memblock_virt_alloc_internal - allocate boot memory block 125626f09e9bSSantosh Shilimkar * @size: size of memory block to be allocated in bytes 125726f09e9bSSantosh Shilimkar * @align: alignment of the region and block's size 125826f09e9bSSantosh Shilimkar * @min_addr: the lower bound of the memory region to allocate (phys address) 125926f09e9bSSantosh Shilimkar * @max_addr: the upper bound of the memory region to allocate (phys address) 126026f09e9bSSantosh Shilimkar * @nid: nid of the free area to find, %NUMA_NO_NODE for any node 126126f09e9bSSantosh Shilimkar * 126226f09e9bSSantosh Shilimkar * The @min_addr limit is dropped if it can not be satisfied and the allocation 126326f09e9bSSantosh Shilimkar * will fall back to memory below @min_addr. Also, allocation may fall back 126426f09e9bSSantosh Shilimkar * to any node in the system if the specified node can not 126526f09e9bSSantosh Shilimkar * hold the requested memory. 126626f09e9bSSantosh Shilimkar * 126726f09e9bSSantosh Shilimkar * The allocation is performed from memory region limited by 126826f09e9bSSantosh Shilimkar * memblock.current_limit if @max_addr == %BOOTMEM_ALLOC_ACCESSIBLE. 126926f09e9bSSantosh Shilimkar * 127026f09e9bSSantosh Shilimkar * The memory block is aligned on SMP_CACHE_BYTES if @align == 0. 127126f09e9bSSantosh Shilimkar * 127226f09e9bSSantosh Shilimkar * The phys address of allocated boot memory block is converted to virtual and 127326f09e9bSSantosh Shilimkar * allocated memory is reset to 0. 127426f09e9bSSantosh Shilimkar * 127526f09e9bSSantosh Shilimkar * In addition, function sets the min_count to 0 using kmemleak_alloc for 127626f09e9bSSantosh Shilimkar * allocated boot memory block, so that it is never reported as leaks. 127726f09e9bSSantosh Shilimkar * 127826f09e9bSSantosh Shilimkar * RETURNS: 127926f09e9bSSantosh Shilimkar * Virtual address of allocated memory block on success, NULL on failure. 128026f09e9bSSantosh Shilimkar */ 128126f09e9bSSantosh Shilimkar static void * __init memblock_virt_alloc_internal( 128226f09e9bSSantosh Shilimkar phys_addr_t size, phys_addr_t align, 128326f09e9bSSantosh Shilimkar phys_addr_t min_addr, phys_addr_t max_addr, 128426f09e9bSSantosh Shilimkar int nid) 128526f09e9bSSantosh Shilimkar { 128626f09e9bSSantosh Shilimkar phys_addr_t alloc; 128726f09e9bSSantosh Shilimkar void *ptr; 1288a3f5bafcSTony Luck ulong flags = choose_memblock_flags(); 128926f09e9bSSantosh Shilimkar 1290560dca27SGrygorii Strashko if (WARN_ONCE(nid == MAX_NUMNODES, "Usage of MAX_NUMNODES is deprecated. Use NUMA_NO_NODE instead\n")) 1291560dca27SGrygorii Strashko nid = NUMA_NO_NODE; 129226f09e9bSSantosh Shilimkar 129326f09e9bSSantosh Shilimkar /* 129426f09e9bSSantosh Shilimkar * Detect any accidental use of these APIs after slab is ready, as at 129526f09e9bSSantosh Shilimkar * this moment memblock may be deinitialized already and its 129626f09e9bSSantosh Shilimkar * internal data may be destroyed (after execution of free_all_bootmem) 129726f09e9bSSantosh Shilimkar */ 129826f09e9bSSantosh Shilimkar if (WARN_ON_ONCE(slab_is_available())) 129926f09e9bSSantosh Shilimkar return kzalloc_node(size, GFP_NOWAIT, nid); 130026f09e9bSSantosh Shilimkar 130126f09e9bSSantosh Shilimkar if (!align) 130226f09e9bSSantosh Shilimkar align = SMP_CACHE_BYTES; 130326f09e9bSSantosh Shilimkar 1304f544e14fSYinghai Lu if (max_addr > memblock.current_limit) 1305f544e14fSYinghai Lu max_addr = memblock.current_limit; 130626f09e9bSSantosh Shilimkar again: 130726f09e9bSSantosh Shilimkar alloc = memblock_find_in_range_node(size, align, min_addr, max_addr, 1308a3f5bafcSTony Luck nid, flags); 13097d41c03eSWei Yang if (alloc && !memblock_reserve(alloc, size)) 131026f09e9bSSantosh Shilimkar goto done; 131126f09e9bSSantosh Shilimkar 131226f09e9bSSantosh Shilimkar if (nid != NUMA_NO_NODE) { 131326f09e9bSSantosh Shilimkar alloc = memblock_find_in_range_node(size, align, min_addr, 1314fc6daaf9STony Luck max_addr, NUMA_NO_NODE, 1315a3f5bafcSTony Luck flags); 13167d41c03eSWei Yang if (alloc && !memblock_reserve(alloc, size)) 131726f09e9bSSantosh Shilimkar goto done; 131826f09e9bSSantosh Shilimkar } 131926f09e9bSSantosh Shilimkar 132026f09e9bSSantosh Shilimkar if (min_addr) { 132126f09e9bSSantosh Shilimkar min_addr = 0; 132226f09e9bSSantosh Shilimkar goto again; 132326f09e9bSSantosh Shilimkar } 132426f09e9bSSantosh Shilimkar 1325a3f5bafcSTony Luck if (flags & MEMBLOCK_MIRROR) { 1326a3f5bafcSTony Luck flags &= ~MEMBLOCK_MIRROR; 1327a3f5bafcSTony Luck pr_warn("Could not allocate %pap bytes of mirrored memory\n", 1328a3f5bafcSTony Luck &size); 1329a3f5bafcSTony Luck goto again; 1330a3f5bafcSTony Luck } 1331a3f5bafcSTony Luck 1332a3f5bafcSTony Luck return NULL; 133326f09e9bSSantosh Shilimkar done: 133426f09e9bSSantosh Shilimkar ptr = phys_to_virt(alloc); 133526f09e9bSSantosh Shilimkar memset(ptr, 0, size); 133626f09e9bSSantosh Shilimkar 133726f09e9bSSantosh Shilimkar /* 133826f09e9bSSantosh Shilimkar * The min_count is set to 0 so that bootmem allocated blocks 133926f09e9bSSantosh Shilimkar * are never reported as leaks. This is because many of these blocks 134026f09e9bSSantosh Shilimkar * are only referred via the physical address which is not 134126f09e9bSSantosh Shilimkar * looked up by kmemleak. 134226f09e9bSSantosh Shilimkar */ 134326f09e9bSSantosh Shilimkar kmemleak_alloc(ptr, size, 0, 0); 134426f09e9bSSantosh Shilimkar 134526f09e9bSSantosh Shilimkar return ptr; 134626f09e9bSSantosh Shilimkar } 134726f09e9bSSantosh Shilimkar 134826f09e9bSSantosh Shilimkar /** 134926f09e9bSSantosh Shilimkar * memblock_virt_alloc_try_nid_nopanic - allocate boot memory block 135026f09e9bSSantosh Shilimkar * @size: size of memory block to be allocated in bytes 135126f09e9bSSantosh Shilimkar * @align: alignment of the region and block's size 135226f09e9bSSantosh Shilimkar * @min_addr: the lower bound of the memory region from where the allocation 135326f09e9bSSantosh Shilimkar * is preferred (phys address) 135426f09e9bSSantosh Shilimkar * @max_addr: the upper bound of the memory region from where the allocation 135526f09e9bSSantosh Shilimkar * is preferred (phys address), or %BOOTMEM_ALLOC_ACCESSIBLE to 135626f09e9bSSantosh Shilimkar * allocate only from memory limited by memblock.current_limit value 135726f09e9bSSantosh Shilimkar * @nid: nid of the free area to find, %NUMA_NO_NODE for any node 135826f09e9bSSantosh Shilimkar * 135926f09e9bSSantosh Shilimkar * Public version of _memblock_virt_alloc_try_nid_nopanic() which provides 136026f09e9bSSantosh Shilimkar * additional debug information (including caller info), if enabled. 136126f09e9bSSantosh Shilimkar * 136226f09e9bSSantosh Shilimkar * RETURNS: 136326f09e9bSSantosh Shilimkar * Virtual address of allocated memory block on success, NULL on failure. 136426f09e9bSSantosh Shilimkar */ 136526f09e9bSSantosh Shilimkar void * __init memblock_virt_alloc_try_nid_nopanic( 136626f09e9bSSantosh Shilimkar phys_addr_t size, phys_addr_t align, 136726f09e9bSSantosh Shilimkar phys_addr_t min_addr, phys_addr_t max_addr, 136826f09e9bSSantosh Shilimkar int nid) 136926f09e9bSSantosh Shilimkar { 137026f09e9bSSantosh Shilimkar memblock_dbg("%s: %llu bytes align=0x%llx nid=%d from=0x%llx max_addr=0x%llx %pF\n", 137126f09e9bSSantosh Shilimkar __func__, (u64)size, (u64)align, nid, (u64)min_addr, 137226f09e9bSSantosh Shilimkar (u64)max_addr, (void *)_RET_IP_); 137326f09e9bSSantosh Shilimkar return memblock_virt_alloc_internal(size, align, min_addr, 137426f09e9bSSantosh Shilimkar max_addr, nid); 137526f09e9bSSantosh Shilimkar } 137626f09e9bSSantosh Shilimkar 137726f09e9bSSantosh Shilimkar /** 137826f09e9bSSantosh Shilimkar * memblock_virt_alloc_try_nid - allocate boot memory block with panicking 137926f09e9bSSantosh Shilimkar * @size: size of memory block to be allocated in bytes 138026f09e9bSSantosh Shilimkar * @align: alignment of the region and block's size 138126f09e9bSSantosh Shilimkar * @min_addr: the lower bound of the memory region from where the allocation 138226f09e9bSSantosh Shilimkar * is preferred (phys address) 138326f09e9bSSantosh Shilimkar * @max_addr: the upper bound of the memory region from where the allocation 138426f09e9bSSantosh Shilimkar * is preferred (phys address), or %BOOTMEM_ALLOC_ACCESSIBLE to 138526f09e9bSSantosh Shilimkar * allocate only from memory limited by memblock.current_limit value 138626f09e9bSSantosh Shilimkar * @nid: nid of the free area to find, %NUMA_NO_NODE for any node 138726f09e9bSSantosh Shilimkar * 138826f09e9bSSantosh Shilimkar * Public panicking version of _memblock_virt_alloc_try_nid_nopanic() 138926f09e9bSSantosh Shilimkar * which provides debug information (including caller info), if enabled, 139026f09e9bSSantosh Shilimkar * and panics if the request can not be satisfied. 139126f09e9bSSantosh Shilimkar * 139226f09e9bSSantosh Shilimkar * RETURNS: 139326f09e9bSSantosh Shilimkar * Virtual address of allocated memory block on success, NULL on failure. 139426f09e9bSSantosh Shilimkar */ 139526f09e9bSSantosh Shilimkar void * __init memblock_virt_alloc_try_nid( 139626f09e9bSSantosh Shilimkar phys_addr_t size, phys_addr_t align, 139726f09e9bSSantosh Shilimkar phys_addr_t min_addr, phys_addr_t max_addr, 139826f09e9bSSantosh Shilimkar int nid) 139926f09e9bSSantosh Shilimkar { 140026f09e9bSSantosh Shilimkar void *ptr; 140126f09e9bSSantosh Shilimkar 140226f09e9bSSantosh Shilimkar memblock_dbg("%s: %llu bytes align=0x%llx nid=%d from=0x%llx max_addr=0x%llx %pF\n", 140326f09e9bSSantosh Shilimkar __func__, (u64)size, (u64)align, nid, (u64)min_addr, 140426f09e9bSSantosh Shilimkar (u64)max_addr, (void *)_RET_IP_); 140526f09e9bSSantosh Shilimkar ptr = memblock_virt_alloc_internal(size, align, 140626f09e9bSSantosh Shilimkar min_addr, max_addr, nid); 140726f09e9bSSantosh Shilimkar if (ptr) 140826f09e9bSSantosh Shilimkar return ptr; 140926f09e9bSSantosh Shilimkar 141026f09e9bSSantosh Shilimkar panic("%s: Failed to allocate %llu bytes align=0x%llx nid=%d from=0x%llx max_addr=0x%llx\n", 141126f09e9bSSantosh Shilimkar __func__, (u64)size, (u64)align, nid, (u64)min_addr, 141226f09e9bSSantosh Shilimkar (u64)max_addr); 141326f09e9bSSantosh Shilimkar return NULL; 141426f09e9bSSantosh Shilimkar } 141526f09e9bSSantosh Shilimkar 141626f09e9bSSantosh Shilimkar /** 141726f09e9bSSantosh Shilimkar * __memblock_free_early - free boot memory block 141826f09e9bSSantosh Shilimkar * @base: phys starting address of the boot memory block 141926f09e9bSSantosh Shilimkar * @size: size of the boot memory block in bytes 142026f09e9bSSantosh Shilimkar * 142126f09e9bSSantosh Shilimkar * Free boot memory block previously allocated by memblock_virt_alloc_xx() API. 142226f09e9bSSantosh Shilimkar * The freeing memory will not be released to the buddy allocator. 142326f09e9bSSantosh Shilimkar */ 142426f09e9bSSantosh Shilimkar void __init __memblock_free_early(phys_addr_t base, phys_addr_t size) 142526f09e9bSSantosh Shilimkar { 142626f09e9bSSantosh Shilimkar memblock_dbg("%s: [%#016llx-%#016llx] %pF\n", 142726f09e9bSSantosh Shilimkar __func__, (u64)base, (u64)base + size - 1, 142826f09e9bSSantosh Shilimkar (void *)_RET_IP_); 14299099daedSCatalin Marinas kmemleak_free_part_phys(base, size); 1430f1af9d3aSPhilipp Hachtmann memblock_remove_range(&memblock.reserved, base, size); 143126f09e9bSSantosh Shilimkar } 143226f09e9bSSantosh Shilimkar 143326f09e9bSSantosh Shilimkar /* 143426f09e9bSSantosh Shilimkar * __memblock_free_late - free bootmem block pages directly to buddy allocator 143526f09e9bSSantosh Shilimkar * @addr: phys starting address of the boot memory block 143626f09e9bSSantosh Shilimkar * @size: size of the boot memory block in bytes 143726f09e9bSSantosh Shilimkar * 143826f09e9bSSantosh Shilimkar * This is only useful when the bootmem allocator has already been torn 143926f09e9bSSantosh Shilimkar * down, but we are still initializing the system. Pages are released directly 144026f09e9bSSantosh Shilimkar * to the buddy allocator, no bootmem metadata is updated because it is gone. 144126f09e9bSSantosh Shilimkar */ 144226f09e9bSSantosh Shilimkar void __init __memblock_free_late(phys_addr_t base, phys_addr_t size) 144326f09e9bSSantosh Shilimkar { 144426f09e9bSSantosh Shilimkar u64 cursor, end; 144526f09e9bSSantosh Shilimkar 144626f09e9bSSantosh Shilimkar memblock_dbg("%s: [%#016llx-%#016llx] %pF\n", 144726f09e9bSSantosh Shilimkar __func__, (u64)base, (u64)base + size - 1, 144826f09e9bSSantosh Shilimkar (void *)_RET_IP_); 14499099daedSCatalin Marinas kmemleak_free_part_phys(base, size); 145026f09e9bSSantosh Shilimkar cursor = PFN_UP(base); 145126f09e9bSSantosh Shilimkar end = PFN_DOWN(base + size); 145226f09e9bSSantosh Shilimkar 145326f09e9bSSantosh Shilimkar for (; cursor < end; cursor++) { 1454d70ddd7aSMel Gorman __free_pages_bootmem(pfn_to_page(cursor), cursor, 0); 145526f09e9bSSantosh Shilimkar totalram_pages++; 145626f09e9bSSantosh Shilimkar } 145726f09e9bSSantosh Shilimkar } 14589d1e2492SBenjamin Herrenschmidt 14599d1e2492SBenjamin Herrenschmidt /* 14609d1e2492SBenjamin Herrenschmidt * Remaining API functions 14619d1e2492SBenjamin Herrenschmidt */ 14629d1e2492SBenjamin Herrenschmidt 14631f1ffb8aSDavid Gibson phys_addr_t __init_memblock memblock_phys_mem_size(void) 146495f72d1eSYinghai Lu { 14651440c4e2STejun Heo return memblock.memory.total_size; 146695f72d1eSYinghai Lu } 146795f72d1eSYinghai Lu 14688907de5dSSrikar Dronamraju phys_addr_t __init_memblock memblock_reserved_size(void) 14698907de5dSSrikar Dronamraju { 14708907de5dSSrikar Dronamraju return memblock.reserved.total_size; 14718907de5dSSrikar Dronamraju } 14728907de5dSSrikar Dronamraju 1473595ad9afSYinghai Lu phys_addr_t __init memblock_mem_size(unsigned long limit_pfn) 1474595ad9afSYinghai Lu { 1475595ad9afSYinghai Lu unsigned long pages = 0; 1476595ad9afSYinghai Lu struct memblock_region *r; 1477595ad9afSYinghai Lu unsigned long start_pfn, end_pfn; 1478595ad9afSYinghai Lu 1479595ad9afSYinghai Lu for_each_memblock(memory, r) { 1480595ad9afSYinghai Lu start_pfn = memblock_region_memory_base_pfn(r); 1481595ad9afSYinghai Lu end_pfn = memblock_region_memory_end_pfn(r); 1482595ad9afSYinghai Lu start_pfn = min_t(unsigned long, start_pfn, limit_pfn); 1483595ad9afSYinghai Lu end_pfn = min_t(unsigned long, end_pfn, limit_pfn); 1484595ad9afSYinghai Lu pages += end_pfn - start_pfn; 1485595ad9afSYinghai Lu } 1486595ad9afSYinghai Lu 148716763230SFabian Frederick return PFN_PHYS(pages); 1488595ad9afSYinghai Lu } 1489595ad9afSYinghai Lu 14900a93ebefSSam Ravnborg /* lowest address */ 14910a93ebefSSam Ravnborg phys_addr_t __init_memblock memblock_start_of_DRAM(void) 14920a93ebefSSam Ravnborg { 14930a93ebefSSam Ravnborg return memblock.memory.regions[0].base; 14940a93ebefSSam Ravnborg } 14950a93ebefSSam Ravnborg 149610d06439SYinghai Lu phys_addr_t __init_memblock memblock_end_of_DRAM(void) 149795f72d1eSYinghai Lu { 149895f72d1eSYinghai Lu int idx = memblock.memory.cnt - 1; 149995f72d1eSYinghai Lu 1500e3239ff9SBenjamin Herrenschmidt return (memblock.memory.regions[idx].base + memblock.memory.regions[idx].size); 150195f72d1eSYinghai Lu } 150295f72d1eSYinghai Lu 1503a571d4ebSDennis Chen static phys_addr_t __init_memblock __find_max_addr(phys_addr_t limit) 150495f72d1eSYinghai Lu { 1505c0ce8fefSTejun Heo phys_addr_t max_addr = (phys_addr_t)ULLONG_MAX; 1506136199f0SEmil Medve struct memblock_region *r; 150795f72d1eSYinghai Lu 1508a571d4ebSDennis Chen /* 1509a571d4ebSDennis Chen * translate the memory @limit size into the max address within one of 1510a571d4ebSDennis Chen * the memory memblock regions, if the @limit exceeds the total size 1511a571d4ebSDennis Chen * of those regions, max_addr will keep original value ULLONG_MAX 1512a571d4ebSDennis Chen */ 1513136199f0SEmil Medve for_each_memblock(memory, r) { 1514c0ce8fefSTejun Heo if (limit <= r->size) { 1515c0ce8fefSTejun Heo max_addr = r->base + limit; 151695f72d1eSYinghai Lu break; 151795f72d1eSYinghai Lu } 1518c0ce8fefSTejun Heo limit -= r->size; 151995f72d1eSYinghai Lu } 1520c0ce8fefSTejun Heo 1521a571d4ebSDennis Chen return max_addr; 1522a571d4ebSDennis Chen } 1523a571d4ebSDennis Chen 1524a571d4ebSDennis Chen void __init memblock_enforce_memory_limit(phys_addr_t limit) 1525a571d4ebSDennis Chen { 1526a571d4ebSDennis Chen phys_addr_t max_addr = (phys_addr_t)ULLONG_MAX; 1527a571d4ebSDennis Chen 1528a571d4ebSDennis Chen if (!limit) 1529a571d4ebSDennis Chen return; 1530a571d4ebSDennis Chen 1531a571d4ebSDennis Chen max_addr = __find_max_addr(limit); 1532a571d4ebSDennis Chen 1533a571d4ebSDennis Chen /* @limit exceeds the total size of the memory, do nothing */ 1534a571d4ebSDennis Chen if (max_addr == (phys_addr_t)ULLONG_MAX) 1535a571d4ebSDennis Chen return; 1536a571d4ebSDennis Chen 1537c0ce8fefSTejun Heo /* truncate both memory and reserved regions */ 1538f1af9d3aSPhilipp Hachtmann memblock_remove_range(&memblock.memory, max_addr, 1539f1af9d3aSPhilipp Hachtmann (phys_addr_t)ULLONG_MAX); 1540f1af9d3aSPhilipp Hachtmann memblock_remove_range(&memblock.reserved, max_addr, 1541f1af9d3aSPhilipp Hachtmann (phys_addr_t)ULLONG_MAX); 154295f72d1eSYinghai Lu } 154395f72d1eSYinghai Lu 1544a571d4ebSDennis Chen void __init memblock_mem_limit_remove_map(phys_addr_t limit) 1545a571d4ebSDennis Chen { 1546a571d4ebSDennis Chen struct memblock_type *type = &memblock.memory; 1547a571d4ebSDennis Chen phys_addr_t max_addr; 1548a571d4ebSDennis Chen int i, ret, start_rgn, end_rgn; 1549a571d4ebSDennis Chen 1550a571d4ebSDennis Chen if (!limit) 1551a571d4ebSDennis Chen return; 1552a571d4ebSDennis Chen 1553a571d4ebSDennis Chen max_addr = __find_max_addr(limit); 1554a571d4ebSDennis Chen 1555a571d4ebSDennis Chen /* @limit exceeds the total size of the memory, do nothing */ 1556a571d4ebSDennis Chen if (max_addr == (phys_addr_t)ULLONG_MAX) 1557a571d4ebSDennis Chen return; 1558a571d4ebSDennis Chen 1559a571d4ebSDennis Chen ret = memblock_isolate_range(type, max_addr, (phys_addr_t)ULLONG_MAX, 1560a571d4ebSDennis Chen &start_rgn, &end_rgn); 1561a571d4ebSDennis Chen if (ret) 1562a571d4ebSDennis Chen return; 1563a571d4ebSDennis Chen 1564a571d4ebSDennis Chen /* remove all the MAP regions above the limit */ 1565a571d4ebSDennis Chen for (i = end_rgn - 1; i >= start_rgn; i--) { 1566a571d4ebSDennis Chen if (!memblock_is_nomap(&type->regions[i])) 1567a571d4ebSDennis Chen memblock_remove_region(type, i); 1568a571d4ebSDennis Chen } 1569a571d4ebSDennis Chen /* truncate the reserved regions */ 1570a571d4ebSDennis Chen memblock_remove_range(&memblock.reserved, max_addr, 1571a571d4ebSDennis Chen (phys_addr_t)ULLONG_MAX); 1572a571d4ebSDennis Chen } 1573a571d4ebSDennis Chen 1574cd79481dSYinghai Lu static int __init_memblock memblock_search(struct memblock_type *type, phys_addr_t addr) 157572d4b0b4SBenjamin Herrenschmidt { 157672d4b0b4SBenjamin Herrenschmidt unsigned int left = 0, right = type->cnt; 157772d4b0b4SBenjamin Herrenschmidt 157872d4b0b4SBenjamin Herrenschmidt do { 157972d4b0b4SBenjamin Herrenschmidt unsigned int mid = (right + left) / 2; 158072d4b0b4SBenjamin Herrenschmidt 158172d4b0b4SBenjamin Herrenschmidt if (addr < type->regions[mid].base) 158272d4b0b4SBenjamin Herrenschmidt right = mid; 158372d4b0b4SBenjamin Herrenschmidt else if (addr >= (type->regions[mid].base + 158472d4b0b4SBenjamin Herrenschmidt type->regions[mid].size)) 158572d4b0b4SBenjamin Herrenschmidt left = mid + 1; 158672d4b0b4SBenjamin Herrenschmidt else 158772d4b0b4SBenjamin Herrenschmidt return mid; 158872d4b0b4SBenjamin Herrenschmidt } while (left < right); 158972d4b0b4SBenjamin Herrenschmidt return -1; 159072d4b0b4SBenjamin Herrenschmidt } 159172d4b0b4SBenjamin Herrenschmidt 1592b4ad0c7eSYaowei Bai bool __init memblock_is_reserved(phys_addr_t addr) 159395f72d1eSYinghai Lu { 159472d4b0b4SBenjamin Herrenschmidt return memblock_search(&memblock.reserved, addr) != -1; 159595f72d1eSYinghai Lu } 159672d4b0b4SBenjamin Herrenschmidt 1597b4ad0c7eSYaowei Bai bool __init_memblock memblock_is_memory(phys_addr_t addr) 159872d4b0b4SBenjamin Herrenschmidt { 159972d4b0b4SBenjamin Herrenschmidt return memblock_search(&memblock.memory, addr) != -1; 160072d4b0b4SBenjamin Herrenschmidt } 160172d4b0b4SBenjamin Herrenschmidt 1602bf3d3cc5SArd Biesheuvel int __init_memblock memblock_is_map_memory(phys_addr_t addr) 1603bf3d3cc5SArd Biesheuvel { 1604bf3d3cc5SArd Biesheuvel int i = memblock_search(&memblock.memory, addr); 1605bf3d3cc5SArd Biesheuvel 1606bf3d3cc5SArd Biesheuvel if (i == -1) 1607bf3d3cc5SArd Biesheuvel return false; 1608bf3d3cc5SArd Biesheuvel return !memblock_is_nomap(&memblock.memory.regions[i]); 1609bf3d3cc5SArd Biesheuvel } 1610bf3d3cc5SArd Biesheuvel 1611e76b63f8SYinghai Lu #ifdef CONFIG_HAVE_MEMBLOCK_NODE_MAP 1612e76b63f8SYinghai Lu int __init_memblock memblock_search_pfn_nid(unsigned long pfn, 1613e76b63f8SYinghai Lu unsigned long *start_pfn, unsigned long *end_pfn) 1614e76b63f8SYinghai Lu { 1615e76b63f8SYinghai Lu struct memblock_type *type = &memblock.memory; 161616763230SFabian Frederick int mid = memblock_search(type, PFN_PHYS(pfn)); 1617e76b63f8SYinghai Lu 1618e76b63f8SYinghai Lu if (mid == -1) 1619e76b63f8SYinghai Lu return -1; 1620e76b63f8SYinghai Lu 1621f7e2f7e8SFabian Frederick *start_pfn = PFN_DOWN(type->regions[mid].base); 1622f7e2f7e8SFabian Frederick *end_pfn = PFN_DOWN(type->regions[mid].base + type->regions[mid].size); 1623e76b63f8SYinghai Lu 1624e76b63f8SYinghai Lu return type->regions[mid].nid; 1625e76b63f8SYinghai Lu } 1626e76b63f8SYinghai Lu #endif 1627e76b63f8SYinghai Lu 1628eab30949SStephen Boyd /** 1629eab30949SStephen Boyd * memblock_is_region_memory - check if a region is a subset of memory 1630eab30949SStephen Boyd * @base: base of region to check 1631eab30949SStephen Boyd * @size: size of region to check 1632eab30949SStephen Boyd * 1633eab30949SStephen Boyd * Check if the region [@base, @base+@size) is a subset of a memory block. 1634eab30949SStephen Boyd * 1635eab30949SStephen Boyd * RETURNS: 1636eab30949SStephen Boyd * 0 if false, non-zero if true 1637eab30949SStephen Boyd */ 16383661ca66SYinghai Lu int __init_memblock memblock_is_region_memory(phys_addr_t base, phys_addr_t size) 163972d4b0b4SBenjamin Herrenschmidt { 1640abb65272STomi Valkeinen int idx = memblock_search(&memblock.memory, base); 1641eb18f1b5STejun Heo phys_addr_t end = base + memblock_cap_size(base, &size); 164272d4b0b4SBenjamin Herrenschmidt 164372d4b0b4SBenjamin Herrenschmidt if (idx == -1) 164495f72d1eSYinghai Lu return 0; 1645ef415ef4SWei Yang return (memblock.memory.regions[idx].base + 1646eb18f1b5STejun Heo memblock.memory.regions[idx].size) >= end; 164795f72d1eSYinghai Lu } 164895f72d1eSYinghai Lu 1649eab30949SStephen Boyd /** 1650eab30949SStephen Boyd * memblock_is_region_reserved - check if a region intersects reserved memory 1651eab30949SStephen Boyd * @base: base of region to check 1652eab30949SStephen Boyd * @size: size of region to check 1653eab30949SStephen Boyd * 1654eab30949SStephen Boyd * Check if the region [@base, @base+@size) intersects a reserved memory block. 1655eab30949SStephen Boyd * 1656eab30949SStephen Boyd * RETURNS: 1657c5c5c9d1STang Chen * True if they intersect, false if not. 1658eab30949SStephen Boyd */ 1659c5c5c9d1STang Chen bool __init_memblock memblock_is_region_reserved(phys_addr_t base, phys_addr_t size) 166095f72d1eSYinghai Lu { 1661eb18f1b5STejun Heo memblock_cap_size(base, &size); 1662c5c5c9d1STang Chen return memblock_overlaps_region(&memblock.reserved, base, size); 166395f72d1eSYinghai Lu } 166495f72d1eSYinghai Lu 16656ede1fd3SYinghai Lu void __init_memblock memblock_trim_memory(phys_addr_t align) 16666ede1fd3SYinghai Lu { 16676ede1fd3SYinghai Lu phys_addr_t start, end, orig_start, orig_end; 1668136199f0SEmil Medve struct memblock_region *r; 16696ede1fd3SYinghai Lu 1670136199f0SEmil Medve for_each_memblock(memory, r) { 1671136199f0SEmil Medve orig_start = r->base; 1672136199f0SEmil Medve orig_end = r->base + r->size; 16736ede1fd3SYinghai Lu start = round_up(orig_start, align); 16746ede1fd3SYinghai Lu end = round_down(orig_end, align); 16756ede1fd3SYinghai Lu 16766ede1fd3SYinghai Lu if (start == orig_start && end == orig_end) 16776ede1fd3SYinghai Lu continue; 16786ede1fd3SYinghai Lu 16796ede1fd3SYinghai Lu if (start < end) { 1680136199f0SEmil Medve r->base = start; 1681136199f0SEmil Medve r->size = end - start; 16826ede1fd3SYinghai Lu } else { 1683136199f0SEmil Medve memblock_remove_region(&memblock.memory, 1684136199f0SEmil Medve r - memblock.memory.regions); 1685136199f0SEmil Medve r--; 16866ede1fd3SYinghai Lu } 16876ede1fd3SYinghai Lu } 16886ede1fd3SYinghai Lu } 1689e63075a3SBenjamin Herrenschmidt 16903661ca66SYinghai Lu void __init_memblock memblock_set_current_limit(phys_addr_t limit) 1691e63075a3SBenjamin Herrenschmidt { 1692e63075a3SBenjamin Herrenschmidt memblock.current_limit = limit; 1693e63075a3SBenjamin Herrenschmidt } 1694e63075a3SBenjamin Herrenschmidt 1695fec51014SLaura Abbott phys_addr_t __init_memblock memblock_get_current_limit(void) 1696fec51014SLaura Abbott { 1697fec51014SLaura Abbott return memblock.current_limit; 1698fec51014SLaura Abbott } 1699fec51014SLaura Abbott 17007c0caeb8STejun Heo static void __init_memblock memblock_dump(struct memblock_type *type, char *name) 17016ed311b2SBenjamin Herrenschmidt { 17025d63f81cSMiles Chen phys_addr_t base, end, size; 170366a20757STang Chen unsigned long flags; 17048c9c1701SAlexander Kuleshov int idx; 17058c9c1701SAlexander Kuleshov struct memblock_region *rgn; 17066ed311b2SBenjamin Herrenschmidt 17077c0caeb8STejun Heo pr_info(" %s.cnt = 0x%lx\n", name, type->cnt); 17086ed311b2SBenjamin Herrenschmidt 17098c9c1701SAlexander Kuleshov for_each_memblock_type(type, rgn) { 17107c0caeb8STejun Heo char nid_buf[32] = ""; 17116ed311b2SBenjamin Herrenschmidt 17127c0caeb8STejun Heo base = rgn->base; 17137c0caeb8STejun Heo size = rgn->size; 17145d63f81cSMiles Chen end = base + size - 1; 171566a20757STang Chen flags = rgn->flags; 17167c0caeb8STejun Heo #ifdef CONFIG_HAVE_MEMBLOCK_NODE_MAP 17177c0caeb8STejun Heo if (memblock_get_region_node(rgn) != MAX_NUMNODES) 17187c0caeb8STejun Heo snprintf(nid_buf, sizeof(nid_buf), " on node %d", 17197c0caeb8STejun Heo memblock_get_region_node(rgn)); 17207c0caeb8STejun Heo #endif 17215d63f81cSMiles Chen pr_info(" %s[%#x]\t[%pa-%pa], %pa bytes%s flags: %#lx\n", 17225d63f81cSMiles Chen name, idx, &base, &end, &size, nid_buf, flags); 17236ed311b2SBenjamin Herrenschmidt } 17246ed311b2SBenjamin Herrenschmidt } 17256ed311b2SBenjamin Herrenschmidt 17264ff7b82fSTejun Heo void __init_memblock __memblock_dump_all(void) 17276ed311b2SBenjamin Herrenschmidt { 17286ed311b2SBenjamin Herrenschmidt pr_info("MEMBLOCK configuration:\n"); 17295d63f81cSMiles Chen pr_info(" memory size = %pa reserved size = %pa\n", 17305d63f81cSMiles Chen &memblock.memory.total_size, 17315d63f81cSMiles Chen &memblock.reserved.total_size); 17326ed311b2SBenjamin Herrenschmidt 17336ed311b2SBenjamin Herrenschmidt memblock_dump(&memblock.memory, "memory"); 17346ed311b2SBenjamin Herrenschmidt memblock_dump(&memblock.reserved, "reserved"); 1735*409efd4cSHeiko Carstens #ifdef CONFIG_HAVE_MEMBLOCK_PHYS_MAP 1736*409efd4cSHeiko Carstens memblock_dump(&memblock.physmem, "physmem"); 1737*409efd4cSHeiko Carstens #endif 17386ed311b2SBenjamin Herrenschmidt } 17396ed311b2SBenjamin Herrenschmidt 17401aadc056STejun Heo void __init memblock_allow_resize(void) 17416ed311b2SBenjamin Herrenschmidt { 1742142b45a7SBenjamin Herrenschmidt memblock_can_resize = 1; 17436ed311b2SBenjamin Herrenschmidt } 17446ed311b2SBenjamin Herrenschmidt 17456ed311b2SBenjamin Herrenschmidt static int __init early_memblock(char *p) 17466ed311b2SBenjamin Herrenschmidt { 17476ed311b2SBenjamin Herrenschmidt if (p && strstr(p, "debug")) 17486ed311b2SBenjamin Herrenschmidt memblock_debug = 1; 17496ed311b2SBenjamin Herrenschmidt return 0; 17506ed311b2SBenjamin Herrenschmidt } 17516ed311b2SBenjamin Herrenschmidt early_param("memblock", early_memblock); 17526ed311b2SBenjamin Herrenschmidt 1753c378ddd5STejun Heo #if defined(CONFIG_DEBUG_FS) && !defined(CONFIG_ARCH_DISCARD_MEMBLOCK) 17546d03b885SBenjamin Herrenschmidt 17556d03b885SBenjamin Herrenschmidt static int memblock_debug_show(struct seq_file *m, void *private) 17566d03b885SBenjamin Herrenschmidt { 17576d03b885SBenjamin Herrenschmidt struct memblock_type *type = m->private; 17586d03b885SBenjamin Herrenschmidt struct memblock_region *reg; 17596d03b885SBenjamin Herrenschmidt int i; 17605d63f81cSMiles Chen phys_addr_t end; 17616d03b885SBenjamin Herrenschmidt 17626d03b885SBenjamin Herrenschmidt for (i = 0; i < type->cnt; i++) { 17636d03b885SBenjamin Herrenschmidt reg = &type->regions[i]; 17645d63f81cSMiles Chen end = reg->base + reg->size - 1; 17656d03b885SBenjamin Herrenschmidt 17665d63f81cSMiles Chen seq_printf(m, "%4d: ", i); 17675d63f81cSMiles Chen seq_printf(m, "%pa..%pa\n", ®->base, &end); 17686d03b885SBenjamin Herrenschmidt } 17696d03b885SBenjamin Herrenschmidt return 0; 17706d03b885SBenjamin Herrenschmidt } 17716d03b885SBenjamin Herrenschmidt 17726d03b885SBenjamin Herrenschmidt static int memblock_debug_open(struct inode *inode, struct file *file) 17736d03b885SBenjamin Herrenschmidt { 17746d03b885SBenjamin Herrenschmidt return single_open(file, memblock_debug_show, inode->i_private); 17756d03b885SBenjamin Herrenschmidt } 17766d03b885SBenjamin Herrenschmidt 17776d03b885SBenjamin Herrenschmidt static const struct file_operations memblock_debug_fops = { 17786d03b885SBenjamin Herrenschmidt .open = memblock_debug_open, 17796d03b885SBenjamin Herrenschmidt .read = seq_read, 17806d03b885SBenjamin Herrenschmidt .llseek = seq_lseek, 17816d03b885SBenjamin Herrenschmidt .release = single_release, 17826d03b885SBenjamin Herrenschmidt }; 17836d03b885SBenjamin Herrenschmidt 17846d03b885SBenjamin Herrenschmidt static int __init memblock_init_debugfs(void) 17856d03b885SBenjamin Herrenschmidt { 17866d03b885SBenjamin Herrenschmidt struct dentry *root = debugfs_create_dir("memblock", NULL); 17876d03b885SBenjamin Herrenschmidt if (!root) 17886d03b885SBenjamin Herrenschmidt return -ENXIO; 17896d03b885SBenjamin Herrenschmidt debugfs_create_file("memory", S_IRUGO, root, &memblock.memory, &memblock_debug_fops); 17906d03b885SBenjamin Herrenschmidt debugfs_create_file("reserved", S_IRUGO, root, &memblock.reserved, &memblock_debug_fops); 179170210ed9SPhilipp Hachtmann #ifdef CONFIG_HAVE_MEMBLOCK_PHYS_MAP 179270210ed9SPhilipp Hachtmann debugfs_create_file("physmem", S_IRUGO, root, &memblock.physmem, &memblock_debug_fops); 179370210ed9SPhilipp Hachtmann #endif 17946d03b885SBenjamin Herrenschmidt 17956d03b885SBenjamin Herrenschmidt return 0; 17966d03b885SBenjamin Herrenschmidt } 17976d03b885SBenjamin Herrenschmidt __initcall(memblock_init_debugfs); 17986d03b885SBenjamin Herrenschmidt 17996d03b885SBenjamin Herrenschmidt #endif /* CONFIG_DEBUG_FS */ 1800