blob: 89be3c8820474f2da17b75e257f6927fd4ccc029 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
#include <generic/bootloader/bootloader.hpp>
#include <generic/lock/spinlock.hpp>
#if defined(__x86_64__)
#include <arch/x86_64/cpu_local.hpp>
#endif
#include <generic/time.hpp>
#include <generic/arch.hpp>
#include <generic/mp.hpp>
#include <klibc/stdio.hpp>
#include <cstdint>
#include <atomic>
std::uint32_t balance_how_much_cpus = 1;
std::uint32_t how_much_cpus = 0;
static mp::barrier mp_barrier;
locks::spinlock smp_lock;
std::uint32_t mp::cpu_count() {
return how_much_cpus;
}
void mp::sync() {
if (how_much_cpus <= 1) return;
uint32_t current_gen = mp_barrier.generation.load(std::memory_order_acquire);
if (mp_barrier.count.fetch_add(1, std::memory_order_acq_rel) == (how_much_cpus - 1)) {
mp_barrier.count.store(0, std::memory_order_relaxed);
mp_barrier.generation.fetch_add(1, std::memory_order_release);
} else {
while (mp_barrier.generation.load(std::memory_order_acquire) == current_gen) {
arch::pause();
}
}
}
void smptrampoline(limine_mp_info* smp_info) {
smp_lock.lock();
std::uint32_t enum_cpu = smp_info->processor_id;
arch::init(ARCH_INIT_MP);
#if defined(__x86_64__)
x86_64::cpu_data()->cpu = balance_how_much_cpus++;
enum_cpu = x86_64::cpu_data()->cpu;
#endif
log("smp", "cpu %d is online (%d)",smp_info->lapic_id,enum_cpu);
smp_lock.unlock();
mp::sync();
if(time::timer) time::timer->sleep(10000);
mp::sync();
arch::enable_interrupts();
while(true) {
arch::wait_for_interrupt();
}
}
void mp::init() {
limine_mp_response* mp_info = bootloader::bootloader->get_mp_info();
if(!mp_info)
return;
how_much_cpus = mp_info->cpu_count;
for(std::uint16_t i = 0;i < mp_info->cpu_count;i++) {
if(mp_info->bsp_lapic_id != i) {
mp_info->cpus[i]->goto_address = smptrampoline;
}
}
log("mp", "detected %d cpus", mp_info->cpu_count);
}
|
