src/smpdetect.c

   1 // CPU count detection
   2 //
   3 // Copyright (C) 2008  Kevin O'Connor <kevin@koconnor.net>
   4 // Copyright (C) 2006 Fabrice Bellard
   5 //
   6 // This file may be distributed under the terms of the GNU GPLv3 license.
   7
   8 #include "util.h" // dprintf
   9 #include "config.h" // CONFIG_*
  10 #include "cmos.h" // CMOS_BIOS_SMP_COUNT
  11
  12 #define CPUID_APIC (1 << 9)
  13
  14 #define APIC_ICR_LOW ((u8*)BUILD_APIC_ADDR + 0x300)
  15 #define APIC_SVR     ((u8*)BUILD_APIC_ADDR + 0x0F0)
  16
  17 #define APIC_ENABLED 0x0100
  18
  19 static inline void writel(void *addr, u32 val)
  20 {
  21     *(volatile u32 *)addr = val;
  22 }
  23
  24 static inline void writew(void *addr, u16 val)
  25 {
  26     *(volatile u16 *)addr = val;
  27 }
  28
  29 static inline void writeb(void *addr, u8 val)
  30 {
  31     *(volatile u8 *)addr = val;
  32 }
  33
  34 static inline u32 readl(const void *addr)
  35 {
  36     return *(volatile const u32 *)addr;
  37 }
  38
  39 static inline u16 readw(const void *addr)
  40 {
  41     return *(volatile const u16 *)addr;
  42 }
  43
  44 static inline u8 readb(const void *addr)
  45 {
  46     return *(volatile const u8 *)addr;
  47 }
  48
  49 extern void smp_ap_boot_code();
  50 extern u32 smp_cpus;
  51 #if MODE16
  52 u32 smp_cpus VISIBLE16;
  53 asm(
  54     "  .global smp_ap_boot_code\n"
  55     "smp_ap_boot_code:\n"
  56     // Increment the cpu counter
  57     "  movw $" __stringify(SEG_BIOS) ", %ax\n"
  58     "  movw %ax, %ds\n"
  59     "  lock incl smp_cpus\n"
  60     // Halt the processor.
  61     "  jmp permanent_halt\n"
  62     );
  63 #endif
  64
  65 /* find the number of CPUs by launching a SIPI to them */
  66 int
  67 smp_probe(void)
  68 {
  69     if (smp_cpus)
  70         return smp_cpus;
  71
  72     u32 eax, ebx, ecx, cpuid_features;
  73     cpuid(1, &eax, &ebx, &ecx, &cpuid_features);
  74     if (! (cpuid_features & CPUID_APIC)) {
  75         // No apic - only the main cpu is present.
  76         smp_cpus = 1;
  77         return 1;
  78     }
  79
  80     // Init the counter.
  81     writel(&smp_cpus, 1);
  82
  83     // Setup jump trampoline to counter code.
  84     u64 old = *(u64*)BUILD_AP_BOOT_ADDR;
  85     // ljmpw $SEG_BIOS, $(smp_ap_boot_code - BUILD_BIOS_ADDR)
  86     u64 new = (0xea | ((u64)SEG_BIOS<<24)
  87                | (((u32)smp_ap_boot_code - BUILD_BIOS_ADDR) << 8));
  88     *(u64*)BUILD_AP_BOOT_ADDR = new;
  89
  90     // enable local APIC
  91     u32 val = readl(APIC_SVR);
  92     writel(APIC_SVR, val | APIC_ENABLED);
  93
  94     // broadcast SIPI
  95     writel(APIC_ICR_LOW, 0x000C4500);
  96     u32 sipi_vector = BUILD_AP_BOOT_ADDR >> 12;
  97     writel(APIC_ICR_LOW, 0x000C4600 | sipi_vector);
  98
  99     // Wait for other CPUs to process the SIPI.
 100     if (CONFIG_COREBOOT)
 101         mdelay(10);
 102     else
 103         while (inb_cmos(CMOS_BIOS_SMP_COUNT) + 1 != readl(&smp_cpus))
 104             ;
 105
 106     // Restore memory.
 107     *(u64*)BUILD_AP_BOOT_ADDR = old;
 108
 109     u32 count = readl(&smp_cpus);
 110     dprintf(1, "Found %d cpu(s)\n", count);
 111     return count;
 112 }
 113
 114 // Reset smp_cpus to zero (forces a recheck on reboots).
 115 void
 116 smp_probe_setup(void)
 117 {
 118     smp_cpus = 0;
 119 }