LCOV - code coverage report
Current view: top level - mm - memblock.c (source / functions) Hit Total Coverage
Test: Real Lines: 289 508 56.9 %
Date: 2020-10-17 15:46:43 Functions: 0 63 0.0 %
Legend: Neither, QEMU, Real, Both Branches: 0 0 -

           Branch data     Line data    Source code
       1                 :            : // SPDX-License-Identifier: GPL-2.0-or-later
       2                 :            : /*
       3                 :            :  * Procedures for maintaining information about logical memory blocks.
       4                 :            :  *
       5                 :            :  * Peter Bergner, IBM Corp.     June 2001.
       6                 :            :  * Copyright (C) 2001 Peter Bergner.
       7                 :            :  */
       8                 :            : 
       9                 :            : #include <linux/kernel.h>
      10                 :            : #include <linux/slab.h>
      11                 :            : #include <linux/init.h>
      12                 :            : #include <linux/bitops.h>
      13                 :            : #include <linux/poison.h>
      14                 :            : #include <linux/pfn.h>
      15                 :            : #include <linux/debugfs.h>
      16                 :            : #include <linux/kmemleak.h>
      17                 :            : #include <linux/seq_file.h>
      18                 :            : #include <linux/memblock.h>
      19                 :            : 
      20                 :            : #include <asm/sections.h>
      21                 :            : #include <linux/io.h>
      22                 :            : 
      23                 :            : #include "internal.h"
      24                 :            : 
      25                 :            : #define INIT_MEMBLOCK_REGIONS                   128
      26                 :            : #define INIT_PHYSMEM_REGIONS                    4
      27                 :            : 
      28                 :            : #ifndef INIT_MEMBLOCK_RESERVED_REGIONS
      29                 :            : # define INIT_MEMBLOCK_RESERVED_REGIONS         INIT_MEMBLOCK_REGIONS
      30                 :            : #endif
      31                 :            : 
      32                 :            : /**
      33                 :            :  * DOC: memblock overview
      34                 :            :  *
      35                 :            :  * Memblock is a method of managing memory regions during the early
      36                 :            :  * boot period when the usual kernel memory allocators are not up and
      37                 :            :  * running.
      38                 :            :  *
      39                 :            :  * Memblock views the system memory as collections of contiguous
      40                 :            :  * regions. There are several types of these collections:
      41                 :            :  *
      42                 :            :  * * ``memory`` - describes the physical memory available to the
      43                 :            :  *   kernel; this may differ from the actual physical memory installed
      44                 :            :  *   in the system, for instance when the memory is restricted with
      45                 :            :  *   ``mem=`` command line parameter
      46                 :            :  * * ``reserved`` - describes the regions that were allocated
      47                 :            :  * * ``physmap`` - describes the actual physical memory regardless of
      48                 :            :  *   the possible restrictions; the ``physmap`` type is only available
      49                 :            :  *   on some architectures.
      50                 :            :  *
      51                 :            :  * Each region is represented by :c:type:`struct memblock_region` that
      52                 :            :  * defines the region extents, its attributes and NUMA node id on NUMA
      53                 :            :  * systems. Every memory type is described by the :c:type:`struct
      54                 :            :  * memblock_type` which contains an array of memory regions along with
      55                 :            :  * the allocator metadata. The memory types are nicely wrapped with
      56                 :            :  * :c:type:`struct memblock`. This structure is statically initialzed
      57                 :            :  * at build time. The region arrays for the "memory" and "reserved"
      58                 :            :  * types are initially sized to %INIT_MEMBLOCK_REGIONS and for the
      59                 :            :  * "physmap" type to %INIT_PHYSMEM_REGIONS.
      60                 :            :  * The :c:func:`memblock_allow_resize` enables automatic resizing of
      61                 :            :  * the region arrays during addition of new regions. This feature
      62                 :            :  * should be used with care so that memory allocated for the region
      63                 :            :  * array will not overlap with areas that should be reserved, for
      64                 :            :  * example initrd.
      65                 :            :  *
      66                 :            :  * The early architecture setup should tell memblock what the physical
      67                 :            :  * memory layout is by using :c:func:`memblock_add` or
      68                 :            :  * :c:func:`memblock_add_node` functions. The first function does not
      69                 :            :  * assign the region to a NUMA node and it is appropriate for UMA
      70                 :            :  * systems. Yet, it is possible to use it on NUMA systems as well and
      71                 :            :  * assign the region to a NUMA node later in the setup process using
      72                 :            :  * :c:func:`memblock_set_node`. The :c:func:`memblock_add_node`
      73                 :            :  * performs such an assignment directly.
      74                 :            :  *
      75                 :            :  * Once memblock is setup the memory can be allocated using one of the
      76                 :            :  * API variants:
      77                 :            :  *
      78                 :            :  * * :c:func:`memblock_phys_alloc*` - these functions return the
      79                 :            :  *   **physical** address of the allocated memory
      80                 :            :  * * :c:func:`memblock_alloc*` - these functions return the **virtual**
      81                 :            :  *   address of the allocated memory.
      82                 :            :  *
      83                 :            :  * Note, that both API variants use implict assumptions about allowed
      84                 :            :  * memory ranges and the fallback methods. Consult the documentation
      85                 :            :  * of :c:func:`memblock_alloc_internal` and
      86                 :            :  * :c:func:`memblock_alloc_range_nid` functions for more elaboarte
      87                 :            :  * description.
      88                 :            :  *
      89                 :            :  * As the system boot progresses, the architecture specific
      90                 :            :  * :c:func:`mem_init` function frees all the memory to the buddy page
      91                 :            :  * allocator.
      92                 :            :  *
      93                 :            :  * Unless an architecure enables %CONFIG_ARCH_KEEP_MEMBLOCK, the
      94                 :            :  * memblock data structures will be discarded after the system
      95                 :            :  * initialization compltes.
      96                 :            :  */
      97                 :            : 
      98                 :            : #ifndef CONFIG_NEED_MULTIPLE_NODES
      99                 :            : struct pglist_data __refdata contig_page_data;
     100                 :            : EXPORT_SYMBOL(contig_page_data);
     101                 :            : #endif
     102                 :            : 
     103                 :            : unsigned long max_low_pfn;
     104                 :            : unsigned long min_low_pfn;
     105                 :            : unsigned long max_pfn;
     106                 :            : unsigned long long max_possible_pfn;
     107                 :            : 
     108                 :            : static struct memblock_region memblock_memory_init_regions[INIT_MEMBLOCK_REGIONS] __initdata_memblock;
     109                 :            : static struct memblock_region memblock_reserved_init_regions[INIT_MEMBLOCK_RESERVED_REGIONS] __initdata_memblock;
     110                 :            : #ifdef CONFIG_HAVE_MEMBLOCK_PHYS_MAP
     111                 :            : static struct memblock_region memblock_physmem_init_regions[INIT_PHYSMEM_REGIONS] __initdata_memblock;
     112                 :            : #endif
     113                 :            : 
     114                 :            : struct memblock memblock __initdata_memblock = {
     115                 :            :         .memory.regions         = memblock_memory_init_regions,
     116                 :            :         .memory.cnt             = 1,    /* empty dummy entry */
     117                 :            :         .memory.max             = INIT_MEMBLOCK_REGIONS,
     118                 :            :         .memory.name            = "memory",
     119                 :            : 
     120                 :            :         .reserved.regions       = memblock_reserved_init_regions,
     121                 :            :         .reserved.cnt           = 1,    /* empty dummy entry */
     122                 :            :         .reserved.max           = INIT_MEMBLOCK_RESERVED_REGIONS,
     123                 :            :         .reserved.name          = "reserved",
     124                 :            : 
     125                 :            : #ifdef CONFIG_HAVE_MEMBLOCK_PHYS_MAP
     126                 :            :         .physmem.regions        = memblock_physmem_init_regions,
     127                 :            :         .physmem.cnt            = 1,    /* empty dummy entry */
     128                 :            :         .physmem.max            = INIT_PHYSMEM_REGIONS,
     129                 :            :         .physmem.name           = "physmem",
     130                 :            : #endif
     131                 :            : 
     132                 :            :         .bottom_up              = false,
     133                 :            :         .current_limit          = MEMBLOCK_ALLOC_ANYWHERE,
     134                 :            : };
     135                 :            : 
     136                 :            : int memblock_debug __initdata_memblock;
     137                 :            : static bool system_has_some_mirror __initdata_memblock = false;
     138                 :            : static int memblock_can_resize __initdata_memblock;
     139                 :            : static int memblock_memory_in_slab __initdata_memblock = 0;
     140                 :            : static int memblock_reserved_in_slab __initdata_memblock = 0;
     141                 :            : 
     142                 :            : static enum memblock_flags __init_memblock choose_memblock_flags(void)
     143                 :            : {
     144                 :          3 :         return system_has_some_mirror ? MEMBLOCK_MIRROR : MEMBLOCK_NONE;
     145                 :            : }
     146                 :            : 
     147                 :            : /* adjust *@size so that (@base + *@size) doesn't overflow, return new size */
     148                 :            : static inline phys_addr_t memblock_cap_size(phys_addr_t base, phys_addr_t *size)
     149                 :            : {
     150                 :          3 :         return *size = min(*size, PHYS_ADDR_MAX - base);
     151                 :            : }
     152                 :            : 
     153                 :            : /*
     154                 :            :  * Address comparison utilities
     155                 :            :  */
     156                 :            : static unsigned long __init_memblock memblock_addrs_overlap(phys_addr_t base1, phys_addr_t size1,
     157                 :            :                                        phys_addr_t base2, phys_addr_t size2)
     158                 :            : {
     159                 :          3 :         return ((base1 < (base2 + size2)) && (base2 < (base1 + size1)));
     160                 :            : }
     161                 :            : 
     162                 :          3 : bool __init_memblock memblock_overlaps_region(struct memblock_type *type,
     163                 :            :                                         phys_addr_t base, phys_addr_t size)
     164                 :            : {
     165                 :            :         unsigned long i;
     166                 :            : 
     167                 :          3 :         for (i = 0; i < type->cnt; i++)
     168                 :          3 :                 if (memblock_addrs_overlap(base, size, type->regions[i].base,
     169                 :            :                                            type->regions[i].size))
     170                 :            :                         break;
     171                 :          3 :         return i < type->cnt;
     172                 :            : }
     173                 :            : 
     174                 :            : /**
     175                 :            :  * __memblock_find_range_bottom_up - find free area utility in bottom-up
     176                 :            :  * @start: start of candidate range
     177                 :            :  * @end: end of candidate range, can be %MEMBLOCK_ALLOC_ANYWHERE or
     178                 :            :  *       %MEMBLOCK_ALLOC_ACCESSIBLE
     179                 :            :  * @size: size of free area to find
     180                 :            :  * @align: alignment of free area to find
     181                 :            :  * @nid: nid of the free area to find, %NUMA_NO_NODE for any node
     182                 :            :  * @flags: pick from blocks based on memory attributes
     183                 :            :  *
     184                 :            :  * Utility called from memblock_find_in_range_node(), find free area bottom-up.
     185                 :            :  *
     186                 :            :  * Return:
     187                 :            :  * Found address on success, 0 on failure.
     188                 :            :  */
     189                 :            : static phys_addr_t __init_memblock
     190                 :          0 : __memblock_find_range_bottom_up(phys_addr_t start, phys_addr_t end,
     191                 :            :                                 phys_addr_t size, phys_addr_t align, int nid,
     192                 :            :                                 enum memblock_flags flags)
     193                 :            : {
     194                 :            :         phys_addr_t this_start, this_end, cand;
     195                 :            :         u64 i;
     196                 :            : 
     197                 :          0 :         for_each_free_mem_range(i, nid, flags, &this_start, &this_end, NULL) {
     198                 :          0 :                 this_start = clamp(this_start, start, end);
     199                 :          0 :                 this_end = clamp(this_end, start, end);
     200                 :            : 
     201                 :          0 :                 cand = round_up(this_start, align);
     202                 :          0 :                 if (cand < this_end && this_end - cand >= size)
     203                 :          0 :                         return cand;
     204                 :            :         }
     205                 :            : 
     206                 :            :         return 0;
     207                 :            : }
     208                 :            : 
     209                 :            : /**
     210                 :            :  * __memblock_find_range_top_down - find free area utility, in top-down
     211                 :            :  * @start: start of candidate range
     212                 :            :  * @end: end of candidate range, can be %MEMBLOCK_ALLOC_ANYWHERE or
     213                 :            :  *       %MEMBLOCK_ALLOC_ACCESSIBLE
     214                 :            :  * @size: size of free area to find
     215                 :            :  * @align: alignment of free area to find
     216                 :            :  * @nid: nid of the free area to find, %NUMA_NO_NODE for any node
     217                 :            :  * @flags: pick from blocks based on memory attributes
     218                 :            :  *
     219                 :            :  * Utility called from memblock_find_in_range_node(), find free area top-down.
     220                 :            :  *
     221                 :            :  * Return:
     222                 :            :  * Found address on success, 0 on failure.
     223                 :            :  */
     224                 :            : static phys_addr_t __init_memblock
     225                 :          3 : __memblock_find_range_top_down(phys_addr_t start, phys_addr_t end,
     226                 :            :                                phys_addr_t size, phys_addr_t align, int nid,
     227                 :            :                                enum memblock_flags flags)
     228                 :            : {
     229                 :            :         phys_addr_t this_start, this_end, cand;
     230                 :            :         u64 i;
     231                 :            : 
     232                 :          3 :         for_each_free_mem_range_reverse(i, nid, flags, &this_start, &this_end,
     233                 :            :                                         NULL) {
     234                 :          3 :                 this_start = clamp(this_start, start, end);
     235                 :          3 :                 this_end = clamp(this_end, start, end);
     236                 :            : 
     237                 :          3 :                 if (this_end < size)
     238                 :          0 :                         continue;
     239                 :            : 
     240                 :          3 :                 cand = round_down(this_end - size, align);
     241                 :          3 :                 if (cand >= this_start)
     242                 :          3 :                         return cand;
     243                 :            :         }
     244                 :            : 
     245                 :            :         return 0;
     246                 :            : }
     247                 :            : 
     248                 :            : /**
     249                 :            :  * memblock_find_in_range_node - find free area in given range and node
     250                 :            :  * @size: size of free area to find
     251                 :            :  * @align: alignment of free area to find
     252                 :            :  * @start: start of candidate range
     253                 :            :  * @end: end of candidate range, can be %MEMBLOCK_ALLOC_ANYWHERE or
     254                 :            :  *       %MEMBLOCK_ALLOC_ACCESSIBLE
     255                 :            :  * @nid: nid of the free area to find, %NUMA_NO_NODE for any node
     256                 :            :  * @flags: pick from blocks based on memory attributes
     257                 :            :  *
     258                 :            :  * Find @size free area aligned to @align in the specified range and node.
     259                 :            :  *
     260                 :            :  * When allocation direction is bottom-up, the @start should be greater
     261                 :            :  * than the end of the kernel image. Otherwise, it will be trimmed. The
     262                 :            :  * reason is that we want the bottom-up allocation just near the kernel
     263                 :            :  * image so it is highly likely that the allocated memory and the kernel
     264                 :            :  * will reside in the same node.
     265                 :            :  *
     266                 :            :  * If bottom-up allocation failed, will try to allocate memory top-down.
     267                 :            :  *
     268                 :            :  * Return:
     269                 :            :  * Found address on success, 0 on failure.
     270                 :            :  */
     271                 :          3 : static phys_addr_t __init_memblock memblock_find_in_range_node(phys_addr_t size,
     272                 :            :                                         phys_addr_t align, phys_addr_t start,
     273                 :            :                                         phys_addr_t end, int nid,
     274                 :            :                                         enum memblock_flags flags)
     275                 :            : {
     276                 :            :         phys_addr_t kernel_end, ret;
     277                 :            : 
     278                 :            :         /* pump up @end */
     279                 :          3 :         if (end == MEMBLOCK_ALLOC_ACCESSIBLE ||
     280                 :            :             end == MEMBLOCK_ALLOC_KASAN)
     281                 :          3 :                 end = memblock.current_limit;
     282                 :            : 
     283                 :            :         /* avoid allocating the first page */
     284                 :          3 :         start = max_t(phys_addr_t, start, PAGE_SIZE);
     285                 :          3 :         end = max(start, end);
     286                 :          3 :         kernel_end = __pa_symbol(_end);
     287                 :            : 
     288                 :            :         /*
     289                 :            :          * try bottom-up allocation only when bottom-up mode
     290                 :            :          * is set and @end is above the kernel image.
     291                 :            :          */
     292                 :          3 :         if (memblock_bottom_up() && end > kernel_end) {
     293                 :            :                 phys_addr_t bottom_up_start;
     294                 :            : 
     295                 :            :                 /* make sure we will allocate above the kernel */
     296                 :          0 :                 bottom_up_start = max(start, kernel_end);
     297                 :            : 
     298                 :            :                 /* ok, try bottom-up allocation first */
     299                 :          0 :                 ret = __memblock_find_range_bottom_up(bottom_up_start, end,
     300                 :            :                                                       size, align, nid, flags);
     301                 :          0 :                 if (ret)
     302                 :            :                         return ret;
     303                 :            : 
     304                 :            :                 /*
     305                 :            :                  * we always limit bottom-up allocation above the kernel,
     306                 :            :                  * but top-down allocation doesn't have the limit, so
     307                 :            :                  * retrying top-down allocation may succeed when bottom-up
     308                 :            :                  * allocation failed.
     309                 :            :                  *
     310                 :            :                  * bottom-up allocation is expected to be fail very rarely,
     311                 :            :                  * so we use WARN_ONCE() here to see the stack trace if
     312                 :            :                  * fail happens.
     313                 :            :                  */
     314                 :            :                 WARN_ONCE(IS_ENABLED(CONFIG_MEMORY_HOTREMOVE),
     315                 :            :                           "memblock: bottom-up allocation failed, memory hotremove may be affected\n");
     316                 :            :         }
     317                 :            : 
     318                 :          3 :         return __memblock_find_range_top_down(start, end, size, align, nid,
     319                 :            :                                               flags);
     320                 :            : }
     321                 :            : 
     322                 :            : /**
     323                 :            :  * memblock_find_in_range - find free area in given range
     324                 :            :  * @start: start of candidate range
     325                 :            :  * @end: end of candidate range, can be %MEMBLOCK_ALLOC_ANYWHERE or
     326                 :            :  *       %MEMBLOCK_ALLOC_ACCESSIBLE
     327                 :            :  * @size: size of free area to find
     328                 :            :  * @align: alignment of free area to find
     329                 :            :  *
     330                 :            :  * Find @size free area aligned to @align in the specified range.
     331                 :            :  *
     332                 :            :  * Return:
     333                 :            :  * Found address on success, 0 on failure.
     334                 :            :  */
     335                 :          3 : phys_addr_t __init_memblock memblock_find_in_range(phys_addr_t start,
     336                 :            :                                         phys_addr_t end, phys_addr_t size,
     337                 :            :                                         phys_addr_t align)
     338                 :            : {
     339                 :            :         phys_addr_t ret;
     340                 :            :         enum memblock_flags flags = choose_memblock_flags();
     341                 :            : 
     342                 :            : again:
     343                 :          3 :         ret = memblock_find_in_range_node(size, align, start, end,
     344                 :            :                                             NUMA_NO_NODE, flags);
     345                 :            : 
     346                 :          3 :         if (!ret && (flags & MEMBLOCK_MIRROR)) {
     347                 :          0 :                 pr_warn("Could not allocate %pap bytes of mirrored memory\n",
     348                 :            :                         &size);
     349                 :            :                 flags &= ~MEMBLOCK_MIRROR;
     350                 :          0 :                 goto again;
     351                 :            :         }
     352                 :            : 
     353                 :          3 :         return ret;
     354                 :            : }
     355                 :            : 
     356                 :          3 : static void __init_memblock memblock_remove_region(struct memblock_type *type, unsigned long r)
     357                 :            : {
     358                 :          3 :         type->total_size -= type->regions[r].size;
     359                 :          3 :         memmove(&type->regions[r], &type->regions[r + 1],
     360                 :          3 :                 (type->cnt - (r + 1)) * sizeof(type->regions[r]));
     361                 :          3 :         type->cnt--;
     362                 :            : 
     363                 :            :         /* Special case for empty arrays */
     364                 :          3 :         if (type->cnt == 0) {
     365                 :          0 :                 WARN_ON(type->total_size != 0);
     366                 :          0 :                 type->cnt = 1;
     367                 :          0 :                 type->regions[0].base = 0;
     368                 :          0 :                 type->regions[0].size = 0;
     369                 :          0 :                 type->regions[0].flags = 0;
     370                 :            :                 memblock_set_region_node(&type->regions[0], MAX_NUMNODES);
     371                 :            :         }
     372                 :          3 : }
     373                 :            : 
     374                 :            : #ifndef CONFIG_ARCH_KEEP_MEMBLOCK
     375                 :            : /**
     376                 :            :  * memblock_discard - discard memory and reserved arrays if they were allocated
     377                 :            :  */
     378                 :            : void __init memblock_discard(void)
     379                 :            : {
     380                 :            :         phys_addr_t addr, size;
     381                 :            : 
     382                 :            :         if (memblock.reserved.regions != memblock_reserved_init_regions) {
     383                 :            :                 addr = __pa(memblock.reserved.regions);
     384                 :            :                 size = PAGE_ALIGN(sizeof(struct memblock_region) *
     385                 :            :                                   memblock.reserved.max);
     386                 :            :                 __memblock_free_late(addr, size);
     387                 :            :         }
     388                 :            : 
     389                 :            :         if (memblock.memory.regions != memblock_memory_init_regions) {
     390                 :            :                 addr = __pa(memblock.memory.regions);
     391                 :            :                 size = PAGE_ALIGN(sizeof(struct memblock_region) *
     392                 :            :                                   memblock.memory.max);
     393                 :            :                 __memblock_free_late(addr, size);
     394                 :            :         }
     395                 :            : }
     396                 :            : #endif
     397                 :            : 
     398                 :            : /**
     399                 :            :  * memblock_double_array - double the size of the memblock regions array
     400                 :            :  * @type: memblock type of the regions array being doubled
     401                 :            :  * @new_area_start: starting address of memory range to avoid overlap with
     402                 :            :  * @new_area_size: size of memory range to avoid overlap with
     403                 :            :  *
     404                 :            :  * Double the size of the @type regions array. If memblock is being used to
     405                 :            :  * allocate memory for a new reserved regions array and there is a previously
     406                 :            :  * allocated memory range [@new_area_start, @new_area_start + @new_area_size]
     407                 :            :  * waiting to be reserved, ensure the memory used by the new array does
     408                 :            :  * not overlap.
     409                 :            :  *
     410                 :            :  * Return:
     411                 :            :  * 0 on success, -1 on failure.
     412                 :            :  */
     413                 :          0 : static int __init_memblock memblock_double_array(struct memblock_type *type,
     414                 :            :                                                 phys_addr_t new_area_start,
     415                 :            :                                                 phys_addr_t new_area_size)
     416                 :            : {
     417                 :            :         struct memblock_region *new_array, *old_array;
     418                 :            :         phys_addr_t old_alloc_size, new_alloc_size;
     419                 :            :         phys_addr_t old_size, new_size, addr, new_end;
     420                 :          0 :         int use_slab = slab_is_available();
     421                 :            :         int *in_slab;
     422                 :            : 
     423                 :            :         /* We don't allow resizing until we know about the reserved regions
     424                 :            :          * of memory that aren't suitable for allocation
     425                 :            :          */
     426                 :          0 :         if (!memblock_can_resize)
     427                 :            :                 return -1;
     428                 :            : 
     429                 :            :         /* Calculate new doubled size */
     430                 :          0 :         old_size = type->max * sizeof(struct memblock_region);
     431                 :          0 :         new_size = old_size << 1;
     432                 :            :         /*
     433                 :            :          * We need to allocated new one align to PAGE_SIZE,
     434                 :            :          *   so we can free them completely later.
     435                 :            :          */
     436                 :          0 :         old_alloc_size = PAGE_ALIGN(old_size);
     437                 :          0 :         new_alloc_size = PAGE_ALIGN(new_size);
     438                 :            : 
     439                 :            :         /* Retrieve the slab flag */
     440                 :          0 :         if (type == &memblock.memory)
     441                 :            :                 in_slab = &memblock_memory_in_slab;
     442                 :            :         else
     443                 :            :                 in_slab = &memblock_reserved_in_slab;
     444                 :            : 
     445                 :            :         /* Try to find some space for it */
     446                 :          0 :         if (use_slab) {
     447                 :            :                 new_array = kmalloc(new_size, GFP_KERNEL);
     448                 :          0 :                 addr = new_array ? __pa(new_array) : 0;
     449                 :            :         } else {
     450                 :            :                 /* only exclude range when trying to double reserved.regions */
     451                 :          0 :                 if (type != &memblock.reserved)
     452                 :            :                         new_area_start = new_area_size = 0;
     453                 :            : 
     454                 :          0 :                 addr = memblock_find_in_range(new_area_start + new_area_size,
     455                 :            :                                                 memblock.current_limit,
     456                 :            :                                                 new_alloc_size, PAGE_SIZE);
     457                 :          0 :                 if (!addr && new_area_size)
     458                 :          0 :                         addr = memblock_find_in_range(0,
     459                 :          0 :                                 min(new_area_start, memblock.current_limit),
     460                 :            :                                 new_alloc_size, PAGE_SIZE);
     461                 :            : 
     462                 :          0 :                 new_array = addr ? __va(addr) : NULL;
     463                 :            :         }
     464                 :          0 :         if (!addr) {
     465                 :          0 :                 pr_err("memblock: Failed to double %s array from %ld to %ld entries !\n",
     466                 :            :                        type->name, type->max, type->max * 2);
     467                 :          0 :                 return -1;
     468                 :            :         }
     469                 :            : 
     470                 :          0 :         new_end = addr + new_size - 1;
     471                 :          0 :         memblock_dbg("memblock: %s is doubled to %ld at [%pa-%pa]",
     472                 :            :                         type->name, type->max * 2, &addr, &new_end);
     473                 :            : 
     474                 :            :         /*
     475                 :            :          * Found space, we now need to move the array over before we add the
     476                 :            :          * reserved region since it may be our reserved array itself that is
     477                 :            :          * full.
     478                 :            :          */
     479                 :          0 :         memcpy(new_array, type->regions, old_size);
     480                 :          0 :         memset(new_array + type->max, 0, old_size);
     481                 :          0 :         old_array = type->regions;
     482                 :          0 :         type->regions = new_array;
     483                 :          0 :         type->max <<= 1;
     484                 :            : 
     485                 :            :         /* Free old array. We needn't free it if the array is the static one */
     486                 :          0 :         if (*in_slab)
     487                 :          0 :                 kfree(old_array);
     488                 :          0 :         else if (old_array != memblock_memory_init_regions &&
     489                 :            :                  old_array != memblock_reserved_init_regions)
     490                 :          0 :                 memblock_free(__pa(old_array), old_alloc_size);
     491                 :            : 
     492                 :            :         /*
     493                 :            :          * Reserve the new array if that comes from the memblock.  Otherwise, we
     494                 :            :          * needn't do it
     495                 :            :          */
     496                 :          0 :         if (!use_slab)
     497                 :          0 :                 BUG_ON(memblock_reserve(addr, new_alloc_size));
     498                 :            : 
     499                 :            :         /* Update slab flag */
     500                 :          0 :         *in_slab = use_slab;
     501                 :            : 
     502                 :          0 :         return 0;
     503                 :            : }
     504                 :            : 
     505                 :            : /**
     506                 :            :  * memblock_merge_regions - merge neighboring compatible regions
     507                 :            :  * @type: memblock type to scan
     508                 :            :  *
     509                 :            :  * Scan @type and merge neighboring compatible regions.
     510                 :            :  */
     511                 :          3 : static void __init_memblock memblock_merge_regions(struct memblock_type *type)
     512                 :            : {
     513                 :            :         int i = 0;
     514                 :            : 
     515                 :            :         /* cnt never goes below 1 */
     516                 :          3 :         while (i < type->cnt - 1) {
     517                 :          3 :                 struct memblock_region *this = &type->regions[i];
     518                 :          3 :                 struct memblock_region *next = &type->regions[i + 1];
     519                 :            : 
     520                 :          3 :                 if (this->base + this->size != next->base ||
     521                 :            :                     memblock_get_region_node(this) !=
     522                 :          3 :                     memblock_get_region_node(next) ||
     523                 :          3 :                     this->flags != next->flags) {
     524                 :          3 :                         BUG_ON(this->base + this->size > next->base);
     525                 :          3 :                         i++;
     526                 :          3 :                         continue;
     527                 :            :                 }
     528                 :            : 
     529                 :          3 :                 this->size += next->size;
     530                 :            :                 /* move forward from next + 1, index of which is i + 2 */
     531                 :          3 :                 memmove(next, next + 1, (type->cnt - (i + 2)) * sizeof(*next));
     532                 :          3 :                 type->cnt--;
     533                 :            :         }
     534                 :          3 : }
     535                 :            : 
     536                 :            : /**
     537                 :            :  * memblock_insert_region - insert new memblock region
     538                 :            :  * @type:       memblock type to insert into
     539                 :            :  * @idx:        index for the insertion point
     540                 :            :  * @base:       base address of the new region
     541                 :            :  * @size:       size of the new region
     542                 :            :  * @nid:        node id of the new region
     543                 :            :  * @flags:      flags of the new region
     544                 :            :  *
     545                 :            :  * Insert new memblock region [@base, @base + @size) into @type at @idx.
     546                 :            :  * @type must already have extra room to accommodate the new region.
     547                 :            :  */
     548                 :          3 : static void __init_memblock memblock_insert_region(struct memblock_type *type,
     549                 :            :                                                    int idx, phys_addr_t base,
     550                 :            :                                                    phys_addr_t size,
     551                 :            :                                                    int nid,
     552                 :            :                                                    enum memblock_flags flags)
     553                 :            : {
     554                 :          3 :         struct memblock_region *rgn = &type->regions[idx];
     555                 :            : 
     556                 :          3 :         BUG_ON(type->cnt >= type->max);
     557                 :          3 :         memmove(rgn + 1, rgn, (type->cnt - idx) * sizeof(*rgn));
     558                 :          3 :         rgn->base = base;
     559                 :          3 :         rgn->size = size;
     560                 :          3 :         rgn->flags = flags;
     561                 :            :         memblock_set_region_node(rgn, nid);
     562                 :          3 :         type->cnt++;
     563                 :          3 :         type->total_size += size;
     564                 :          3 : }
     565                 :            : 
     566                 :            : /**
     567                 :            :  * memblock_add_range - add new memblock region
     568                 :            :  * @type: memblock type to add new region into
     569                 :            :  * @base: base address of the new region
     570                 :            :  * @size: size of the new region
     571                 :            :  * @nid: nid of the new region
     572                 :            :  * @flags: flags of the new region
     573                 :            :  *
     574                 :            :  * Add new memblock region [@base, @base + @size) into @type.  The new region
     575                 :            :  * is allowed to overlap with existing ones - overlaps don't affect already
     576                 :            :  * existing regions.  @type is guaranteed to be minimal (all neighbouring
     577                 :            :  * compatible regions are merged) after the addition.
     578                 :            :  *
     579                 :            :  * Return:
     580                 :            :  * 0 on success, -errno on failure.
     581                 :            :  */
     582                 :          3 : int __init_memblock memblock_add_range(struct memblock_type *type,
     583                 :            :                                 phys_addr_t base, phys_addr_t size,
     584                 :            :                                 int nid, enum memblock_flags flags)
     585                 :            : {
     586                 :            :         bool insert = false;
     587                 :            :         phys_addr_t obase = base;
     588                 :          3 :         phys_addr_t end = base + memblock_cap_size(base, &size);
     589                 :            :         int idx, nr_new;
     590                 :            :         struct memblock_region *rgn;
     591                 :            : 
     592                 :          3 :         if (!size)
     593                 :            :                 return 0;
     594                 :            : 
     595                 :            :         /* special case for empty array */
     596                 :          3 :         if (type->regions[0].size == 0) {
     597                 :          3 :                 WARN_ON(type->cnt != 1 || type->total_size);
     598                 :          3 :                 type->regions[0].base = base;
     599                 :          3 :                 type->regions[0].size = size;
     600                 :          3 :                 type->regions[0].flags = flags;
     601                 :            :                 memblock_set_region_node(&type->regions[0], nid);
     602                 :          3 :                 type->total_size = size;
     603                 :          3 :                 return 0;
     604                 :            :         }
     605                 :            : repeat:
     606                 :            :         /*
     607                 :            :          * The following is executed twice.  Once with %false @insert and
     608                 :            :          * then with %true.  The first counts the number of regions needed
     609                 :            :          * to accommodate the new area.  The second actually inserts them.
     610                 :            :          */
     611                 :            :         base = obase;
     612                 :            :         nr_new = 0;
     613                 :            : 
     614                 :          3 :         for_each_memblock_type(idx, type, rgn) {
     615                 :          3 :                 phys_addr_t rbase = rgn->base;
     616                 :          3 :                 phys_addr_t rend = rbase + rgn->size;
     617                 :            : 
     618                 :          3 :                 if (rbase >= end)
     619                 :            :                         break;
     620                 :          3 :                 if (rend <= base)
     621                 :          3 :                         continue;
     622                 :            :                 /*
     623                 :            :                  * @rgn overlaps.  If it separates the lower part of new
     624                 :            :                  * area, insert that portion.
     625                 :            :                  */
     626                 :          0 :                 if (rbase > base) {
     627                 :            : #ifdef CONFIG_HAVE_MEMBLOCK_NODE_MAP
     628                 :            :                         WARN_ON(nid != memblock_get_region_node(rgn));
     629                 :            : #endif
     630                 :          0 :                         WARN_ON(flags != rgn->flags);
     631                 :          0 :                         nr_new++;
     632                 :          0 :                         if (insert)
     633                 :          0 :                                 memblock_insert_region(type, idx++, base,
     634                 :            :                                                        rbase - base, nid,
     635                 :            :                                                        flags);
     636                 :            :                 }
     637                 :            :                 /* area below @rend is dealt with, forget about it */
     638                 :          0 :                 base = min(rend, end);
     639                 :            :         }
     640                 :            : 
     641                 :            :         /* insert the remaining portion */
     642                 :          3 :         if (base < end) {
     643                 :          3 :                 nr_new++;
     644                 :          3 :                 if (insert)
     645                 :          3 :                         memblock_insert_region(type, idx, base, end - base,
     646                 :            :                                                nid, flags);
     647                 :            :         }
     648                 :            : 
     649                 :          3 :         if (!nr_new)
     650                 :            :                 return 0;
     651                 :            : 
     652                 :            :         /*
     653                 :            :          * If this was the first round, resize array and repeat for actual
     654                 :            :          * insertions; otherwise, merge and return.
     655                 :            :          */
     656                 :          3 :         if (!insert) {
     657                 :          3 :                 while (type->cnt + nr_new > type->max)
     658                 :          0 :                         if (memblock_double_array(type, obase, size) < 0)
     659                 :            :                                 return -ENOMEM;
     660                 :            :                 insert = true;
     661                 :            :                 goto repeat;
     662                 :            :         } else {
     663                 :          3 :                 memblock_merge_regions(type);
     664                 :          3 :                 return 0;
     665                 :            :         }
     666                 :            : }
     667                 :            : 
     668                 :            : /**
     669                 :            :  * memblock_add_node - add new memblock region within a NUMA node
     670                 :            :  * @base: base address of the new region
     671                 :            :  * @size: size of the new region
     672                 :            :  * @nid: nid of the new region
     673                 :            :  *
     674                 :            :  * Add new memblock region [@base, @base + @size) to the "memory"
     675                 :            :  * type. See memblock_add_range() description for mode details
     676                 :            :  *
     677                 :            :  * Return:
     678                 :            :  * 0 on success, -errno on failure.
     679                 :            :  */
     680                 :          0 : int __init_memblock memblock_add_node(phys_addr_t base, phys_addr_t size,
     681                 :            :                                        int nid)
     682                 :            : {
     683                 :          0 :         return memblock_add_range(&memblock.memory, base, size, nid, 0);
     684                 :            : }
     685                 :            : 
     686                 :            : /**
     687                 :            :  * memblock_add - add new memblock region
     688                 :            :  * @base: base address of the new region
     689                 :            :  * @size: size of the new region
     690                 :            :  *
     691                 :            :  * Add new memblock region [@base, @base + @size) to the "memory"
     692                 :            :  * type. See memblock_add_range() description for mode details
     693                 :            :  *
     694                 :            :  * Return:
     695                 :            :  * 0 on success, -errno on failure.
     696                 :            :  */
     697                 :          3 : int __init_memblock memblock_add(phys_addr_t base, phys_addr_t size)
     698                 :            : {
     699                 :          3 :         phys_addr_t end = base + size - 1;
     700                 :            : 
     701                 :          3 :         memblock_dbg("memblock_add: [%pa-%pa] %pS\n",
     702                 :            :                      &base, &end, (void *)_RET_IP_);
     703                 :            : 
     704                 :          3 :         return memblock_add_range(&memblock.memory, base, size, MAX_NUMNODES, 0);
     705                 :            : }
     706                 :            : 
     707                 :            : /**
     708                 :            :  * memblock_isolate_range - isolate given range into disjoint memblocks
     709                 :            :  * @type: memblock type to isolate range for
     710                 :            :  * @base: base of range to isolate
     711                 :            :  * @size: size of range to isolate
     712                 :            :  * @start_rgn: out parameter for the start of isolated region
     713                 :            :  * @end_rgn: out parameter for the end of isolated region
     714                 :            :  *
     715                 :            :  * Walk @type and ensure that regions don't cross the boundaries defined by
     716                 :            :  * [@base, @base + @size).  Crossing regions are split at the boundaries,
     717                 :            :  * which may create at most two more regions.  The index of the first
     718                 :            :  * region inside the range is returned in *@start_rgn and end in *@end_rgn.
     719                 :            :  *
     720                 :            :  * Return:
     721                 :            :  * 0 on success, -errno on failure.
     722                 :            :  */
     723                 :          3 : static int __init_memblock memblock_isolate_range(struct memblock_type *type,
     724                 :            :                                         phys_addr_t base, phys_addr_t size,
     725                 :            :                                         int *start_rgn, int *end_rgn)
     726                 :            : {
     727                 :          3 :         phys_addr_t end = base + memblock_cap_size(base, &size);
     728                 :            :         int idx;
     729                 :            :         struct memblock_region *rgn;
     730                 :            : 
     731                 :          3 :         *start_rgn = *end_rgn = 0;
     732                 :            : 
     733                 :          3 :         if (!size)
     734                 :            :                 return 0;
     735                 :            : 
     736                 :            :         /* we'll create at most two more regions */
     737                 :          3 :         while (type->cnt + 2 > type->max)
     738                 :          0 :                 if (memblock_double_array(type, base, size) < 0)
     739                 :            :                         return -ENOMEM;
     740                 :            : 
     741                 :          3 :         for_each_memblock_type(idx, type, rgn) {
     742                 :          3 :                 phys_addr_t rbase = rgn->base;
     743                 :          3 :                 phys_addr_t rend = rbase + rgn->size;
     744                 :            : 
     745                 :          3 :                 if (rbase >= end)
     746                 :            :                         break;
     747                 :          3 :                 if (rend <= base)
     748                 :          3 :                         continue;
     749                 :            : 
     750                 :          3 :                 if (rbase < base) {
     751                 :            :                         /*
     752                 :            :                          * @rgn intersects from below.  Split and continue
     753                 :            :                          * to process the next region - the new top half.
     754                 :            :                          */
     755                 :          3 :                         rgn->base = base;
     756                 :          3 :                         rgn->size -= base - rbase;
     757                 :          3 :                         type->total_size -= base - rbase;
     758                 :          3 :                         memblock_insert_region(type, idx, rbase, base - rbase,
     759                 :            :                                                memblock_get_region_node(rgn),
     760                 :            :                                                rgn->flags);
     761                 :          3 :                 } else if (rend > end) {
     762                 :            :                         /*
     763                 :            :                          * @rgn intersects from above.  Split and redo the
     764                 :            :                          * current region - the new bottom half.
     765                 :            :                          */
     766                 :          0 :                         rgn->base = end;
     767                 :          0 :                         rgn->size -= end - rbase;
     768                 :          0 :                         type->total_size -= end - rbase;
     769                 :          0 :                         memblock_insert_region(type, idx--, rbase, end - rbase,
     770                 :            :                                                memblock_get_region_node(rgn),
     771                 :            :                                                rgn->flags);
     772                 :            :                 } else {
     773                 :            :                         /* @rgn is fully contained, record it */
     774                 :          3 :                         if (!*end_rgn)
     775                 :          3 :                                 *start_rgn = idx;
     776                 :          3 :                         *end_rgn = idx + 1;
     777                 :            :                 }
     778                 :            :         }
     779                 :            : 
     780                 :            :         return 0;
     781                 :            : }
     782                 :            : 
     783                 :          3 : static int __init_memblock memblock_remove_range(struct memblock_type *type,
     784                 :            :                                           phys_addr_t base, phys_addr_t size)
     785                 :            : {
     786                 :            :         int start_rgn, end_rgn;
     787                 :            :         int i, ret;
     788                 :            : 
     789                 :          3 :         ret = memblock_isolate_range(type, base, size, &start_rgn, &end_rgn);
     790                 :          3 :         if (ret)
     791                 :            :                 return ret;
     792                 :            : 
     793                 :          3 :         for (i = end_rgn - 1; i >= start_rgn; i--)
     794                 :          3 :                 memblock_remove_region(type, i);
     795                 :            :         return 0;
     796                 :            : }
     797                 :            : 
     798                 :          0 : int __init_memblock memblock_remove(phys_addr_t base, phys_addr_t size)
     799                 :            : {
     800                 :          0 :         phys_addr_t end = base + size - 1;
     801                 :            : 
     802                 :          0 :         memblock_dbg("memblock_remove: [%pa-%pa] %pS\n",
     803                 :            :                      &base, &end, (void *)_RET_IP_);
     804                 :            : 
     805                 :          0 :         return memblock_remove_range(&memblock.memory, base, size);
     806                 :            : }
     807                 :            : 
     808                 :            : /**
     809                 :            :  * memblock_free - free boot memory block
     810                 :            :  * @base: phys starting address of the  boot memory block
     811                 :            :  * @size: size of the boot memory block in bytes
     812                 :            :  *
     813                 :            :  * Free boot memory block previously allocated by memblock_alloc_xx() API.
     814                 :            :  * The freeing memory will not be released to the buddy allocator.
     815                 :            :  */
     816                 :          3 : int __init_memblock memblock_free(phys_addr_t base, phys_addr_t size)
     817                 :            : {
     818                 :          3 :         phys_addr_t end = base + size - 1;
     819                 :            : 
     820                 :          3 :         memblock_dbg("   memblock_free: [%pa-%pa] %pS\n",
     821                 :            :                      &base, &end, (void *)_RET_IP_);
     822                 :            : 
     823                 :            :         kmemleak_free_part_phys(base, size);
     824                 :          3 :         return memblock_remove_range(&memblock.reserved, base, size);
     825                 :            : }
     826                 :            : 
     827                 :          3 : int __init_memblock memblock_reserve(phys_addr_t base, phys_addr_t size)
     828                 :            : {
     829                 :          3 :         phys_addr_t end = base + size - 1;
     830                 :            : 
     831                 :          3 :         memblock_dbg("memblock_reserve: [%pa-%pa] %pS\n",
     832                 :            :                      &base, &end, (void *)_RET_IP_);
     833                 :            : 
     834                 :          3 :         return memblock_add_range(&memblock.reserved, base, size, MAX_NUMNODES, 0);
     835                 :            : }
     836                 :            : 
     837                 :            : /**
     838                 :            :  * memblock_setclr_flag - set or clear flag for a memory region
     839                 :            :  * @base: base address of the region
     840                 :            :  * @size: size of the region
     841                 :            :  * @set: set or clear the flag
     842                 :            :  * @flag: the flag to udpate
     843                 :            :  *
     844                 :            :  * This function isolates region [@base, @base + @size), and sets/clears flag
     845                 :            :  *
     846                 :            :  * Return: 0 on success, -errno on failure.
     847                 :            :  */
     848                 :          3 : static int __init_memblock memblock_setclr_flag(phys_addr_t base,
     849                 :            :                                 phys_addr_t size, int set, int flag)
     850                 :            : {
     851                 :            :         struct memblock_type *type = &memblock.memory;
     852                 :            :         int i, ret, start_rgn, end_rgn;
     853                 :            : 
     854                 :          3 :         ret = memblock_isolate_range(type, base, size, &start_rgn, &end_rgn);
     855                 :          3 :         if (ret)
     856                 :            :                 return ret;
     857                 :            : 
     858                 :          3 :         for (i = start_rgn; i < end_rgn; i++) {
     859                 :          3 :                 struct memblock_region *r = &type->regions[i];
     860                 :            : 
     861                 :          3 :                 if (set)
     862                 :          0 :                         r->flags |= flag;
     863                 :            :                 else
     864                 :          3 :                         r->flags &= ~flag;
     865                 :            :         }
     866                 :            : 
     867                 :          3 :         memblock_merge_regions(type);
     868                 :          3 :         return 0;
     869                 :            : }
     870                 :            : 
     871                 :            : /**
     872                 :            :  * memblock_mark_hotplug - Mark hotpluggable memory with flag MEMBLOCK_HOTPLUG.
     873                 :            :  * @base: the base phys addr of the region
     874                 :            :  * @size: the size of the region
     875                 :            :  *
     876                 :            :  * Return: 0 on success, -errno on failure.
     877                 :            :  */
     878                 :          0 : int __init_memblock memblock_mark_hotplug(phys_addr_t base, phys_addr_t size)
     879                 :            : {
     880                 :          0 :         return memblock_setclr_flag(base, size, 1, MEMBLOCK_HOTPLUG);
     881                 :            : }
     882                 :            : 
     883                 :            : /**
     884                 :            :  * memblock_clear_hotplug - Clear flag MEMBLOCK_HOTPLUG for a specified region.
     885                 :            :  * @base: the base phys addr of the region
     886                 :            :  * @size: the size of the region
     887                 :            :  *
     888                 :            :  * Return: 0 on success, -errno on failure.
     889                 :            :  */
     890                 :          0 : int __init_memblock memblock_clear_hotplug(phys_addr_t base, phys_addr_t size)
     891                 :            : {
     892                 :          3 :         return memblock_setclr_flag(base, size, 0, MEMBLOCK_HOTPLUG);
     893                 :            : }
     894                 :            : 
     895                 :            : /**
     896                 :            :  * memblock_mark_mirror - Mark mirrored memory with flag MEMBLOCK_MIRROR.
     897                 :            :  * @base: the base phys addr of the region
     898                 :            :  * @size: the size of the region
     899                 :            :  *
     900                 :            :  * Return: 0 on success, -errno on failure.
     901                 :            :  */
     902                 :          0 : int __init_memblock memblock_mark_mirror(phys_addr_t base, phys_addr_t size)
     903                 :            : {
     904                 :          0 :         system_has_some_mirror = true;
     905                 :            : 
     906                 :          0 :         return memblock_setclr_flag(base, size, 1, MEMBLOCK_MIRROR);
     907                 :            : }
     908                 :            : 
     909                 :            : /**
     910                 :            :  * memblock_mark_nomap - Mark a memory region with flag MEMBLOCK_NOMAP.
     911                 :            :  * @base: the base phys addr of the region
     912                 :            :  * @size: the size of the region
     913                 :            :  *
     914                 :            :  * Return: 0 on success, -errno on failure.
     915                 :            :  */
     916                 :          0 : int __init_memblock memblock_mark_nomap(phys_addr_t base, phys_addr_t size)
     917                 :            : {
     918                 :          0 :         return memblock_setclr_flag(base, size, 1, MEMBLOCK_NOMAP);
     919                 :            : }
     920                 :            : 
     921                 :            : /**
     922                 :            :  * memblock_clear_nomap - Clear flag MEMBLOCK_NOMAP for a specified region.
     923                 :            :  * @base: the base phys addr of the region
     924                 :            :  * @size: the size of the region
     925                 :            :  *
     926                 :            :  * Return: 0 on success, -errno on failure.
     927                 :            :  */
     928                 :          0 : int __init_memblock memblock_clear_nomap(phys_addr_t base, phys_addr_t size)
     929                 :            : {
     930                 :          0 :         return memblock_setclr_flag(base, size, 0, MEMBLOCK_NOMAP);
     931                 :            : }
     932                 :            : 
     933                 :            : /**
     934                 :            :  * __next_reserved_mem_region - next function for for_each_reserved_region()
     935                 :            :  * @idx: pointer to u64 loop variable
     936                 :            :  * @out_start: ptr to phys_addr_t for start address of the region, can be %NULL
     937                 :            :  * @out_end: ptr to phys_addr_t for end address of the region, can be %NULL
     938                 :            :  *
     939                 :            :  * Iterate over all reserved memory regions.
     940                 :            :  */
     941                 :          3 : void __init_memblock __next_reserved_mem_region(u64 *idx,
     942                 :            :                                            phys_addr_t *out_start,
     943                 :            :                                            phys_addr_t *out_end)
     944                 :            : {
     945                 :            :         struct memblock_type *type = &memblock.reserved;
     946                 :            : 
     947                 :          3 :         if (*idx < type->cnt) {
     948                 :          3 :                 struct memblock_region *r = &type->regions[*idx];
     949                 :          3 :                 phys_addr_t base = r->base;
     950                 :          3 :                 phys_addr_t size = r->size;
     951                 :            : 
     952                 :          3 :                 if (out_start)
     953                 :          3 :                         *out_start = base;
     954                 :          3 :                 if (out_end)
     955                 :          3 :                         *out_end = base + size - 1;
     956                 :            : 
     957                 :          3 :                 *idx += 1;
     958                 :          3 :                 return;
     959                 :            :         }
     960                 :            : 
     961                 :            :         /* signal end of iteration */
     962                 :          3 :         *idx = ULLONG_MAX;
     963                 :            : }
     964                 :            : 
     965                 :          3 : static bool should_skip_region(struct memblock_region *m, int nid, int flags)
     966                 :            : {
     967                 :            :         int m_nid = memblock_get_region_node(m);
     968                 :            : 
     969                 :            :         /* only memory regions are associated with nodes, check it */
     970                 :          3 :         if (nid != NUMA_NO_NODE && nid != m_nid)
     971                 :            :                 return true;
     972                 :            : 
     973                 :            :         /* skip hotpluggable memory regions if needed */
     974                 :            :         if (movable_node_is_enabled() && memblock_is_hotpluggable(m))
     975                 :            :                 return true;
     976                 :            : 
     977                 :            :         /* if we want mirror memory skip non-mirror memory regions */
     978                 :          3 :         if ((flags & MEMBLOCK_MIRROR) && !memblock_is_mirror(m))
     979                 :            :                 return true;
     980                 :            : 
     981                 :            :         /* skip nomap memory unless we were asked for it explicitly */
     982                 :          3 :         if (!(flags & MEMBLOCK_NOMAP) && memblock_is_nomap(m))
     983                 :            :                 return true;
     984                 :            : 
     985                 :          3 :         return false;
     986                 :            : }
     987                 :            : 
     988                 :            : /**
     989                 :            :  * __next_mem_range - next function for for_each_free_mem_range() etc.
     990                 :            :  * @idx: pointer to u64 loop variable
     991                 :            :  * @nid: node selector, %NUMA_NO_NODE for all nodes
     992                 :            :  * @flags: pick from blocks based on memory attributes
     993                 :            :  * @type_a: pointer to memblock_type from where the range is taken
     994                 :            :  * @type_b: pointer to memblock_type which excludes memory from being taken
     995                 :            :  * @out_start: ptr to phys_addr_t for start address of the range, can be %NULL
     996                 :            :  * @out_end: ptr to phys_addr_t for end address of the range, can be %NULL
     997                 :            :  * @out_nid: ptr to int for nid of the range, can be %NULL
     998                 :            :  *
     999                 :            :  * Find the first area from *@idx which matches @nid, fill the out
    1000                 :            :  * parameters, and update *@idx for the next iteration.  The lower 32bit of
    1001                 :            :  * *@idx contains index into type_a and the upper 32bit indexes the
    1002                 :            :  * areas before each region in type_b.  For example, if type_b regions
    1003                 :            :  * look like the following,
    1004                 :            :  *
    1005                 :            :  *      0:[0-16), 1:[32-48), 2:[128-130)
    1006                 :            :  *
    1007                 :            :  * The upper 32bit indexes the following regions.
    1008                 :            :  *
    1009                 :            :  *      0:[0-0), 1:[16-32), 2:[48-128), 3:[130-MAX)
    1010                 :            :  *
    1011                 :            :  * As both region arrays are sorted, the function advances the two indices
    1012                 :            :  * in lockstep and returns each intersection.
    1013                 :            :  */
    1014                 :          3 : void __init_memblock __next_mem_range(u64 *idx, int nid,
    1015                 :            :                                       enum memblock_flags flags,
    1016                 :            :                                       struct memblock_type *type_a,
    1017                 :            :                                       struct memblock_type *type_b,
    1018                 :            :                                       phys_addr_t *out_start,
    1019                 :            :                                       phys_addr_t *out_end, int *out_nid)
    1020                 :            : {
    1021                 :          3 :         int idx_a = *idx & 0xffffffff;
    1022                 :          3 :         int idx_b = *idx >> 32;
    1023                 :            : 
    1024                 :          3 :         if (WARN_ONCE(nid == MAX_NUMNODES,
    1025                 :            :         "Usage of MAX_NUMNODES is deprecated. Use NUMA_NO_NODE instead\n"))
    1026                 :            :                 nid = NUMA_NO_NODE;
    1027                 :            : 
    1028                 :          3 :         for (; idx_a < type_a->cnt; idx_a++) {
    1029                 :          3 :                 struct memblock_region *m = &type_a->regions[idx_a];
    1030                 :            : 
    1031                 :          3 :                 phys_addr_t m_start = m->base;
    1032                 :          3 :                 phys_addr_t m_end = m->base + m->size;
    1033                 :            :                 int         m_nid = memblock_get_region_node(m);
    1034                 :            : 
    1035                 :          3 :                 if (should_skip_region(m, nid, flags))
    1036                 :          0 :                         continue;
    1037                 :            : 
    1038                 :          3 :                 if (!type_b) {
    1039                 :          0 :                         if (out_start)
    1040                 :          0 :                                 *out_start = m_start;
    1041                 :          0 :                         if (out_end)
    1042                 :          0 :                                 *out_end = m_end;
    1043                 :          0 :                         if (out_nid)
    1044                 :          0 :                                 *out_nid = m_nid;
    1045                 :          0 :                         idx_a++;
    1046                 :          0 :                         *idx = (u32)idx_a | (u64)idx_b << 32;
    1047                 :          0 :                         return;
    1048                 :            :                 }
    1049                 :            : 
    1050                 :            :                 /* scan areas before each reservation */
    1051                 :          3 :                 for (; idx_b < type_b->cnt + 1; idx_b++) {
    1052                 :            :                         struct memblock_region *r;
    1053                 :            :                         phys_addr_t r_start;
    1054                 :            :                         phys_addr_t r_end;
    1055                 :            : 
    1056                 :          3 :                         r = &type_b->regions[idx_b];
    1057                 :          3 :                         r_start = idx_b ? r[-1].base + r[-1].size : 0;
    1058                 :            :                         r_end = idx_b < type_b->cnt ?
    1059                 :          3 :                                 r->base : PHYS_ADDR_MAX;
    1060                 :            : 
    1061                 :            :                         /*
    1062                 :            :                          * if idx_b advanced past idx_a,
    1063                 :            :                          * break out to advance idx_a
    1064                 :            :                          */
    1065                 :          3 :                         if (r_start >= m_end)
    1066                 :            :                                 break;
    1067                 :            :                         /* if the two regions intersect, we're done */
    1068                 :          3 :                         if (m_start < r_end) {
    1069                 :          3 :                                 if (out_start)
    1070                 :          3 :                                         *out_start =
    1071                 :          3 :                                                 max(m_start, r_start);
    1072                 :          3 :                                 if (out_end)
    1073                 :          3 :                                         *out_end = min(m_end, r_end);
    1074                 :          3 :                                 if (out_nid)
    1075                 :          0 :                                         *out_nid = m_nid;
    1076                 :            :                                 /*
    1077                 :            :                                  * The region which ends first is
    1078                 :            :                                  * advanced for the next iteration.
    1079                 :            :                                  */
    1080                 :          3 :                                 if (m_end <= r_end)
    1081                 :          0 :                                         idx_a++;
    1082                 :            :                                 else
    1083                 :          3 :                                         idx_b++;
    1084                 :          3 :                                 *idx = (u32)idx_a | (u64)idx_b << 32;
    1085                 :          3 :                                 return;
    1086                 :            :                         }
    1087                 :            :                 }
    1088                 :            :         }
    1089                 :            : 
    1090                 :            :         /* signal end of iteration */
    1091                 :          3 :         *idx = ULLONG_MAX;
    1092                 :            : }
    1093                 :            : 
    1094                 :            : /**
    1095                 :            :  * __next_mem_range_rev - generic next function for for_each_*_range_rev()
    1096                 :            :  *
    1097                 :            :  * @idx: pointer to u64 loop variable
    1098                 :            :  * @nid: node selector, %NUMA_NO_NODE for all nodes
    1099                 :            :  * @flags: pick from blocks based on memory attributes
    1100                 :            :  * @type_a: pointer to memblock_type from where the range is taken
    1101                 :            :  * @type_b: pointer to memblock_type which excludes memory from being taken
    1102                 :            :  * @out_start: ptr to phys_addr_t for start address of the range, can be %NULL
    1103                 :            :  * @out_end: ptr to phys_addr_t for end address of the range, can be %NULL
    1104                 :            :  * @out_nid: ptr to int for nid of the range, can be %NULL
    1105                 :            :  *
    1106                 :            :  * Finds the next range from type_a which is not marked as unsuitable
    1107                 :            :  * in type_b.
    1108                 :            :  *
    1109                 :            :  * Reverse of __next_mem_range().
    1110                 :            :  */
    1111                 :          3 : void __init_memblock __next_mem_range_rev(u64 *idx, int nid,
    1112                 :            :                                           enum memblock_flags flags,
    1113                 :            :                                           struct memblock_type *type_a,
    1114                 :            :                                           struct memblock_type *type_b,
    1115                 :            :                                           phys_addr_t *out_start,
    1116                 :            :                                           phys_addr_t *out_end, int *out_nid)
    1117                 :            : {
    1118                 :          3 :         int idx_a = *idx & 0xffffffff;
    1119                 :          3 :         int idx_b = *idx >> 32;
    1120                 :            : 
    1121                 :          3 :         if (WARN_ONCE(nid == MAX_NUMNODES, "Usage of MAX_NUMNODES is deprecated. Use NUMA_NO_NODE instead\n"))
    1122                 :            :                 nid = NUMA_NO_NODE;
    1123                 :            : 
    1124                 :          3 :         if (*idx == (u64)ULLONG_MAX) {
    1125                 :          3 :                 idx_a = type_a->cnt - 1;
    1126                 :          3 :                 if (type_b != NULL)
    1127                 :          3 :                         idx_b = type_b->cnt;
    1128                 :            :                 else
    1129                 :            :                         idx_b = 0;
    1130                 :            :         }
    1131                 :            : 
    1132                 :          3 :         for (; idx_a >= 0; idx_a--) {
    1133                 :          3 :                 struct memblock_region *m = &type_a->regions[idx_a];
    1134                 :            : 
    1135                 :          3 :                 phys_addr_t m_start = m->base;
    1136                 :          3 :                 phys_addr_t m_end = m->base + m->size;
    1137                 :            :                 int m_nid = memblock_get_region_node(m);
    1138                 :            : 
    1139                 :          3 :                 if (should_skip_region(m, nid, flags))
    1140                 :          0 :                         continue;
    1141                 :            : 
    1142                 :          3 :                 if (!type_b) {
    1143                 :          0 :                         if (out_start)
    1144                 :          0 :                                 *out_start = m_start;
    1145                 :          0 :                         if (out_end)
    1146                 :          0 :                                 *out_end = m_end;
    1147                 :          0 :                         if (out_nid)
    1148                 :          0 :                                 *out_nid = m_nid;
    1149                 :          0 :                         idx_a--;
    1150                 :          0 :                         *idx = (u32)idx_a | (u64)idx_b << 32;
    1151                 :          0 :                         return;
    1152                 :            :                 }
    1153                 :            : 
    1154                 :            :                 /* scan areas before each reservation */
    1155                 :          3 :                 for (; idx_b >= 0; idx_b--) {
    1156                 :            :                         struct memblock_region *r;
    1157                 :            :                         phys_addr_t r_start;
    1158                 :            :                         phys_addr_t r_end;
    1159                 :            : 
    1160                 :          3 :                         r = &type_b->regions[idx_b];
    1161                 :          3 :                         r_start = idx_b ? r[-1].base + r[-1].size : 0;
    1162                 :          3 :                         r_end = idx_b < type_b->cnt ?
    1163                 :          3 :                                 r->base : PHYS_ADDR_MAX;
    1164                 :            :                         /*
    1165                 :            :                          * if idx_b advanced past idx_a,
    1166                 :            :                          * break out to advance idx_a
    1167                 :            :                          */
    1168                 :            : 
    1169                 :          3 :                         if (r_end <= m_start)
    1170                 :            :                                 break;
    1171                 :            :                         /* if the two regions intersect, we're done */
    1172                 :          3 :                         if (m_end > r_start) {
    1173                 :          3 :                                 if (out_start)
    1174                 :          3 :                                         *out_start = max(m_start, r_start);
    1175                 :          3 :                                 if (out_end)
    1176                 :          3 :                                         *out_end = min(m_end, r_end);
    1177                 :          3 :                                 if (out_nid)
    1178                 :          0 :                                         *out_nid = m_nid;
    1179                 :          3 :                                 if (m_start >= r_start)
    1180                 :          0 :                                         idx_a--;
    1181                 :            :                                 else
    1182                 :          3 :                                         idx_b--;
    1183                 :          3 :                                 *idx = (u32)idx_a | (u64)idx_b << 32;
    1184                 :          3 :                                 return;
    1185                 :            :                         }
    1186                 :            :                 }
    1187                 :            :         }
    1188                 :            :         /* signal end of iteration */
    1189                 :          3 :         *idx = ULLONG_MAX;
    1190                 :            : }
    1191                 :            : 
    1192                 :            : #ifdef CONFIG_HAVE_MEMBLOCK_NODE_MAP
    1193                 :            : /*
    1194                 :            :  * Common iterator interface used to define for_each_mem_pfn_range().
    1195                 :            :  */
    1196                 :            : void __init_memblock __next_mem_pfn_range(int *idx, int nid,
    1197                 :            :                                 unsigned long *out_start_pfn,
    1198                 :            :                                 unsigned long *out_end_pfn, int *out_nid)
    1199                 :            : {
    1200                 :            :         struct memblock_type *type = &memblock.memory;
    1201                 :            :         struct memblock_region *r;
    1202                 :            : 
    1203                 :            :         while (++*idx < type->cnt) {
    1204                 :            :                 r = &type->regions[*idx];
    1205                 :            : 
    1206                 :            :                 if (PFN_UP(r->base) >= PFN_DOWN(r->base + r->size))
    1207                 :            :                         continue;
    1208                 :            :                 if (nid == MAX_NUMNODES || nid == r->nid)
    1209                 :            :                         break;
    1210                 :            :         }
    1211                 :            :         if (*idx >= type->cnt) {
    1212                 :            :                 *idx = -1;
    1213                 :            :                 return;
    1214                 :            :         }
    1215                 :            : 
    1216                 :            :         if (out_start_pfn)
    1217                 :            :                 *out_start_pfn = PFN_UP(r->base);
    1218                 :            :         if (out_end_pfn)
    1219                 :            :                 *out_end_pfn = PFN_DOWN(r->base + r->size);
    1220                 :            :         if (out_nid)
    1221                 :            :                 *out_nid = r->nid;
    1222                 :            : }
    1223                 :            : 
    1224                 :            : /**
    1225                 :            :  * memblock_set_node - set node ID on memblock regions
    1226                 :            :  * @base: base of area to set node ID for
    1227                 :            :  * @size: size of area to set node ID for
    1228                 :            :  * @type: memblock type to set node ID for
    1229                 :            :  * @nid: node ID to set
    1230                 :            :  *
    1231                 :            :  * Set the nid of memblock @type regions in [@base, @base + @size) to @nid.
    1232                 :            :  * Regions which cross the area boundaries are split as necessary.
    1233                 :            :  *
    1234                 :            :  * Return:
    1235                 :            :  * 0 on success, -errno on failure.
    1236                 :            :  */
    1237                 :            : int __init_memblock memblock_set_node(phys_addr_t base, phys_addr_t size,
    1238                 :            :                                       struct memblock_type *type, int nid)
    1239                 :            : {
    1240                 :            :         int start_rgn, end_rgn;
    1241                 :            :         int i, ret;
    1242                 :            : 
    1243                 :            :         ret = memblock_isolate_range(type, base, size, &start_rgn, &end_rgn);
    1244                 :            :         if (ret)
    1245                 :            :                 return ret;
    1246                 :            : 
    1247                 :            :         for (i = start_rgn; i < end_rgn; i++)
    1248                 :            :                 memblock_set_region_node(&type->regions[i], nid);
    1249                 :            : 
    1250                 :            :         memblock_merge_regions(type);
    1251                 :            :         return 0;
    1252                 :            : }
    1253                 :            : #endif /* CONFIG_HAVE_MEMBLOCK_NODE_MAP */
    1254                 :            : #ifdef CONFIG_DEFERRED_STRUCT_PAGE_INIT
    1255                 :            : /**
    1256                 :            :  * __next_mem_pfn_range_in_zone - iterator for for_each_*_range_in_zone()
    1257                 :            :  *
    1258                 :            :  * @idx: pointer to u64 loop variable
    1259                 :            :  * @zone: zone in which all of the memory blocks reside
    1260                 :            :  * @out_spfn: ptr to ulong for start pfn of the range, can be %NULL
    1261                 :            :  * @out_epfn: ptr to ulong for end pfn of the range, can be %NULL
    1262                 :            :  *
    1263                 :            :  * This function is meant to be a zone/pfn specific wrapper for the
    1264                 :            :  * for_each_mem_range type iterators. Specifically they are used in the
    1265                 :            :  * deferred memory init routines and as such we were duplicating much of
    1266                 :            :  * this logic throughout the code. So instead of having it in multiple
    1267                 :            :  * locations it seemed like it would make more sense to centralize this to
    1268                 :            :  * one new iterator that does everything they need.
    1269                 :            :  */
    1270                 :            : void __init_memblock
    1271                 :            : __next_mem_pfn_range_in_zone(u64 *idx, struct zone *zone,
    1272                 :            :                              unsigned long *out_spfn, unsigned long *out_epfn)
    1273                 :            : {
    1274                 :            :         int zone_nid = zone_to_nid(zone);
    1275                 :            :         phys_addr_t spa, epa;
    1276                 :            :         int nid;
    1277                 :            : 
    1278                 :            :         __next_mem_range(idx, zone_nid, MEMBLOCK_NONE,
    1279                 :            :                          &memblock.memory, &memblock.reserved,
    1280                 :            :                          &spa, &epa, &nid);
    1281                 :            : 
    1282                 :            :         while (*idx != U64_MAX) {
    1283                 :            :                 unsigned long epfn = PFN_DOWN(epa);
    1284                 :            :                 unsigned long spfn = PFN_UP(spa);
    1285                 :            : 
    1286                 :            :                 /*
    1287                 :            :                  * Verify the end is at least past the start of the zone and
    1288                 :            :                  * that we have at least one PFN to initialize.
    1289                 :            :                  */
    1290                 :            :                 if (zone->zone_start_pfn < epfn && spfn < epfn) {
    1291                 :            :                         /* if we went too far just stop searching */
    1292                 :            :                         if (zone_end_pfn(zone) <= spfn) {
    1293                 :            :                                 *idx = U64_MAX;
    1294                 :            :                                 break;
    1295                 :            :                         }
    1296                 :            : 
    1297                 :            :                         if (out_spfn)
    1298                 :            :                                 *out_spfn = max(zone->zone_start_pfn, spfn);
    1299                 :            :                         if (out_epfn)
    1300                 :            :                                 *out_epfn = min(zone_end_pfn(zone), epfn);
    1301                 :            : 
    1302                 :            :                         return;
    1303                 :            :                 }
    1304                 :            : 
    1305                 :            :                 __next_mem_range(idx, zone_nid, MEMBLOCK_NONE,
    1306                 :            :                                  &memblock.memory, &memblock.reserved,
    1307                 :            :                                  &spa, &epa, &nid);
    1308                 :            :         }
    1309                 :            : 
    1310                 :            :         /* signal end of iteration */
    1311                 :            :         if (out_spfn)
    1312                 :            :                 *out_spfn = ULONG_MAX;
    1313                 :            :         if (out_epfn)
    1314                 :            :                 *out_epfn = 0;
    1315                 :            : }
    1316                 :            : 
    1317                 :            : #endif /* CONFIG_DEFERRED_STRUCT_PAGE_INIT */
    1318                 :            : 
    1319                 :            : /**
    1320                 :            :  * memblock_alloc_range_nid - allocate boot memory block
    1321                 :            :  * @size: size of memory block to be allocated in bytes
    1322                 :            :  * @align: alignment of the region and block's size
    1323                 :            :  * @start: the lower bound of the memory region to allocate (phys address)
    1324                 :            :  * @end: the upper bound of the memory region to allocate (phys address)
    1325                 :            :  * @nid: nid of the free area to find, %NUMA_NO_NODE for any node
    1326                 :            :  *
    1327                 :            :  * The allocation is performed from memory region limited by
    1328                 :            :  * memblock.current_limit if @max_addr == %MEMBLOCK_ALLOC_ACCESSIBLE.
    1329                 :            :  *
    1330                 :            :  * If the specified node can not hold the requested memory the
    1331                 :            :  * allocation falls back to any node in the system
    1332                 :            :  *
    1333                 :            :  * For systems with memory mirroring, the allocation is attempted first
    1334                 :            :  * from the regions with mirroring enabled and then retried from any
    1335                 :            :  * memory region.
    1336                 :            :  *
    1337                 :            :  * In addition, function sets the min_count to 0 using kmemleak_alloc_phys for
    1338                 :            :  * allocated boot memory block, so that it is never reported as leaks.
    1339                 :            :  *
    1340                 :            :  * Return:
    1341                 :            :  * Physical address of allocated memory block on success, %0 on failure.
    1342                 :            :  */
    1343                 :          3 : static phys_addr_t __init memblock_alloc_range_nid(phys_addr_t size,
    1344                 :            :                                         phys_addr_t align, phys_addr_t start,
    1345                 :            :                                         phys_addr_t end, int nid)
    1346                 :            : {
    1347                 :            :         enum memblock_flags flags = choose_memblock_flags();
    1348                 :            :         phys_addr_t found;
    1349                 :            : 
    1350                 :          3 :         if (WARN_ONCE(nid == MAX_NUMNODES, "Usage of MAX_NUMNODES is deprecated. Use NUMA_NO_NODE instead\n"))
    1351                 :            :                 nid = NUMA_NO_NODE;
    1352                 :            : 
    1353                 :          3 :         if (!align) {
    1354                 :            :                 /* Can't use WARNs this early in boot on powerpc */
    1355                 :          0 :                 dump_stack();
    1356                 :            :                 align = SMP_CACHE_BYTES;
    1357                 :            :         }
    1358                 :            : 
    1359                 :            : again:
    1360                 :          3 :         found = memblock_find_in_range_node(size, align, start, end, nid,
    1361                 :            :                                             flags);
    1362                 :          3 :         if (found && !memblock_reserve(found, size))
    1363                 :            :                 goto done;
    1364                 :            : 
    1365                 :          3 :         if (nid != NUMA_NO_NODE) {
    1366                 :          0 :                 found = memblock_find_in_range_node(size, align, start,
    1367                 :            :                                                     end, NUMA_NO_NODE,
    1368                 :            :                                                     flags);
    1369                 :          0 :                 if (found && !memblock_reserve(found, size))
    1370                 :            :                         goto done;
    1371                 :            :         }
    1372                 :            : 
    1373                 :          3 :         if (flags & MEMBLOCK_MIRROR) {
    1374                 :            :                 flags &= ~MEMBLOCK_MIRROR;
    1375                 :          0 :                 pr_warn("Could not allocate %pap bytes of mirrored memory\n",
    1376                 :            :                         &size);
    1377                 :          0 :                 goto again;
    1378                 :            :         }
    1379                 :            : 
    1380                 :            :         return 0;
    1381                 :            : 
    1382                 :            : done:
    1383                 :            :         /* Skip kmemleak for kasan_init() due to high volume. */
    1384                 :            :         if (end != MEMBLOCK_ALLOC_KASAN)
    1385                 :            :                 /*
    1386                 :            :                  * The min_count is set to 0 so that memblock allocated
    1387                 :            :                  * blocks are never reported as leaks. This is because many
    1388                 :            :                  * of these blocks are only referred via the physical
    1389                 :            :                  * address which is not looked up by kmemleak.
    1390                 :            :                  */
    1391                 :            :                 kmemleak_alloc_phys(found, size, 0, 0);
    1392                 :            : 
    1393                 :          3 :         return found;
    1394                 :            : }
    1395                 :            : 
    1396                 :            : /**
    1397                 :            :  * memblock_phys_alloc_range - allocate a memory block inside specified range
    1398                 :            :  * @size: size of memory block to be allocated in bytes
    1399                 :            :  * @align: alignment of the region and block's size
    1400                 :            :  * @start: the lower bound of the memory region to allocate (physical address)
    1401                 :            :  * @end: the upper bound of the memory region to allocate (physical address)
    1402                 :            :  *
    1403                 :            :  * Allocate @size bytes in the between @start and @end.
    1404                 :            :  *
    1405                 :            :  * Return: physical address of the allocated memory block on success,
    1406                 :            :  * %0 on failure.
    1407                 :            :  */
    1408                 :          0 : phys_addr_t __init memblock_phys_alloc_range(phys_addr_t size,
    1409                 :            :                                              phys_addr_t align,
    1410                 :            :                                              phys_addr_t start,
    1411                 :            :                                              phys_addr_t end)
    1412                 :            : {
    1413                 :          0 :         return memblock_alloc_range_nid(size, align, start, end, NUMA_NO_NODE);
    1414                 :            : }
    1415                 :            : 
    1416                 :            : /**
    1417                 :            :  * memblock_phys_alloc_try_nid - allocate a memory block from specified MUMA node
    1418                 :            :  * @size: size of memory block to be allocated in bytes
    1419                 :            :  * @align: alignment of the region and block's size
    1420                 :            :  * @nid: nid of the free area to find, %NUMA_NO_NODE for any node
    1421                 :            :  *
    1422                 :            :  * Allocates memory block from the specified NUMA node. If the node
    1423                 :            :  * has no available memory, attempts to allocated from any node in the
    1424                 :            :  * system.
    1425                 :            :  *
    1426                 :            :  * Return: physical address of the allocated memory block on success,
    1427                 :            :  * %0 on failure.
    1428                 :            :  */
    1429                 :          0 : phys_addr_t __init memblock_phys_alloc_try_nid(phys_addr_t size, phys_addr_t align, int nid)
    1430                 :            : {
    1431                 :          0 :         return memblock_alloc_range_nid(size, align, 0,
    1432                 :            :                                         MEMBLOCK_ALLOC_ACCESSIBLE, nid);
    1433                 :            : }
    1434                 :            : 
    1435                 :            : /**
    1436                 :            :  * memblock_alloc_internal - allocate boot memory block
    1437                 :            :  * @size: size of memory block to be allocated in bytes
    1438                 :            :  * @align: alignment of the region and block's size
    1439                 :            :  * @min_addr: the lower bound of the memory region to allocate (phys address)
    1440                 :            :  * @max_addr: the upper bound of the memory region to allocate (phys address)
    1441                 :            :  * @nid: nid of the free area to find, %NUMA_NO_NODE for any node
    1442                 :            :  *
    1443                 :            :  * Allocates memory block using memblock_alloc_range_nid() and
    1444                 :            :  * converts the returned physical address to virtual.
    1445                 :            :  *
    1446                 :            :  * The @min_addr limit is dropped if it can not be satisfied and the allocation
    1447                 :            :  * will fall back to memory below @min_addr. Other constraints, such
    1448                 :            :  * as node and mirrored memory will be handled again in
    1449                 :            :  * memblock_alloc_range_nid().
    1450                 :            :  *
    1451                 :            :  * Return:
    1452                 :            :  * Virtual address of allocated memory block on success, NULL on failure.
    1453                 :            :  */
    1454                 :          3 : static void * __init memblock_alloc_internal(
    1455                 :            :                                 phys_addr_t size, phys_addr_t align,
    1456                 :            :                                 phys_addr_t min_addr, phys_addr_t max_addr,
    1457                 :            :                                 int nid)
    1458                 :            : {
    1459                 :            :         phys_addr_t alloc;
    1460                 :            : 
    1461                 :            :         /*
    1462                 :            :          * Detect any accidental use of these APIs after slab is ready, as at
    1463                 :            :          * this moment memblock may be deinitialized already and its
    1464                 :            :          * internal data may be destroyed (after execution of memblock_free_all)
    1465                 :            :          */
    1466                 :          3 :         if (WARN_ON_ONCE(slab_is_available()))
    1467                 :          0 :                 return kzalloc_node(size, GFP_NOWAIT, nid);
    1468                 :            : 
    1469                 :          3 :         if (max_addr > memblock.current_limit)
    1470                 :            :                 max_addr = memblock.current_limit;
    1471                 :            : 
    1472                 :          3 :         alloc = memblock_alloc_range_nid(size, align, min_addr, max_addr, nid);
    1473                 :            : 
    1474                 :            :         /* retry allocation without lower limit */
    1475                 :          3 :         if (!alloc && min_addr)
    1476                 :          3 :                 alloc = memblock_alloc_range_nid(size, align, 0, max_addr, nid);
    1477                 :            : 
    1478                 :          3 :         if (!alloc)
    1479                 :            :                 return NULL;
    1480                 :            : 
    1481                 :          3 :         return phys_to_virt(alloc);
    1482                 :            : }
    1483                 :            : 
    1484                 :            : /**
    1485                 :            :  * memblock_alloc_try_nid_raw - allocate boot memory block without zeroing
    1486                 :            :  * memory and without panicking
    1487                 :            :  * @size: size of memory block to be allocated in bytes
    1488                 :            :  * @align: alignment of the region and block's size
    1489                 :            :  * @min_addr: the lower bound of the memory region from where the allocation
    1490                 :            :  *        is preferred (phys address)
    1491                 :            :  * @max_addr: the upper bound of the memory region from where the allocation
    1492                 :            :  *            is preferred (phys address), or %MEMBLOCK_ALLOC_ACCESSIBLE to
    1493                 :            :  *            allocate only from memory limited by memblock.current_limit value
    1494                 :            :  * @nid: nid of the free area to find, %NUMA_NO_NODE for any node
    1495                 :            :  *
    1496                 :            :  * Public function, provides additional debug information (including caller
    1497                 :            :  * info), if enabled. Does not zero allocated memory, does not panic if request
    1498                 :            :  * cannot be satisfied.
    1499                 :            :  *
    1500                 :            :  * Return:
    1501                 :            :  * Virtual address of allocated memory block on success, NULL on failure.
    1502                 :            :  */
    1503                 :          0 : void * __init memblock_alloc_try_nid_raw(
    1504                 :            :                         phys_addr_t size, phys_addr_t align,
    1505                 :            :                         phys_addr_t min_addr, phys_addr_t max_addr,
    1506                 :            :                         int nid)
    1507                 :            : {
    1508                 :            :         void *ptr;
    1509                 :            : 
    1510                 :          0 :         memblock_dbg("%s: %llu bytes align=0x%llx nid=%d from=%pa max_addr=%pa %pS\n",
    1511                 :            :                      __func__, (u64)size, (u64)align, nid, &min_addr,
    1512                 :            :                      &max_addr, (void *)_RET_IP_);
    1513                 :            : 
    1514                 :          0 :         ptr = memblock_alloc_internal(size, align,
    1515                 :            :                                            min_addr, max_addr, nid);
    1516                 :            :         if (ptr && size > 0)
    1517                 :            :                 page_init_poison(ptr, size);
    1518                 :            : 
    1519                 :          0 :         return ptr;
    1520                 :            : }
    1521                 :            : 
    1522                 :            : /**
    1523                 :            :  * memblock_alloc_try_nid - allocate boot memory block
    1524                 :            :  * @size: size of memory block to be allocated in bytes
    1525                 :            :  * @align: alignment of the region and block's size
    1526                 :            :  * @min_addr: the lower bound of the memory region from where the allocation
    1527                 :            :  *        is preferred (phys address)
    1528                 :            :  * @max_addr: the upper bound of the memory region from where the allocation
    1529                 :            :  *            is preferred (phys address), or %MEMBLOCK_ALLOC_ACCESSIBLE to
    1530                 :            :  *            allocate only from memory limited by memblock.current_limit value
    1531                 :            :  * @nid: nid of the free area to find, %NUMA_NO_NODE for any node
    1532                 :            :  *
    1533                 :            :  * Public function, provides additional debug information (including caller
    1534                 :            :  * info), if enabled. This function zeroes the allocated memory.
    1535                 :            :  *
    1536                 :            :  * Return:
    1537                 :            :  * Virtual address of allocated memory block on success, NULL on failure.
    1538                 :            :  */
    1539                 :          3 : void * __init memblock_alloc_try_nid(
    1540                 :            :                         phys_addr_t size, phys_addr_t align,
    1541                 :            :                         phys_addr_t min_addr, phys_addr_t max_addr,
    1542                 :            :                         int nid)
    1543                 :            : {
    1544                 :            :         void *ptr;
    1545                 :            : 
    1546                 :          3 :         memblock_dbg("%s: %llu bytes align=0x%llx nid=%d from=%pa max_addr=%pa %pS\n",
    1547                 :            :                      __func__, (u64)size, (u64)align, nid, &min_addr,
    1548                 :            :                      &max_addr, (void *)_RET_IP_);
    1549                 :          3 :         ptr = memblock_alloc_internal(size, align,
    1550                 :            :                                            min_addr, max_addr, nid);
    1551                 :          3 :         if (ptr)
    1552                 :          3 :                 memset(ptr, 0, size);
    1553                 :            : 
    1554                 :          3 :         return ptr;
    1555                 :            : }
    1556                 :            : 
    1557                 :            : /**
    1558                 :            :  * __memblock_free_late - free pages directly to buddy allocator
    1559                 :            :  * @base: phys starting address of the  boot memory block
    1560                 :            :  * @size: size of the boot memory block in bytes
    1561                 :            :  *
    1562                 :            :  * This is only useful when the memblock allocator has already been torn
    1563                 :            :  * down, but we are still initializing the system.  Pages are released directly
    1564                 :            :  * to the buddy allocator.
    1565                 :            :  */
    1566                 :          0 : void __init __memblock_free_late(phys_addr_t base, phys_addr_t size)
    1567                 :            : {
    1568                 :            :         phys_addr_t cursor, end;
    1569                 :            : 
    1570                 :          0 :         end = base + size - 1;
    1571                 :          0 :         memblock_dbg("%s: [%pa-%pa] %pS\n",
    1572                 :            :                      __func__, &base, &end, (void *)_RET_IP_);
    1573                 :            :         kmemleak_free_part_phys(base, size);
    1574                 :          0 :         cursor = PFN_UP(base);
    1575                 :          0 :         end = PFN_DOWN(base + size);
    1576                 :            : 
    1577                 :          0 :         for (; cursor < end; cursor++) {
    1578                 :          0 :                 memblock_free_pages(pfn_to_page(cursor), cursor, 0);
    1579                 :          0 :                 totalram_pages_inc();
    1580                 :            :         }
    1581                 :          0 : }
    1582                 :            : 
    1583                 :            : /*
    1584                 :            :  * Remaining API functions
    1585                 :            :  */
    1586                 :            : 
    1587                 :          0 : phys_addr_t __init_memblock memblock_phys_mem_size(void)
    1588                 :            : {
    1589                 :          0 :         return memblock.memory.total_size;
    1590                 :            : }
    1591                 :            : 
    1592                 :          0 : phys_addr_t __init_memblock memblock_reserved_size(void)
    1593                 :            : {
    1594                 :          0 :         return memblock.reserved.total_size;
    1595                 :            : }
    1596                 :            : 
    1597                 :          0 : phys_addr_t __init memblock_mem_size(unsigned long limit_pfn)
    1598                 :            : {
    1599                 :            :         unsigned long pages = 0;
    1600                 :            :         struct memblock_region *r;
    1601                 :            :         unsigned long start_pfn, end_pfn;
    1602                 :            : 
    1603                 :          0 :         for_each_memblock(memory, r) {
    1604                 :            :                 start_pfn = memblock_region_memory_base_pfn(r);
    1605                 :            :                 end_pfn = memblock_region_memory_end_pfn(r);
    1606                 :          0 :                 start_pfn = min_t(unsigned long, start_pfn, limit_pfn);
    1607                 :          0 :                 end_pfn = min_t(unsigned long, end_pfn, limit_pfn);
    1608                 :          0 :                 pages += end_pfn - start_pfn;
    1609                 :            :         }
    1610                 :            : 
    1611                 :          0 :         return PFN_PHYS(pages);
    1612                 :            : }
    1613                 :            : 
    1614                 :            : /* lowest address */
    1615                 :          3 : phys_addr_t __init_memblock memblock_start_of_DRAM(void)
    1616                 :            : {
    1617                 :          3 :         return memblock.memory.regions[0].base;
    1618                 :            : }
    1619                 :            : 
    1620                 :          3 : phys_addr_t __init_memblock memblock_end_of_DRAM(void)
    1621                 :            : {
    1622                 :          3 :         int idx = memblock.memory.cnt - 1;
    1623                 :            : 
    1624                 :          3 :         return (memblock.memory.regions[idx].base + memblock.memory.regions[idx].size);
    1625                 :            : }
    1626                 :            : 
    1627                 :            : static phys_addr_t __init_memblock __find_max_addr(phys_addr_t limit)
    1628                 :            : {
    1629                 :            :         phys_addr_t max_addr = PHYS_ADDR_MAX;
    1630                 :            :         struct memblock_region *r;
    1631                 :            : 
    1632                 :            :         /*
    1633                 :            :          * translate the memory @limit size into the max address within one of
    1634                 :            :          * the memory memblock regions, if the @limit exceeds the total size
    1635                 :            :          * of those regions, max_addr will keep original value PHYS_ADDR_MAX
    1636                 :            :          */
    1637                 :          0 :         for_each_memblock(memory, r) {
    1638                 :          0 :                 if (limit <= r->size) {
    1639                 :          0 :                         max_addr = r->base + limit;
    1640                 :            :                         break;
    1641                 :            :                 }
    1642                 :          0 :                 limit -= r->size;
    1643                 :            :         }
    1644                 :            : 
    1645                 :            :         return max_addr;
    1646                 :            : }
    1647                 :            : 
    1648                 :          0 : void __init memblock_enforce_memory_limit(phys_addr_t limit)
    1649                 :            : {
    1650                 :            :         phys_addr_t max_addr = PHYS_ADDR_MAX;
    1651                 :            : 
    1652                 :          0 :         if (!limit)
    1653                 :            :                 return;
    1654                 :            : 
    1655                 :            :         max_addr = __find_max_addr(limit);
    1656                 :            : 
    1657                 :            :         /* @limit exceeds the total size of the memory, do nothing */
    1658                 :          0 :         if (max_addr == PHYS_ADDR_MAX)
    1659                 :            :                 return;
    1660                 :            : 
    1661                 :            :         /* truncate both memory and reserved regions */
    1662                 :          0 :         memblock_remove_range(&memblock.memory, max_addr,
    1663                 :            :                               PHYS_ADDR_MAX);
    1664                 :          0 :         memblock_remove_range(&memblock.reserved, max_addr,
    1665                 :            :                               PHYS_ADDR_MAX);
    1666                 :            : }
    1667                 :            : 
    1668                 :          0 : void __init memblock_cap_memory_range(phys_addr_t base, phys_addr_t size)
    1669                 :            : {
    1670                 :            :         int start_rgn, end_rgn;
    1671                 :            :         int i, ret;
    1672                 :            : 
    1673                 :          0 :         if (!size)
    1674                 :          0 :                 return;
    1675                 :            : 
    1676                 :          0 :         ret = memblock_isolate_range(&memblock.memory, base, size,
    1677                 :            :                                                 &start_rgn, &end_rgn);
    1678                 :          0 :         if (ret)
    1679                 :            :                 return;
    1680                 :            : 
    1681                 :            :         /* remove all the MAP regions */
    1682                 :          0 :         for (i = memblock.memory.cnt - 1; i >= end_rgn; i--)
    1683                 :          0 :                 if (!memblock_is_nomap(&memblock.memory.regions[i]))
    1684                 :          0 :                         memblock_remove_region(&memblock.memory, i);
    1685                 :            : 
    1686                 :          0 :         for (i = start_rgn - 1; i >= 0; i--)
    1687                 :          0 :                 if (!memblock_is_nomap(&memblock.memory.regions[i]))
    1688                 :          0 :                         memblock_remove_region(&memblock.memory, i);
    1689                 :            : 
    1690                 :            :         /* truncate the reserved regions */
    1691                 :          0 :         memblock_remove_range(&memblock.reserved, 0, base);
    1692                 :          0 :         memblock_remove_range(&memblock.reserved,
    1693                 :            :                         base + size, PHYS_ADDR_MAX);
    1694                 :            : }
    1695                 :            : 
    1696                 :          0 : void __init memblock_mem_limit_remove_map(phys_addr_t limit)
    1697                 :            : {
    1698                 :            :         phys_addr_t max_addr;
    1699                 :            : 
    1700                 :          0 :         if (!limit)
    1701                 :            :                 return;
    1702                 :            : 
    1703                 :            :         max_addr = __find_max_addr(limit);
    1704                 :            : 
    1705                 :            :         /* @limit exceeds the total size of the memory, do nothing */
    1706                 :          0 :         if (max_addr == PHYS_ADDR_MAX)
    1707                 :            :                 return;
    1708                 :            : 
    1709                 :          0 :         memblock_cap_memory_range(0, max_addr);
    1710                 :            : }
    1711                 :            : 
    1712                 :            : static int __init_memblock memblock_search(struct memblock_type *type, phys_addr_t addr)
    1713                 :            : {
    1714                 :          3 :         unsigned int left = 0, right = type->cnt;
    1715                 :            : 
    1716                 :            :         do {
    1717                 :          3 :                 unsigned int mid = (right + left) / 2;
    1718                 :            : 
    1719                 :          3 :                 if (addr < type->regions[mid].base)
    1720                 :            :                         right = mid;
    1721                 :          3 :                 else if (addr >= (type->regions[mid].base +
    1722                 :          3 :                                   type->regions[mid].size))
    1723                 :          3 :                         left = mid + 1;
    1724                 :            :                 else
    1725                 :          3 :                         return mid;
    1726                 :          3 :         } while (left < right);
    1727                 :            :         return -1;
    1728                 :            : }
    1729                 :            : 
    1730                 :          0 : bool __init_memblock memblock_is_reserved(phys_addr_t addr)
    1731                 :            : {
    1732                 :          0 :         return memblock_search(&memblock.reserved, addr) != -1;
    1733                 :            : }
    1734                 :            : 
    1735                 :          0 : bool __init_memblock memblock_is_memory(phys_addr_t addr)
    1736                 :            : {
    1737                 :          0 :         return memblock_search(&memblock.memory, addr) != -1;
    1738                 :            : }
    1739                 :            : 
    1740                 :          3 : bool __init_memblock memblock_is_map_memory(phys_addr_t addr)
    1741                 :            : {
    1742                 :            :         int i = memblock_search(&memblock.memory, addr);
    1743                 :            : 
    1744                 :          3 :         if (i == -1)
    1745                 :            :                 return false;
    1746                 :          3 :         return !memblock_is_nomap(&memblock.memory.regions[i]);
    1747                 :            : }
    1748                 :            : 
    1749                 :            : #ifdef CONFIG_HAVE_MEMBLOCK_NODE_MAP
    1750                 :            : int __init_memblock memblock_search_pfn_nid(unsigned long pfn,
    1751                 :            :                          unsigned long *start_pfn, unsigned long *end_pfn)
    1752                 :            : {
    1753                 :            :         struct memblock_type *type = &memblock.memory;
    1754                 :            :         int mid = memblock_search(type, PFN_PHYS(pfn));
    1755                 :            : 
    1756                 :            :         if (mid == -1)
    1757                 :            :                 return -1;
    1758                 :            : 
    1759                 :            :         *start_pfn = PFN_DOWN(type->regions[mid].base);
    1760                 :            :         *end_pfn = PFN_DOWN(type->regions[mid].base + type->regions[mid].size);
    1761                 :            : 
    1762                 :            :         return type->regions[mid].nid;
    1763                 :            : }
    1764                 :            : #endif
    1765                 :            : 
    1766                 :            : /**
    1767                 :            :  * memblock_is_region_memory - check if a region is a subset of memory
    1768                 :            :  * @base: base of region to check
    1769                 :            :  * @size: size of region to check
    1770                 :            :  *
    1771                 :            :  * Check if the region [@base, @base + @size) is a subset of a memory block.
    1772                 :            :  *
    1773                 :            :  * Return:
    1774                 :            :  * 0 if false, non-zero if true
    1775                 :            :  */
    1776                 :          0 : bool __init_memblock memblock_is_region_memory(phys_addr_t base, phys_addr_t size)
    1777                 :            : {
    1778                 :            :         int idx = memblock_search(&memblock.memory, base);
    1779                 :          0 :         phys_addr_t end = base + memblock_cap_size(base, &size);
    1780                 :            : 
    1781                 :          0 :         if (idx == -1)
    1782                 :            :                 return false;
    1783                 :          0 :         return (memblock.memory.regions[idx].base +
    1784                 :          0 :                  memblock.memory.regions[idx].size) >= end;
    1785                 :            : }
    1786                 :            : 
    1787                 :            : /**
    1788                 :            :  * memblock_is_region_reserved - check if a region intersects reserved memory
    1789                 :            :  * @base: base of region to check
    1790                 :            :  * @size: size of region to check
    1791                 :            :  *
    1792                 :            :  * Check if the region [@base, @base + @size) intersects a reserved
    1793                 :            :  * memory block.
    1794                 :            :  *
    1795                 :            :  * Return:
    1796                 :            :  * True if they intersect, false if not.
    1797                 :            :  */
    1798                 :          3 : bool __init_memblock memblock_is_region_reserved(phys_addr_t base, phys_addr_t size)
    1799                 :            : {
    1800                 :            :         memblock_cap_size(base, &size);
    1801                 :          3 :         return memblock_overlaps_region(&memblock.reserved, base, size);
    1802                 :            : }
    1803                 :            : 
    1804                 :          0 : void __init_memblock memblock_trim_memory(phys_addr_t align)
    1805                 :            : {
    1806                 :            :         phys_addr_t start, end, orig_start, orig_end;
    1807                 :            :         struct memblock_region *r;
    1808                 :            : 
    1809                 :          0 :         for_each_memblock(memory, r) {
    1810                 :          0 :                 orig_start = r->base;
    1811                 :          0 :                 orig_end = r->base + r->size;
    1812                 :          0 :                 start = round_up(orig_start, align);
    1813                 :          0 :                 end = round_down(orig_end, align);
    1814                 :            : 
    1815                 :          0 :                 if (start == orig_start && end == orig_end)
    1816                 :          0 :                         continue;
    1817                 :            : 
    1818                 :          0 :                 if (start < end) {
    1819                 :          0 :                         r->base = start;
    1820                 :          0 :                         r->size = end - start;
    1821                 :            :                 } else {
    1822                 :          0 :                         memblock_remove_region(&memblock.memory,
    1823                 :          0 :                                                r - memblock.memory.regions);
    1824                 :          0 :                         r--;
    1825                 :            :                 }
    1826                 :            :         }
    1827                 :          0 : }
    1828                 :            : 
    1829                 :          3 : void __init_memblock memblock_set_current_limit(phys_addr_t limit)
    1830                 :            : {
    1831                 :          3 :         memblock.current_limit = limit;
    1832                 :          3 : }
    1833                 :            : 
    1834                 :          3 : phys_addr_t __init_memblock memblock_get_current_limit(void)
    1835                 :            : {
    1836                 :          3 :         return memblock.current_limit;
    1837                 :            : }
    1838                 :            : 
    1839                 :          0 : static void __init_memblock memblock_dump(struct memblock_type *type)
    1840                 :            : {
    1841                 :            :         phys_addr_t base, end, size;
    1842                 :            :         enum memblock_flags flags;
    1843                 :            :         int idx;
    1844                 :            :         struct memblock_region *rgn;
    1845                 :            : 
    1846                 :          0 :         pr_info(" %s.cnt  = 0x%lx\n", type->name, type->cnt);
    1847                 :            : 
    1848                 :          0 :         for_each_memblock_type(idx, type, rgn) {
    1849                 :          0 :                 char nid_buf[32] = "";
    1850                 :            : 
    1851                 :          0 :                 base = rgn->base;
    1852                 :          0 :                 size = rgn->size;
    1853                 :          0 :                 end = base + size - 1;
    1854                 :          0 :                 flags = rgn->flags;
    1855                 :            : #ifdef CONFIG_HAVE_MEMBLOCK_NODE_MAP
    1856                 :            :                 if (memblock_get_region_node(rgn) != MAX_NUMNODES)
    1857                 :            :                         snprintf(nid_buf, sizeof(nid_buf), " on node %d",
    1858                 :            :                                  memblock_get_region_node(rgn));
    1859                 :            : #endif
    1860                 :          0 :                 pr_info(" %s[%#x]\t[%pa-%pa], %pa bytes%s flags: %#x\n",
    1861                 :            :                         type->name, idx, &base, &end, &size, nid_buf, flags);
    1862                 :            :         }
    1863                 :          0 : }
    1864                 :            : 
    1865                 :          0 : void __init_memblock __memblock_dump_all(void)
    1866                 :            : {
    1867                 :          0 :         pr_info("MEMBLOCK configuration:\n");
    1868                 :          0 :         pr_info(" memory size = %pa reserved size = %pa\n",
    1869                 :            :                 &memblock.memory.total_size,
    1870                 :            :                 &memblock.reserved.total_size);
    1871                 :            : 
    1872                 :          0 :         memblock_dump(&memblock.memory);
    1873                 :          0 :         memblock_dump(&memblock.reserved);
    1874                 :            : #ifdef CONFIG_HAVE_MEMBLOCK_PHYS_MAP
    1875                 :            :         memblock_dump(&memblock.physmem);
    1876                 :            : #endif
    1877                 :          0 : }
    1878                 :            : 
    1879                 :          3 : void __init memblock_allow_resize(void)
    1880                 :            : {
    1881                 :          3 :         memblock_can_resize = 1;
    1882                 :          3 : }
    1883                 :            : 
    1884                 :          0 : static int __init early_memblock(char *p)
    1885                 :            : {
    1886                 :          0 :         if (p && strstr(p, "debug"))
    1887                 :          0 :                 memblock_debug = 1;
    1888                 :          0 :         return 0;
    1889                 :            : }
    1890                 :            : early_param("memblock", early_memblock);
    1891                 :            : 
    1892                 :          3 : static void __init __free_pages_memory(unsigned long start, unsigned long end)
    1893                 :            : {
    1894                 :            :         int order;
    1895                 :            : 
    1896                 :          3 :         while (start < end) {
    1897                 :          3 :                 order = min(MAX_ORDER - 1UL, __ffs(start));
    1898                 :            : 
    1899                 :          3 :                 while (start + (1UL << order) > end)
    1900                 :          3 :                         order--;
    1901                 :            : 
    1902                 :          3 :                 memblock_free_pages(pfn_to_page(start), start, order);
    1903                 :            : 
    1904                 :          3 :                 start += (1UL << order);
    1905                 :            :         }
    1906                 :          3 : }
    1907                 :            : 
    1908                 :          3 : static unsigned long __init __free_memory_core(phys_addr_t start,
    1909                 :            :                                  phys_addr_t end)
    1910                 :            : {
    1911                 :          3 :         unsigned long start_pfn = PFN_UP(start);
    1912                 :          3 :         unsigned long end_pfn = min_t(unsigned long,
    1913                 :            :                                       PFN_DOWN(end), max_low_pfn);
    1914                 :            : 
    1915                 :          3 :         if (start_pfn >= end_pfn)
    1916                 :            :                 return 0;
    1917                 :            : 
    1918                 :          3 :         __free_pages_memory(start_pfn, end_pfn);
    1919                 :            : 
    1920                 :          3 :         return end_pfn - start_pfn;
    1921                 :            : }
    1922                 :            : 
    1923                 :          3 : static unsigned long __init free_low_memory_core_early(void)
    1924                 :            : {
    1925                 :            :         unsigned long count = 0;
    1926                 :            :         phys_addr_t start, end;
    1927                 :            :         u64 i;
    1928                 :            : 
    1929                 :            :         memblock_clear_hotplug(0, -1);
    1930                 :            : 
    1931                 :          3 :         for_each_reserved_mem_region(i, &start, &end)
    1932                 :          3 :                 reserve_bootmem_region(start, end);
    1933                 :            : 
    1934                 :            :         /*
    1935                 :            :          * We need to use NUMA_NO_NODE instead of NODE_DATA(0)->node_id
    1936                 :            :          *  because in some case like Node0 doesn't have RAM installed
    1937                 :            :          *  low ram will be on Node1
    1938                 :            :          */
    1939                 :          3 :         for_each_free_mem_range(i, NUMA_NO_NODE, MEMBLOCK_NONE, &start, &end,
    1940                 :            :                                 NULL)
    1941                 :          3 :                 count += __free_memory_core(start, end);
    1942                 :            : 
    1943                 :          3 :         return count;
    1944                 :            : }
    1945                 :            : 
    1946                 :            : static int reset_managed_pages_done __initdata;
    1947                 :            : 
    1948                 :          0 : void reset_node_managed_pages(pg_data_t *pgdat)
    1949                 :            : {
    1950                 :            :         struct zone *z;
    1951                 :            : 
    1952                 :          3 :         for (z = pgdat->node_zones; z < pgdat->node_zones + MAX_NR_ZONES; z++)
    1953                 :            :                 atomic_long_set(&z->managed_pages, 0);
    1954                 :          0 : }
    1955                 :            : 
    1956                 :          3 : void __init reset_all_zones_managed_pages(void)
    1957                 :            : {
    1958                 :            :         struct pglist_data *pgdat;
    1959                 :            : 
    1960                 :          3 :         if (reset_managed_pages_done)
    1961                 :          3 :                 return;
    1962                 :            : 
    1963                 :          3 :         for_each_online_pgdat(pgdat)
    1964                 :            :                 reset_node_managed_pages(pgdat);
    1965                 :            : 
    1966                 :          3 :         reset_managed_pages_done = 1;
    1967                 :            : }
    1968                 :            : 
    1969                 :            : /**
    1970                 :            :  * memblock_free_all - release free pages to the buddy allocator
    1971                 :            :  *
    1972                 :            :  * Return: the number of pages actually released.
    1973                 :            :  */
    1974                 :          3 : unsigned long __init memblock_free_all(void)
    1975                 :            : {
    1976                 :            :         unsigned long pages;
    1977                 :            : 
    1978                 :          3 :         reset_all_zones_managed_pages();
    1979                 :            : 
    1980                 :          3 :         pages = free_low_memory_core_early();
    1981                 :          3 :         totalram_pages_add(pages);
    1982                 :            : 
    1983                 :          3 :         return pages;
    1984                 :            : }
    1985                 :            : 
    1986                 :            : #if defined(CONFIG_DEBUG_FS) && defined(CONFIG_ARCH_KEEP_MEMBLOCK)
    1987                 :            : 
    1988                 :          0 : static int memblock_debug_show(struct seq_file *m, void *private)
    1989                 :            : {
    1990                 :          0 :         struct memblock_type *type = m->private;
    1991                 :            :         struct memblock_region *reg;
    1992                 :            :         int i;
    1993                 :            :         phys_addr_t end;
    1994                 :            : 
    1995                 :          0 :         for (i = 0; i < type->cnt; i++) {
    1996                 :          0 :                 reg = &type->regions[i];
    1997                 :          0 :                 end = reg->base + reg->size - 1;
    1998                 :            : 
    1999                 :          0 :                 seq_printf(m, "%4d: ", i);
    2000                 :          0 :                 seq_printf(m, "%pa..%pa\n", &reg->base, &end);
    2001                 :            :         }
    2002                 :          0 :         return 0;
    2003                 :            : }
    2004                 :          0 : DEFINE_SHOW_ATTRIBUTE(memblock_debug);
    2005                 :            : 
    2006                 :          3 : static int __init memblock_init_debugfs(void)
    2007                 :            : {
    2008                 :          3 :         struct dentry *root = debugfs_create_dir("memblock", NULL);
    2009                 :            : 
    2010                 :          3 :         debugfs_create_file("memory", 0444, root,
    2011                 :            :                             &memblock.memory, &memblock_debug_fops);
    2012                 :          3 :         debugfs_create_file("reserved", 0444, root,
    2013                 :            :                             &memblock.reserved, &memblock_debug_fops);
    2014                 :            : #ifdef CONFIG_HAVE_MEMBLOCK_PHYS_MAP
    2015                 :            :         debugfs_create_file("physmem", 0444, root,
    2016                 :            :                             &memblock.physmem, &memblock_debug_fops);
    2017                 :            : #endif
    2018                 :            : 
    2019                 :          3 :         return 0;
    2020                 :            : }
    2021                 :            : __initcall(memblock_init_debugfs);
    2022                 :            : 
    2023                 :            : #endif /* CONFIG_DEBUG_FS */
    

Generated by: LCOV version 1.14