3 // Copyright (C) 2008 Kevin O'Connor <kevin@koconnor.net>
4 // Copyright (C) 2006 Fabrice Bellard
6 // This file may be distributed under the terms of the GNU LGPLv3 license.
8 #include "util.h" // dprintf
9 #include "config.h" // CONFIG_*
10 #include "cmos.h" // CMOS_BIOS_SMP_COUNT
11 #include "farptr.h" // ASSERT32
13 #define CPUID_APIC (1 << 9)
15 #define APIC_ICR_LOW ((u8*)BUILD_APIC_ADDR + 0x300)
16 #define APIC_SVR ((u8*)BUILD_APIC_ADDR + 0x0F0)
18 #define APIC_ENABLED 0x0100
20 static inline void writel(void *addr, u32 val)
22 *(volatile u32 *)addr = val;
25 static inline void writew(void *addr, u16 val)
27 *(volatile u16 *)addr = val;
30 static inline void writeb(void *addr, u8 val)
32 *(volatile u8 *)addr = val;
35 static inline u32 readl(const void *addr)
37 return *(volatile const u32 *)addr;
40 static inline u16 readw(const void *addr)
42 return *(volatile const u16 *)addr;
45 static inline u8 readb(const void *addr)
47 return *(volatile const u8 *)addr;
50 u32 smp_cpus VAR16_32;
51 extern void smp_ap_boot_code();
53 " .global smp_ap_boot_code\n"
55 // Increment the cpu counter
56 " movw $" __stringify(SEG_BIOS) ", %ax\n"
58 " lock incl smp_cpus\n"
59 // Halt the processor.
64 /* find the number of CPUs by launching a SIPI to them */
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.
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;
91 u32 val = readl(APIC_SVR);
92 writel(APIC_SVR, val | APIC_ENABLED);
95 writel(APIC_ICR_LOW, 0x000C4500);
96 u32 sipi_vector = BUILD_AP_BOOT_ADDR >> 12;
97 writel(APIC_ICR_LOW, 0x000C4600 | sipi_vector);
99 // Wait for other CPUs to process the SIPI.
103 while (inb_cmos(CMOS_BIOS_SMP_COUNT) + 1 != readl(&smp_cpus))
107 *(u64*)BUILD_AP_BOOT_ADDR = old;
109 u32 count = readl(&smp_cpus);
110 dprintf(1, "Found %d cpu(s)\n", count);
114 // Reset smp_cpus to zero (forces a recheck on reboots).
116 smp_probe_setup(void)