Merge branch 'next' of git://git.kernel.org/pub/scm/virt/kvm/kvm

[karo-tx-linux.git] / arch / x86 / kernel / vsmp_64.c

diff --git a/arch/x86/kernel/vsmp_64.c b/arch/x86/kernel/vsmp_64.c
index a1d804bcd48357554cc017f4b63c217eb2b84eaa..8eeb55a551b423fdce43eb432c8f8f5599246848 100644 (file)
--- a/arch/x86/kernel/vsmp_64.c
+++ b/arch/x86/kernel/vsmp_64.c
@@ -15,6 +15,7 @@
 #include <linux/init.h>
 #include <linux/pci_ids.h>
 #include <linux/pci_regs.h>
+#include <linux/smp.h>
 
 #include <asm/apic.h>
 #include <asm/pci-direct.h>
@@ -22,6 +23,8 @@
 #include <asm/paravirt.h>
 #include <asm/setup.h>
 
+#define TOPOLOGY_REGISTER_OFFSET 0x10
+
 #if defined CONFIG_PCI && defined CONFIG_PARAVIRT
 /*
  * Interrupt control on vSMPowered systems:
@@ -149,12 +152,49 @@ int is_vsmp_box(void)
        return 0;
 }
 #endif
+
+static void __init vsmp_cap_cpus(void)
+{
+#if !defined(CONFIG_X86_VSMP) && defined(CONFIG_SMP)
+       void __iomem *address;
+       unsigned int cfg, topology, node_shift, maxcpus;
+
+       /*
+        * CONFIG_X86_VSMP is not configured, so limit the number CPUs to the
+        * ones present in the first board, unless explicitly overridden by
+        * setup_max_cpus
+        */
+       if (setup_max_cpus != NR_CPUS)
+               return;
+
+       /* Read the vSMP Foundation topology register */
+       cfg = read_pci_config(0, 0x1f, 0, PCI_BASE_ADDRESS_0);
+       address = early_ioremap(cfg + TOPOLOGY_REGISTER_OFFSET, 4);
+       if (WARN_ON(!address))
+               return;
+
+       topology = readl(address);
+       node_shift = (topology >> 16) & 0x7;
+       if (!node_shift)
+               /* The value 0 should be decoded as 8 */
+               node_shift = 8;
+       maxcpus = (topology & ((1 << node_shift) - 1)) + 1;
+
+       pr_info("vSMP CTL: Capping CPUs to %d (CONFIG_X86_VSMP is unset)\n",
+               maxcpus);
+       setup_max_cpus = maxcpus;
+       early_iounmap(address, 4);
+#endif
+}
+
 void __init vsmp_init(void)
 {
        detect_vsmp_box();
        if (!is_vsmp_box())
                return;
 
+       vsmp_cap_cpus();
+
        set_vsmp_pv_ops();
        return;
 }