xref: /linux/mm/memblock.c (revision 2bfc2862c4fe38379a2fb2cfba33fad32ccb4ff4)
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 
2379442ed1STang Chen #include <asm-generic/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
571aadc056STejun Heo static int memblock_can_resize __initdata_memblock;
58181eb394SGavin Shan static int memblock_memory_in_slab __initdata_memblock = 0;
59181eb394SGavin Shan static int memblock_reserved_in_slab __initdata_memblock = 0;
6095f72d1eSYinghai Lu 
61142b45a7SBenjamin Herrenschmidt /* inline so we don't get a warning when pr_debug is compiled out */
62c2233116SRaghavendra D Prabhu static __init_memblock const char *
63c2233116SRaghavendra D Prabhu memblock_type_name(struct memblock_type *type)
64142b45a7SBenjamin Herrenschmidt {
65142b45a7SBenjamin Herrenschmidt 	if (type == &memblock.memory)
66142b45a7SBenjamin Herrenschmidt 		return "memory";
67142b45a7SBenjamin Herrenschmidt 	else if (type == &memblock.reserved)
68142b45a7SBenjamin Herrenschmidt 		return "reserved";
69142b45a7SBenjamin Herrenschmidt 	else
70142b45a7SBenjamin Herrenschmidt 		return "unknown";
71142b45a7SBenjamin Herrenschmidt }
72142b45a7SBenjamin Herrenschmidt 
73eb18f1b5STejun Heo /* adjust *@size so that (@base + *@size) doesn't overflow, return new size */
74eb18f1b5STejun Heo static inline phys_addr_t memblock_cap_size(phys_addr_t base, phys_addr_t *size)
75eb18f1b5STejun Heo {
76eb18f1b5STejun Heo 	return *size = min(*size, (phys_addr_t)ULLONG_MAX - base);
77eb18f1b5STejun Heo }
78eb18f1b5STejun Heo 
796ed311b2SBenjamin Herrenschmidt /*
806ed311b2SBenjamin Herrenschmidt  * Address comparison utilities
816ed311b2SBenjamin Herrenschmidt  */
8210d06439SYinghai Lu static unsigned long __init_memblock memblock_addrs_overlap(phys_addr_t base1, phys_addr_t size1,
832898cc4cSBenjamin Herrenschmidt 				       phys_addr_t base2, phys_addr_t size2)
8495f72d1eSYinghai Lu {
8595f72d1eSYinghai Lu 	return ((base1 < (base2 + size2)) && (base2 < (base1 + size1)));
8695f72d1eSYinghai Lu }
8795f72d1eSYinghai Lu 
882d7d3eb2SH Hartley Sweeten static long __init_memblock memblock_overlaps_region(struct memblock_type *type,
892d7d3eb2SH Hartley Sweeten 					phys_addr_t base, phys_addr_t size)
906ed311b2SBenjamin Herrenschmidt {
916ed311b2SBenjamin Herrenschmidt 	unsigned long i;
926ed311b2SBenjamin Herrenschmidt 
936ed311b2SBenjamin Herrenschmidt 	for (i = 0; i < type->cnt; i++) {
946ed311b2SBenjamin Herrenschmidt 		phys_addr_t rgnbase = type->regions[i].base;
956ed311b2SBenjamin Herrenschmidt 		phys_addr_t rgnsize = type->regions[i].size;
966ed311b2SBenjamin Herrenschmidt 		if (memblock_addrs_overlap(base, size, rgnbase, rgnsize))
976ed311b2SBenjamin Herrenschmidt 			break;
986ed311b2SBenjamin Herrenschmidt 	}
996ed311b2SBenjamin Herrenschmidt 
1006ed311b2SBenjamin Herrenschmidt 	return (i < type->cnt) ? i : -1;
1016ed311b2SBenjamin Herrenschmidt }
1026ed311b2SBenjamin Herrenschmidt 
10379442ed1STang Chen /*
10479442ed1STang Chen  * __memblock_find_range_bottom_up - find free area utility in bottom-up
10579442ed1STang Chen  * @start: start of candidate range
10679442ed1STang Chen  * @end: end of candidate range, can be %MEMBLOCK_ALLOC_{ANYWHERE|ACCESSIBLE}
10779442ed1STang Chen  * @size: size of free area to find
10879442ed1STang Chen  * @align: alignment of free area to find
109b1154233SGrygorii Strashko  * @nid: nid of the free area to find, %NUMA_NO_NODE for any node
11079442ed1STang Chen  *
11179442ed1STang Chen  * Utility called from memblock_find_in_range_node(), find free area bottom-up.
11279442ed1STang Chen  *
11379442ed1STang Chen  * RETURNS:
11479442ed1STang Chen  * Found address on success, 0 on failure.
11579442ed1STang Chen  */
11679442ed1STang Chen static phys_addr_t __init_memblock
11779442ed1STang Chen __memblock_find_range_bottom_up(phys_addr_t start, phys_addr_t end,
11879442ed1STang Chen 				phys_addr_t size, phys_addr_t align, int nid)
11979442ed1STang Chen {
12079442ed1STang Chen 	phys_addr_t this_start, this_end, cand;
12179442ed1STang Chen 	u64 i;
12279442ed1STang Chen 
12379442ed1STang Chen 	for_each_free_mem_range(i, nid, &this_start, &this_end, NULL) {
12479442ed1STang Chen 		this_start = clamp(this_start, start, end);
12579442ed1STang Chen 		this_end = clamp(this_end, start, end);
12679442ed1STang Chen 
12779442ed1STang Chen 		cand = round_up(this_start, align);
12879442ed1STang Chen 		if (cand < this_end && this_end - cand >= size)
12979442ed1STang Chen 			return cand;
13079442ed1STang Chen 	}
13179442ed1STang Chen 
13279442ed1STang Chen 	return 0;
13379442ed1STang Chen }
13479442ed1STang Chen 
1357bd0b0f0STejun Heo /**
1361402899eSTang Chen  * __memblock_find_range_top_down - find free area utility, in top-down
1371402899eSTang Chen  * @start: start of candidate range
1381402899eSTang Chen  * @end: end of candidate range, can be %MEMBLOCK_ALLOC_{ANYWHERE|ACCESSIBLE}
1391402899eSTang Chen  * @size: size of free area to find
1401402899eSTang Chen  * @align: alignment of free area to find
141b1154233SGrygorii Strashko  * @nid: nid of the free area to find, %NUMA_NO_NODE for any node
1421402899eSTang Chen  *
1431402899eSTang Chen  * Utility called from memblock_find_in_range_node(), find free area top-down.
1441402899eSTang Chen  *
1451402899eSTang Chen  * RETURNS:
14679442ed1STang Chen  * Found address on success, 0 on failure.
1471402899eSTang Chen  */
1481402899eSTang Chen static phys_addr_t __init_memblock
1491402899eSTang Chen __memblock_find_range_top_down(phys_addr_t start, phys_addr_t end,
1501402899eSTang Chen 			       phys_addr_t size, phys_addr_t align, int nid)
1511402899eSTang Chen {
1521402899eSTang Chen 	phys_addr_t this_start, this_end, cand;
1531402899eSTang Chen 	u64 i;
1541402899eSTang Chen 
1551402899eSTang Chen 	for_each_free_mem_range_reverse(i, nid, &this_start, &this_end, NULL) {
1561402899eSTang Chen 		this_start = clamp(this_start, start, end);
1571402899eSTang Chen 		this_end = clamp(this_end, start, end);
1581402899eSTang Chen 
1591402899eSTang Chen 		if (this_end < size)
1601402899eSTang Chen 			continue;
1611402899eSTang Chen 
1621402899eSTang Chen 		cand = round_down(this_end - size, align);
1631402899eSTang Chen 		if (cand >= this_start)
1641402899eSTang Chen 			return cand;
1651402899eSTang Chen 	}
1661402899eSTang Chen 
1671402899eSTang Chen 	return 0;
1681402899eSTang Chen }
1691402899eSTang Chen 
1701402899eSTang Chen /**
1717bd0b0f0STejun Heo  * memblock_find_in_range_node - find free area in given range and node
1727bd0b0f0STejun Heo  * @size: size of free area to find
1737bd0b0f0STejun Heo  * @align: alignment of free area to find
17487029ee9SGrygorii Strashko  * @start: start of candidate range
17587029ee9SGrygorii Strashko  * @end: end of candidate range, can be %MEMBLOCK_ALLOC_{ANYWHERE|ACCESSIBLE}
176b1154233SGrygorii Strashko  * @nid: nid of the free area to find, %NUMA_NO_NODE for any node
1777bd0b0f0STejun Heo  *
1787bd0b0f0STejun Heo  * Find @size free area aligned to @align in the specified range and node.
1797bd0b0f0STejun Heo  *
18079442ed1STang Chen  * When allocation direction is bottom-up, the @start should be greater
18179442ed1STang Chen  * than the end of the kernel image. Otherwise, it will be trimmed. The
18279442ed1STang Chen  * reason is that we want the bottom-up allocation just near the kernel
18379442ed1STang Chen  * image so it is highly likely that the allocated memory and the kernel
18479442ed1STang Chen  * will reside in the same node.
18579442ed1STang Chen  *
18679442ed1STang Chen  * If bottom-up allocation failed, will try to allocate memory top-down.
18779442ed1STang Chen  *
1887bd0b0f0STejun Heo  * RETURNS:
18979442ed1STang Chen  * Found address on success, 0 on failure.
1906ed311b2SBenjamin Herrenschmidt  */
19187029ee9SGrygorii Strashko phys_addr_t __init_memblock memblock_find_in_range_node(phys_addr_t size,
19287029ee9SGrygorii Strashko 					phys_addr_t align, phys_addr_t start,
19387029ee9SGrygorii Strashko 					phys_addr_t end, int nid)
194f7210e6cSTang Chen {
19579442ed1STang Chen 	int ret;
19679442ed1STang Chen 	phys_addr_t kernel_end;
19779442ed1STang Chen 
198f7210e6cSTang Chen 	/* pump up @end */
199f7210e6cSTang Chen 	if (end == MEMBLOCK_ALLOC_ACCESSIBLE)
200f7210e6cSTang Chen 		end = memblock.current_limit;
201f7210e6cSTang Chen 
202f7210e6cSTang Chen 	/* avoid allocating the first page */
203f7210e6cSTang Chen 	start = max_t(phys_addr_t, start, PAGE_SIZE);
204f7210e6cSTang Chen 	end = max(start, end);
20579442ed1STang Chen 	kernel_end = __pa_symbol(_end);
20679442ed1STang Chen 
20779442ed1STang Chen 	/*
20879442ed1STang Chen 	 * try bottom-up allocation only when bottom-up mode
20979442ed1STang Chen 	 * is set and @end is above the kernel image.
21079442ed1STang Chen 	 */
21179442ed1STang Chen 	if (memblock_bottom_up() && end > kernel_end) {
21279442ed1STang Chen 		phys_addr_t bottom_up_start;
21379442ed1STang Chen 
21479442ed1STang Chen 		/* make sure we will allocate above the kernel */
21579442ed1STang Chen 		bottom_up_start = max(start, kernel_end);
21679442ed1STang Chen 
21779442ed1STang Chen 		/* ok, try bottom-up allocation first */
21879442ed1STang Chen 		ret = __memblock_find_range_bottom_up(bottom_up_start, end,
21979442ed1STang Chen 						      size, align, nid);
22079442ed1STang Chen 		if (ret)
22179442ed1STang Chen 			return ret;
22279442ed1STang Chen 
22379442ed1STang Chen 		/*
22479442ed1STang Chen 		 * we always limit bottom-up allocation above the kernel,
22579442ed1STang Chen 		 * but top-down allocation doesn't have the limit, so
22679442ed1STang Chen 		 * retrying top-down allocation may succeed when bottom-up
22779442ed1STang Chen 		 * allocation failed.
22879442ed1STang Chen 		 *
22979442ed1STang Chen 		 * bottom-up allocation is expected to be fail very rarely,
23079442ed1STang Chen 		 * so we use WARN_ONCE() here to see the stack trace if
23179442ed1STang Chen 		 * fail happens.
23279442ed1STang Chen 		 */
23379442ed1STang Chen 		WARN_ONCE(1, "memblock: bottom-up allocation failed, "
23479442ed1STang Chen 			     "memory hotunplug may be affected\n");
23579442ed1STang Chen 	}
236f7210e6cSTang Chen 
2371402899eSTang Chen 	return __memblock_find_range_top_down(start, end, size, align, nid);
238f7210e6cSTang Chen }
2396ed311b2SBenjamin Herrenschmidt 
2407bd0b0f0STejun Heo /**
2417bd0b0f0STejun Heo  * memblock_find_in_range - find free area in given range
2427bd0b0f0STejun Heo  * @start: start of candidate range
2437bd0b0f0STejun Heo  * @end: end of candidate range, can be %MEMBLOCK_ALLOC_{ANYWHERE|ACCESSIBLE}
2447bd0b0f0STejun Heo  * @size: size of free area to find
2457bd0b0f0STejun Heo  * @align: alignment of free area to find
2467bd0b0f0STejun Heo  *
2477bd0b0f0STejun Heo  * Find @size free area aligned to @align in the specified range.
2487bd0b0f0STejun Heo  *
2497bd0b0f0STejun Heo  * RETURNS:
25079442ed1STang Chen  * Found address on success, 0 on failure.
2517bd0b0f0STejun Heo  */
2527bd0b0f0STejun Heo phys_addr_t __init_memblock memblock_find_in_range(phys_addr_t start,
2537bd0b0f0STejun Heo 					phys_addr_t end, phys_addr_t size,
2547bd0b0f0STejun Heo 					phys_addr_t align)
2557bd0b0f0STejun Heo {
25687029ee9SGrygorii Strashko 	return memblock_find_in_range_node(size, align, start, end,
257b1154233SGrygorii Strashko 					    NUMA_NO_NODE);
2587bd0b0f0STejun Heo }
2597bd0b0f0STejun Heo 
26010d06439SYinghai Lu static void __init_memblock memblock_remove_region(struct memblock_type *type, unsigned long r)
26195f72d1eSYinghai Lu {
2621440c4e2STejun Heo 	type->total_size -= type->regions[r].size;
2637c0caeb8STejun Heo 	memmove(&type->regions[r], &type->regions[r + 1],
2647c0caeb8STejun Heo 		(type->cnt - (r + 1)) * sizeof(type->regions[r]));
265e3239ff9SBenjamin Herrenschmidt 	type->cnt--;
26695f72d1eSYinghai Lu 
2678f7a6605SBenjamin Herrenschmidt 	/* Special case for empty arrays */
2688f7a6605SBenjamin Herrenschmidt 	if (type->cnt == 0) {
2691440c4e2STejun Heo 		WARN_ON(type->total_size != 0);
2708f7a6605SBenjamin Herrenschmidt 		type->cnt = 1;
2718f7a6605SBenjamin Herrenschmidt 		type->regions[0].base = 0;
2728f7a6605SBenjamin Herrenschmidt 		type->regions[0].size = 0;
27366a20757STang Chen 		type->regions[0].flags = 0;
2747c0caeb8STejun Heo 		memblock_set_region_node(&type->regions[0], MAX_NUMNODES);
2758f7a6605SBenjamin Herrenschmidt 	}
27695f72d1eSYinghai Lu }
27795f72d1eSYinghai Lu 
278354f17e1SPhilipp Hachtmann #ifdef CONFIG_ARCH_DISCARD_MEMBLOCK
279354f17e1SPhilipp Hachtmann 
28029f67386SYinghai Lu phys_addr_t __init_memblock get_allocated_memblock_reserved_regions_info(
28129f67386SYinghai Lu 					phys_addr_t *addr)
28229f67386SYinghai Lu {
28329f67386SYinghai Lu 	if (memblock.reserved.regions == memblock_reserved_init_regions)
28429f67386SYinghai Lu 		return 0;
28529f67386SYinghai Lu 
28629f67386SYinghai Lu 	*addr = __pa(memblock.reserved.regions);
28729f67386SYinghai Lu 
28829f67386SYinghai Lu 	return PAGE_ALIGN(sizeof(struct memblock_region) *
28929f67386SYinghai Lu 			  memblock.reserved.max);
29029f67386SYinghai Lu }
29129f67386SYinghai Lu 
2925e270e25SPhilipp Hachtmann phys_addr_t __init_memblock get_allocated_memblock_memory_regions_info(
2935e270e25SPhilipp Hachtmann 					phys_addr_t *addr)
2945e270e25SPhilipp Hachtmann {
2955e270e25SPhilipp Hachtmann 	if (memblock.memory.regions == memblock_memory_init_regions)
2965e270e25SPhilipp Hachtmann 		return 0;
2975e270e25SPhilipp Hachtmann 
2985e270e25SPhilipp Hachtmann 	*addr = __pa(memblock.memory.regions);
2995e270e25SPhilipp Hachtmann 
3005e270e25SPhilipp Hachtmann 	return PAGE_ALIGN(sizeof(struct memblock_region) *
3015e270e25SPhilipp Hachtmann 			  memblock.memory.max);
3025e270e25SPhilipp Hachtmann }
3035e270e25SPhilipp Hachtmann 
3045e270e25SPhilipp Hachtmann #endif
3055e270e25SPhilipp Hachtmann 
30648c3b583SGreg Pearson /**
30748c3b583SGreg Pearson  * memblock_double_array - double the size of the memblock regions array
30848c3b583SGreg Pearson  * @type: memblock type of the regions array being doubled
30948c3b583SGreg Pearson  * @new_area_start: starting address of memory range to avoid overlap with
31048c3b583SGreg Pearson  * @new_area_size: size of memory range to avoid overlap with
31148c3b583SGreg Pearson  *
31248c3b583SGreg Pearson  * Double the size of the @type regions array. If memblock is being used to
31348c3b583SGreg Pearson  * allocate memory for a new reserved regions array and there is a previously
31448c3b583SGreg Pearson  * allocated memory range [@new_area_start,@new_area_start+@new_area_size]
31548c3b583SGreg Pearson  * waiting to be reserved, ensure the memory used by the new array does
31648c3b583SGreg Pearson  * not overlap.
31748c3b583SGreg Pearson  *
31848c3b583SGreg Pearson  * RETURNS:
31948c3b583SGreg Pearson  * 0 on success, -1 on failure.
32048c3b583SGreg Pearson  */
32148c3b583SGreg Pearson static int __init_memblock memblock_double_array(struct memblock_type *type,
32248c3b583SGreg Pearson 						phys_addr_t new_area_start,
32348c3b583SGreg Pearson 						phys_addr_t new_area_size)
324142b45a7SBenjamin Herrenschmidt {
325142b45a7SBenjamin Herrenschmidt 	struct memblock_region *new_array, *old_array;
32629f67386SYinghai Lu 	phys_addr_t old_alloc_size, new_alloc_size;
327142b45a7SBenjamin Herrenschmidt 	phys_addr_t old_size, new_size, addr;
328142b45a7SBenjamin Herrenschmidt 	int use_slab = slab_is_available();
329181eb394SGavin Shan 	int *in_slab;
330142b45a7SBenjamin Herrenschmidt 
331142b45a7SBenjamin Herrenschmidt 	/* We don't allow resizing until we know about the reserved regions
332142b45a7SBenjamin Herrenschmidt 	 * of memory that aren't suitable for allocation
333142b45a7SBenjamin Herrenschmidt 	 */
334142b45a7SBenjamin Herrenschmidt 	if (!memblock_can_resize)
335142b45a7SBenjamin Herrenschmidt 		return -1;
336142b45a7SBenjamin Herrenschmidt 
337142b45a7SBenjamin Herrenschmidt 	/* Calculate new doubled size */
338142b45a7SBenjamin Herrenschmidt 	old_size = type->max * sizeof(struct memblock_region);
339142b45a7SBenjamin Herrenschmidt 	new_size = old_size << 1;
34029f67386SYinghai Lu 	/*
34129f67386SYinghai Lu 	 * We need to allocated new one align to PAGE_SIZE,
34229f67386SYinghai Lu 	 *   so we can free them completely later.
34329f67386SYinghai Lu 	 */
34429f67386SYinghai Lu 	old_alloc_size = PAGE_ALIGN(old_size);
34529f67386SYinghai Lu 	new_alloc_size = PAGE_ALIGN(new_size);
346142b45a7SBenjamin Herrenschmidt 
347181eb394SGavin Shan 	/* Retrieve the slab flag */
348181eb394SGavin Shan 	if (type == &memblock.memory)
349181eb394SGavin Shan 		in_slab = &memblock_memory_in_slab;
350181eb394SGavin Shan 	else
351181eb394SGavin Shan 		in_slab = &memblock_reserved_in_slab;
352181eb394SGavin Shan 
353142b45a7SBenjamin Herrenschmidt 	/* Try to find some space for it.
354142b45a7SBenjamin Herrenschmidt 	 *
355142b45a7SBenjamin Herrenschmidt 	 * WARNING: We assume that either slab_is_available() and we use it or
356fd07383bSAndrew Morton 	 * we use MEMBLOCK for allocations. That means that this is unsafe to
357fd07383bSAndrew Morton 	 * use when bootmem is currently active (unless bootmem itself is
358fd07383bSAndrew Morton 	 * implemented on top of MEMBLOCK which isn't the case yet)
359142b45a7SBenjamin Herrenschmidt 	 *
360142b45a7SBenjamin Herrenschmidt 	 * This should however not be an issue for now, as we currently only
361fd07383bSAndrew Morton 	 * call into MEMBLOCK while it's still active, or much later when slab
362fd07383bSAndrew Morton 	 * is active for memory hotplug operations
363142b45a7SBenjamin Herrenschmidt 	 */
364142b45a7SBenjamin Herrenschmidt 	if (use_slab) {
365142b45a7SBenjamin Herrenschmidt 		new_array = kmalloc(new_size, GFP_KERNEL);
3661f5026a7STejun Heo 		addr = new_array ? __pa(new_array) : 0;
3674e2f0775SGavin Shan 	} else {
36848c3b583SGreg Pearson 		/* only exclude range when trying to double reserved.regions */
36948c3b583SGreg Pearson 		if (type != &memblock.reserved)
37048c3b583SGreg Pearson 			new_area_start = new_area_size = 0;
37148c3b583SGreg Pearson 
37248c3b583SGreg Pearson 		addr = memblock_find_in_range(new_area_start + new_area_size,
37348c3b583SGreg Pearson 						memblock.current_limit,
37429f67386SYinghai Lu 						new_alloc_size, PAGE_SIZE);
37548c3b583SGreg Pearson 		if (!addr && new_area_size)
37648c3b583SGreg Pearson 			addr = memblock_find_in_range(0,
37748c3b583SGreg Pearson 				min(new_area_start, memblock.current_limit),
37829f67386SYinghai Lu 				new_alloc_size, PAGE_SIZE);
37948c3b583SGreg Pearson 
38015674868SSachin Kamat 		new_array = addr ? __va(addr) : NULL;
3814e2f0775SGavin Shan 	}
3821f5026a7STejun Heo 	if (!addr) {
383142b45a7SBenjamin Herrenschmidt 		pr_err("memblock: Failed to double %s array from %ld to %ld entries !\n",
384142b45a7SBenjamin Herrenschmidt 		       memblock_type_name(type), type->max, type->max * 2);
385142b45a7SBenjamin Herrenschmidt 		return -1;
386142b45a7SBenjamin Herrenschmidt 	}
387142b45a7SBenjamin Herrenschmidt 
388fd07383bSAndrew Morton 	memblock_dbg("memblock: %s is doubled to %ld at [%#010llx-%#010llx]",
389fd07383bSAndrew Morton 			memblock_type_name(type), type->max * 2, (u64)addr,
390fd07383bSAndrew Morton 			(u64)addr + new_size - 1);
391ea9e4376SYinghai Lu 
392fd07383bSAndrew Morton 	/*
393fd07383bSAndrew Morton 	 * Found space, we now need to move the array over before we add the
394fd07383bSAndrew Morton 	 * reserved region since it may be our reserved array itself that is
395fd07383bSAndrew Morton 	 * full.
396142b45a7SBenjamin Herrenschmidt 	 */
397142b45a7SBenjamin Herrenschmidt 	memcpy(new_array, type->regions, old_size);
398142b45a7SBenjamin Herrenschmidt 	memset(new_array + type->max, 0, old_size);
399142b45a7SBenjamin Herrenschmidt 	old_array = type->regions;
400142b45a7SBenjamin Herrenschmidt 	type->regions = new_array;
401142b45a7SBenjamin Herrenschmidt 	type->max <<= 1;
402142b45a7SBenjamin Herrenschmidt 
403fd07383bSAndrew Morton 	/* Free old array. We needn't free it if the array is the static one */
404181eb394SGavin Shan 	if (*in_slab)
405181eb394SGavin Shan 		kfree(old_array);
406181eb394SGavin Shan 	else if (old_array != memblock_memory_init_regions &&
407142b45a7SBenjamin Herrenschmidt 		 old_array != memblock_reserved_init_regions)
40829f67386SYinghai Lu 		memblock_free(__pa(old_array), old_alloc_size);
409142b45a7SBenjamin Herrenschmidt 
410fd07383bSAndrew Morton 	/*
411fd07383bSAndrew Morton 	 * Reserve the new array if that comes from the memblock.  Otherwise, we
412fd07383bSAndrew Morton 	 * needn't do it
413181eb394SGavin Shan 	 */
414181eb394SGavin Shan 	if (!use_slab)
41529f67386SYinghai Lu 		BUG_ON(memblock_reserve(addr, new_alloc_size));
416181eb394SGavin Shan 
417181eb394SGavin Shan 	/* Update slab flag */
418181eb394SGavin Shan 	*in_slab = use_slab;
419181eb394SGavin Shan 
420142b45a7SBenjamin Herrenschmidt 	return 0;
421142b45a7SBenjamin Herrenschmidt }
422142b45a7SBenjamin Herrenschmidt 
423784656f9STejun Heo /**
424784656f9STejun Heo  * memblock_merge_regions - merge neighboring compatible regions
425784656f9STejun Heo  * @type: memblock type to scan
426784656f9STejun Heo  *
427784656f9STejun Heo  * Scan @type and merge neighboring compatible regions.
428784656f9STejun Heo  */
429784656f9STejun Heo static void __init_memblock memblock_merge_regions(struct memblock_type *type)
430784656f9STejun Heo {
431784656f9STejun Heo 	int i = 0;
432784656f9STejun Heo 
433784656f9STejun Heo 	/* cnt never goes below 1 */
434784656f9STejun Heo 	while (i < type->cnt - 1) {
435784656f9STejun Heo 		struct memblock_region *this = &type->regions[i];
436784656f9STejun Heo 		struct memblock_region *next = &type->regions[i + 1];
437784656f9STejun Heo 
4387c0caeb8STejun Heo 		if (this->base + this->size != next->base ||
4397c0caeb8STejun Heo 		    memblock_get_region_node(this) !=
44066a20757STang Chen 		    memblock_get_region_node(next) ||
44166a20757STang Chen 		    this->flags != next->flags) {
442784656f9STejun Heo 			BUG_ON(this->base + this->size > next->base);
443784656f9STejun Heo 			i++;
444784656f9STejun Heo 			continue;
445784656f9STejun Heo 		}
446784656f9STejun Heo 
447784656f9STejun Heo 		this->size += next->size;
448c0232ae8SLin Feng 		/* move forward from next + 1, index of which is i + 2 */
449c0232ae8SLin Feng 		memmove(next, next + 1, (type->cnt - (i + 2)) * sizeof(*next));
450784656f9STejun Heo 		type->cnt--;
451784656f9STejun Heo 	}
452784656f9STejun Heo }
453784656f9STejun Heo 
454784656f9STejun Heo /**
455784656f9STejun Heo  * memblock_insert_region - insert new memblock region
456784656f9STejun Heo  * @type:	memblock type to insert into
457784656f9STejun Heo  * @idx:	index for the insertion point
458784656f9STejun Heo  * @base:	base address of the new region
459784656f9STejun Heo  * @size:	size of the new region
460209ff86dSTang Chen  * @nid:	node id of the new region
46166a20757STang Chen  * @flags:	flags of the new region
462784656f9STejun Heo  *
463784656f9STejun Heo  * Insert new memblock region [@base,@base+@size) into @type at @idx.
464784656f9STejun Heo  * @type must already have extra room to accomodate the new region.
465784656f9STejun Heo  */
466784656f9STejun Heo static void __init_memblock memblock_insert_region(struct memblock_type *type,
467784656f9STejun Heo 						   int idx, phys_addr_t base,
46866a20757STang Chen 						   phys_addr_t size,
46966a20757STang Chen 						   int nid, unsigned long flags)
470784656f9STejun Heo {
471784656f9STejun Heo 	struct memblock_region *rgn = &type->regions[idx];
472784656f9STejun Heo 
473784656f9STejun Heo 	BUG_ON(type->cnt >= type->max);
474784656f9STejun Heo 	memmove(rgn + 1, rgn, (type->cnt - idx) * sizeof(*rgn));
475784656f9STejun Heo 	rgn->base = base;
476784656f9STejun Heo 	rgn->size = size;
47766a20757STang Chen 	rgn->flags = flags;
4787c0caeb8STejun Heo 	memblock_set_region_node(rgn, nid);
479784656f9STejun Heo 	type->cnt++;
4801440c4e2STejun Heo 	type->total_size += size;
481784656f9STejun Heo }
482784656f9STejun Heo 
483784656f9STejun Heo /**
484f1af9d3aSPhilipp Hachtmann  * memblock_add_range - add new memblock region
485784656f9STejun Heo  * @type: memblock type to add new region into
486784656f9STejun Heo  * @base: base address of the new region
487784656f9STejun Heo  * @size: size of the new region
4887fb0bc3fSTejun Heo  * @nid: nid of the new region
48966a20757STang Chen  * @flags: flags of the new region
490784656f9STejun Heo  *
491784656f9STejun Heo  * Add new memblock region [@base,@base+@size) into @type.  The new region
492784656f9STejun Heo  * is allowed to overlap with existing ones - overlaps don't affect already
493784656f9STejun Heo  * existing regions.  @type is guaranteed to be minimal (all neighbouring
494784656f9STejun Heo  * compatible regions are merged) after the addition.
495784656f9STejun Heo  *
496784656f9STejun Heo  * RETURNS:
497784656f9STejun Heo  * 0 on success, -errno on failure.
498784656f9STejun Heo  */
499f1af9d3aSPhilipp Hachtmann int __init_memblock memblock_add_range(struct memblock_type *type,
50066a20757STang Chen 				phys_addr_t base, phys_addr_t size,
50166a20757STang Chen 				int nid, unsigned long flags)
50295f72d1eSYinghai Lu {
503784656f9STejun Heo 	bool insert = false;
504eb18f1b5STejun Heo 	phys_addr_t obase = base;
505eb18f1b5STejun Heo 	phys_addr_t end = base + memblock_cap_size(base, &size);
506784656f9STejun Heo 	int i, nr_new;
50795f72d1eSYinghai Lu 
508b3dc627cSTejun Heo 	if (!size)
509b3dc627cSTejun Heo 		return 0;
510b3dc627cSTejun Heo 
511784656f9STejun Heo 	/* special case for empty array */
512784656f9STejun Heo 	if (type->regions[0].size == 0) {
5131440c4e2STejun Heo 		WARN_ON(type->cnt != 1 || type->total_size);
514784656f9STejun Heo 		type->regions[0].base = base;
515784656f9STejun Heo 		type->regions[0].size = size;
51666a20757STang Chen 		type->regions[0].flags = flags;
5177fb0bc3fSTejun Heo 		memblock_set_region_node(&type->regions[0], nid);
5181440c4e2STejun Heo 		type->total_size = size;
519784656f9STejun Heo 		return 0;
520784656f9STejun Heo 	}
521784656f9STejun Heo repeat:
522784656f9STejun Heo 	/*
523784656f9STejun Heo 	 * The following is executed twice.  Once with %false @insert and
524784656f9STejun Heo 	 * then with %true.  The first counts the number of regions needed
525784656f9STejun Heo 	 * to accomodate the new area.  The second actually inserts them.
526784656f9STejun Heo 	 */
527784656f9STejun Heo 	base = obase;
528784656f9STejun Heo 	nr_new = 0;
529784656f9STejun Heo 
5308f7a6605SBenjamin Herrenschmidt 	for (i = 0; i < type->cnt; i++) {
5318f7a6605SBenjamin Herrenschmidt 		struct memblock_region *rgn = &type->regions[i];
532784656f9STejun Heo 		phys_addr_t rbase = rgn->base;
533784656f9STejun Heo 		phys_addr_t rend = rbase + rgn->size;
5348f7a6605SBenjamin Herrenschmidt 
535784656f9STejun Heo 		if (rbase >= end)
5368f7a6605SBenjamin Herrenschmidt 			break;
537784656f9STejun Heo 		if (rend <= base)
538784656f9STejun Heo 			continue;
539784656f9STejun Heo 		/*
540784656f9STejun Heo 		 * @rgn overlaps.  If it separates the lower part of new
541784656f9STejun Heo 		 * area, insert that portion.
5428f7a6605SBenjamin Herrenschmidt 		 */
543784656f9STejun Heo 		if (rbase > base) {
544784656f9STejun Heo 			nr_new++;
545784656f9STejun Heo 			if (insert)
546784656f9STejun Heo 				memblock_insert_region(type, i++, base,
54766a20757STang Chen 						       rbase - base, nid,
54866a20757STang Chen 						       flags);
549784656f9STejun Heo 		}
550784656f9STejun Heo 		/* area below @rend is dealt with, forget about it */
551784656f9STejun Heo 		base = min(rend, end);
5528f7a6605SBenjamin Herrenschmidt 	}
5538f7a6605SBenjamin Herrenschmidt 
554784656f9STejun Heo 	/* insert the remaining portion */
555784656f9STejun Heo 	if (base < end) {
556784656f9STejun Heo 		nr_new++;
557784656f9STejun Heo 		if (insert)
55866a20757STang Chen 			memblock_insert_region(type, i, base, end - base,
55966a20757STang Chen 					       nid, flags);
5608f7a6605SBenjamin Herrenschmidt 	}
5618f7a6605SBenjamin Herrenschmidt 
562784656f9STejun Heo 	/*
563784656f9STejun Heo 	 * If this was the first round, resize array and repeat for actual
564784656f9STejun Heo 	 * insertions; otherwise, merge and return.
5658f7a6605SBenjamin Herrenschmidt 	 */
566784656f9STejun Heo 	if (!insert) {
567784656f9STejun Heo 		while (type->cnt + nr_new > type->max)
56848c3b583SGreg Pearson 			if (memblock_double_array(type, obase, size) < 0)
569784656f9STejun Heo 				return -ENOMEM;
570784656f9STejun Heo 		insert = true;
571784656f9STejun Heo 		goto repeat;
57295f72d1eSYinghai Lu 	} else {
573784656f9STejun Heo 		memblock_merge_regions(type);
57495f72d1eSYinghai Lu 		return 0;
57595f72d1eSYinghai Lu 	}
576784656f9STejun Heo }
57795f72d1eSYinghai Lu 
5787fb0bc3fSTejun Heo int __init_memblock memblock_add_node(phys_addr_t base, phys_addr_t size,
5797fb0bc3fSTejun Heo 				       int nid)
5807fb0bc3fSTejun Heo {
581f1af9d3aSPhilipp Hachtmann 	return memblock_add_range(&memblock.memory, base, size, nid, 0);
5827fb0bc3fSTejun Heo }
5837fb0bc3fSTejun Heo 
584581adcbeSTejun Heo int __init_memblock memblock_add(phys_addr_t base, phys_addr_t size)
58595f72d1eSYinghai Lu {
586f1af9d3aSPhilipp Hachtmann 	return memblock_add_range(&memblock.memory, base, size,
58766a20757STang Chen 				   MAX_NUMNODES, 0);
58895f72d1eSYinghai Lu }
58995f72d1eSYinghai Lu 
5906a9ceb31STejun Heo /**
5916a9ceb31STejun Heo  * memblock_isolate_range - isolate given range into disjoint memblocks
5926a9ceb31STejun Heo  * @type: memblock type to isolate range for
5936a9ceb31STejun Heo  * @base: base of range to isolate
5946a9ceb31STejun Heo  * @size: size of range to isolate
5956a9ceb31STejun Heo  * @start_rgn: out parameter for the start of isolated region
5966a9ceb31STejun Heo  * @end_rgn: out parameter for the end of isolated region
5976a9ceb31STejun Heo  *
5986a9ceb31STejun Heo  * Walk @type and ensure that regions don't cross the boundaries defined by
5996a9ceb31STejun Heo  * [@base,@base+@size).  Crossing regions are split at the boundaries,
6006a9ceb31STejun Heo  * which may create at most two more regions.  The index of the first
6016a9ceb31STejun Heo  * region inside the range is returned in *@start_rgn and end in *@end_rgn.
6026a9ceb31STejun Heo  *
6036a9ceb31STejun Heo  * RETURNS:
6046a9ceb31STejun Heo  * 0 on success, -errno on failure.
6056a9ceb31STejun Heo  */
6066a9ceb31STejun Heo static int __init_memblock memblock_isolate_range(struct memblock_type *type,
6076a9ceb31STejun Heo 					phys_addr_t base, phys_addr_t size,
6086a9ceb31STejun Heo 					int *start_rgn, int *end_rgn)
6096a9ceb31STejun Heo {
610eb18f1b5STejun Heo 	phys_addr_t end = base + memblock_cap_size(base, &size);
6116a9ceb31STejun Heo 	int i;
6126a9ceb31STejun Heo 
6136a9ceb31STejun Heo 	*start_rgn = *end_rgn = 0;
6146a9ceb31STejun Heo 
615b3dc627cSTejun Heo 	if (!size)
616b3dc627cSTejun Heo 		return 0;
617b3dc627cSTejun Heo 
6186a9ceb31STejun Heo 	/* we'll create at most two more regions */
6196a9ceb31STejun Heo 	while (type->cnt + 2 > type->max)
62048c3b583SGreg Pearson 		if (memblock_double_array(type, base, size) < 0)
6216a9ceb31STejun Heo 			return -ENOMEM;
6226a9ceb31STejun Heo 
6236a9ceb31STejun Heo 	for (i = 0; i < type->cnt; i++) {
6246a9ceb31STejun Heo 		struct memblock_region *rgn = &type->regions[i];
6256a9ceb31STejun Heo 		phys_addr_t rbase = rgn->base;
6266a9ceb31STejun Heo 		phys_addr_t rend = rbase + rgn->size;
6276a9ceb31STejun Heo 
6286a9ceb31STejun Heo 		if (rbase >= end)
6296a9ceb31STejun Heo 			break;
6306a9ceb31STejun Heo 		if (rend <= base)
6316a9ceb31STejun Heo 			continue;
6326a9ceb31STejun Heo 
6336a9ceb31STejun Heo 		if (rbase < base) {
6346a9ceb31STejun Heo 			/*
6356a9ceb31STejun Heo 			 * @rgn intersects from below.  Split and continue
6366a9ceb31STejun Heo 			 * to process the next region - the new top half.
6376a9ceb31STejun Heo 			 */
6386a9ceb31STejun Heo 			rgn->base = base;
6391440c4e2STejun Heo 			rgn->size -= base - rbase;
6401440c4e2STejun Heo 			type->total_size -= base - rbase;
6416a9ceb31STejun Heo 			memblock_insert_region(type, i, rbase, base - rbase,
64266a20757STang Chen 					       memblock_get_region_node(rgn),
64366a20757STang Chen 					       rgn->flags);
6446a9ceb31STejun Heo 		} else if (rend > end) {
6456a9ceb31STejun Heo 			/*
6466a9ceb31STejun Heo 			 * @rgn intersects from above.  Split and redo the
6476a9ceb31STejun Heo 			 * current region - the new bottom half.
6486a9ceb31STejun Heo 			 */
6496a9ceb31STejun Heo 			rgn->base = end;
6501440c4e2STejun Heo 			rgn->size -= end - rbase;
6511440c4e2STejun Heo 			type->total_size -= end - rbase;
6526a9ceb31STejun Heo 			memblock_insert_region(type, i--, rbase, end - rbase,
65366a20757STang Chen 					       memblock_get_region_node(rgn),
65466a20757STang Chen 					       rgn->flags);
6556a9ceb31STejun Heo 		} else {
6566a9ceb31STejun Heo 			/* @rgn is fully contained, record it */
6576a9ceb31STejun Heo 			if (!*end_rgn)
6586a9ceb31STejun Heo 				*start_rgn = i;
6596a9ceb31STejun Heo 			*end_rgn = i + 1;
6606a9ceb31STejun Heo 		}
6616a9ceb31STejun Heo 	}
6626a9ceb31STejun Heo 
6636a9ceb31STejun Heo 	return 0;
6646a9ceb31STejun Heo }
6656a9ceb31STejun Heo 
666f1af9d3aSPhilipp Hachtmann int __init_memblock memblock_remove_range(struct memblock_type *type,
6678f7a6605SBenjamin Herrenschmidt 					  phys_addr_t base, phys_addr_t size)
66895f72d1eSYinghai Lu {
66971936180STejun Heo 	int start_rgn, end_rgn;
67071936180STejun Heo 	int i, ret;
67195f72d1eSYinghai Lu 
67271936180STejun Heo 	ret = memblock_isolate_range(type, base, size, &start_rgn, &end_rgn);
67371936180STejun Heo 	if (ret)
67471936180STejun Heo 		return ret;
67595f72d1eSYinghai Lu 
67671936180STejun Heo 	for (i = end_rgn - 1; i >= start_rgn; i--)
67771936180STejun Heo 		memblock_remove_region(type, i);
67895f72d1eSYinghai Lu 	return 0;
67995f72d1eSYinghai Lu }
68095f72d1eSYinghai Lu 
681581adcbeSTejun Heo int __init_memblock memblock_remove(phys_addr_t base, phys_addr_t size)
68295f72d1eSYinghai Lu {
683f1af9d3aSPhilipp Hachtmann 	return memblock_remove_range(&memblock.memory, base, size);
68495f72d1eSYinghai Lu }
68595f72d1eSYinghai Lu 
686f1af9d3aSPhilipp Hachtmann 
687581adcbeSTejun Heo int __init_memblock memblock_free(phys_addr_t base, phys_addr_t size)
68895f72d1eSYinghai Lu {
68924aa0788STejun Heo 	memblock_dbg("   memblock_free: [%#016llx-%#016llx] %pF\n",
690a150439cSH. Peter Anvin 		     (unsigned long long)base,
691931d13f5SGrygorii Strashko 		     (unsigned long long)base + size - 1,
692a150439cSH. Peter Anvin 		     (void *)_RET_IP_);
69324aa0788STejun Heo 
694f1af9d3aSPhilipp Hachtmann 	return memblock_remove_range(&memblock.reserved, base, size);
69595f72d1eSYinghai Lu }
69695f72d1eSYinghai Lu 
69766a20757STang Chen static int __init_memblock memblock_reserve_region(phys_addr_t base,
69866a20757STang Chen 						   phys_addr_t size,
69966a20757STang Chen 						   int nid,
70066a20757STang Chen 						   unsigned long flags)
70195f72d1eSYinghai Lu {
702e3239ff9SBenjamin Herrenschmidt 	struct memblock_type *_rgn = &memblock.reserved;
70395f72d1eSYinghai Lu 
70466a20757STang Chen 	memblock_dbg("memblock_reserve: [%#016llx-%#016llx] flags %#02lx %pF\n",
705a150439cSH. Peter Anvin 		     (unsigned long long)base,
706931d13f5SGrygorii Strashko 		     (unsigned long long)base + size - 1,
70766a20757STang Chen 		     flags, (void *)_RET_IP_);
70895f72d1eSYinghai Lu 
709f1af9d3aSPhilipp Hachtmann 	return memblock_add_range(_rgn, base, size, nid, flags);
71066a20757STang Chen }
71166a20757STang Chen 
71266a20757STang Chen int __init_memblock memblock_reserve(phys_addr_t base, phys_addr_t size)
71366a20757STang Chen {
71466a20757STang Chen 	return memblock_reserve_region(base, size, MAX_NUMNODES, 0);
71595f72d1eSYinghai Lu }
71695f72d1eSYinghai Lu 
71735fd0808STejun Heo /**
71866b16edfSTang Chen  * memblock_mark_hotplug - Mark hotpluggable memory with flag MEMBLOCK_HOTPLUG.
71966b16edfSTang Chen  * @base: the base phys addr of the region
72066b16edfSTang Chen  * @size: the size of the region
72166b16edfSTang Chen  *
72266b16edfSTang Chen  * This function isolates region [@base, @base + @size), and mark it with flag
72366b16edfSTang Chen  * MEMBLOCK_HOTPLUG.
72466b16edfSTang Chen  *
72566b16edfSTang Chen  * Return 0 on succees, -errno on failure.
72666b16edfSTang Chen  */
72766b16edfSTang Chen int __init_memblock memblock_mark_hotplug(phys_addr_t base, phys_addr_t size)
72866b16edfSTang Chen {
72966b16edfSTang Chen 	struct memblock_type *type = &memblock.memory;
73066b16edfSTang Chen 	int i, ret, start_rgn, end_rgn;
73166b16edfSTang Chen 
73266b16edfSTang Chen 	ret = memblock_isolate_range(type, base, size, &start_rgn, &end_rgn);
73366b16edfSTang Chen 	if (ret)
73466b16edfSTang Chen 		return ret;
73566b16edfSTang Chen 
73666b16edfSTang Chen 	for (i = start_rgn; i < end_rgn; i++)
73766b16edfSTang Chen 		memblock_set_region_flags(&type->regions[i], MEMBLOCK_HOTPLUG);
73866b16edfSTang Chen 
73966b16edfSTang Chen 	memblock_merge_regions(type);
74066b16edfSTang Chen 	return 0;
74166b16edfSTang Chen }
74266b16edfSTang Chen 
74366b16edfSTang Chen /**
74466b16edfSTang Chen  * memblock_clear_hotplug - Clear flag MEMBLOCK_HOTPLUG for a specified region.
74566b16edfSTang Chen  * @base: the base phys addr of the region
74666b16edfSTang Chen  * @size: the size of the region
74766b16edfSTang Chen  *
74866b16edfSTang Chen  * This function isolates region [@base, @base + @size), and clear flag
74966b16edfSTang Chen  * MEMBLOCK_HOTPLUG for the isolated regions.
75066b16edfSTang Chen  *
75166b16edfSTang Chen  * Return 0 on succees, -errno on failure.
75266b16edfSTang Chen  */
75366b16edfSTang Chen int __init_memblock memblock_clear_hotplug(phys_addr_t base, phys_addr_t size)
75466b16edfSTang Chen {
75566b16edfSTang Chen 	struct memblock_type *type = &memblock.memory;
75666b16edfSTang Chen 	int i, ret, start_rgn, end_rgn;
75766b16edfSTang Chen 
75866b16edfSTang Chen 	ret = memblock_isolate_range(type, base, size, &start_rgn, &end_rgn);
75966b16edfSTang Chen 	if (ret)
76066b16edfSTang Chen 		return ret;
76166b16edfSTang Chen 
76266b16edfSTang Chen 	for (i = start_rgn; i < end_rgn; i++)
76366b16edfSTang Chen 		memblock_clear_region_flags(&type->regions[i],
76466b16edfSTang Chen 					    MEMBLOCK_HOTPLUG);
76566b16edfSTang Chen 
76666b16edfSTang Chen 	memblock_merge_regions(type);
76766b16edfSTang Chen 	return 0;
76866b16edfSTang Chen }
76966b16edfSTang Chen 
77066b16edfSTang Chen /**
771f1af9d3aSPhilipp Hachtmann  * __next__mem_range - next function for for_each_free_mem_range() etc.
77235fd0808STejun Heo  * @idx: pointer to u64 loop variable
773b1154233SGrygorii Strashko  * @nid: node selector, %NUMA_NO_NODE for all nodes
774f1af9d3aSPhilipp Hachtmann  * @type_a: pointer to memblock_type from where the range is taken
775f1af9d3aSPhilipp Hachtmann  * @type_b: pointer to memblock_type which excludes memory from being taken
776dad7557eSWanpeng Li  * @out_start: ptr to phys_addr_t for start address of the range, can be %NULL
777dad7557eSWanpeng Li  * @out_end: ptr to phys_addr_t for end address of the range, can be %NULL
778dad7557eSWanpeng Li  * @out_nid: ptr to int for nid of the range, can be %NULL
77935fd0808STejun Heo  *
780f1af9d3aSPhilipp Hachtmann  * Find the first area from *@idx which matches @nid, fill the out
78135fd0808STejun Heo  * parameters, and update *@idx for the next iteration.  The lower 32bit of
782f1af9d3aSPhilipp Hachtmann  * *@idx contains index into type_a and the upper 32bit indexes the
783f1af9d3aSPhilipp Hachtmann  * areas before each region in type_b.	For example, if type_b regions
78435fd0808STejun Heo  * look like the following,
78535fd0808STejun Heo  *
78635fd0808STejun Heo  *	0:[0-16), 1:[32-48), 2:[128-130)
78735fd0808STejun Heo  *
78835fd0808STejun Heo  * The upper 32bit indexes the following regions.
78935fd0808STejun Heo  *
79035fd0808STejun Heo  *	0:[0-0), 1:[16-32), 2:[48-128), 3:[130-MAX)
79135fd0808STejun Heo  *
79235fd0808STejun Heo  * As both region arrays are sorted, the function advances the two indices
79335fd0808STejun Heo  * in lockstep and returns each intersection.
79435fd0808STejun Heo  */
795f1af9d3aSPhilipp Hachtmann void __init_memblock __next_mem_range(u64 *idx, int nid,
796f1af9d3aSPhilipp Hachtmann 				      struct memblock_type *type_a,
797f1af9d3aSPhilipp Hachtmann 				      struct memblock_type *type_b,
79835fd0808STejun Heo 				      phys_addr_t *out_start,
79935fd0808STejun Heo 				      phys_addr_t *out_end, int *out_nid)
80035fd0808STejun Heo {
801f1af9d3aSPhilipp Hachtmann 	int idx_a = *idx & 0xffffffff;
802f1af9d3aSPhilipp Hachtmann 	int idx_b = *idx >> 32;
803b1154233SGrygorii Strashko 
804f1af9d3aSPhilipp Hachtmann 	if (WARN_ONCE(nid == MAX_NUMNODES,
805f1af9d3aSPhilipp Hachtmann 	"Usage of MAX_NUMNODES is deprecated. Use NUMA_NO_NODE instead\n"))
806560dca27SGrygorii Strashko 		nid = NUMA_NO_NODE;
80735fd0808STejun Heo 
808f1af9d3aSPhilipp Hachtmann 	for (; idx_a < type_a->cnt; idx_a++) {
809f1af9d3aSPhilipp Hachtmann 		struct memblock_region *m = &type_a->regions[idx_a];
810f1af9d3aSPhilipp Hachtmann 
81135fd0808STejun Heo 		phys_addr_t m_start = m->base;
81235fd0808STejun Heo 		phys_addr_t m_end = m->base + m->size;
813f1af9d3aSPhilipp Hachtmann 		int	    m_nid = memblock_get_region_node(m);
81435fd0808STejun Heo 
81535fd0808STejun Heo 		/* only memory regions are associated with nodes, check it */
816f1af9d3aSPhilipp Hachtmann 		if (nid != NUMA_NO_NODE && nid != m_nid)
81735fd0808STejun Heo 			continue;
81835fd0808STejun Heo 
819f1af9d3aSPhilipp Hachtmann 		if (!type_b) {
820f1af9d3aSPhilipp Hachtmann 			if (out_start)
821f1af9d3aSPhilipp Hachtmann 				*out_start = m_start;
822f1af9d3aSPhilipp Hachtmann 			if (out_end)
823f1af9d3aSPhilipp Hachtmann 				*out_end = m_end;
824f1af9d3aSPhilipp Hachtmann 			if (out_nid)
825f1af9d3aSPhilipp Hachtmann 				*out_nid = m_nid;
826f1af9d3aSPhilipp Hachtmann 			idx_a++;
827f1af9d3aSPhilipp Hachtmann 			*idx = (u32)idx_a | (u64)idx_b << 32;
828f1af9d3aSPhilipp Hachtmann 			return;
829f1af9d3aSPhilipp Hachtmann 		}
83035fd0808STejun Heo 
831f1af9d3aSPhilipp Hachtmann 		/* scan areas before each reservation */
832f1af9d3aSPhilipp Hachtmann 		for (; idx_b < type_b->cnt + 1; idx_b++) {
833f1af9d3aSPhilipp Hachtmann 			struct memblock_region *r;
834f1af9d3aSPhilipp Hachtmann 			phys_addr_t r_start;
835f1af9d3aSPhilipp Hachtmann 			phys_addr_t r_end;
836f1af9d3aSPhilipp Hachtmann 
837f1af9d3aSPhilipp Hachtmann 			r = &type_b->regions[idx_b];
838f1af9d3aSPhilipp Hachtmann 			r_start = idx_b ? r[-1].base + r[-1].size : 0;
839f1af9d3aSPhilipp Hachtmann 			r_end = idx_b < type_b->cnt ?
840f1af9d3aSPhilipp Hachtmann 				r->base : ULLONG_MAX;
841f1af9d3aSPhilipp Hachtmann 
842f1af9d3aSPhilipp Hachtmann 			/*
843f1af9d3aSPhilipp Hachtmann 			 * if idx_b advanced past idx_a,
844f1af9d3aSPhilipp Hachtmann 			 * break out to advance idx_a
845f1af9d3aSPhilipp Hachtmann 			 */
84635fd0808STejun Heo 			if (r_start >= m_end)
84735fd0808STejun Heo 				break;
84835fd0808STejun Heo 			/* if the two regions intersect, we're done */
84935fd0808STejun Heo 			if (m_start < r_end) {
85035fd0808STejun Heo 				if (out_start)
851f1af9d3aSPhilipp Hachtmann 					*out_start =
852f1af9d3aSPhilipp Hachtmann 						max(m_start, r_start);
85335fd0808STejun Heo 				if (out_end)
85435fd0808STejun Heo 					*out_end = min(m_end, r_end);
85535fd0808STejun Heo 				if (out_nid)
856f1af9d3aSPhilipp Hachtmann 					*out_nid = m_nid;
85735fd0808STejun Heo 				/*
858f1af9d3aSPhilipp Hachtmann 				 * The region which ends first is
859f1af9d3aSPhilipp Hachtmann 				 * advanced for the next iteration.
86035fd0808STejun Heo 				 */
86135fd0808STejun Heo 				if (m_end <= r_end)
862f1af9d3aSPhilipp Hachtmann 					idx_a++;
86335fd0808STejun Heo 				else
864f1af9d3aSPhilipp Hachtmann 					idx_b++;
865f1af9d3aSPhilipp Hachtmann 				*idx = (u32)idx_a | (u64)idx_b << 32;
86635fd0808STejun Heo 				return;
86735fd0808STejun Heo 			}
86835fd0808STejun Heo 		}
86935fd0808STejun Heo 	}
87035fd0808STejun Heo 
87135fd0808STejun Heo 	/* signal end of iteration */
87235fd0808STejun Heo 	*idx = ULLONG_MAX;
87335fd0808STejun Heo }
87435fd0808STejun Heo 
8757bd0b0f0STejun Heo /**
876f1af9d3aSPhilipp Hachtmann  * __next_mem_range_rev - generic next function for for_each_*_range_rev()
877f1af9d3aSPhilipp Hachtmann  *
878f1af9d3aSPhilipp Hachtmann  * Finds the next range from type_a which is not marked as unsuitable
879f1af9d3aSPhilipp Hachtmann  * in type_b.
880f1af9d3aSPhilipp Hachtmann  *
8817bd0b0f0STejun Heo  * @idx: pointer to u64 loop variable
882b1154233SGrygorii Strashko  * @nid: nid: node selector, %NUMA_NO_NODE for all nodes
883f1af9d3aSPhilipp Hachtmann  * @type_a: pointer to memblock_type from where the range is taken
884f1af9d3aSPhilipp Hachtmann  * @type_b: pointer to memblock_type which excludes memory from being taken
885dad7557eSWanpeng Li  * @out_start: ptr to phys_addr_t for start address of the range, can be %NULL
886dad7557eSWanpeng Li  * @out_end: ptr to phys_addr_t for end address of the range, can be %NULL
887dad7557eSWanpeng Li  * @out_nid: ptr to int for nid of the range, can be %NULL
8887bd0b0f0STejun Heo  *
889f1af9d3aSPhilipp Hachtmann  * Reverse of __next_mem_range().
8907bd0b0f0STejun Heo  */
891f1af9d3aSPhilipp Hachtmann void __init_memblock __next_mem_range_rev(u64 *idx, int nid,
892f1af9d3aSPhilipp Hachtmann 					  struct memblock_type *type_a,
893f1af9d3aSPhilipp Hachtmann 					  struct memblock_type *type_b,
8947bd0b0f0STejun Heo 					  phys_addr_t *out_start,
8957bd0b0f0STejun Heo 					  phys_addr_t *out_end, int *out_nid)
8967bd0b0f0STejun Heo {
897f1af9d3aSPhilipp Hachtmann 	int idx_a = *idx & 0xffffffff;
898f1af9d3aSPhilipp Hachtmann 	int idx_b = *idx >> 32;
899b1154233SGrygorii Strashko 
900560dca27SGrygorii Strashko 	if (WARN_ONCE(nid == MAX_NUMNODES, "Usage of MAX_NUMNODES is deprecated. Use NUMA_NO_NODE instead\n"))
901560dca27SGrygorii Strashko 		nid = NUMA_NO_NODE;
9027bd0b0f0STejun Heo 
9037bd0b0f0STejun Heo 	if (*idx == (u64)ULLONG_MAX) {
904f1af9d3aSPhilipp Hachtmann 		idx_a = type_a->cnt - 1;
905f1af9d3aSPhilipp Hachtmann 		idx_b = type_b->cnt;
9067bd0b0f0STejun Heo 	}
9077bd0b0f0STejun Heo 
908f1af9d3aSPhilipp Hachtmann 	for (; idx_a >= 0; idx_a--) {
909f1af9d3aSPhilipp Hachtmann 		struct memblock_region *m = &type_a->regions[idx_a];
910f1af9d3aSPhilipp Hachtmann 
9117bd0b0f0STejun Heo 		phys_addr_t m_start = m->base;
9127bd0b0f0STejun Heo 		phys_addr_t m_end = m->base + m->size;
913f1af9d3aSPhilipp Hachtmann 		int m_nid = memblock_get_region_node(m);
9147bd0b0f0STejun Heo 
9157bd0b0f0STejun Heo 		/* only memory regions are associated with nodes, check it */
916f1af9d3aSPhilipp Hachtmann 		if (nid != NUMA_NO_NODE && nid != m_nid)
9177bd0b0f0STejun Heo 			continue;
9187bd0b0f0STejun Heo 
91955ac590cSTang Chen 		/* skip hotpluggable memory regions if needed */
92055ac590cSTang Chen 		if (movable_node_is_enabled() && memblock_is_hotpluggable(m))
92155ac590cSTang Chen 			continue;
92255ac590cSTang Chen 
923f1af9d3aSPhilipp Hachtmann 		if (!type_b) {
924f1af9d3aSPhilipp Hachtmann 			if (out_start)
925f1af9d3aSPhilipp Hachtmann 				*out_start = m_start;
926f1af9d3aSPhilipp Hachtmann 			if (out_end)
927f1af9d3aSPhilipp Hachtmann 				*out_end = m_end;
928f1af9d3aSPhilipp Hachtmann 			if (out_nid)
929f1af9d3aSPhilipp Hachtmann 				*out_nid = m_nid;
930f1af9d3aSPhilipp Hachtmann 			idx_a++;
931f1af9d3aSPhilipp Hachtmann 			*idx = (u32)idx_a | (u64)idx_b << 32;
932f1af9d3aSPhilipp Hachtmann 			return;
933f1af9d3aSPhilipp Hachtmann 		}
9347bd0b0f0STejun Heo 
935f1af9d3aSPhilipp Hachtmann 		/* scan areas before each reservation */
936f1af9d3aSPhilipp Hachtmann 		for (; idx_b >= 0; idx_b--) {
937f1af9d3aSPhilipp Hachtmann 			struct memblock_region *r;
938f1af9d3aSPhilipp Hachtmann 			phys_addr_t r_start;
939f1af9d3aSPhilipp Hachtmann 			phys_addr_t r_end;
940f1af9d3aSPhilipp Hachtmann 
941f1af9d3aSPhilipp Hachtmann 			r = &type_b->regions[idx_b];
942f1af9d3aSPhilipp Hachtmann 			r_start = idx_b ? r[-1].base + r[-1].size : 0;
943f1af9d3aSPhilipp Hachtmann 			r_end = idx_b < type_b->cnt ?
944f1af9d3aSPhilipp Hachtmann 				r->base : ULLONG_MAX;
945f1af9d3aSPhilipp Hachtmann 			/*
946f1af9d3aSPhilipp Hachtmann 			 * if idx_b advanced past idx_a,
947f1af9d3aSPhilipp Hachtmann 			 * break out to advance idx_a
948f1af9d3aSPhilipp Hachtmann 			 */
949f1af9d3aSPhilipp Hachtmann 
9507bd0b0f0STejun Heo 			if (r_end <= m_start)
9517bd0b0f0STejun Heo 				break;
9527bd0b0f0STejun Heo 			/* if the two regions intersect, we're done */
9537bd0b0f0STejun Heo 			if (m_end > r_start) {
9547bd0b0f0STejun Heo 				if (out_start)
9557bd0b0f0STejun Heo 					*out_start = max(m_start, r_start);
9567bd0b0f0STejun Heo 				if (out_end)
9577bd0b0f0STejun Heo 					*out_end = min(m_end, r_end);
9587bd0b0f0STejun Heo 				if (out_nid)
959f1af9d3aSPhilipp Hachtmann 					*out_nid = m_nid;
9607bd0b0f0STejun Heo 				if (m_start >= r_start)
961f1af9d3aSPhilipp Hachtmann 					idx_a--;
9627bd0b0f0STejun Heo 				else
963f1af9d3aSPhilipp Hachtmann 					idx_b--;
964f1af9d3aSPhilipp Hachtmann 				*idx = (u32)idx_a | (u64)idx_b << 32;
9657bd0b0f0STejun Heo 				return;
9667bd0b0f0STejun Heo 			}
9677bd0b0f0STejun Heo 		}
9687bd0b0f0STejun Heo 	}
969f1af9d3aSPhilipp Hachtmann 	/* signal end of iteration */
9707bd0b0f0STejun Heo 	*idx = ULLONG_MAX;
9717bd0b0f0STejun Heo }
9727bd0b0f0STejun Heo 
9737c0caeb8STejun Heo #ifdef CONFIG_HAVE_MEMBLOCK_NODE_MAP
9747c0caeb8STejun Heo /*
9757c0caeb8STejun Heo  * Common iterator interface used to define for_each_mem_range().
9767c0caeb8STejun Heo  */
9777c0caeb8STejun Heo void __init_memblock __next_mem_pfn_range(int *idx, int nid,
9787c0caeb8STejun Heo 				unsigned long *out_start_pfn,
9797c0caeb8STejun Heo 				unsigned long *out_end_pfn, int *out_nid)
9807c0caeb8STejun Heo {
9817c0caeb8STejun Heo 	struct memblock_type *type = &memblock.memory;
9827c0caeb8STejun Heo 	struct memblock_region *r;
9837c0caeb8STejun Heo 
9847c0caeb8STejun Heo 	while (++*idx < type->cnt) {
9857c0caeb8STejun Heo 		r = &type->regions[*idx];
9867c0caeb8STejun Heo 
9877c0caeb8STejun Heo 		if (PFN_UP(r->base) >= PFN_DOWN(r->base + r->size))
9887c0caeb8STejun Heo 			continue;
9897c0caeb8STejun Heo 		if (nid == MAX_NUMNODES || nid == r->nid)
9907c0caeb8STejun Heo 			break;
9917c0caeb8STejun Heo 	}
9927c0caeb8STejun Heo 	if (*idx >= type->cnt) {
9937c0caeb8STejun Heo 		*idx = -1;
9947c0caeb8STejun Heo 		return;
9957c0caeb8STejun Heo 	}
9967c0caeb8STejun Heo 
9977c0caeb8STejun Heo 	if (out_start_pfn)
9987c0caeb8STejun Heo 		*out_start_pfn = PFN_UP(r->base);
9997c0caeb8STejun Heo 	if (out_end_pfn)
10007c0caeb8STejun Heo 		*out_end_pfn = PFN_DOWN(r->base + r->size);
10017c0caeb8STejun Heo 	if (out_nid)
10027c0caeb8STejun Heo 		*out_nid = r->nid;
10037c0caeb8STejun Heo }
10047c0caeb8STejun Heo 
10057c0caeb8STejun Heo /**
10067c0caeb8STejun Heo  * memblock_set_node - set node ID on memblock regions
10077c0caeb8STejun Heo  * @base: base of area to set node ID for
10087c0caeb8STejun Heo  * @size: size of area to set node ID for
1009e7e8de59STang Chen  * @type: memblock type to set node ID for
10107c0caeb8STejun Heo  * @nid: node ID to set
10117c0caeb8STejun Heo  *
1012e7e8de59STang Chen  * Set the nid of memblock @type regions in [@base,@base+@size) to @nid.
10137c0caeb8STejun Heo  * Regions which cross the area boundaries are split as necessary.
10147c0caeb8STejun Heo  *
10157c0caeb8STejun Heo  * RETURNS:
10167c0caeb8STejun Heo  * 0 on success, -errno on failure.
10177c0caeb8STejun Heo  */
10187c0caeb8STejun Heo int __init_memblock memblock_set_node(phys_addr_t base, phys_addr_t size,
1019e7e8de59STang Chen 				      struct memblock_type *type, int nid)
10207c0caeb8STejun Heo {
10216a9ceb31STejun Heo 	int start_rgn, end_rgn;
10226a9ceb31STejun Heo 	int i, ret;
10237c0caeb8STejun Heo 
10246a9ceb31STejun Heo 	ret = memblock_isolate_range(type, base, size, &start_rgn, &end_rgn);
10256a9ceb31STejun Heo 	if (ret)
10266a9ceb31STejun Heo 		return ret;
10277c0caeb8STejun Heo 
10286a9ceb31STejun Heo 	for (i = start_rgn; i < end_rgn; i++)
1029e9d24ad3SWanpeng Li 		memblock_set_region_node(&type->regions[i], nid);
10307c0caeb8STejun Heo 
10317c0caeb8STejun Heo 	memblock_merge_regions(type);
10327c0caeb8STejun Heo 	return 0;
10337c0caeb8STejun Heo }
10347c0caeb8STejun Heo #endif /* CONFIG_HAVE_MEMBLOCK_NODE_MAP */
10357c0caeb8STejun Heo 
1036*2bfc2862SAkinobu Mita static phys_addr_t __init memblock_alloc_range_nid(phys_addr_t size,
1037*2bfc2862SAkinobu Mita 					phys_addr_t align, phys_addr_t start,
1038*2bfc2862SAkinobu Mita 					phys_addr_t end, int nid)
103995f72d1eSYinghai Lu {
10406ed311b2SBenjamin Herrenschmidt 	phys_addr_t found;
104195f72d1eSYinghai Lu 
104279f40fabSGrygorii Strashko 	if (!align)
104379f40fabSGrygorii Strashko 		align = SMP_CACHE_BYTES;
104494f3d3afSVineet Gupta 
1045*2bfc2862SAkinobu Mita 	found = memblock_find_in_range_node(size, align, start, end, nid);
10469c8c27e2STejun Heo 	if (found && !memblock_reserve(found, size))
10476ed311b2SBenjamin Herrenschmidt 		return found;
10486ed311b2SBenjamin Herrenschmidt 
10496ed311b2SBenjamin Herrenschmidt 	return 0;
105095f72d1eSYinghai Lu }
105195f72d1eSYinghai Lu 
1052*2bfc2862SAkinobu Mita phys_addr_t __init memblock_alloc_range(phys_addr_t size, phys_addr_t align,
1053*2bfc2862SAkinobu Mita 					phys_addr_t start, phys_addr_t end)
1054*2bfc2862SAkinobu Mita {
1055*2bfc2862SAkinobu Mita 	return memblock_alloc_range_nid(size, align, start, end, NUMA_NO_NODE);
1056*2bfc2862SAkinobu Mita }
1057*2bfc2862SAkinobu Mita 
1058*2bfc2862SAkinobu Mita static phys_addr_t __init memblock_alloc_base_nid(phys_addr_t size,
1059*2bfc2862SAkinobu Mita 					phys_addr_t align, phys_addr_t max_addr,
1060*2bfc2862SAkinobu Mita 					int nid)
1061*2bfc2862SAkinobu Mita {
1062*2bfc2862SAkinobu Mita 	return memblock_alloc_range_nid(size, align, 0, max_addr, nid);
1063*2bfc2862SAkinobu Mita }
1064*2bfc2862SAkinobu Mita 
10657bd0b0f0STejun Heo phys_addr_t __init memblock_alloc_nid(phys_addr_t size, phys_addr_t align, int nid)
10667bd0b0f0STejun Heo {
10677bd0b0f0STejun Heo 	return memblock_alloc_base_nid(size, align, MEMBLOCK_ALLOC_ACCESSIBLE, nid);
10687bd0b0f0STejun Heo }
10697bd0b0f0STejun Heo 
10707bd0b0f0STejun Heo phys_addr_t __init __memblock_alloc_base(phys_addr_t size, phys_addr_t align, phys_addr_t max_addr)
10717bd0b0f0STejun Heo {
1072b1154233SGrygorii Strashko 	return memblock_alloc_base_nid(size, align, max_addr, NUMA_NO_NODE);
10737bd0b0f0STejun Heo }
10747bd0b0f0STejun Heo 
10756ed311b2SBenjamin Herrenschmidt phys_addr_t __init memblock_alloc_base(phys_addr_t size, phys_addr_t align, phys_addr_t max_addr)
107695f72d1eSYinghai Lu {
10776ed311b2SBenjamin Herrenschmidt 	phys_addr_t alloc;
10786ed311b2SBenjamin Herrenschmidt 
10796ed311b2SBenjamin Herrenschmidt 	alloc = __memblock_alloc_base(size, align, max_addr);
10806ed311b2SBenjamin Herrenschmidt 
10816ed311b2SBenjamin Herrenschmidt 	if (alloc == 0)
10826ed311b2SBenjamin Herrenschmidt 		panic("ERROR: Failed to allocate 0x%llx bytes below 0x%llx.\n",
10836ed311b2SBenjamin Herrenschmidt 		      (unsigned long long) size, (unsigned long long) max_addr);
10846ed311b2SBenjamin Herrenschmidt 
10856ed311b2SBenjamin Herrenschmidt 	return alloc;
108695f72d1eSYinghai Lu }
108795f72d1eSYinghai Lu 
10886ed311b2SBenjamin Herrenschmidt phys_addr_t __init memblock_alloc(phys_addr_t size, phys_addr_t align)
108995f72d1eSYinghai Lu {
10906ed311b2SBenjamin Herrenschmidt 	return memblock_alloc_base(size, align, MEMBLOCK_ALLOC_ACCESSIBLE);
109195f72d1eSYinghai Lu }
109295f72d1eSYinghai Lu 
10939d1e2492SBenjamin Herrenschmidt phys_addr_t __init memblock_alloc_try_nid(phys_addr_t size, phys_addr_t align, int nid)
10949d1e2492SBenjamin Herrenschmidt {
10959d1e2492SBenjamin Herrenschmidt 	phys_addr_t res = memblock_alloc_nid(size, align, nid);
10969d1e2492SBenjamin Herrenschmidt 
10979d1e2492SBenjamin Herrenschmidt 	if (res)
10989d1e2492SBenjamin Herrenschmidt 		return res;
109915fb0972STejun Heo 	return memblock_alloc_base(size, align, MEMBLOCK_ALLOC_ACCESSIBLE);
110095f72d1eSYinghai Lu }
110195f72d1eSYinghai Lu 
110226f09e9bSSantosh Shilimkar /**
110326f09e9bSSantosh Shilimkar  * memblock_virt_alloc_internal - allocate boot memory block
110426f09e9bSSantosh Shilimkar  * @size: size of memory block to be allocated in bytes
110526f09e9bSSantosh Shilimkar  * @align: alignment of the region and block's size
110626f09e9bSSantosh Shilimkar  * @min_addr: the lower bound of the memory region to allocate (phys address)
110726f09e9bSSantosh Shilimkar  * @max_addr: the upper bound of the memory region to allocate (phys address)
110826f09e9bSSantosh Shilimkar  * @nid: nid of the free area to find, %NUMA_NO_NODE for any node
110926f09e9bSSantosh Shilimkar  *
111026f09e9bSSantosh Shilimkar  * The @min_addr limit is dropped if it can not be satisfied and the allocation
111126f09e9bSSantosh Shilimkar  * will fall back to memory below @min_addr. Also, allocation may fall back
111226f09e9bSSantosh Shilimkar  * to any node in the system if the specified node can not
111326f09e9bSSantosh Shilimkar  * hold the requested memory.
111426f09e9bSSantosh Shilimkar  *
111526f09e9bSSantosh Shilimkar  * The allocation is performed from memory region limited by
111626f09e9bSSantosh Shilimkar  * memblock.current_limit if @max_addr == %BOOTMEM_ALLOC_ACCESSIBLE.
111726f09e9bSSantosh Shilimkar  *
111826f09e9bSSantosh Shilimkar  * The memory block is aligned on SMP_CACHE_BYTES if @align == 0.
111926f09e9bSSantosh Shilimkar  *
112026f09e9bSSantosh Shilimkar  * The phys address of allocated boot memory block is converted to virtual and
112126f09e9bSSantosh Shilimkar  * allocated memory is reset to 0.
112226f09e9bSSantosh Shilimkar  *
112326f09e9bSSantosh Shilimkar  * In addition, function sets the min_count to 0 using kmemleak_alloc for
112426f09e9bSSantosh Shilimkar  * allocated boot memory block, so that it is never reported as leaks.
112526f09e9bSSantosh Shilimkar  *
112626f09e9bSSantosh Shilimkar  * RETURNS:
112726f09e9bSSantosh Shilimkar  * Virtual address of allocated memory block on success, NULL on failure.
112826f09e9bSSantosh Shilimkar  */
112926f09e9bSSantosh Shilimkar static void * __init memblock_virt_alloc_internal(
113026f09e9bSSantosh Shilimkar 				phys_addr_t size, phys_addr_t align,
113126f09e9bSSantosh Shilimkar 				phys_addr_t min_addr, phys_addr_t max_addr,
113226f09e9bSSantosh Shilimkar 				int nid)
113326f09e9bSSantosh Shilimkar {
113426f09e9bSSantosh Shilimkar 	phys_addr_t alloc;
113526f09e9bSSantosh Shilimkar 	void *ptr;
113626f09e9bSSantosh Shilimkar 
1137560dca27SGrygorii Strashko 	if (WARN_ONCE(nid == MAX_NUMNODES, "Usage of MAX_NUMNODES is deprecated. Use NUMA_NO_NODE instead\n"))
1138560dca27SGrygorii Strashko 		nid = NUMA_NO_NODE;
113926f09e9bSSantosh Shilimkar 
114026f09e9bSSantosh Shilimkar 	/*
114126f09e9bSSantosh Shilimkar 	 * Detect any accidental use of these APIs after slab is ready, as at
114226f09e9bSSantosh Shilimkar 	 * this moment memblock may be deinitialized already and its
114326f09e9bSSantosh Shilimkar 	 * internal data may be destroyed (after execution of free_all_bootmem)
114426f09e9bSSantosh Shilimkar 	 */
114526f09e9bSSantosh Shilimkar 	if (WARN_ON_ONCE(slab_is_available()))
114626f09e9bSSantosh Shilimkar 		return kzalloc_node(size, GFP_NOWAIT, nid);
114726f09e9bSSantosh Shilimkar 
114826f09e9bSSantosh Shilimkar 	if (!align)
114926f09e9bSSantosh Shilimkar 		align = SMP_CACHE_BYTES;
115026f09e9bSSantosh Shilimkar 
1151f544e14fSYinghai Lu 	if (max_addr > memblock.current_limit)
1152f544e14fSYinghai Lu 		max_addr = memblock.current_limit;
1153f544e14fSYinghai Lu 
115426f09e9bSSantosh Shilimkar again:
115526f09e9bSSantosh Shilimkar 	alloc = memblock_find_in_range_node(size, align, min_addr, max_addr,
115626f09e9bSSantosh Shilimkar 					    nid);
115726f09e9bSSantosh Shilimkar 	if (alloc)
115826f09e9bSSantosh Shilimkar 		goto done;
115926f09e9bSSantosh Shilimkar 
116026f09e9bSSantosh Shilimkar 	if (nid != NUMA_NO_NODE) {
116126f09e9bSSantosh Shilimkar 		alloc = memblock_find_in_range_node(size, align, min_addr,
116226f09e9bSSantosh Shilimkar 						    max_addr,  NUMA_NO_NODE);
116326f09e9bSSantosh Shilimkar 		if (alloc)
116426f09e9bSSantosh Shilimkar 			goto done;
116526f09e9bSSantosh Shilimkar 	}
116626f09e9bSSantosh Shilimkar 
116726f09e9bSSantosh Shilimkar 	if (min_addr) {
116826f09e9bSSantosh Shilimkar 		min_addr = 0;
116926f09e9bSSantosh Shilimkar 		goto again;
117026f09e9bSSantosh Shilimkar 	} else {
117126f09e9bSSantosh Shilimkar 		goto error;
117226f09e9bSSantosh Shilimkar 	}
117326f09e9bSSantosh Shilimkar 
117426f09e9bSSantosh Shilimkar done:
117526f09e9bSSantosh Shilimkar 	memblock_reserve(alloc, size);
117626f09e9bSSantosh Shilimkar 	ptr = phys_to_virt(alloc);
117726f09e9bSSantosh Shilimkar 	memset(ptr, 0, size);
117826f09e9bSSantosh Shilimkar 
117926f09e9bSSantosh Shilimkar 	/*
118026f09e9bSSantosh Shilimkar 	 * The min_count is set to 0 so that bootmem allocated blocks
118126f09e9bSSantosh Shilimkar 	 * are never reported as leaks. This is because many of these blocks
118226f09e9bSSantosh Shilimkar 	 * are only referred via the physical address which is not
118326f09e9bSSantosh Shilimkar 	 * looked up by kmemleak.
118426f09e9bSSantosh Shilimkar 	 */
118526f09e9bSSantosh Shilimkar 	kmemleak_alloc(ptr, size, 0, 0);
118626f09e9bSSantosh Shilimkar 
118726f09e9bSSantosh Shilimkar 	return ptr;
118826f09e9bSSantosh Shilimkar 
118926f09e9bSSantosh Shilimkar error:
119026f09e9bSSantosh Shilimkar 	return NULL;
119126f09e9bSSantosh Shilimkar }
119226f09e9bSSantosh Shilimkar 
119326f09e9bSSantosh Shilimkar /**
119426f09e9bSSantosh Shilimkar  * memblock_virt_alloc_try_nid_nopanic - allocate boot memory block
119526f09e9bSSantosh Shilimkar  * @size: size of memory block to be allocated in bytes
119626f09e9bSSantosh Shilimkar  * @align: alignment of the region and block's size
119726f09e9bSSantosh Shilimkar  * @min_addr: the lower bound of the memory region from where the allocation
119826f09e9bSSantosh Shilimkar  *	  is preferred (phys address)
119926f09e9bSSantosh Shilimkar  * @max_addr: the upper bound of the memory region from where the allocation
120026f09e9bSSantosh Shilimkar  *	      is preferred (phys address), or %BOOTMEM_ALLOC_ACCESSIBLE to
120126f09e9bSSantosh Shilimkar  *	      allocate only from memory limited by memblock.current_limit value
120226f09e9bSSantosh Shilimkar  * @nid: nid of the free area to find, %NUMA_NO_NODE for any node
120326f09e9bSSantosh Shilimkar  *
120426f09e9bSSantosh Shilimkar  * Public version of _memblock_virt_alloc_try_nid_nopanic() which provides
120526f09e9bSSantosh Shilimkar  * additional debug information (including caller info), if enabled.
120626f09e9bSSantosh Shilimkar  *
120726f09e9bSSantosh Shilimkar  * RETURNS:
120826f09e9bSSantosh Shilimkar  * Virtual address of allocated memory block on success, NULL on failure.
120926f09e9bSSantosh Shilimkar  */
121026f09e9bSSantosh Shilimkar void * __init memblock_virt_alloc_try_nid_nopanic(
121126f09e9bSSantosh Shilimkar 				phys_addr_t size, phys_addr_t align,
121226f09e9bSSantosh Shilimkar 				phys_addr_t min_addr, phys_addr_t max_addr,
121326f09e9bSSantosh Shilimkar 				int nid)
121426f09e9bSSantosh Shilimkar {
121526f09e9bSSantosh Shilimkar 	memblock_dbg("%s: %llu bytes align=0x%llx nid=%d from=0x%llx max_addr=0x%llx %pF\n",
121626f09e9bSSantosh Shilimkar 		     __func__, (u64)size, (u64)align, nid, (u64)min_addr,
121726f09e9bSSantosh Shilimkar 		     (u64)max_addr, (void *)_RET_IP_);
121826f09e9bSSantosh Shilimkar 	return memblock_virt_alloc_internal(size, align, min_addr,
121926f09e9bSSantosh Shilimkar 					     max_addr, nid);
122026f09e9bSSantosh Shilimkar }
122126f09e9bSSantosh Shilimkar 
122226f09e9bSSantosh Shilimkar /**
122326f09e9bSSantosh Shilimkar  * memblock_virt_alloc_try_nid - allocate boot memory block with panicking
122426f09e9bSSantosh Shilimkar  * @size: size of memory block to be allocated in bytes
122526f09e9bSSantosh Shilimkar  * @align: alignment of the region and block's size
122626f09e9bSSantosh Shilimkar  * @min_addr: the lower bound of the memory region from where the allocation
122726f09e9bSSantosh Shilimkar  *	  is preferred (phys address)
122826f09e9bSSantosh Shilimkar  * @max_addr: the upper bound of the memory region from where the allocation
122926f09e9bSSantosh Shilimkar  *	      is preferred (phys address), or %BOOTMEM_ALLOC_ACCESSIBLE to
123026f09e9bSSantosh Shilimkar  *	      allocate only from memory limited by memblock.current_limit value
123126f09e9bSSantosh Shilimkar  * @nid: nid of the free area to find, %NUMA_NO_NODE for any node
123226f09e9bSSantosh Shilimkar  *
123326f09e9bSSantosh Shilimkar  * Public panicking version of _memblock_virt_alloc_try_nid_nopanic()
123426f09e9bSSantosh Shilimkar  * which provides debug information (including caller info), if enabled,
123526f09e9bSSantosh Shilimkar  * and panics if the request can not be satisfied.
123626f09e9bSSantosh Shilimkar  *
123726f09e9bSSantosh Shilimkar  * RETURNS:
123826f09e9bSSantosh Shilimkar  * Virtual address of allocated memory block on success, NULL on failure.
123926f09e9bSSantosh Shilimkar  */
124026f09e9bSSantosh Shilimkar void * __init memblock_virt_alloc_try_nid(
124126f09e9bSSantosh Shilimkar 			phys_addr_t size, phys_addr_t align,
124226f09e9bSSantosh Shilimkar 			phys_addr_t min_addr, phys_addr_t max_addr,
124326f09e9bSSantosh Shilimkar 			int nid)
124426f09e9bSSantosh Shilimkar {
124526f09e9bSSantosh Shilimkar 	void *ptr;
124626f09e9bSSantosh Shilimkar 
124726f09e9bSSantosh Shilimkar 	memblock_dbg("%s: %llu bytes align=0x%llx nid=%d from=0x%llx max_addr=0x%llx %pF\n",
124826f09e9bSSantosh Shilimkar 		     __func__, (u64)size, (u64)align, nid, (u64)min_addr,
124926f09e9bSSantosh Shilimkar 		     (u64)max_addr, (void *)_RET_IP_);
125026f09e9bSSantosh Shilimkar 	ptr = memblock_virt_alloc_internal(size, align,
125126f09e9bSSantosh Shilimkar 					   min_addr, max_addr, nid);
125226f09e9bSSantosh Shilimkar 	if (ptr)
125326f09e9bSSantosh Shilimkar 		return ptr;
125426f09e9bSSantosh Shilimkar 
125526f09e9bSSantosh Shilimkar 	panic("%s: Failed to allocate %llu bytes align=0x%llx nid=%d from=0x%llx max_addr=0x%llx\n",
125626f09e9bSSantosh Shilimkar 	      __func__, (u64)size, (u64)align, nid, (u64)min_addr,
125726f09e9bSSantosh Shilimkar 	      (u64)max_addr);
125826f09e9bSSantosh Shilimkar 	return NULL;
125926f09e9bSSantosh Shilimkar }
126026f09e9bSSantosh Shilimkar 
126126f09e9bSSantosh Shilimkar /**
126226f09e9bSSantosh Shilimkar  * __memblock_free_early - free boot memory block
126326f09e9bSSantosh Shilimkar  * @base: phys starting address of the  boot memory block
126426f09e9bSSantosh Shilimkar  * @size: size of the boot memory block in bytes
126526f09e9bSSantosh Shilimkar  *
126626f09e9bSSantosh Shilimkar  * Free boot memory block previously allocated by memblock_virt_alloc_xx() API.
126726f09e9bSSantosh Shilimkar  * The freeing memory will not be released to the buddy allocator.
126826f09e9bSSantosh Shilimkar  */
126926f09e9bSSantosh Shilimkar void __init __memblock_free_early(phys_addr_t base, phys_addr_t size)
127026f09e9bSSantosh Shilimkar {
127126f09e9bSSantosh Shilimkar 	memblock_dbg("%s: [%#016llx-%#016llx] %pF\n",
127226f09e9bSSantosh Shilimkar 		     __func__, (u64)base, (u64)base + size - 1,
127326f09e9bSSantosh Shilimkar 		     (void *)_RET_IP_);
127426f09e9bSSantosh Shilimkar 	kmemleak_free_part(__va(base), size);
1275f1af9d3aSPhilipp Hachtmann 	memblock_remove_range(&memblock.reserved, base, size);
127626f09e9bSSantosh Shilimkar }
127726f09e9bSSantosh Shilimkar 
127826f09e9bSSantosh Shilimkar /*
127926f09e9bSSantosh Shilimkar  * __memblock_free_late - free bootmem block pages directly to buddy allocator
128026f09e9bSSantosh Shilimkar  * @addr: phys starting address of the  boot memory block
128126f09e9bSSantosh Shilimkar  * @size: size of the boot memory block in bytes
128226f09e9bSSantosh Shilimkar  *
128326f09e9bSSantosh Shilimkar  * This is only useful when the bootmem allocator has already been torn
128426f09e9bSSantosh Shilimkar  * down, but we are still initializing the system.  Pages are released directly
128526f09e9bSSantosh Shilimkar  * to the buddy allocator, no bootmem metadata is updated because it is gone.
128626f09e9bSSantosh Shilimkar  */
128726f09e9bSSantosh Shilimkar void __init __memblock_free_late(phys_addr_t base, phys_addr_t size)
128826f09e9bSSantosh Shilimkar {
128926f09e9bSSantosh Shilimkar 	u64 cursor, end;
129026f09e9bSSantosh Shilimkar 
129126f09e9bSSantosh Shilimkar 	memblock_dbg("%s: [%#016llx-%#016llx] %pF\n",
129226f09e9bSSantosh Shilimkar 		     __func__, (u64)base, (u64)base + size - 1,
129326f09e9bSSantosh Shilimkar 		     (void *)_RET_IP_);
129426f09e9bSSantosh Shilimkar 	kmemleak_free_part(__va(base), size);
129526f09e9bSSantosh Shilimkar 	cursor = PFN_UP(base);
129626f09e9bSSantosh Shilimkar 	end = PFN_DOWN(base + size);
129726f09e9bSSantosh Shilimkar 
129826f09e9bSSantosh Shilimkar 	for (; cursor < end; cursor++) {
129926f09e9bSSantosh Shilimkar 		__free_pages_bootmem(pfn_to_page(cursor), 0);
130026f09e9bSSantosh Shilimkar 		totalram_pages++;
130126f09e9bSSantosh Shilimkar 	}
130226f09e9bSSantosh Shilimkar }
13039d1e2492SBenjamin Herrenschmidt 
13049d1e2492SBenjamin Herrenschmidt /*
13059d1e2492SBenjamin Herrenschmidt  * Remaining API functions
13069d1e2492SBenjamin Herrenschmidt  */
13079d1e2492SBenjamin Herrenschmidt 
13082898cc4cSBenjamin Herrenschmidt phys_addr_t __init memblock_phys_mem_size(void)
130995f72d1eSYinghai Lu {
13101440c4e2STejun Heo 	return memblock.memory.total_size;
131195f72d1eSYinghai Lu }
131295f72d1eSYinghai Lu 
1313595ad9afSYinghai Lu phys_addr_t __init memblock_mem_size(unsigned long limit_pfn)
1314595ad9afSYinghai Lu {
1315595ad9afSYinghai Lu 	unsigned long pages = 0;
1316595ad9afSYinghai Lu 	struct memblock_region *r;
1317595ad9afSYinghai Lu 	unsigned long start_pfn, end_pfn;
1318595ad9afSYinghai Lu 
1319595ad9afSYinghai Lu 	for_each_memblock(memory, r) {
1320595ad9afSYinghai Lu 		start_pfn = memblock_region_memory_base_pfn(r);
1321595ad9afSYinghai Lu 		end_pfn = memblock_region_memory_end_pfn(r);
1322595ad9afSYinghai Lu 		start_pfn = min_t(unsigned long, start_pfn, limit_pfn);
1323595ad9afSYinghai Lu 		end_pfn = min_t(unsigned long, end_pfn, limit_pfn);
1324595ad9afSYinghai Lu 		pages += end_pfn - start_pfn;
1325595ad9afSYinghai Lu 	}
1326595ad9afSYinghai Lu 
132716763230SFabian Frederick 	return PFN_PHYS(pages);
1328595ad9afSYinghai Lu }
1329595ad9afSYinghai Lu 
13300a93ebefSSam Ravnborg /* lowest address */
13310a93ebefSSam Ravnborg phys_addr_t __init_memblock memblock_start_of_DRAM(void)
13320a93ebefSSam Ravnborg {
13330a93ebefSSam Ravnborg 	return memblock.memory.regions[0].base;
13340a93ebefSSam Ravnborg }
13350a93ebefSSam Ravnborg 
133610d06439SYinghai Lu phys_addr_t __init_memblock memblock_end_of_DRAM(void)
133795f72d1eSYinghai Lu {
133895f72d1eSYinghai Lu 	int idx = memblock.memory.cnt - 1;
133995f72d1eSYinghai Lu 
1340e3239ff9SBenjamin Herrenschmidt 	return (memblock.memory.regions[idx].base + memblock.memory.regions[idx].size);
134195f72d1eSYinghai Lu }
134295f72d1eSYinghai Lu 
1343c0ce8fefSTejun Heo void __init memblock_enforce_memory_limit(phys_addr_t limit)
134495f72d1eSYinghai Lu {
1345c0ce8fefSTejun Heo 	phys_addr_t max_addr = (phys_addr_t)ULLONG_MAX;
1346136199f0SEmil Medve 	struct memblock_region *r;
134795f72d1eSYinghai Lu 
1348c0ce8fefSTejun Heo 	if (!limit)
134995f72d1eSYinghai Lu 		return;
135095f72d1eSYinghai Lu 
1351c0ce8fefSTejun Heo 	/* find out max address */
1352136199f0SEmil Medve 	for_each_memblock(memory, r) {
1353c0ce8fefSTejun Heo 		if (limit <= r->size) {
1354c0ce8fefSTejun Heo 			max_addr = r->base + limit;
135595f72d1eSYinghai Lu 			break;
135695f72d1eSYinghai Lu 		}
1357c0ce8fefSTejun Heo 		limit -= r->size;
135895f72d1eSYinghai Lu 	}
1359c0ce8fefSTejun Heo 
1360c0ce8fefSTejun Heo 	/* truncate both memory and reserved regions */
1361f1af9d3aSPhilipp Hachtmann 	memblock_remove_range(&memblock.memory, max_addr,
1362f1af9d3aSPhilipp Hachtmann 			      (phys_addr_t)ULLONG_MAX);
1363f1af9d3aSPhilipp Hachtmann 	memblock_remove_range(&memblock.reserved, max_addr,
1364f1af9d3aSPhilipp Hachtmann 			      (phys_addr_t)ULLONG_MAX);
136595f72d1eSYinghai Lu }
136695f72d1eSYinghai Lu 
1367cd79481dSYinghai Lu static int __init_memblock memblock_search(struct memblock_type *type, phys_addr_t addr)
136872d4b0b4SBenjamin Herrenschmidt {
136972d4b0b4SBenjamin Herrenschmidt 	unsigned int left = 0, right = type->cnt;
137072d4b0b4SBenjamin Herrenschmidt 
137172d4b0b4SBenjamin Herrenschmidt 	do {
137272d4b0b4SBenjamin Herrenschmidt 		unsigned int mid = (right + left) / 2;
137372d4b0b4SBenjamin Herrenschmidt 
137472d4b0b4SBenjamin Herrenschmidt 		if (addr < type->regions[mid].base)
137572d4b0b4SBenjamin Herrenschmidt 			right = mid;
137672d4b0b4SBenjamin Herrenschmidt 		else if (addr >= (type->regions[mid].base +
137772d4b0b4SBenjamin Herrenschmidt 				  type->regions[mid].size))
137872d4b0b4SBenjamin Herrenschmidt 			left = mid + 1;
137972d4b0b4SBenjamin Herrenschmidt 		else
138072d4b0b4SBenjamin Herrenschmidt 			return mid;
138172d4b0b4SBenjamin Herrenschmidt 	} while (left < right);
138272d4b0b4SBenjamin Herrenschmidt 	return -1;
138372d4b0b4SBenjamin Herrenschmidt }
138472d4b0b4SBenjamin Herrenschmidt 
13852898cc4cSBenjamin Herrenschmidt int __init memblock_is_reserved(phys_addr_t addr)
138695f72d1eSYinghai Lu {
138772d4b0b4SBenjamin Herrenschmidt 	return memblock_search(&memblock.reserved, addr) != -1;
138895f72d1eSYinghai Lu }
138972d4b0b4SBenjamin Herrenschmidt 
13903661ca66SYinghai Lu int __init_memblock memblock_is_memory(phys_addr_t addr)
139172d4b0b4SBenjamin Herrenschmidt {
139272d4b0b4SBenjamin Herrenschmidt 	return memblock_search(&memblock.memory, addr) != -1;
139372d4b0b4SBenjamin Herrenschmidt }
139472d4b0b4SBenjamin Herrenschmidt 
1395e76b63f8SYinghai Lu #ifdef CONFIG_HAVE_MEMBLOCK_NODE_MAP
1396e76b63f8SYinghai Lu int __init_memblock memblock_search_pfn_nid(unsigned long pfn,
1397e76b63f8SYinghai Lu 			 unsigned long *start_pfn, unsigned long *end_pfn)
1398e76b63f8SYinghai Lu {
1399e76b63f8SYinghai Lu 	struct memblock_type *type = &memblock.memory;
140016763230SFabian Frederick 	int mid = memblock_search(type, PFN_PHYS(pfn));
1401e76b63f8SYinghai Lu 
1402e76b63f8SYinghai Lu 	if (mid == -1)
1403e76b63f8SYinghai Lu 		return -1;
1404e76b63f8SYinghai Lu 
1405e76b63f8SYinghai Lu 	*start_pfn = type->regions[mid].base >> PAGE_SHIFT;
1406e76b63f8SYinghai Lu 	*end_pfn = (type->regions[mid].base + type->regions[mid].size)
1407e76b63f8SYinghai Lu 			>> PAGE_SHIFT;
1408e76b63f8SYinghai Lu 
1409e76b63f8SYinghai Lu 	return type->regions[mid].nid;
1410e76b63f8SYinghai Lu }
1411e76b63f8SYinghai Lu #endif
1412e76b63f8SYinghai Lu 
1413eab30949SStephen Boyd /**
1414eab30949SStephen Boyd  * memblock_is_region_memory - check if a region is a subset of memory
1415eab30949SStephen Boyd  * @base: base of region to check
1416eab30949SStephen Boyd  * @size: size of region to check
1417eab30949SStephen Boyd  *
1418eab30949SStephen Boyd  * Check if the region [@base, @base+@size) is a subset of a memory block.
1419eab30949SStephen Boyd  *
1420eab30949SStephen Boyd  * RETURNS:
1421eab30949SStephen Boyd  * 0 if false, non-zero if true
1422eab30949SStephen Boyd  */
14233661ca66SYinghai Lu int __init_memblock memblock_is_region_memory(phys_addr_t base, phys_addr_t size)
142472d4b0b4SBenjamin Herrenschmidt {
1425abb65272STomi Valkeinen 	int idx = memblock_search(&memblock.memory, base);
1426eb18f1b5STejun Heo 	phys_addr_t end = base + memblock_cap_size(base, &size);
142772d4b0b4SBenjamin Herrenschmidt 
142872d4b0b4SBenjamin Herrenschmidt 	if (idx == -1)
142995f72d1eSYinghai Lu 		return 0;
1430abb65272STomi Valkeinen 	return memblock.memory.regions[idx].base <= base &&
1431abb65272STomi Valkeinen 		(memblock.memory.regions[idx].base +
1432eb18f1b5STejun Heo 		 memblock.memory.regions[idx].size) >= end;
143395f72d1eSYinghai Lu }
143495f72d1eSYinghai Lu 
1435eab30949SStephen Boyd /**
1436eab30949SStephen Boyd  * memblock_is_region_reserved - check if a region intersects reserved memory
1437eab30949SStephen Boyd  * @base: base of region to check
1438eab30949SStephen Boyd  * @size: size of region to check
1439eab30949SStephen Boyd  *
1440eab30949SStephen Boyd  * Check if the region [@base, @base+@size) intersects a reserved memory block.
1441eab30949SStephen Boyd  *
1442eab30949SStephen Boyd  * RETURNS:
1443eab30949SStephen Boyd  * 0 if false, non-zero if true
1444eab30949SStephen Boyd  */
144510d06439SYinghai Lu int __init_memblock memblock_is_region_reserved(phys_addr_t base, phys_addr_t size)
144695f72d1eSYinghai Lu {
1447eb18f1b5STejun Heo 	memblock_cap_size(base, &size);
1448f1c2c19cSBenjamin Herrenschmidt 	return memblock_overlaps_region(&memblock.reserved, base, size) >= 0;
144995f72d1eSYinghai Lu }
145095f72d1eSYinghai Lu 
14516ede1fd3SYinghai Lu void __init_memblock memblock_trim_memory(phys_addr_t align)
14526ede1fd3SYinghai Lu {
14536ede1fd3SYinghai Lu 	phys_addr_t start, end, orig_start, orig_end;
1454136199f0SEmil Medve 	struct memblock_region *r;
14556ede1fd3SYinghai Lu 
1456136199f0SEmil Medve 	for_each_memblock(memory, r) {
1457136199f0SEmil Medve 		orig_start = r->base;
1458136199f0SEmil Medve 		orig_end = r->base + r->size;
14596ede1fd3SYinghai Lu 		start = round_up(orig_start, align);
14606ede1fd3SYinghai Lu 		end = round_down(orig_end, align);
14616ede1fd3SYinghai Lu 
14626ede1fd3SYinghai Lu 		if (start == orig_start && end == orig_end)
14636ede1fd3SYinghai Lu 			continue;
14646ede1fd3SYinghai Lu 
14656ede1fd3SYinghai Lu 		if (start < end) {
1466136199f0SEmil Medve 			r->base = start;
1467136199f0SEmil Medve 			r->size = end - start;
14686ede1fd3SYinghai Lu 		} else {
1469136199f0SEmil Medve 			memblock_remove_region(&memblock.memory,
1470136199f0SEmil Medve 					       r - memblock.memory.regions);
1471136199f0SEmil Medve 			r--;
14726ede1fd3SYinghai Lu 		}
14736ede1fd3SYinghai Lu 	}
14746ede1fd3SYinghai Lu }
1475e63075a3SBenjamin Herrenschmidt 
14763661ca66SYinghai Lu void __init_memblock memblock_set_current_limit(phys_addr_t limit)
1477e63075a3SBenjamin Herrenschmidt {
1478e63075a3SBenjamin Herrenschmidt 	memblock.current_limit = limit;
1479e63075a3SBenjamin Herrenschmidt }
1480e63075a3SBenjamin Herrenschmidt 
1481fec51014SLaura Abbott phys_addr_t __init_memblock memblock_get_current_limit(void)
1482fec51014SLaura Abbott {
1483fec51014SLaura Abbott 	return memblock.current_limit;
1484fec51014SLaura Abbott }
1485fec51014SLaura Abbott 
14867c0caeb8STejun Heo static void __init_memblock memblock_dump(struct memblock_type *type, char *name)
14876ed311b2SBenjamin Herrenschmidt {
14886ed311b2SBenjamin Herrenschmidt 	unsigned long long base, size;
148966a20757STang Chen 	unsigned long flags;
14906ed311b2SBenjamin Herrenschmidt 	int i;
14916ed311b2SBenjamin Herrenschmidt 
14927c0caeb8STejun Heo 	pr_info(" %s.cnt  = 0x%lx\n", name, type->cnt);
14936ed311b2SBenjamin Herrenschmidt 
14947c0caeb8STejun Heo 	for (i = 0; i < type->cnt; i++) {
14957c0caeb8STejun Heo 		struct memblock_region *rgn = &type->regions[i];
14967c0caeb8STejun Heo 		char nid_buf[32] = "";
14976ed311b2SBenjamin Herrenschmidt 
14987c0caeb8STejun Heo 		base = rgn->base;
14997c0caeb8STejun Heo 		size = rgn->size;
150066a20757STang Chen 		flags = rgn->flags;
15017c0caeb8STejun Heo #ifdef CONFIG_HAVE_MEMBLOCK_NODE_MAP
15027c0caeb8STejun Heo 		if (memblock_get_region_node(rgn) != MAX_NUMNODES)
15037c0caeb8STejun Heo 			snprintf(nid_buf, sizeof(nid_buf), " on node %d",
15047c0caeb8STejun Heo 				 memblock_get_region_node(rgn));
15057c0caeb8STejun Heo #endif
150666a20757STang Chen 		pr_info(" %s[%#x]\t[%#016llx-%#016llx], %#llx bytes%s flags: %#lx\n",
150766a20757STang Chen 			name, i, base, base + size - 1, size, nid_buf, flags);
15086ed311b2SBenjamin Herrenschmidt 	}
15096ed311b2SBenjamin Herrenschmidt }
15106ed311b2SBenjamin Herrenschmidt 
15114ff7b82fSTejun Heo void __init_memblock __memblock_dump_all(void)
15126ed311b2SBenjamin Herrenschmidt {
15136ed311b2SBenjamin Herrenschmidt 	pr_info("MEMBLOCK configuration:\n");
15141440c4e2STejun Heo 	pr_info(" memory size = %#llx reserved size = %#llx\n",
15151440c4e2STejun Heo 		(unsigned long long)memblock.memory.total_size,
15161440c4e2STejun Heo 		(unsigned long long)memblock.reserved.total_size);
15176ed311b2SBenjamin Herrenschmidt 
15186ed311b2SBenjamin Herrenschmidt 	memblock_dump(&memblock.memory, "memory");
15196ed311b2SBenjamin Herrenschmidt 	memblock_dump(&memblock.reserved, "reserved");
15206ed311b2SBenjamin Herrenschmidt }
15216ed311b2SBenjamin Herrenschmidt 
15221aadc056STejun Heo void __init memblock_allow_resize(void)
15236ed311b2SBenjamin Herrenschmidt {
1524142b45a7SBenjamin Herrenschmidt 	memblock_can_resize = 1;
15256ed311b2SBenjamin Herrenschmidt }
15266ed311b2SBenjamin Herrenschmidt 
15276ed311b2SBenjamin Herrenschmidt static int __init early_memblock(char *p)
15286ed311b2SBenjamin Herrenschmidt {
15296ed311b2SBenjamin Herrenschmidt 	if (p && strstr(p, "debug"))
15306ed311b2SBenjamin Herrenschmidt 		memblock_debug = 1;
15316ed311b2SBenjamin Herrenschmidt 	return 0;
15326ed311b2SBenjamin Herrenschmidt }
15336ed311b2SBenjamin Herrenschmidt early_param("memblock", early_memblock);
15346ed311b2SBenjamin Herrenschmidt 
1535c378ddd5STejun Heo #if defined(CONFIG_DEBUG_FS) && !defined(CONFIG_ARCH_DISCARD_MEMBLOCK)
15366d03b885SBenjamin Herrenschmidt 
15376d03b885SBenjamin Herrenschmidt static int memblock_debug_show(struct seq_file *m, void *private)
15386d03b885SBenjamin Herrenschmidt {
15396d03b885SBenjamin Herrenschmidt 	struct memblock_type *type = m->private;
15406d03b885SBenjamin Herrenschmidt 	struct memblock_region *reg;
15416d03b885SBenjamin Herrenschmidt 	int i;
15426d03b885SBenjamin Herrenschmidt 
15436d03b885SBenjamin Herrenschmidt 	for (i = 0; i < type->cnt; i++) {
15446d03b885SBenjamin Herrenschmidt 		reg = &type->regions[i];
15456d03b885SBenjamin Herrenschmidt 		seq_printf(m, "%4d: ", i);
15466d03b885SBenjamin Herrenschmidt 		if (sizeof(phys_addr_t) == 4)
15476d03b885SBenjamin Herrenschmidt 			seq_printf(m, "0x%08lx..0x%08lx\n",
15486d03b885SBenjamin Herrenschmidt 				   (unsigned long)reg->base,
15496d03b885SBenjamin Herrenschmidt 				   (unsigned long)(reg->base + reg->size - 1));
15506d03b885SBenjamin Herrenschmidt 		else
15516d03b885SBenjamin Herrenschmidt 			seq_printf(m, "0x%016llx..0x%016llx\n",
15526d03b885SBenjamin Herrenschmidt 				   (unsigned long long)reg->base,
15536d03b885SBenjamin Herrenschmidt 				   (unsigned long long)(reg->base + reg->size - 1));
15546d03b885SBenjamin Herrenschmidt 
15556d03b885SBenjamin Herrenschmidt 	}
15566d03b885SBenjamin Herrenschmidt 	return 0;
15576d03b885SBenjamin Herrenschmidt }
15586d03b885SBenjamin Herrenschmidt 
15596d03b885SBenjamin Herrenschmidt static int memblock_debug_open(struct inode *inode, struct file *file)
15606d03b885SBenjamin Herrenschmidt {
15616d03b885SBenjamin Herrenschmidt 	return single_open(file, memblock_debug_show, inode->i_private);
15626d03b885SBenjamin Herrenschmidt }
15636d03b885SBenjamin Herrenschmidt 
15646d03b885SBenjamin Herrenschmidt static const struct file_operations memblock_debug_fops = {
15656d03b885SBenjamin Herrenschmidt 	.open = memblock_debug_open,
15666d03b885SBenjamin Herrenschmidt 	.read = seq_read,
15676d03b885SBenjamin Herrenschmidt 	.llseek = seq_lseek,
15686d03b885SBenjamin Herrenschmidt 	.release = single_release,
15696d03b885SBenjamin Herrenschmidt };
15706d03b885SBenjamin Herrenschmidt 
15716d03b885SBenjamin Herrenschmidt static int __init memblock_init_debugfs(void)
15726d03b885SBenjamin Herrenschmidt {
15736d03b885SBenjamin Herrenschmidt 	struct dentry *root = debugfs_create_dir("memblock", NULL);
15746d03b885SBenjamin Herrenschmidt 	if (!root)
15756d03b885SBenjamin Herrenschmidt 		return -ENXIO;
15766d03b885SBenjamin Herrenschmidt 	debugfs_create_file("memory", S_IRUGO, root, &memblock.memory, &memblock_debug_fops);
15776d03b885SBenjamin Herrenschmidt 	debugfs_create_file("reserved", S_IRUGO, root, &memblock.reserved, &memblock_debug_fops);
157870210ed9SPhilipp Hachtmann #ifdef CONFIG_HAVE_MEMBLOCK_PHYS_MAP
157970210ed9SPhilipp Hachtmann 	debugfs_create_file("physmem", S_IRUGO, root, &memblock.physmem, &memblock_debug_fops);
158070210ed9SPhilipp Hachtmann #endif
15816d03b885SBenjamin Herrenschmidt 
15826d03b885SBenjamin Herrenschmidt 	return 0;
15836d03b885SBenjamin Herrenschmidt }
15846d03b885SBenjamin Herrenschmidt __initcall(memblock_init_debugfs);
15856d03b885SBenjamin Herrenschmidt 
15866d03b885SBenjamin Herrenschmidt #endif /* CONFIG_DEBUG_FS */
1587