Pileus Git - ~andy/linux/blob - arch/x86/mm/numa_64.c

   1 /*
   2  * Generic VM initialization for x86-64 NUMA setups.
   3  * Copyright 2002,2003 Andi Kleen, SuSE Labs.
   4  */
   5 #include <linux/kernel.h>
   6 #include <linux/mm.h>
   7 #include <linux/string.h>
   8 #include <linux/init.h>
   9 #include <linux/bootmem.h>
  10 #include <linux/memblock.h>
  11 #include <linux/mmzone.h>
  12 #include <linux/ctype.h>
  13 #include <linux/module.h>
  14 #include <linux/nodemask.h>
  15 #include <linux/sched.h>
  16 #include <linux/acpi.h>
  17
  18 #include <asm/e820.h>
  19 #include <asm/proto.h>
  20 #include <asm/dma.h>
  21 #include <asm/numa.h>
  22 #include <asm/acpi.h>
  23 #include <asm/amd_nb.h>
  24
  25 struct numa_memblk {
  26         u64                     start;
  27         u64                     end;
  28         int                     nid;
  29 };
  30
  31 struct numa_meminfo {
  32         int                     nr_blks;
  33         struct numa_memblk      blk[NR_NODE_MEMBLKS];
  34 };
  35
  36 struct pglist_data *node_data[MAX_NUMNODES] __read_mostly;
  37 EXPORT_SYMBOL(node_data);
  38
  39 nodemask_t numa_nodes_parsed __initdata;
  40
  41 struct memnode memnode;
  42
  43 static unsigned long __initdata nodemap_addr;
  44 static unsigned long __initdata nodemap_size;
  45
  46 static struct numa_meminfo numa_meminfo __initdata;
  47
  48 static int numa_distance_cnt;
  49 static u8 *numa_distance;
  50
  51 /*
  52  * Given a shift value, try to populate memnodemap[]
  53  * Returns :
  54  * 1 if OK
  55  * 0 if memnodmap[] too small (of shift too small)
  56  * -1 if node overlap or lost ram (shift too big)
  57  */
  58 static int __init populate_memnodemap(const struct numa_meminfo *mi, int shift)
  59 {
  60         unsigned long addr, end;
  61         int i, res = -1;
  62
  63         memset(memnodemap, 0xff, sizeof(s16)*memnodemapsize);
  64         for (i = 0; i < mi->nr_blks; i++) {
  65                 addr = mi->blk[i].start;
  66                 end = mi->blk[i].end;
  67                 if (addr >= end)
  68                         continue;
  69                 if ((end >> shift) >= memnodemapsize)
  70                         return 0;
  71                 do {
  72                         if (memnodemap[addr >> shift] != NUMA_NO_NODE)
  73                                 return -1;
  74                         memnodemap[addr >> shift] = mi->blk[i].nid;
  75                         addr += (1UL << shift);
  76                 } while (addr < end);
  77                 res = 1;
  78         }
  79         return res;
  80 }
  81
  82 static int __init allocate_cachealigned_memnodemap(void)
  83 {
  84         unsigned long addr;
  85
  86         memnodemap = memnode.embedded_map;
  87         if (memnodemapsize <= ARRAY_SIZE(memnode.embedded_map))
  88                 return 0;
  89
  90         addr = 0x8000;
  91         nodemap_size = roundup(sizeof(s16) * memnodemapsize, L1_CACHE_BYTES);
  92         nodemap_addr = memblock_find_in_range(addr, get_max_mapped(),
  93                                       nodemap_size, L1_CACHE_BYTES);
  94         if (nodemap_addr == MEMBLOCK_ERROR) {
  95                 printk(KERN_ERR
  96                        "NUMA: Unable to allocate Memory to Node hash map\n");
  97                 nodemap_addr = nodemap_size = 0;
  98                 return -1;
  99         }
 100         memnodemap = phys_to_virt(nodemap_addr);
 101         memblock_x86_reserve_range(nodemap_addr, nodemap_addr + nodemap_size, "MEMNODEMAP");
 102
 103         printk(KERN_DEBUG "NUMA: Allocated memnodemap from %lx - %lx\n",
 104                nodemap_addr, nodemap_addr + nodemap_size);
 105         return 0;
 106 }
 107
 108 /*
 109  * The LSB of all start and end addresses in the node map is the value of the
 110  * maximum possible shift.
 111  */
 112 static int __init extract_lsb_from_nodes(const struct numa_meminfo *mi)
 113 {
 114         int i, nodes_used = 0;
 115         unsigned long start, end;
 116         unsigned long bitfield = 0, memtop = 0;
 117
 118         for (i = 0; i < mi->nr_blks; i++) {
 119                 start = mi->blk[i].start;
 120                 end = mi->blk[i].end;
 121                 if (start >= end)
 122                         continue;
 123                 bitfield |= start;
 124                 nodes_used++;
 125                 if (end > memtop)
 126                         memtop = end;
 127         }
 128         if (nodes_used <= 1)
 129                 i = 63;
 130         else
 131                 i = find_first_bit(&bitfield, sizeof(unsigned long)*8);
 132         memnodemapsize = (memtop >> i)+1;
 133         return i;
 134 }
 135
 136 static int __init compute_hash_shift(const struct numa_meminfo *mi)
 137 {
 138         int shift;
 139
 140         shift = extract_lsb_from_nodes(mi);
 141         if (allocate_cachealigned_memnodemap())
 142                 return -1;
 143         printk(KERN_DEBUG "NUMA: Using %d for the hash shift.\n",
 144                 shift);
 145
 146         if (populate_memnodemap(mi, shift) != 1) {
 147                 printk(KERN_INFO "Your memory is not aligned you need to "
 148                        "rebuild your kernel with a bigger NODEMAPSIZE "
 149                        "shift=%d\n", shift);
 150                 return -1;
 151         }
 152         return shift;
 153 }
 154
 155 int __meminit  __early_pfn_to_nid(unsigned long pfn)
 156 {
 157         return phys_to_nid(pfn << PAGE_SHIFT);
 158 }
 159
 160 static void * __init early_node_mem(int nodeid, unsigned long start,
 161                                     unsigned long end, unsigned long size,
 162                                     unsigned long align)
 163 {
 164         unsigned long mem;
 165
 166         /*
 167          * put it on high as possible
 168          * something will go with NODE_DATA
 169          */
 170         if (start < (MAX_DMA_PFN<<PAGE_SHIFT))
 171                 start = MAX_DMA_PFN<<PAGE_SHIFT;
 172         if (start < (MAX_DMA32_PFN<<PAGE_SHIFT) &&
 173             end > (MAX_DMA32_PFN<<PAGE_SHIFT))
 174                 start = MAX_DMA32_PFN<<PAGE_SHIFT;
 175         mem = memblock_x86_find_in_range_node(nodeid, start, end, size, align);
 176         if (mem != MEMBLOCK_ERROR)
 177                 return __va(mem);
 178
 179         /* extend the search scope */
 180         end = max_pfn_mapped << PAGE_SHIFT;
 181         start = MAX_DMA_PFN << PAGE_SHIFT;
 182         mem = memblock_find_in_range(start, end, size, align);
 183         if (mem != MEMBLOCK_ERROR)
 184                 return __va(mem);
 185
 186         printk(KERN_ERR "Cannot find %lu bytes in node %d\n",
 187                        size, nodeid);
 188
 189         return NULL;
 190 }
 191
 192 static int __init numa_add_memblk_to(int nid, u64 start, u64 end,
 193                                      struct numa_meminfo *mi)
 194 {
 195         /* ignore zero length blks */
 196         if (start == end)
 197                 return 0;
 198
 199         /* whine about and ignore invalid blks */
 200         if (start > end || nid < 0 || nid >= MAX_NUMNODES) {
 201                 pr_warning("NUMA: Warning: invalid memblk node %d (%Lx-%Lx)\n",
 202                            nid, start, end);
 203                 return 0;
 204         }
 205
 206         if (mi->nr_blks >= NR_NODE_MEMBLKS) {
 207                 pr_err("NUMA: too many memblk ranges\n");
 208                 return -EINVAL;
 209         }
 210
 211         mi->blk[mi->nr_blks].start = start;
 212         mi->blk[mi->nr_blks].end = end;
 213         mi->blk[mi->nr_blks].nid = nid;
 214         mi->nr_blks++;
 215         return 0;
 216 }
 217
 218 static void __init numa_remove_memblk_from(int idx, struct numa_meminfo *mi)
 219 {
 220         mi->nr_blks--;
 221         memmove(&mi->blk[idx], &mi->blk[idx + 1],
 222                 (mi->nr_blks - idx) * sizeof(mi->blk[0]));
 223 }
 224
 225 int __init numa_add_memblk(int nid, u64 start, u64 end)
 226 {
 227         return numa_add_memblk_to(nid, start, end, &numa_meminfo);
 228 }
 229
 230 /* Initialize bootmem allocator for a node */
 231 void __init
 232 setup_node_bootmem(int nodeid, unsigned long start, unsigned long end)
 233 {
 234         unsigned long start_pfn, last_pfn, nodedata_phys;
 235         const int pgdat_size = roundup(sizeof(pg_data_t), PAGE_SIZE);
 236         int nid;
 237
 238         if (!end)
 239                 return;
 240
 241         /*
 242          * Don't confuse VM with a node that doesn't have the
 243          * minimum amount of memory:
 244          */
 245         if (end && (end - start) < NODE_MIN_SIZE)
 246                 return;
 247
 248         start = roundup(start, ZONE_ALIGN);
 249
 250         printk(KERN_INFO "Initmem setup node %d %016lx-%016lx\n", nodeid,
 251                start, end);
 252
 253         start_pfn = start >> PAGE_SHIFT;
 254         last_pfn = end >> PAGE_SHIFT;
 255
 256         node_data[nodeid] = early_node_mem(nodeid, start, end, pgdat_size,
 257                                            SMP_CACHE_BYTES);
 258         if (node_data[nodeid] == NULL)
 259                 return;
 260         nodedata_phys = __pa(node_data[nodeid]);
 261         memblock_x86_reserve_range(nodedata_phys, nodedata_phys + pgdat_size, "NODE_DATA");
 262         printk(KERN_INFO "  NODE_DATA [%016lx - %016lx]\n", nodedata_phys,
 263                 nodedata_phys + pgdat_size - 1);
 264         nid = phys_to_nid(nodedata_phys);
 265         if (nid != nodeid)
 266                 printk(KERN_INFO "    NODE_DATA(%d) on node %d\n", nodeid, nid);
 267
 268         memset(NODE_DATA(nodeid), 0, sizeof(pg_data_t));
 269         NODE_DATA(nodeid)->node_id = nodeid;
 270         NODE_DATA(nodeid)->node_start_pfn = start_pfn;
 271         NODE_DATA(nodeid)->node_spanned_pages = last_pfn - start_pfn;
 272
 273         node_set_online(nodeid);
 274 }
 275
 276 static int __init numa_cleanup_meminfo(struct numa_meminfo *mi)
 277 {
 278         const u64 low = 0;
 279         const u64 high = (u64)max_pfn << PAGE_SHIFT;
 280         int i, j, k;
 281
 282         for (i = 0; i < mi->nr_blks; i++) {
 283                 struct numa_memblk *bi = &mi->blk[i];
 284
 285                 /* make sure all blocks are inside the limits */
 286                 bi->start = max(bi->start, low);
 287                 bi->end = min(bi->end, high);
 288
 289                 /* and there's no empty block */
 290                 if (bi->start == bi->end) {
 291                         numa_remove_memblk_from(i--, mi);
 292                         continue;
 293                 }
 294
 295                 for (j = i + 1; j < mi->nr_blks; j++) {
 296                         struct numa_memblk *bj = &mi->blk[j];
 297                         unsigned long start, end;
 298
 299                         /*
 300                          * See whether there are overlapping blocks.  Whine
 301                          * about but allow overlaps of the same nid.  They
 302                          * will be merged below.
 303                          */
 304                         if (bi->end > bj->start && bi->start < bj->end) {
 305                                 if (bi->nid != bj->nid) {
 306                                         pr_err("NUMA: node %d (%Lx-%Lx) overlaps with node %d (%Lx-%Lx)\n",
 307                                                bi->nid, bi->start, bi->end,
 308                                                bj->nid, bj->start, bj->end);
 309                                         return -EINVAL;
 310                                 }
 311                                 pr_warning("NUMA: Warning: node %d (%Lx-%Lx) overlaps with itself (%Lx-%Lx)\n",
 312                                            bi->nid, bi->start, bi->end,
 313                                            bj->start, bj->end);
 314                         }
 315
 316                         /*
 317                          * Join together blocks on the same node, holes
 318                          * between which don't overlap with memory on other
 319                          * nodes.
 320                          */
 321                         if (bi->nid != bj->nid)
 322                                 continue;
 323                         start = max(min(bi->start, bj->start), low);
 324                         end = min(max(bi->end, bj->end), high);
 325                         for (k = 0; k < mi->nr_blks; k++) {
 326                                 struct numa_memblk *bk = &mi->blk[k];
 327
 328                                 if (bi->nid == bk->nid)
 329                                         continue;
 330                                 if (start < bk->end && end > bk->start)
 331                                         break;
 332                         }
 333                         if (k < mi->nr_blks)
 334                                 continue;
 335                         printk(KERN_INFO "NUMA: Node %d [%Lx,%Lx) + [%Lx,%Lx) -> [%lx,%lx)\n",
 336                                bi->nid, bi->start, bi->end, bj->start, bj->end,
 337                                start, end);
 338                         bi->start = start;
 339                         bi->end = end;
 340                         numa_remove_memblk_from(j--, mi);
 341                 }
 342         }
 343
 344         for (i = mi->nr_blks; i < ARRAY_SIZE(mi->blk); i++) {
 345                 mi->blk[i].start = mi->blk[i].end = 0;
 346                 mi->blk[i].nid = NUMA_NO_NODE;
 347         }
 348
 349         return 0;
 350 }
 351
 352 /*
 353  * Set nodes, which have memory in @mi, in *@nodemask.
 354  */
 355 static void __init numa_nodemask_from_meminfo(nodemask_t *nodemask,
 356                                               const struct numa_meminfo *mi)
 357 {
 358         int i;
 359
 360         for (i = 0; i < ARRAY_SIZE(mi->blk); i++)
 361                 if (mi->blk[i].start != mi->blk[i].end &&
 362                     mi->blk[i].nid != NUMA_NO_NODE)
 363                         node_set(mi->blk[i].nid, *nodemask);
 364 }
 365
 366 /*
 367  * Reset distance table.  The current table is freed.  The next
 368  * numa_set_distance() call will create a new one.
 369  */
 370 static void __init numa_reset_distance(void)
 371 {
 372         size_t size;
 373
 374         if (numa_distance_cnt) {
 375                 size = numa_distance_cnt * sizeof(numa_distance[0]);
 376                 memblock_x86_free_range(__pa(numa_distance),
 377                                         __pa(numa_distance) + size);
 378                 numa_distance_cnt = 0;
 379         }
 380         numa_distance = NULL;
 381 }
 382
 383 /*
 384  * Set the distance between node @from to @to to @distance.  If distance
 385  * table doesn't exist, one which is large enough to accomodate all the
 386  * currently known nodes will be created.
 387  */
 388 void __init numa_set_distance(int from, int to, int distance)
 389 {
 390         if (!numa_distance) {
 391                 nodemask_t nodes_parsed;
 392                 size_t size;
 393                 int i, j, cnt = 0;
 394                 u64 phys;
 395
 396                 /* size the new table and allocate it */
 397                 nodes_parsed = numa_nodes_parsed;
 398                 numa_nodemask_from_meminfo(&nodes_parsed, &numa_meminfo);
 399
 400                 for_each_node_mask(i, nodes_parsed)
 401                         cnt = i;
 402                 size = ++cnt * sizeof(numa_distance[0]);
 403
 404                 phys = memblock_find_in_range(0,
 405                                               (u64)max_pfn_mapped << PAGE_SHIFT,
 406                                               size, PAGE_SIZE);
 407                 if (phys == MEMBLOCK_ERROR) {
 408                         pr_warning("NUMA: Warning: can't allocate distance table!\n");
 409                         /* don't retry until explicitly reset */
 410                         numa_distance = (void *)1LU;
 411                         return;
 412                 }
 413                 memblock_x86_reserve_range(phys, phys + size, "NUMA DIST");
 414
 415                 numa_distance = __va(phys);
 416                 numa_distance_cnt = cnt;
 417
 418                 /* fill with the default distances */
 419                 for (i = 0; i < cnt; i++)
 420                         for (j = 0; j < cnt; j++)
 421                                 numa_distance[i * cnt + j] = i == j ?
 422                                         LOCAL_DISTANCE : REMOTE_DISTANCE;
 423                 printk(KERN_DEBUG "NUMA: Initialized distance table, cnt=%d\n", cnt);
 424         }
 425
 426         if (from >= numa_distance_cnt || to >= numa_distance_cnt) {
 427                 printk_once(KERN_DEBUG "NUMA: Debug: distance out of bound, from=%d to=%d distance=%d\n",
 428                             from, to, distance);
 429                 return;
 430         }
 431
 432         if ((u8)distance != distance ||
 433             (from == to && distance != LOCAL_DISTANCE)) {
 434                 pr_warn_once("NUMA: Warning: invalid distance parameter, from=%d to=%d distance=%d\n",
 435                              from, to, distance);
 436                 return;
 437         }
 438
 439         numa_distance[from * numa_distance_cnt + to] = distance;
 440 }
 441
 442 int __node_distance(int from, int to)
 443 {
 444         if (from >= numa_distance_cnt || to >= numa_distance_cnt)
 445                 return from == to ? LOCAL_DISTANCE : REMOTE_DISTANCE;
 446         return numa_distance[from * numa_distance_cnt + to];
 447 }
 448 EXPORT_SYMBOL(__node_distance);
 449
 450 /*
 451  * Sanity check to catch more bad NUMA configurations (they are amazingly
 452  * common).  Make sure the nodes cover all memory.
 453  */
 454 static bool __init numa_meminfo_cover_memory(const struct numa_meminfo *mi)
 455 {
 456         unsigned long numaram, e820ram;
 457         int i;
 458
 459         numaram = 0;
 460         for (i = 0; i < mi->nr_blks; i++) {
 461                 unsigned long s = mi->blk[i].start >> PAGE_SHIFT;
 462                 unsigned long e = mi->blk[i].end >> PAGE_SHIFT;
 463                 numaram += e - s;
 464                 numaram -= __absent_pages_in_range(mi->blk[i].nid, s, e);
 465                 if ((long)numaram < 0)
 466                         numaram = 0;
 467         }
 468
 469         e820ram = max_pfn - (memblock_x86_hole_size(0,
 470                                         max_pfn << PAGE_SHIFT) >> PAGE_SHIFT);
 471         /* We seem to lose 3 pages somewhere. Allow 1M of slack. */
 472         if ((long)(e820ram - numaram) >= (1 << (20 - PAGE_SHIFT))) {
 473                 printk(KERN_ERR "NUMA: nodes only cover %luMB of your %luMB e820 RAM. Not used.\n",
 474                        (numaram << PAGE_SHIFT) >> 20,
 475                        (e820ram << PAGE_SHIFT) >> 20);
 476                 return false;
 477         }
 478         return true;
 479 }
 480
 481 static int __init numa_register_memblks(struct numa_meminfo *mi)
 482 {
 483         int i, nid;
 484
 485         /* Account for nodes with cpus and no memory */
 486         node_possible_map = numa_nodes_parsed;
 487         numa_nodemask_from_meminfo(&node_possible_map, mi);
 488         if (WARN_ON(nodes_empty(node_possible_map)))
 489                 return -EINVAL;
 490
 491         memnode_shift = compute_hash_shift(mi);
 492         if (memnode_shift < 0) {
 493                 printk(KERN_ERR "NUMA: No NUMA node hash function found. Contact maintainer\n");
 494                 return -EINVAL;
 495         }
 496
 497         for (i = 0; i < mi->nr_blks; i++)
 498                 memblock_x86_register_active_regions(mi->blk[i].nid,
 499                                         mi->blk[i].start >> PAGE_SHIFT,
 500                                         mi->blk[i].end >> PAGE_SHIFT);
 501
 502         /* for out of order entries */
 503         sort_node_map();
 504         if (!numa_meminfo_cover_memory(mi))
 505                 return -EINVAL;
 506
 507         init_memory_mapping_high();
 508
 509         /* Finally register nodes. */
 510         for_each_node_mask(nid, node_possible_map) {
 511                 u64 start = (u64)max_pfn << PAGE_SHIFT;
 512                 u64 end = 0;
 513
 514                 for (i = 0; i < mi->nr_blks; i++) {
 515                         if (nid != mi->blk[i].nid)
 516                                 continue;
 517                         start = min(mi->blk[i].start, start);
 518                         end = max(mi->blk[i].end, end);
 519                 }
 520
 521                 if (start < end)
 522                         setup_node_bootmem(nid, start, end);
 523         }
 524
 525         return 0;
 526 }
 527
 528 #ifdef CONFIG_NUMA_EMU
 529 /* Numa emulation */
 530 static int emu_nid_to_phys[MAX_NUMNODES] __cpuinitdata;
 531 static char *emu_cmdline __initdata;
 532
 533 void __init numa_emu_cmdline(char *str)
 534 {
 535         emu_cmdline = str;
 536 }
 537
 538 static int __init emu_find_memblk_by_nid(int nid, const struct numa_meminfo *mi)
 539 {
 540         int i;
 541
 542         for (i = 0; i < mi->nr_blks; i++)
 543                 if (mi->blk[i].nid == nid)
 544                         return i;
 545         return -ENOENT;
 546 }
 547
 548 /*
 549  * Sets up nid to range from @start to @end.  The return value is -errno if
 550  * something went wrong, 0 otherwise.
 551  */
 552 static int __init emu_setup_memblk(struct numa_meminfo *ei,
 553                                    struct numa_meminfo *pi,
 554                                    int nid, int phys_blk, u64 size)
 555 {
 556         struct numa_memblk *eb = &ei->blk[ei->nr_blks];
 557         struct numa_memblk *pb = &pi->blk[phys_blk];
 558
 559         if (ei->nr_blks >= NR_NODE_MEMBLKS) {
 560                 pr_err("NUMA: Too many emulated memblks, failing emulation\n");
 561                 return -EINVAL;
 562         }
 563
 564         ei->nr_blks++;
 565         eb->start = pb->start;
 566         eb->end = pb->start + size;
 567         eb->nid = nid;
 568
 569         if (emu_nid_to_phys[nid] == NUMA_NO_NODE)
 570                 emu_nid_to_phys[nid] = pb->nid;
 571
 572         pb->start += size;
 573         if (pb->start >= pb->end) {
 574                 WARN_ON_ONCE(pb->start > pb->end);
 575                 numa_remove_memblk_from(phys_blk, pi);
 576         }
 577
 578         printk(KERN_INFO "Faking node %d at %016Lx-%016Lx (%LuMB)\n", nid,
 579                eb->start, eb->end, (eb->end - eb->start) >> 20);
 580         return 0;
 581 }
 582
 583 /*
 584  * Sets up nr_nodes fake nodes interleaved over physical nodes ranging from addr
 585  * to max_addr.  The return value is the number of nodes allocated.
 586  */
 587 static int __init split_nodes_interleave(struct numa_meminfo *ei,
 588                                          struct numa_meminfo *pi,
 589                                          u64 addr, u64 max_addr, int nr_nodes)
 590 {
 591         nodemask_t physnode_mask = NODE_MASK_NONE;
 592         u64 size;
 593         int big;
 594         int nid = 0;
 595         int i, ret;
 596
 597         if (nr_nodes <= 0)
 598                 return -1;
 599         if (nr_nodes > MAX_NUMNODES) {
 600                 pr_info("numa=fake=%d too large, reducing to %d\n",
 601                         nr_nodes, MAX_NUMNODES);
 602                 nr_nodes = MAX_NUMNODES;
 603         }
 604
 605         size = (max_addr - addr - memblock_x86_hole_size(addr, max_addr)) / nr_nodes;
 606         /*
 607          * Calculate the number of big nodes that can be allocated as a result
 608          * of consolidating the remainder.
 609          */
 610         big = ((size & ~FAKE_NODE_MIN_HASH_MASK) * nr_nodes) /
 611                 FAKE_NODE_MIN_SIZE;
 612
 613         size &= FAKE_NODE_MIN_HASH_MASK;
 614         if (!size) {
 615                 pr_err("Not enough memory for each node.  "
 616                         "NUMA emulation disabled.\n");
 617                 return -1;
 618         }
 619
 620         for (i = 0; i < pi->nr_blks; i++)
 621                 node_set(pi->blk[i].nid, physnode_mask);
 622
 623         /*
 624          * Continue to fill physical nodes with fake nodes until there is no
 625          * memory left on any of them.
 626          */
 627         while (nodes_weight(physnode_mask)) {
 628                 for_each_node_mask(i, physnode_mask) {
 629                         u64 dma32_end = PFN_PHYS(MAX_DMA32_PFN);
 630                         u64 start, limit, end;
 631                         int phys_blk;
 632
 633                         phys_blk = emu_find_memblk_by_nid(i, pi);
 634                         if (phys_blk < 0) {
 635                                 node_clear(i, physnode_mask);
 636                                 continue;
 637                         }
 638                         start = pi->blk[phys_blk].start;
 639                         limit = pi->blk[phys_blk].end;
 640                         end = start + size;
 641
 642                         if (nid < big)
 643                                 end += FAKE_NODE_MIN_SIZE;
 644
 645                         /*
 646                          * Continue to add memory to this fake node if its
 647                          * non-reserved memory is less than the per-node size.
 648                          */
 649                         while (end - start -
 650                                memblock_x86_hole_size(start, end) < size) {
 651                                 end += FAKE_NODE_MIN_SIZE;
 652                                 if (end > limit) {
 653                                         end = limit;
 654                                         break;
 655                                 }
 656                         }
 657
 658                         /*
 659                          * If there won't be at least FAKE_NODE_MIN_SIZE of
 660                          * non-reserved memory in ZONE_DMA32 for the next node,
 661                          * this one must extend to the boundary.
 662                          */
 663                         if (end < dma32_end && dma32_end - end -
 664                             memblock_x86_hole_size(end, dma32_end) < FAKE_NODE_MIN_SIZE)
 665                                 end = dma32_end;
 666
 667                         /*
 668                          * If there won't be enough non-reserved memory for the
 669                          * next node, this one must extend to the end of the
 670                          * physical node.
 671                          */
 672                         if (limit - end -
 673                             memblock_x86_hole_size(end, limit) < size)
 674                                 end = limit;
 675
 676                         ret = emu_setup_memblk(ei, pi, nid++ % nr_nodes,
 677                                                phys_blk,
 678                                                min(end, limit) - start);
 679                         if (ret < 0)
 680                                 return ret;
 681                 }
 682         }
 683         return 0;
 684 }
 685
 686 /*
 687  * Returns the end address of a node so that there is at least `size' amount of
 688  * non-reserved memory or `max_addr' is reached.
 689  */
 690 static u64 __init find_end_of_node(u64 start, u64 max_addr, u64 size)
 691 {
 692         u64 end = start + size;
 693
 694         while (end - start - memblock_x86_hole_size(start, end) < size) {
 695                 end += FAKE_NODE_MIN_SIZE;
 696                 if (end > max_addr) {
 697                         end = max_addr;
 698                         break;
 699                 }
 700         }
 701         return end;
 702 }
 703
 704 /*
 705  * Sets up fake nodes of `size' interleaved over physical nodes ranging from
 706  * `addr' to `max_addr'.  The return value is the number of nodes allocated.
 707  */
 708 static int __init split_nodes_size_interleave(struct numa_meminfo *ei,
 709                                               struct numa_meminfo *pi,
 710                                               u64 addr, u64 max_addr, u64 size)
 711 {
 712         nodemask_t physnode_mask = NODE_MASK_NONE;
 713         u64 min_size;
 714         int nid = 0;
 715         int i, ret;
 716
 717         if (!size)
 718                 return -1;
 719         /*
 720          * The limit on emulated nodes is MAX_NUMNODES, so the size per node is
 721          * increased accordingly if the requested size is too small.  This
 722          * creates a uniform distribution of node sizes across the entire
 723          * machine (but not necessarily over physical nodes).
 724          */
 725         min_size = (max_addr - addr - memblock_x86_hole_size(addr, max_addr)) /
 726                                                 MAX_NUMNODES;
 727         min_size = max(min_size, FAKE_NODE_MIN_SIZE);
 728         if ((min_size & FAKE_NODE_MIN_HASH_MASK) < min_size)
 729                 min_size = (min_size + FAKE_NODE_MIN_SIZE) &
 730                                                 FAKE_NODE_MIN_HASH_MASK;
 731         if (size < min_size) {
 732                 pr_err("Fake node size %LuMB too small, increasing to %LuMB\n",
 733                         size >> 20, min_size >> 20);
 734                 size = min_size;
 735         }
 736         size &= FAKE_NODE_MIN_HASH_MASK;
 737
 738         for (i = 0; i < pi->nr_blks; i++)
 739                 node_set(pi->blk[i].nid, physnode_mask);
 740
 741         /*
 742          * Fill physical nodes with fake nodes of size until there is no memory
 743          * left on any of them.
 744          */
 745         while (nodes_weight(physnode_mask)) {
 746                 for_each_node_mask(i, physnode_mask) {
 747                         u64 dma32_end = MAX_DMA32_PFN << PAGE_SHIFT;
 748                         u64 start, limit, end;
 749                         int phys_blk;
 750
 751                         phys_blk = emu_find_memblk_by_nid(i, pi);
 752                         if (phys_blk < 0) {
 753                                 node_clear(i, physnode_mask);
 754                                 continue;
 755                         }
 756                         start = pi->blk[phys_blk].start;
 757                         limit = pi->blk[phys_blk].end;
 758
 759                         end = find_end_of_node(start, limit, size);
 760                         /*
 761                          * If there won't be at least FAKE_NODE_MIN_SIZE of
 762                          * non-reserved memory in ZONE_DMA32 for the next node,
 763                          * this one must extend to the boundary.
 764                          */
 765                         if (end < dma32_end && dma32_end - end -
 766                             memblock_x86_hole_size(end, dma32_end) < FAKE_NODE_MIN_SIZE)
 767                                 end = dma32_end;
 768
 769                         /*
 770                          * If there won't be enough non-reserved memory for the
 771                          * next node, this one must extend to the end of the
 772                          * physical node.
 773                          */
 774                         if (limit - end -
 775                             memblock_x86_hole_size(end, limit) < size)
 776                                 end = limit;
 777
 778                         ret = emu_setup_memblk(ei, pi, nid++ % MAX_NUMNODES,
 779                                                phys_blk,
 780                                                min(end, limit) - start);
 781                         if (ret < 0)
 782                                 return ret;
 783                 }
 784         }
 785         return 0;
 786 }
 787
 788 /*
 789  * Sets up the system RAM area from start_pfn to last_pfn according to the
 790  * numa=fake command-line option.
 791  */
 792 static void __init numa_emulation(struct numa_meminfo *numa_meminfo,
 793                                   int numa_dist_cnt)
 794 {
 795         static struct numa_meminfo ei __initdata;
 796         static struct numa_meminfo pi __initdata;
 797         const u64 max_addr = max_pfn << PAGE_SHIFT;
 798         u8 *phys_dist = NULL;
 799         int i, j, ret;
 800
 801         if (!emu_cmdline)
 802                 goto no_emu;
 803
 804         memset(&ei, 0, sizeof(ei));
 805         pi = *numa_meminfo;
 806
 807         for (i = 0; i < MAX_NUMNODES; i++)
 808                 emu_nid_to_phys[i] = NUMA_NO_NODE;
 809
 810         /*
 811          * If the numa=fake command-line contains a 'M' or 'G', it represents
 812          * the fixed node size.  Otherwise, if it is just a single number N,
 813          * split the system RAM into N fake nodes.
 814          */
 815         if (strchr(emu_cmdline, 'M') || strchr(emu_cmdline, 'G')) {
 816                 u64 size;
 817
 818                 size = memparse(emu_cmdline, &emu_cmdline);
 819                 ret = split_nodes_size_interleave(&ei, &pi, 0, max_addr, size);
 820         } else {
 821                 unsigned long n;
 822
 823                 n = simple_strtoul(emu_cmdline, NULL, 0);
 824                 ret = split_nodes_interleave(&ei, &pi, 0, max_addr, n);
 825         }
 826
 827         if (ret < 0)
 828                 goto no_emu;
 829
 830         if (numa_cleanup_meminfo(&ei) < 0) {
 831                 pr_warning("NUMA: Warning: constructed meminfo invalid, disabling emulation\n");
 832                 goto no_emu;
 833         }
 834
 835         /*
 836          * Copy the original distance table.  It's temporary so no need to
 837          * reserve it.
 838          */
 839         if (numa_dist_cnt) {
 840                 size_t size = numa_dist_cnt * sizeof(phys_dist[0]);
 841                 u64 phys;
 842
 843                 phys = memblock_find_in_range(0,
 844                                               (u64)max_pfn_mapped << PAGE_SHIFT,
 845                                               size, PAGE_SIZE);
 846                 if (phys == MEMBLOCK_ERROR) {
 847                         pr_warning("NUMA: Warning: can't allocate copy of distance table, disabling emulation\n");
 848                         goto no_emu;
 849                 }
 850                 phys_dist = __va(phys);
 851
 852                 for (i = 0; i < numa_dist_cnt; i++)
 853                         for (j = 0; j < numa_dist_cnt; j++)
 854                                 phys_dist[i * numa_dist_cnt + j] =
 855                                         node_distance(i, j);
 856         }
 857
 858         /* commit */
 859         *numa_meminfo = ei;
 860
 861         /*
 862          * Transform __apicid_to_node table to use emulated nids by
 863          * reverse-mapping phys_nid.  The maps should always exist but fall
 864          * back to zero just in case.
 865          */
 866         for (i = 0; i < ARRAY_SIZE(__apicid_to_node); i++) {
 867                 if (__apicid_to_node[i] == NUMA_NO_NODE)
 868                         continue;
 869                 for (j = 0; j < ARRAY_SIZE(emu_nid_to_phys); j++)
 870                         if (__apicid_to_node[i] == emu_nid_to_phys[j])
 871                                 break;
 872                 __apicid_to_node[i] = j < ARRAY_SIZE(emu_nid_to_phys) ? j : 0;
 873         }
 874
 875         /* make sure all emulated nodes are mapped to a physical node */
 876         for (i = 0; i < ARRAY_SIZE(emu_nid_to_phys); i++)
 877                 if (emu_nid_to_phys[i] == NUMA_NO_NODE)
 878                         emu_nid_to_phys[i] = 0;
 879
 880         /* transform distance table */
 881         numa_reset_distance();
 882         for (i = 0; i < MAX_NUMNODES; i++) {
 883                 for (j = 0; j < MAX_NUMNODES; j++) {
 884                         int physi = emu_nid_to_phys[i];
 885                         int physj = emu_nid_to_phys[j];
 886                         int dist;
 887
 888                         if (physi >= numa_dist_cnt || physj >= numa_dist_cnt)
 889                                 dist = physi == physj ?
 890                                         LOCAL_DISTANCE : REMOTE_DISTANCE;
 891                         else
 892                                 dist = phys_dist[physi * numa_dist_cnt + physj];
 893
 894                         numa_set_distance(i, j, dist);
 895                 }
 896         }
 897         return;
 898
 899 no_emu:
 900         /* No emulation.  Build identity emu_nid_to_phys[] for numa_add_cpu() */
 901         for (i = 0; i < ARRAY_SIZE(emu_nid_to_phys); i++)
 902                 emu_nid_to_phys[i] = i;
 903 }
 904 #else   /* CONFIG_NUMA_EMU */
 905 static inline void numa_emulation(struct numa_meminfo *numa_meminfo,
 906                                   int numa_dist_cnt)
 907 { }
 908 #endif  /* CONFIG_NUMA_EMU */
 909
 910 static int __init dummy_numa_init(void)
 911 {
 912         printk(KERN_INFO "%s\n",
 913                numa_off ? "NUMA turned off" : "No NUMA configuration found");
 914         printk(KERN_INFO "Faking a node at %016lx-%016lx\n",
 915                0LU, max_pfn << PAGE_SHIFT);
 916
 917         node_set(0, numa_nodes_parsed);
 918         numa_add_memblk(0, 0, (u64)max_pfn << PAGE_SHIFT);
 919
 920         return 0;
 921 }
 922
 923 void __init initmem_init(void)
 924 {
 925         int (*numa_init[])(void) = { [2] = dummy_numa_init };
 926         int i, j;
 927
 928         if (!numa_off) {
 929 #ifdef CONFIG_ACPI_NUMA
 930                 numa_init[0] = x86_acpi_numa_init;
 931 #endif
 932 #ifdef CONFIG_AMD_NUMA
 933                 numa_init[1] = amd_numa_init;
 934 #endif
 935         }
 936
 937         for (i = 0; i < ARRAY_SIZE(numa_init); i++) {
 938                 if (!numa_init[i])
 939                         continue;
 940
 941                 for (j = 0; j < MAX_LOCAL_APIC; j++)
 942                         set_apicid_to_node(j, NUMA_NO_NODE);
 943
 944                 nodes_clear(numa_nodes_parsed);
 945                 nodes_clear(node_possible_map);
 946                 nodes_clear(node_online_map);
 947                 memset(&numa_meminfo, 0, sizeof(numa_meminfo));
 948                 remove_all_active_ranges();
 949                 numa_reset_distance();
 950
 951                 if (numa_init[i]() < 0)
 952                         continue;
 953
 954                 if (numa_cleanup_meminfo(&numa_meminfo) < 0)
 955                         continue;
 956
 957                 numa_emulation(&numa_meminfo, numa_distance_cnt);
 958
 959                 if (numa_register_memblks(&numa_meminfo) < 0)
 960                         continue;
 961
 962                 for (j = 0; j < nr_cpu_ids; j++) {
 963                         int nid = early_cpu_to_node(j);
 964
 965                         if (nid == NUMA_NO_NODE)
 966                                 continue;
 967                         if (!node_online(nid))
 968                                 numa_clear_node(j);
 969                 }
 970                 numa_init_array();
 971                 return;
 972         }
 973         BUG();
 974 }
 975
 976 unsigned long __init numa_free_all_bootmem(void)
 977 {
 978         unsigned long pages = 0;
 979         int i;
 980
 981         for_each_online_node(i)
 982                 pages += free_all_bootmem_node(NODE_DATA(i));
 983
 984         pages += free_all_memory_core_early(MAX_NUMNODES);
 985
 986         return pages;
 987 }
 988
 989 int __cpuinit numa_cpu_node(int cpu)
 990 {
 991         int apicid = early_per_cpu(x86_cpu_to_apicid, cpu);
 992
 993         if (apicid != BAD_APICID)
 994                 return __apicid_to_node[apicid];
 995         return NUMA_NO_NODE;
 996 }
 997
 998 /*
 999  * UGLINESS AHEAD: Currently, CONFIG_NUMA_EMU is 64bit only and makes use
1000  * of 64bit specific data structures.  The distinction is artificial and
1001  * should be removed.  numa_{add|remove}_cpu() are implemented in numa.c
1002  * for both 32 and 64bit when CONFIG_NUMA_EMU is disabled but here when
1003  * enabled.
1004  *
1005  * NUMA emulation is planned to be made generic and the following and other
1006  * related code should be moved to numa.c.
1007  */
1008 #ifdef CONFIG_NUMA_EMU
1009 # ifndef CONFIG_DEBUG_PER_CPU_MAPS
1010 void __cpuinit numa_add_cpu(int cpu)
1011 {
1012         int physnid, nid;
1013
1014         nid = numa_cpu_node(cpu);
1015         if (nid == NUMA_NO_NODE)
1016                 nid = early_cpu_to_node(cpu);
1017         BUG_ON(nid == NUMA_NO_NODE || !node_online(nid));
1018
1019         physnid = emu_nid_to_phys[nid];
1020
1021         /*
1022          * Map the cpu to each emulated node that is allocated on the physical
1023          * node of the cpu's apic id.
1024          */
1025         for_each_online_node(nid)
1026                 if (emu_nid_to_phys[nid] == physnid)
1027                         cpumask_set_cpu(cpu, node_to_cpumask_map[nid]);
1028 }
1029
1030 void __cpuinit numa_remove_cpu(int cpu)
1031 {
1032         int i;
1033
1034         for_each_online_node(i)
1035                 cpumask_clear_cpu(cpu, node_to_cpumask_map[i]);
1036 }
1037 # else  /* !CONFIG_DEBUG_PER_CPU_MAPS */
1038 static void __cpuinit numa_set_cpumask(int cpu, int enable)
1039 {
1040         struct cpumask *mask;
1041         int nid, physnid, i;
1042
1043         nid = early_cpu_to_node(cpu);
1044         if (nid == NUMA_NO_NODE) {
1045                 /* early_cpu_to_node() already emits a warning and trace */
1046                 return;
1047         }
1048
1049         physnid = emu_nid_to_phys[nid];
1050
1051         for_each_online_node(i) {
1052                 if (emu_nid_to_phys[nid] != physnid)
1053                         continue;
1054
1055                 mask = debug_cpumask_set_cpu(cpu, enable);
1056                 if (!mask)
1057                         return;
1058
1059                 if (enable)
1060                         cpumask_set_cpu(cpu, mask);
1061                 else
1062                         cpumask_clear_cpu(cpu, mask);
1063         }
1064 }
1065
1066 void __cpuinit numa_add_cpu(int cpu)
1067 {
1068         numa_set_cpumask(cpu, 1);
1069 }
1070
1071 void __cpuinit numa_remove_cpu(int cpu)
1072 {
1073         numa_set_cpumask(cpu, 0);
1074 }
1075 # endif /* !CONFIG_DEBUG_PER_CPU_MAPS */
1076 #endif  /* CONFIG_NUMA_EMU */